LAFICNN: A Novel Convolutional Adaptive Fusion Framework for Fault Diagnosis of Rotating Machinery

Tianzhuang Yu,Zhaohui Ren,Yongchao Zhang,Shihua Zhou,Zeyu Jiang,Xin Zhou
DOI: https://doi.org/10.1109/tim.2024.3379402
IF: 5.6
2024-03-30
IEEE Transactions on Instrumentation and Measurement
Abstract:Convolutional neural network (CNN)-based approaches have been widely developed and applied in intelligent diagnosis. However, only some CNN models make sufficient use of various favorable information in images, limiting adequate feature extraction. Moreover, existing models are still dominated by single-channel types, with limited multisensor collaborative diagnostic framework development. To alleviate the above problems, we propose a diagnostic framework called lightweight adaptive fusion improved CNN (LAFICNN). Specifically, a global spatial attention (GSA) mechanism is first developed, which considers the nature of images and enhances the model's attention to the long-range dependencies of signals. Then, a novel Conv Block that can better utilize the spatial information in the image is designed, and a single-channel lightweight improved CNN (LICNN) is constructed. In addition, an adaptive fusion module (AFM) is proposed, which can adaptively assign learnable weights to features from different branches at a shallow layer of the network without increasing the training parameters. Finally, the LAFICNN is proposed by embedding the AFM into the LICNN to realize multisensor collaborative diagnosis. The experimental results on a rolling bearing dataset show that LAFICNN improves the performance by 0.8%–4.2% compared to the existing works. The stability of the LAFICNN is verified using a gearbox dataset.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?