Computer Engineering and Applications ›› 2020, Vol. 56 ›› Issue (12): 273-278.DOI: 10.3778/j.issn.1002-8331.1904-0256

Previous Articles    

Application of Cooperative Training Algorithm in Fault Diagnosis of Rolling Bearing

WANG Dexue, LIN Yi, CHEN Junjie   

  1. 1.School of Digital Media, Jiangnan University, Wuxi, Jiangsu 214122, China
    2.Jiangsu Key Laboratory of Media Design and Software Technology, Wuxi, Jiangsu 214122, China
    3.SIEMENS, China Institute, Beijing 100102, China
  • Online:2020-06-15 Published:2020-06-09



  1. 1.江南大学 数字媒体学院,江苏 无锡 214122
    2.江苏省媒体设计与软件技术重点实验室,江苏 无锡 214122
    3.西门子中国研究院,北京 100102


Aiming at the problems of single classifier method in rolling bearing fault diagnosis, such as low accuracy, scarce fault sample mark and high feature space dimension, a co-forest bearing fault diagnosis algorithm combining collaborative training and ensemble learning is proposed. Co-forest is a collaborative training algorithm in semi-supervised learning, which contains multiple base classifiers and realizes confidence estimation in collaborative training through voting. The time domain and frequency domain characteristic indexes are extracted from the vibration signal of rolling bearing. A small number of labeled and large number of unlabeled samples are used to train the base classifier repeatedly. The fault diagnosis of rolling bearing is realized by integrating base classifier. Experimental results show that the co-forest algorithm has a higher accuracy in bearing fault diagnosis than the same type of cooperative training algorithm (co-training, tri-training). Compared with the ISS-LPP algorithm and SS-LLTSA algorithm for the problem of high dimension of feature vectors and scarcity of labeled samples, the co-forest algorithm has certain practical application value under the condition of maintaining a high diagnostic accuracy rate without dimensional reduction and simple parameter setting.

Key words: cooperative training, ensemble learning, fault diagnosis, confidence



关键词: 协同训练, 集成学习, 故障诊断, 置信度