English

中国农机化学报

中国农机化学报 ›› 2024, Vol. 45 ›› Issue (6): 178-183.DOI: 10.13733/j.jcam.issn.2095-5553.2024.06.027

• 农业信息化工程 • 上一篇    下一篇

基于改进对抗蒸馏的轴承故障分类方法

李星逸,付波,范秀香,权轶   

  1. (湖北工业大学电气与电子工程学院,武汉市,430068)
  • 出版日期:2024-06-15 发布日期:2024-06-08
  • 基金资助:
    湖北省重点研发计划项目(2021BAA193)

Bearing fault classification method based on improved countermeasures distillation

Li Xingyi, Fu Bo, Fan Xiuxiang, Quan Yi   

  1. (School of Electrical and Electronic Engineering, Hubei University of Technology, Wuhan, 430068, China)
  • Online:2024-06-15 Published:2024-06-08

摘要:

针对工业现场条件下和农业机械设备故障数据典型样本不充足导致轴承故障诊断精度低的问题,提出一种基于改进对抗蒸馏的轴承故障分类方法。使用对抗蒸馏方法进行轴承故障分类,让学生网络通过对抗学习教师网络的软标签所提供的信息,同时生成器输出与学生网络输出相似的样本提供给判别器后修改学生网络参数。提出退火改进对抗蒸馏方法,在对抗蒸馏中使用动态温度进行训练,增加生成器制作样本难度,使教师网络输出的信息被更好地利用,以提高学生网络泛化能力和鲁棒性。试验使用美国凯斯西储大学轴承故障数据集验证方法的有效性,利用所提出的方法训练出的学生网络在模拟现场轴承故障诊断分类任务中仅使用214 602个参数参与计算,准确率可达91.85%,提高故障诊断精度并节省设备的计算资源。

关键词: 轴承, 故障诊断, 知识蒸馏, 对抗学习, 模拟退火算法

Abstract:

Aiming at  the problem of low accuracy in bearing fault diagnosis caused by insufficient typical samples of fault data in the industrial domain and agricultural machinery,  a bearing fault classification method based on improved adversarial distillation is proposed.  The adversarial knowledge distillation method is used to classify the bearing faults. Based on the soft labels of the teacher network, the student network produces samples similar to those output by the student network. The modification of the parameters of the student network proceeds as the discriminator evaluates the samples. In this paper,  an annealed modified adversarial distillation method is proposed to improve the robustness and generalization ability of the student network. With dynamic temperature training in adversarial distillation, the difficulty of generating samples is increased for more efficient utilization of information from the teacher network. The effectiveness of the method is verified through experiments based on the bearing fault dataset from Case Western Reserve University in the United States. The student network trained with the proposed method achieves an accuracy of 91.85% in the simulation of onsite bearing fault diagnosis classification task, with only 214602 parameters involved in the computation, which not only improves the accuracy of fault diagnosis but also saves computing resources of the equipment.

Key words: bearing, fault diagnosis, knowledge distillation, adversarial learning, simulated annealing algorithm

中图分类号: