Bearing Fault Diagnosis Method Based on Small Sample Data under Unbalanced Loads

被引:5
作者
He Q. [1 ]
Tang X. [1 ,2 ,3 ]
Li C. [1 ,2 ,3 ]
Lu J. [1 ,2 ,3 ]
Chen J. [1 ]
机构
[1] Key Laboratory of Advanced Manufacturing Technology, Ministry of Education, Guizhou University, Guiyang
[2] School of Mechanical Engineering, Guizhou University, Guiyang
[3] State Key Laboratory of Public Big Data, Guizhou University, Guiyang
来源
Zhongguo Jixie Gongcheng/China Mechanical Engineering | 2021年 / 32卷 / 10期
关键词
Bearing fault diagnosis; Convolutional neural network with self-attention mechanism(SeCNN); Gradient penalty Wasserstein generative adversarial network(WGAN-GP); Short-time Fourier transform; Small sample; Unbalanced load;
D O I
10.3969/j.issn.1004-132X.2021.10.004
中图分类号
学科分类号
摘要
Aiming at the problems that the bearing vibration signals were easily disturbed by unbalanced load and the small number of bearing fault samples, a bearing fault diagnosis method based on WGAN-GP and SeCNN was proposed. The bearing vibration signals were processed by short-time Fourier transform to get the time-spectrum samples that were easy to be processed by WGAN-GP, which were divided into training set, validation set and test set. Then the training set was inputted into WGAN-GP for adversarial training, new samples were generated with similar distribution to the training samples, and added to the training set to expand the training set. The expanded training set was input into SeCNN for learning, and the trained model was applied to the test set and output the fault recognition results. The analysis of the CUT-2 platform unbalanced load bearing data was carried out, and the experimental results show that the proposed method may accurately and effectively classify the bearing faults. © 2021, China Mechanical Engineering Magazine Office. All right reserved.
引用
收藏
页码:1164 / 1171and1180
相关论文
共 18 条
[1]  
YANG Z, GAO L., Wavelet Analysis and Fault Feature Extraction of Rolling Bearing, (2014)
[2]  
HE M, HE D., Deep Learning Based Approach for Bearing Fault Diagnosis, Industry Applications IEEE Transactions on, 53, 3, pp. 3057-3065, (2017)
[3]  
WANG Lihua, XIE Yangyang, ZHOU Zixian, Et al., Motor Fault Diagnosis Based on Convolutional Neural Networks, Vibration, Test and Diagnosis, 37, 6, pp. 1208-1215, (2017)
[4]  
WANG Z, WANG J, WANG Y., An Intelligent Diagnosis Scheme Based on Generative Adversarial Learning Deep Neural Networks and Its Application to Planetary Gearbox Fault Pattern Recognition, Neurocomputing, 310, 8, pp. 213-222, (2018)
[5]  
SHAO S, WANG P, YAN R., Generative Adversarial Networks for Data Augmentation in Machine Fault Diagnosis, Computers in Industry, 106, pp. 85-93, (2019)
[6]  
GOODFELLOW I J, POUGET-ABADIE J, MIRZA M, Et al., Generative Adversarial Networks, Advances in Neural Information Processing Systems, 3, pp. 2672-2680, (2014)
[7]  
CABRERA D, SANCHO F, LONG J, Et al., Generative Adversarial Networks Selection Approach for Extremely Imbalanced Fault Diagnosis of Reciprocating Machinery, IEEE Access, 7, pp. 70643-70653, (2019)
[8]  
DING Y, MA L, MA J, Et al., A Generative Adversarial Network-based Intelligent Fault Diagnosis Method for Rotating Machinery under Small Sample Size Conditions, IEEE Access, 7, pp. 149736-149749, (2019)
[9]  
ARJOVSKY M, CHINTALA S, BOTTOU L., Wasserstein Generative Adversarial Networks, International Conference on Machine Learning, pp. 214-223, (2017)
[10]  
GULRAJANI I, AHMED F, ARJOVSKY M, Et al., Improved Training of Wasserstein Gans