不均衡样本下轴承故障的LSGAN‑Swin Transformer诊断方法

LSGAN-Swin Transformer diagnosis method of bearing fault under unbalanced samples

  • 摘要: 针对轴承在复杂环境下工作时故障数据难以大量获取,正常数据与故障数据比例严重失衡造成的深度模型训练不充分、诊断精度低等问题,提出一种基于LSGAN‑Swin Transformer的轴承故障诊断方法,利用最小二乘生成对抗网络(LSGAN)扩充不均衡或缺少的轴承数据集,引入窗口自注意力网络进行轴承故障状态识别,使用两种数据集验证所提方法的有效性,并分别与SGAN、WGAN进行对比,证明LSGAN生成的数据训练模型具有更高的准确率。在小样本条件下训练LSGAN,将所提Swin Transformer(Swin‑T)模型与CNN、AlexNe和SqueezeNet进行对比,诊断准确率分别提升了34.85%、13.45%和12.95%。通过t‑SNE可视化分析对模型分类效果进行评估,结果表明,LSGAN‑Swin‑T模型在训练样本数量较少时仍能较好地满足故障诊断中的需求,为不均衡数据下的轴承故障诊断研究提供思路。

     

    Abstract: Aiming at the problems of bearings working in complex environments, where fault data are difficult to obtain in large quantities and the serious imbalance between the ratio of normal data and fault data resulting in insufficient in-depth model training and low diagnostic accuracy, a bearing fault diagnosis method based on LSGAN-Swin Transformer is proposed. The least-squares generative adversarial network is utilized to expand the imbalanced or lack of bearing dataset, and the windowed self-attentive network is introduced for bearing fault state identification. The proposed method is validated by using two date sets, and compared with SGAN and WGAN respectively. It is demonstrated that LSGAN generates data training models with higher accuracy. The proposed Swin Transformer (Swin-T) model is compared with CNN, AlexNet and SqueezeNet under small sample conditions, and the accuracy is improved by 34.85%, 13.45%, and 12.95%, respectively. The classification effect of the model is evaluated by t-SNE visualization, and the results show that the LSGAN-Swin-T model can still meet the requirements in fault diagnosis better when the number of training samples is small, which provides a new idea for the research of bearing fault diagnosis under unbalanced data.

     

/

返回文章
返回