Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model
Aiming at the problems of complex diesel engine cylinder head signals, difficulty in extracting fault information, and existing deep learning fault diagnosis algorithms with many training parameters, high time cost, and high data volume requirements, a small-sample transfer learning fault diagnosis...
Main Authors: | , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
De Gruyter
2022-10-01
|
Series: | Open Physics |
Subjects: | |
Online Access: | https://doi.org/10.1515/phys-2022-0197 |
_version_ | 1811246779112357888 |
---|---|
author | Liu Yangshuo Kang Jianshe Guo Chiming Bai Yunjie |
author_facet | Liu Yangshuo Kang Jianshe Guo Chiming Bai Yunjie |
author_sort | Liu Yangshuo |
collection | DOAJ |
description | Aiming at the problems of complex diesel engine cylinder head signals, difficulty in extracting fault information, and existing deep learning fault diagnosis algorithms with many training parameters, high time cost, and high data volume requirements, a small-sample transfer learning fault diagnosis algorithm is proposed in this article. First, the fault vibration signal of the diesel engine is converted into a three-channel red green blue (RGB) short-time Fourier transform time–frequency diagram, which reduces the randomness of artificially extracted features. Then, for the problem of slow network training and large sample size requirements, the AlexNet convolutional network and the ResNet-18 convolutional network are fine-tuned on the diesel engine time–frequency map samples as pre-training models with the transfer diagnosis strategy. In addition, to improve the training effect of the network, a surrogate model is introduced to autonomously optimize the hyperparameters of the network. Experiments show that, when compared to other commonly used methods, the transfer fault diagnosis algorithm proposed in this article can obtain high classification accuracy in the diagnosis of diesel engines while maintaining very stable performance under the condition of small samples. |
first_indexed | 2024-04-12T14:58:23Z |
format | Article |
id | doaj.art-ce1bab6a2633463287c97c9ab5b38d5a |
institution | Directory Open Access Journal |
issn | 2391-5471 |
language | English |
last_indexed | 2024-04-12T14:58:23Z |
publishDate | 2022-10-01 |
publisher | De Gruyter |
record_format | Article |
series | Open Physics |
spelling | doaj.art-ce1bab6a2633463287c97c9ab5b38d5a2022-12-22T03:28:09ZengDe GruyterOpen Physics2391-54712022-10-01201993101810.1515/phys-2022-0197Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate modelLiu Yangshuo0Kang Jianshe1Guo Chiming2Bai Yunjie3Department of Equipment Command and Management, Shijiazhuang Campus, Army Engineering University of PLA, Shijiazhuang 050003, ChinaDepartment of Equipment Command and Management, Shijiazhuang Campus, Army Engineering University of PLA, Shijiazhuang 050003, ChinaDepartment of Equipment Command and Management, Shijiazhuang Campus, Army Engineering University of PLA, Shijiazhuang 050003, ChinaSupport detachment, 66029 Unit of the Chinese People’s Liberation Army, Xilinguolemeng 011200, ChinaAiming at the problems of complex diesel engine cylinder head signals, difficulty in extracting fault information, and existing deep learning fault diagnosis algorithms with many training parameters, high time cost, and high data volume requirements, a small-sample transfer learning fault diagnosis algorithm is proposed in this article. First, the fault vibration signal of the diesel engine is converted into a three-channel red green blue (RGB) short-time Fourier transform time–frequency diagram, which reduces the randomness of artificially extracted features. Then, for the problem of slow network training and large sample size requirements, the AlexNet convolutional network and the ResNet-18 convolutional network are fine-tuned on the diesel engine time–frequency map samples as pre-training models with the transfer diagnosis strategy. In addition, to improve the training effect of the network, a surrogate model is introduced to autonomously optimize the hyperparameters of the network. Experiments show that, when compared to other commonly used methods, the transfer fault diagnosis algorithm proposed in this article can obtain high classification accuracy in the diagnosis of diesel engines while maintaining very stable performance under the condition of small samples.https://doi.org/10.1515/phys-2022-0197fault diagnosistransfer learningsurrogate modelhyperparameter optimizationsmall sample |
spellingShingle | Liu Yangshuo Kang Jianshe Guo Chiming Bai Yunjie Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model Open Physics fault diagnosis transfer learning surrogate model hyperparameter optimization small sample |
title | Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model |
title_full | Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model |
title_fullStr | Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model |
title_full_unstemmed | Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model |
title_short | Diesel engine small-sample transfer learning fault diagnosis algorithm based on STFT time–frequency image and hyperparameter autonomous optimization deep convolutional network improved by PSO–GWO–BPNN surrogate model |
title_sort | diesel engine small sample transfer learning fault diagnosis algorithm based on stft time frequency image and hyperparameter autonomous optimization deep convolutional network improved by pso gwo bpnn surrogate model |
topic | fault diagnosis transfer learning surrogate model hyperparameter optimization small sample |
url | https://doi.org/10.1515/phys-2022-0197 |
work_keys_str_mv | AT liuyangshuo dieselenginesmallsampletransferlearningfaultdiagnosisalgorithmbasedonstfttimefrequencyimageandhyperparameterautonomousoptimizationdeepconvolutionalnetworkimprovedbypsogwobpnnsurrogatemodel AT kangjianshe dieselenginesmallsampletransferlearningfaultdiagnosisalgorithmbasedonstfttimefrequencyimageandhyperparameterautonomousoptimizationdeepconvolutionalnetworkimprovedbypsogwobpnnsurrogatemodel AT guochiming dieselenginesmallsampletransferlearningfaultdiagnosisalgorithmbasedonstfttimefrequencyimageandhyperparameterautonomousoptimizationdeepconvolutionalnetworkimprovedbypsogwobpnnsurrogatemodel AT baiyunjie dieselenginesmallsampletransferlearningfaultdiagnosisalgorithmbasedonstfttimefrequencyimageandhyperparameterautonomousoptimizationdeepconvolutionalnetworkimprovedbypsogwobpnnsurrogatemodel |