Skip to main content
SpringerLink
Log in

Intelligent fault diagnosis of rolling bearing based on novel CNN model considering data imbalance

  • Published:
Applied Intelligence Aims and scope Submit manuscript

Abstract

The intelligent fault diagnosis method based on deep learning has become a powerful tool for analyzing mechanical big data. However, a large proportion of collected data belong to the healthy condition, which will provoke data imbalance. In this article, a novel CNN-based intelligent machinery fault diagnosis method is proposed to deal with the data imbalance. This method consists of two modules, including feature extraction and fault identification. The former is optimized by the weighted-center-label loss function to extract discriminative features. The latter uses the distance between extracted features and pattern center vectors for fault identification. Two datasets are constructed to verify the effectiveness and superiority of the proposed method under data imbalance. The experimental results from two datasets show that the proposed method can effectively deal with data imbalance by extracting separable and discriminative features automatically.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10

Similar content being viewed by others

References

  1. Liu R, Yang B, Zio E, Chen X (2018) Artificial intelligence for fault diagnosis of rotating machinery: A review. Mech Syst Signal Process 108:33–47

    Article  Google Scholar 

  2. Zhao R, Yan R, Chen Z, Mao K, Wang P, Gao R X (2019) Deep learning and its applications to machine health monitoring. Mech Syst Signal Process 115:213–237

    Article  Google Scholar 

  3. Wang J, Ma Y, Zhang L, Gao R X, Wu D (2018) Deep learning for smart manufacturing: Methods and applications. J Manuf Syst 48:144–156

    Article  Google Scholar 

  4. Jia F, Lei Y, Lin J, Zhou X, Lu N (2016) Deep neural networks: A promising tool for fault characteristic mining and intelligent diagnosis of rotating machinery with massive data. Mech Syst Signal Process 72–73:303–315

    Article  Google Scholar 

  5. Zhao M, Zhong S, Fu X, Tang B, Pecht M (2020) Deep Residual Shrinkage Networks for Fault Diagnosis. IEEE Trans Ind Inf 16(7):4681–4690

    Article  Google Scholar 

  6. Zhang W, Li X, Ding Q (2019) Deep residual learning-based fault diagnosis method for rotating machinery. ISA Trans 95:295–305

    Article  Google Scholar 

  7. Shao H, Jiang H, Zhao H, Wang F (2017) A novel deep autoencoder feature learning method for rotating machinery fault diagnosis. Mech Syst Signal Process 95:187–204

    Article  Google Scholar 

  8. Han T, Liu C, Yang W, Jiang D (2019) A novel adversarial learning framework in deep convolutional neural network for intelligent diagnosis of mechanical faults. Knowl-Based Syst 165:474–487

    Article  Google Scholar 

  9. Huang W, Cheng J, Yang Y, Guo G (2019) An improved deep convolutional neural network with multi-scale information for bearing fault diagnosis. Neurocomputing 359:77–92

    Article  Google Scholar 

  10. Lei Y, Jia F, Lin J, Xing S, Ding S X (2016) An Intelligent Fault Diagnosis Method Using Unsupervised Feature Learning Towards Mechanical Big Data. IEEE Trans Ind Electron 63(5):3137– 3147

    Article  Google Scholar 

  11. Li Y, Song Y, Jia L, Gao S, Li Q, Qiu M (2021) Intelligent Fault Diagnosis by Fusing Domain Adversarial Training and Maximum Mean Discrepancy via Ensemble Learning. IEEE Trans Ind Inf 17 (4):2833–2841

    Article  Google Scholar 

  12. Ince T, Kiranyaz S, Eren L, Askar M, Gabbouj M (2016) Real-Time Motor Fault Detection by 1-D Convolutional Neural Networks. IEEE Trans Ind Electron 63(11):7067–7075

    Article  Google Scholar 

  13. Xia M, Li T, Xu L, Liu L, de Silva C W (2018) Fault Diagnosis for Rotating Machinery Using Multiple Sensors and Convolutional Neural Networks. IEEE/ASME Trans Mechatron 23(1):101– 110

    Article  Google Scholar 

  14. Wang H, Liu Z, Peng D, Qin Y (2020) Understanding and Learning Discriminant Features based on Multiattention 1DCNN for Wheelset Bearing Fault Diagnosis. IEEE Trans Ind Inf 16(9):5735–5745

    Article  Google Scholar 

  15. shao, Xia M, Han G, Zhang Y, Wan J (2020) Intelligent fault diagnosis of rotor-bearing system under varying working conditions with modified transfer CNN and thermal images. IEEE Trans Ind Inf:1–1

  16. Chang X, Tang B, Tan Q, Deng L, Zhang F (2020) One-dimensional fully decoupled networks for fault diagnosis of planetary gearboxes. Mech Syst Signal Process 141:106482

    Article  Google Scholar 

  17. Zareapoor M, Shamsolmoali P, Yang J (2021) Oversampling adversarial network for class-imbalanced fault diagnosis. Mech Syst Signal Process 149:107175

    Article  Google Scholar 

  18. Zhou F, Yang S, Fujita H, Chen D, Wen C (2020) Deep learning fault diagnosis method based on global optimization GAN for unbalanced data. Knowl-Based Syst 187:104837

    Article  Google Scholar 

  19. Sun J, Li H, Fujita H, Fu B, Ai W (2020) Class-imbalanced dynamic financial distress prediction based on Adaboost-SVM ensemble combined with SMOTE and time weighting. Inf Fusion 54:128–144

    Article  Google Scholar 

  20. Jia F, Lei Y, Lu N, Xing S (2018) Deep normalized convolutional neural network for imbalanced fault classification of machinery and its understanding via visualization. Mech Syst Signal Process 110:349–367

    Article  Google Scholar 

  21. Wang C, Xin C, Xu Z (2021) A novel deep metric learning model for imbalanced fault diagnosis and toward open-set classification. Knowl-Based Syst 220:106925

    Article  Google Scholar 

  22. Tao X, Chen W, Li X, Zhang X, Li Y, Guo J (2021) The ensemble of density-sensitive SVDD classifier based on maximum soft margin for imbalanced datasets. Knowl-Based Syst 219:106897

    Article  Google Scholar 

  23. Mao W, He L, Yan Y, Wang J (2017) Online sequential prediction of bearings imbalanced fault diagnosis by extreme learning machine. Mech Syst Signal Process 83:450–473

    Article  Google Scholar 

  24. Ng W W Y, Hu J, Yeung D S, Yin S, Roli F (2015) Diversified Sensitivity-Based Undersampling for Imbalance Classification Problems. IEEE Trans Cybern 45(11):2402–2412

    Article  Google Scholar 

  25. Zhang W, Li X, Jia X-D, Ma H, Luo Z, Li X (2020) Machinery fault diagnosis with imbalanced data using deep generative adversarial networks. Measurement 152:107377

    Article  Google Scholar 

  26. Li S, Deng W (2020) Deep Facial Expression Recognition: A Survey. IEEE Trans Affect Comput:1–1

  27. An Z, Deng W, Yuan T, Hu J (2018) Deep transfer network with 3D morphable models for face recognition. In: 2018 13th IEEE International Conference on Automatic Face Gesture Recognition (FG 2018), pp 416–422

  28. Deng J, Zhou Y, Zafeiriou S (2017) Marginal loss for deep face recognition. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), pp 2006–2014

  29. Liu W, Wen Y, Yu Z, Yang M (2016) Large-margin softmax loss for convolutional neural networks. In: Proceedings of the 33rd International Conference on International Conference on Machine Learning - Volume 48, ICML’16. JMLR.org, New York, pp 507–516

  30. Wang H, Wang Y, Zhou Z, Ji X, Gong D, Zhou J, Li Z, Liu W (2018) CosFace: Large Margin Cosine Loss for Deep Face Recognition. In: 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. IEEE, Salt Lake City, pp 5265–5274

  31. Deng J, Guo J, Xue N, Zafeiriou S (2019) ArcFace: Additive Angular Margin Loss for Deep Face Recognition. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, Long Beach, pp 4685–4694

  32. Sun J, Lang J, Fujita H, Li H (2018) Imbalanced enterprise credit evaluation with DTE-SBD: Decision tree ensemble based on SMOTE and bagging with differentiated sampling rates. Inf Sci 425:76–91

    Article  MathSciNet  Google Scholar 

  33. He H, Bai Y, Garcia E A, Li S (June 2008) ADASYN: Adaptive synthetic sampling approach for imbalanced learning. In: 2008 IEEE International Joint Conference on Neural Networks (IEEE World Congress on Computational Intelligence), pp 1322–1328

  34. The Case Western Reserve University Bearing Data Center Website

Download references

Acknowledgements

This work is supported by the National Natural Science Foundation of China (No. 51675253).

Author information

Authors and Affiliations

  1. School of Mechanical and Electrical Engineering, Lanzhou University of Technology, LanZhou, 730050, China

    Ziyang Xing, Rongzhen Zhao, Yaochun Wu & Tianjing He

  2. School of Mechanical Engineering, Anyang Institute of Technology, Anyang, 455000, China

    Yaochun Wu

Authors
  1. Ziyang Xing
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Rongzhen Zhao
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Yaochun Wu
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Tianjing He
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Rongzhen Zhao.

Additional information

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Xing, Z., Zhao, R., Wu, Y. et al. Intelligent fault diagnosis of rolling bearing based on novel CNN model considering data imbalance. Appl Intell 52, 16281–16293 (2022). https://doi.org/10.1007/s10489-022-03196-x

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-022-03196-x

Keywords

Search

Navigation