Electronics, Communication & Automation Technology

Meta-DAE Fault Diagnosis Based on Prototype Domain Enhancement in Few-Shot

  • MA Ping ,
  • LIANG Cheng ,
  • WANG Cong ,
  • LI Xinkai ,
  • ZHANG Hongli
Expand
  • School of Electrical Engineering,Xinjiang University,Urumqi 830017,Xinjiang,China
马萍(1994—),女,博士,副教授,主要从事机械系统动态建模与信号处理、智能故障诊断与寿命预测研究。E-mail: maping@xju.edu.cn

Received date: 2023-11-16

  Online published: 2024-06-25

Supported by

the National Natural Science Foundation of China(52065064)

Abstract

Rolling bearings, as a type of precision mechanical component, are widely used in modern industrial machinery and equipment. It is of great significance to diagnose bearing faults using reasonable methods during bea-ring operation. However, in the actual complex and ever-changing environment, the collection of vibration signals often faces challenges such as limited sample sizes, noise interference, and operating condition variations, resulting in low fault diagnosis accuracy. To address the problem of small-sample rolling bearing fault diagnosis under noise interference and variable operating conditions, this paper proposed a meta-learning denoising model based on prototype domain enhancement (Meta-DAE). Firstly, a small-sample fault dataset based on time-frequency diagrams was constructed, and a deep convolutional generative adversarial network was introduced for data preprocessing to gene-rate a pseudo-sample set with a similar distribution. Then, the fault sample set was input into Meta-DAE for adaptive feature extraction. Meta-DAE adopts a prototype domain enhancement strategy to make prototype points of the same category more closely clustered in the embedding space. At the same time, an encoder with noise reduction performance was constructed, and a target function based on prototype domain enhancement and denoising was designed. By fine-tuning the model under small-sample conditions, the noise robustness and classification accuracy of the model were improved. Experimental results of small-sample fault diagnosis under noise interference and variable operating conditions show that, compared to other models, the proposed model demonstrates strong noise robustness. Under -8 dB strong noise interference, the model achieves a classification accuracy improvement of 35.78% to 57.25% using only 10 samples for fine-tuning.

Cite this article

MA Ping , LIANG Cheng , WANG Cong , LI Xinkai , ZHANG Hongli . Meta-DAE Fault Diagnosis Based on Prototype Domain Enhancement in Few-Shot[J]. Journal of South China University of Technology(Natural Science), 2025 , 53(1) : 62 -73 . DOI: 10.12141/j.issn.1000-565X.230715

References

1 HU X S, ZHANG K, LIU K L,et al .Advanced fault diagnosis for lithium-ion battery systems:a review of fault mechanisms,fault features,and diagnosis procedures[J].IEEE Industrial Electronics Magazine202014(3):65-91.
2 KANKAR P K, SHARMA S C, HARSHA S P .Fault diagnosis of ball bearings using machine learning methods[J].Expert Systems with Applications201138(3):1876-1886.
3 LEI Y G, YANG B, JIANG X W,et al .Applications of machine learning to machine fault diagnosis:a review and roadmap[J].Mechanical Systems and Signal Processing2020138:106587/1-39.
4 FENG Y, CHEN J L, XIE J S,et al .Meta-learning as a promising approach for few-shot cross-domain fault diagnosis:algorithms,applications,and prospects[J].Knowledge-Based Systems2022235:107646/1-27.
5 HOANG D T, KANG J .A survey on deep learning based bearing fault diagnosis[J].Neurocomputing2019335:327-335.
6 JIA F, LEI Y, GUO L,et al .A neural network constructed by deep learning technique and its application to intelligent fault diagnosis of machines[J].Neurocomputing2018272:619-628.
7 WEN L, LI X Y, GAO L,et al .A new convolutional neural network-based data-driven fault diagnosis method[J].IEEE Transactions on Industrial Electronics201765(7):5990-5998.
8 ZHANG W, PENF G L, LI C H,et al .A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals[J].Sensors201717(2):425/1-21.
9 ZHAO M H, ZHONG S S, FU X Y,et al .Deep residual shrinkage networks for fault diagnosis[J].IEEE Transactions on Industrial Informatics201916(7):4681-4690.
10 万若青,张纯,江汇强,等 .基于深度自编码器的振动信号盲去噪方法[J].振动与冲击202342(12):118-125.
  WAN Ruoqing, ZHANG Chun, JIANG Huiqiang,et al .A blind denoising method of vibration signals based on a deep autoencoder[J].Journal of Vibration and Shock202342(12):118-125.
11 DONG Y J, LI Y Q, ZHENG H L,et al .A new dynamic model and transfer learning based intelligent fault diagnosis framework for rolling element bearings race faults:solving the small sample problem[J].ISA Transactions2022121:327-348.
12 刘飞,陈仁文,邢凯玲,等 .基于迁移学习与深度残差网络的滚动轴承快速故障诊断算法[J].振动与冲击202241(3):154-164.
  LIU Fei, CHEN Renwen, XING Kailing,et al .Fast fault diagnosis algorithm for rolling bearings based on transfer learning and deep residual network[J].Journal of Vibration and Shock202241(3):154-164.
13 CHEN H, LIU R N, XIE Z X,et al .Majorities help minorities:hierarchical structure guided transfer lear-ning for few-shot fault recognition[J].Pattern Recognition2022123:108383/1-15.
14 ZHANG T C, CHEN J L, HE S L,et al .Prior knowledge-augmented self-supervised feature learning for few-shot intelligent fault diagnosis of machines[J].IEEE Transactions on Industrial Electronics202269(10):10573-10584.
15 强睿儒,赵小强 .基于格拉姆角差场和生成对抗网络的小样本滚动轴承故障诊断方法[J].华南理工大学学报(自然科学版)202452(10):64-75.
  QIANG Ruiru, ZHAO Xiaoqiang.Fault diagnosis method for rolling bearings based on Gram angular field and generative adversarial networks[J].Journal of South China University of Tchnology(Natural Science Edition)202452(10):64-75.
16 胡若晖,张敏,许文鑫 .基于DCGAN和DANN网络的滚动轴承跨域故障诊断[J].振动与冲击202241(6):21-29.
  HU Ruohui, ZHANG Min, XU Wenxin .Cross-domain fault diagnosis of rolling element bearings using DCGAN and DANN[J].Journal of Vibration and Shock202241(6):21-29.
17 LEI T Y, HU H H, LUO Q Y,et al .Adaptive meta-learner via gradient similarity for few-shot text classification[EB/OL].(2022-09-10)[2023-06-20]..
18 CHEN Y D, GUAN C Y, WEI Z K,et al .Meta-Delta:a meta-learning system for few-shot image cla-ssification[J].Proceedings of Machine Learning Research2021140:17-28.
19 HOSPEDALES T, ANTREAS A, MICAELL P,et al .Meta-learning in neural networks:a survey[J].IEEE Transactions on Pattern Analysis and Machine Intelligence202144(9):5149-5169.
20 LI C J, LI S B, ZHANG A S,et al .Meta-learning for few-shot bearing fault diagnosis under complex working conditions[J].Neurocomputing2021439:197-211.
21 WANG D, ZHANG M, XU Y C,et al .Metric-based meta-learning model for few-shot fault diagnosis under multiple limited data conditions[J].Mechanical Systems and Signal Processing2021155:107510/1-15.
22 FENG Y, CHEN J L, ZHANG T C,et al .Semi-supervised meta-learning networks with squeeze-and-excitation attention for few-shot fault diagnosis[J].ISA Transactions2022120:383-401.
23 YAN R Q, GAO R X, CHEN X F .Wavelets for fault diagnosis of rotary machines:a review with applications [J].Signal Processing201496:1-15.
24 RADFORD A, METZ L, CHINTALA S .Unsupervised representation learning with deep convolutional generative adversarial networks[EB/OL].(2015-11-19)[2023-06-20]..
25 SNELL J, SWERSKY K, ZEMEL R .Prototypical networks for few-shot learning[C]∥ Proceedings of the 31st International Conference on Neural Information Processing Systems.Long Beach:ACM,2017:4080-4090.
26 SAUFI S R, AHMAD Z A B, LEONG M S,et al .Gearbox fault diagnosis using a deep learning model with limited data sample[J].IEEE Transactions on Industrial Informatics202016(10):6263-6271.
Outlines

/