《计算机应用》唯一官方网站 ›› 2021, Vol. 41 ›› Issue (12): 3626-3631.DOI: 10.11772/j.issn.1001-9081.2021060974

• 第十八届中国机器学习会议(CCML 2021) • 上一篇    

基于图卷积网络的迁移学习轴承服役故障诊断

彭雪莹, 江永全(), 杨燕   

  1. 西南交通大学 计算机与人工智能学院,成都 611756
  • 收稿日期:2021-05-12 修回日期:2021-06-13 接受日期:2021-06-29 发布日期:2021-12-28 出版日期:2021-12-10
  • 通讯作者: 江永全
  • 作者简介:彭雪莹(1996—),女,四川成都人,硕士研究生,主要研究方向:深度学习
    杨燕(1964—),女,安徽合肥人,教授,博士生导师,博士,CCF会员,主要研究方向:人工智能、大数据分析与挖掘、集成学习与多视图学习、云计算与云服务。
  • 基金资助:
    国家自然科学基金资助项目(61976247)

Transfer learning based on graph convolutional network in bearing service fault diagnosis

Xueying PENG, Yongquan JIANG(), Yan YANG   

  1. School of Computing and Artificial Intelligence,Southwest Jiaotong University,Chengdu Sichuan 611756,China
  • Received:2021-05-12 Revised:2021-06-13 Accepted:2021-06-29 Online:2021-12-28 Published:2021-12-10
  • Contact: Yongquan JIANG
  • About author:PENG Xueying, born in 1996, M. S. candidate. Her research interests include deep learning.
    YANG Yan, born in 1964, Ph. D., professor. Her research interests include artificial intelligence, big data analysis and mining, integrated learning and multi-view learning, cloud computing and cloud services.
  • Supported by:
    the National Natural Science Foundation of China(61976247)

摘要:

深度学习方法被广泛应用于轴承故障诊断,但在实际工程应用中,轴承服役期间的真实服役故障数据不易收集,缺乏数据标签,难以进行充分的训练。针对轴承服役故障诊断困难的问题,提出了一种基于图卷积网络(GCN)的迁移学习轴承服役故障诊断模型。该模型从数据充足的人工模拟损伤故障数据中学习故障知识,并迁移到真实的服役故障上,以提高服役故障的诊断准确率。具体来说,通过将人工模拟损伤故障数据和服役故障数据的原始振动信号由小波变换转换为同时具有时间和频率信息的时频图,并将得到的时频图输入到图卷积层中进行学习,从而有效地提取源域和目标域的故障特征表示;然后计算源域和目标域的数据分布之间的Wasserstein距离来度量两个数据分布之间的差异,通过最小化数据分布差异,构建了一个能诊断轴承服役故障的故障诊断模型。在不同的轴承故障数据集和不同工作条件下设计了多种不同的任务进行实验,实验结果表明,该模型具有诊断轴承服役故障的能力,同时也能从一个工作条件迁移到另一工作条件,在不同组件类型和不同工作条件之间进行故障诊断。

关键词: 轴承故障诊断, 深度学习, 迁移学习, 图卷积网络, 小波变换

Abstract:

Deep learning methods are widely used in bearing fault diagnosis, but in actual engineering applications, real service fault data during bearing service are not easily collected and lack of data labels, which is difficult to train adequately. Focused on the difficulty of bearing service fault diagnosis, a transfer learning model based on Graph Convolutional Network (GCN) in bearing service fault diagnosis was proposed. In the model, the fault knowledge was learned from artificially simulated damage fault data with sufficient data and transferred to real service faults, so as to improve the diagnostic accuracy of service faults. Specifically, the original vibration signals of artificially simulated damage fault data and service fault data were converted into the time-frequency maps with both time and frequency information through wavelet transform, and the obtained maps were input into graph convolutional layers for learning, so as to effectively extract the fault feature representations in the source and target domains. Then the Wasserstein distance between the data distributions of source domain and target domain was calculated to measure the difference between two data distributions, and a fault diagnosis model that can diagnose bearing service faults was constructed by minimizing the difference in data distribution. A variety of different tasks were designed for experiments with different bearing failure data sets and different operating conditions. Experimental results show that the proposed model has the ability to diagnose bearing service faults and also can be transferred from one working condition to another, and perform fault diagnosis between different component types and different working conditions.

Key words: bearing fault diagnosis, deep learning, transfer learning, Graph Convolutional Network (GCN), wavelet transform

中图分类号: