Efficient Gearbox Fault Diagnosis Based on Improved Multi-Scale CNN with Lightweight Convolutional Attention

Abstract

As a core transmission component of modern industrial equipment, the operation status of the gearbox has a significant impact on the reliability and service life of major machinery. In this paper, we propose an intelligent diagnosis framework based on Empirical Mode Decomposition and multimodal feature co-optimization and innovatively construct a fault diagnosis model by fusing a multi-scale convolutional neural network and a lightweight convolutional attention model. The framework extracts the multi-band features of vibration signals through the improved multi-scale convolutional neural network, which significantly enhances adaptability to complex working conditions (variable rotational speed, strong noise); at the same time, the lightweight convolutional attention mechanism is used to replace the multi-attention of the traditional Transformer, which greatly reduces computational complexity while guaranteeing accuracy and realizes highly efficient, lightweight local–global feature modeling. The lightweight convolutional attention is adaptively captured by the dynamic convolutional kernel generation strategy to adaptively capture local features in the time domain, and combined with grouped convolution to enhance the computational efficiency further; in addition, parameterized revised linear units are introduced to retain fault-sensitive negative information, which enhances the model’s ability to detect weak faults. The experimental findings demonstrate that the proposed model achieves an accuracy greater than 98.9%, highlighting its exceptional diagnostic accuracy and robustness. Moreover, compared to other fault diagnosis methods, the model exhibits superior performance under complex working conditions.