A Deep Transfer Model With Wasserstein Distance Guided Multi-Adversarial Networks for Bearing Fault Diagnosis Under Different Working Conditions

被引:109
作者
Zhang, Ming [1 ]
Wang, Duo [2 ]
Lu, Weining [3 ]
Yang, Jun [2 ]
Li, Zhiheng [1 ]
Liang, Bin [2 ,4 ]
机构
[1] Tsinghua Univ, Grad Sch Shenzhen, Ctr Artificial Intelligence & Robot, Shenzhen 518055, Peoples R China
[2] Tsinghua Univ, Dept Automat, Beijing 100084, Peoples R China
[3] Tsinghua Univ, Sch Aerosp Engn, Beijing 100084, Peoples R China
[4] Tsinghua Univ, Res Inst, Shenzhen 518057, Peoples R China
关键词
Transfer learning; fault diagnosis; convolutional neural network; multi-adversarial networks; NEURAL-NETWORKS; MACHINE;
D O I
10.1109/ACCESS.2019.2916935
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In recent years, intelligent fault diagnosis technology with the deep learning algorithm has been widely used in the manufacturing industry for substituting time-consuming human analysis method to enhance the efficiency of fault diagnosis. The rolling bearing as the connection between the rotor and support is the crucial component in rotating equipment. However, the working condition of the rolling bearing is under changing with complex operation demand, which will significantly degrade the performance of the intelligent fault diagnosis method. In this paper, a new deep transfer model based on Wasserstein distance guided multi-adversarial networks (WDMAN) is proposed to address this problem. The WDMAN model exploits complex feature space structures to enable the transfer of different data distributions based on multiple domain critic networks. The essence of our method is learning the shared feature representation by minimizing the Wasserstein distance between the source domain and target domain distribution in an adversarial training way. The experiment results demonstrate that our model outperforms the state-of-the-art methods on rolling bearing fault diagnosis under different working conditions. The t-distributed stochastic neighbor embedding (t-SNE) technology is used to visualize the learned domain invariant feature and investigate the transferability behind the great performance of our proposed model.
引用
收藏
页码:65303 / 65318
页数:16
相关论文
共 47 条
[1]  
[Anonymous], 2014, ABS14123474 ARXIV
[2]  
[Anonymous], P 3 INT C LEARNING R
[3]  
[Anonymous], 2015, J MACH LEARN RES
[4]  
[Anonymous], PROC CVPR IEEE
[5]  
[Anonymous], 2015, Nature, DOI [10.1038/nature14539, DOI 10.1038/NATURE14539]
[6]  
[Anonymous], 2016, DEEP LEARNING
[7]  
[Anonymous], 2016, ADV NEURAL INFORM PR
[8]  
[Anonymous], ADV NEURAL INFORM PR
[9]  
[Anonymous], 2012, TECH REP
[10]  
[Anonymous], 2019, IEEE T SYST MAN CY-S, DOI DOI 10.1109/TSMC.2017.2754287