CA-MSPNet: Multiscale Module and Coordinate Attention Mechanism Strategy Combined With ProtoNet for Gearbox Fault Diagnosis Using Limited Infrared Thermal Images Under Variable Working Conditions

Jian Ge,Di Zhou,Xiao Zhuang,Xiaomin Wang,Weifang Sun,Hongbing Yao,Jiawei Xiang
DOI: https://doi.org/10.1109/tim.2024.3457938
IF: 5.6
2024-09-27
IEEE Transactions on Instrumentation and Measurement
Abstract:Existing deep learning-based infrared image analysis methods have been widely utilized in gearbox faults. However, these methods still face problems such as insufficient amount of effective data and poor adaptability under variable working conditions. In this article, a deep prototype network model combined with a multiscale module and coordinate attention mechanism (CA-MSPNet) is proposed to realize infrared image-based high-precision fault diagnosis of gearboxes. First, the experimental testing platform is established to obtain infrared failure image data with Gaussian noise added to generate six different faults in gearboxes. Second, to solve the difficulty of extracting richer fault features due to the small number of samples in few-shot learning, a multiscale (MS) module is introduced into the base prototype network (ProtoNet) model in this article to reduce the fluctuation range of the feature values so as to obtain and retain the fault features as much as possible. Third, the coordinate attention (CA) mechanism strategy is involved in the model to assign feature weights to enhance the response effect of important features. Finally, extensive experiments are conducted to validate the effectiveness of the CA-MSPNet model. The results of the failure experiments on the gearbox system show that, compared with the other seven benchmark models, the proposed CA-MSPNet has the best performance under multiple working conditions, thus verifying the generalization and superiority of the proposed CA-MSPNet method in few-shot learning.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?