An innovative approach to vibration signal denoising and fault diagnosis using attention-enriched joint learning

Abstract

Vibration signals play a crucial role in mechanical fault diagnosis. However, they are susceptible to various noise disturbances, presenting challenges for reliable fault detection. We propose an end-to-end Cross-task Attention Joint Learning (CTA-JL) model that concurrently denoises and diagnoses faults in noisy signals. This model utilizes a multi-task encoder, composed of task-shared and task-specific feature encoding units, along with a feature information exchange unit with a Cross-task Attention (CTA) mechanism, fostering information exchange across different tasks. By collectively executing diagnosis and denoising tasks and sharing valuable task information, the model enhances prediction accuracy and denoising performance. Under three noise conditions of SNR = −9 dB, −6 dB, and −3 dB, the prediction accuracy of CTA-JL on the rolling bearing datasets reached 91.38%, 97.95%, and 99.69%, respectively. Meanwhile, the result on elevator guide system datasets reached 87.31%, 95.58%, and 99.64%

Keywords

Vibration signal processing fault diagnosis attention mechanisms multi-task learning

Get full access to this article

View all access options for this article.

References

Chen

Liu

(2017) Mechanical state prediction based on lstm neural netwok. In: 2017 36th Chinese Control Conference (CCC), Dalian, China, 26–28 July 2017.

Dataset MFPT (2020) Society for Machinery Failure Prevention Technology.

Dong

Jia

, et al. (2021) Non-iterative denoising algorithm for mechanical vibration signal using spectral graph wavelet transform and detrended fluctuation analysis. Mechanical Systems and Signal Processing 149: 107202.

Feng

, et al. (2023) A novel vibration-based prognostic scheme for gear health management in surface wear progression of the intelligent manufacturing system. Wear 522: 204697.

Han

Wang

Liu

, et al. (2022) Intelligent vibration signal denoising method based on non-local fully convolutional neural network for rolling bearings. ISA Transactions 122: 13–23. DOI: 10.1016/j.isatra.2021.04.022.

Hua

Dai

Liu

, et al. (2022) Transformer quality in linear time. In: International Conference on Machine Learning, Baltimore, 17–23 July 2022.

Jin

Hou

Chen

(2022) A time series transformer based method for the rotating machinery fault diagnosis. Neurocomputing 494: 379–395.

Liao

Dong

Sun

, et al. (2023) Attention-embedded quadratic network (qttention) for effective and interpretable bearing fault diagnosis. IEEE Transactions on Instrumentation and Measurement 72: 1–13.

Liu

Zhou

Zheng

, et al. (2018) Fault diagnosis of rolling bearings with recurrent neural network-based autoencoders. ISA Transactions 77: 167–178.

10.

Mao

, et al. (2017) Bearing fault diagnosis with auto-encoder extreme learning machine: a comparative study. Proceedings of the Institution of Mechanical Engineers - Part C: Journal of Mechanical Engineering Science 231(8): 1560–1578.

11.

Halkon

, et al. (2023) Physics-informed residual network (piresnet) for rolling element bearing fault diagnostics. Mechanical Systems and Signal Processing 200: 110544.

12.

Seryasat

Honarvar

Rahmani

, et al. (2010) Multi-fault diagnosis of ball bearing using fft, wavelet energy entropy mean and root mean square (rms). In: 2010 IEEE International Conference on Systems, Man and Cybernetics, Istanbul, 10–13 October 2010.

13.

Vakharia

Gupta

Kankar

(2015) A multiscale permutation entropy based approach to select wavelet for fault diagnosis of ball bearings. Journal of Vibration and Control 21(16): 3123–3131.

14.

Wang

Xiang

Markert

, et al. (2016) Spectral kurtosis for fault detection, diagnosis and prognostics of rotating machines: a review with applications. Mechanical Systems and Signal Processing 66-67: 679–698.

15.

Wang

Cao

Han

(2022) A combination of dilated self-attention capsule networks and bidirectional long-and short-term memory networks for vibration signal denoising. Machines 10(10): 840.

16.

Xiong

Yuan

, et al. (2023) A novel deep convolutional image-denoiser network for structural vibration signal denoising. Engineering Applications of Artificial Intelligence 117: 105507.

17.

Feng

Yan

, et al. (2023a) Cfcnn: a novel convolutional fusion framework for collaborative fault identification of rotating machinery. Information Fusion 95: 1–16.

18.

Zhang

Peng

, et al. (2017) A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals. Sensors 17(2): 425.

19.

Yan

Sun

, et al. (2023b) Online knowledge distillation based multiscale threshold denoising networks for fault diagnosis of transmission systems. IEEE Transactions on Transportation Electrification 1–1.

20.

Zhang

, et al. (2022) Fault diagnosis for small samples based on attention mechanism. Measurement 187: 110242.

21.

Zhao

, et al. (2020) Deep learning algorithms for rotating machinery intelligent diagnosis: an open source benchmark study. ISA Transactions 107: 224–255.

22.

Zhong

Wang

Ban

(2022) Development of a plug-and-play anti-noise module for fault diagnosis of rotating machines in nuclear power plants. Progress in Nuclear Energy 151: 104344.