基于GWO优化CNN-BiLSTM-Attention的滚动轴承故障诊断研究

Research on rolling bearing fault diagnosis based on CNN-BiLSTM-Attention optimized by GWO

  • 摘要: 为了减少滚动轴承故障特征微弱且容易与干扰耦合误诊漏诊的问题,针对旋转机械,文章提出了一种基于灰狼优化算法(grey wolf optimizer, GWO)全局寻优的CNN-BiLSTM-Attention智能诊断方法。首先,采用卷积神经网络(convolutional neural network,CNN)提取信号局部冲击特征,CNN卷积层能够有选择性地保持局部冲击特征数据,同时抑制噪声干扰;其次,利用双向长短期记忆网络(BiLSTM)捕捉信号的时序长程依赖关系,BiLSTM的隐藏层通过重复操作有选择性地积累重要特征表示,并在最后保留上一次捕捉到的所有有用特征表示;同时,注意力机制(attention mechanism, AM)通过自适应地强化关键故障信息,进一步提升其重要性,进而突出重点特征,减轻背景噪声的扰动,使神经网络聚集于正确的数据,有效提升特征表达能力;最后,GWO自动搜索出CNN卷积层滤波器数量、BiLSTM层神经元数量、AM的dropout率、全连接层神经元数量等超参数及结构,从而避免了人工调参容易陷入局部最优的问题,具有更快的速度且和更强的泛化能力。对比试验表明,所提模型在凯斯西储大学(Case Western Reserve University, CWRU)轴承故障诊断公开数据集上能够有效完成滚动轴承故障诊断,并保持较高的识别率。研究结论对提高滚动轴承故障诊断效率及可靠性具有重要意义,有助于推动旋转机械智能运维以及滚动轴承故障预测性维修技术的发展。

     

    Abstract: To mitigate the challenges of weak fault signatures in rolling bearings and their susceptibility to misdiagnosis or missed detection due to interference coupling, an intelligent diagnostic method for rotating machinery based on CNN-BiLSTM-Attention with global optimization via the grey wolf optimizer (GWO) is proposed. Firstly, Convolutional neural network (CNN) is employed to extract local impulse signatures from the raw vibration signal. The convolutional kernels selectively preserve shock-related features while suppressing noise. Secondly, Bidirectional long short-term memory network (BiLSTM) is utilized to capture long-range temporal dependencies. The hidden layer of BiLSTM selectively accumulates important feature representations through repetitive operations, and retains all the useful feature representations captured in the previous step at the end. Meanwhile, attention mechanism (AM) further reinforces the key fault information by adaptively re-weighting the feature maps, highlighting discriminative signatures and attenuating background disturbances so that the network focuses on the correct data and boosts expressive power. Finally, the GWO automatically searches for the optimal hyper-parameters and architecture, namely the number of CNN filters, BiLSTM hidden neurons, AM dropout rate, and fully-connected layer size,thereby avoiding the local optima that plague manual tuning, accelerating convergence, and enhancing generalization. Comparative experiments on the public bearing fault diagnosis dataset of Case Western Reserve University (CWRU) verify that the proposed model can effectively realize rolling bearing fault diagnosis and maintain high recognition accuracy. The research conclusions are of great significance for improving the efficiency and reliability of rolling bearing fault diagnosis, and help promote the development of intelligent operation and maintenance of rotating machinery as well as predictive maintenance technology for rolling bearing fault.

     

/

返回文章
返回