Rolling Bearing Fault Diagnosis Method Based on Enhanced Knowledge Distillation and Lightweight Convolution Network

Jinlong Yang,Li Feng,Yuanming Lu
DOI: https://doi.org/10.1109/icceic60201.2023.10426666
2023-01-01
Abstract:In order to solve the problems of high time ductility, low calculation accuracy and unfriendly hardware deployment in practical engineering in rolling bearing fault diagnosis methods under strong noise background, a new kind of Improved Knowledge Distillation was proposed. (IKD) lightweight convolutional Neural Network fault diagnosis method. In order to reduce the increase of computing time caused by channel addition in the process of convolution, standard pointwise convolution and channel shuffle are used to design the basic unit of the network. On this basis, the IKD algorithm is used to train the student network (IKDStu) based on the pre-trained teacher network (TeacherNet). Simulation analysis shows that under the background of the given data with strong noise, the student network trained by the proposed method can ensure the accuracy of diagnosis, reduce the size of the model and the number of parameters, and improve the efficiency of diagnosis.
What problem does this paper attempt to address?