WGLformer: a deep learning framework based on weighted differential attention and Gabor-BiLSTM for bearing fault diagnosis in noisy environments
摘要
To solve the problem of low fault diagnosis accuracy in bearing systems under strong noise, this paper proposes a deep learning framework based on weighted differential attention and Gabor-BiLSTM, called WGLformer. First, a Gabor-BiLSTM module is used at the input. The input signal is processed in parallel by learnable Gabor filters and Bidirectional Long Short-Term Memory (BiLSTM) network, and their outputs are summed and passed to the next layer. This module combines local frequency with global time features. Then, in deeper layers of the network, a Multi-Headed Weighted Differential Attention (W-MHDA) mechanism and an Refined Separable Multi-Scale Convolution Block (ReSMCB) are introduced. W-MHDA uses the concept of differential amplifier circuits to suppress noise and enhance features. ReSMCB combines Transformer structure with multi-scale techniques to extract global and local features. Finally, experiments on two datasets show this method achieves higher fault identification accuracy and robustness in high-noise environments.