收稿日期: 2023-11-16
网络出版日期: 2024-06-25
基金资助
国家自然科学基金项目(52065064);新疆维吾尔自治区自然科学基金青年基金项目(2022D01C367)
Meta-DAE Fault Diagnosis Based on Prototype Domain Enhancement in Few-Shot
Received date: 2023-11-16
Online published: 2024-06-25
Supported by
the National Natural Science Foundation of China(52065064)
滚动轴承作为一种精密的机械元件,已广泛运用于现代工业机械设备中。在轴承运行时,采用合理的方法诊断轴承的故障具有重大的意义。但在实际复杂多变环境下,采集振动信号不仅面临样本量少的问题,还受到噪声干扰、工况变换等因素的影响,导致故障诊断的准确率低。因此,针对噪声干扰和变工况下的小样本滚动轴承故障诊断问题,该文提出了一种基于原型域增强的元学习去噪模型(Meta-DAE)。首先,构造基于时频图的小样本故障样本集,引入深度卷积生成对抗网络并对数据进行预处理,生成相似分布的伪样本集;然后,将故障样本集输入Meta-DAE模型进行自适应特征提取,Meta-DAE模型采用原型域增强策略,使同类别原型点在嵌入空间中凝聚更紧密;同时,构建了具有降噪性能的编码器,设计了基于原型域增强和去噪的目标函数,通过在小样本下进行模型微调,以提高小样本下模型的噪声鲁棒性和分类准确率。噪声及变工况下小样本故障诊断实验结果表明,相比于其他模型,所提模型在-8 dB强噪声干扰下,仅用10个样本微调模型,分类准确率提高了35.78~57.25个百分点,具有较强的噪声鲁棒性。
马萍 , 梁城 , 王聪 , 李新凯 , 张宏立 . 小样本下基于原型域增强的Meta-DAE故障诊断[J]. 华南理工大学学报(自然科学版), 2025 , 53(1) : 62 -73 . DOI: 10.12141/j.issn.1000-565X.230715
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.
| 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 Magazine,2020,14(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 Applications,2011,38(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 Processing,2020,138: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 Systems,2022,235:107646/1-27. |
| 5 | HOANG D T, KANG J .A survey on deep learning based bearing fault diagnosis[J].Neurocomputing,2019,335: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].Neurocomputing,2018,272: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 Electronics,2017,65(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].Sensors,2017,17(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 Informatics,2019,16(7):4681-4690. |
| 10 | 万若青,张纯,江汇强,等 .基于深度自编码器的振动信号盲去噪方法[J].振动与冲击,2023,42(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 Shock,2023,42(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 Transactions,2022,121:327-348. |
| 12 | 刘飞,陈仁文,邢凯玲,等 .基于迁移学习与深度残差网络的滚动轴承快速故障诊断算法[J].振动与冲击,2022,41(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 Shock,2022,41(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 Recognition,2022,123: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 Electronics,2022,69(10):10573-10584. |
| 15 | 强睿儒,赵小强 .基于格拉姆角差场和生成对抗网络的小样本滚动轴承故障诊断方法[J].华南理工大学学报(自然科学版),2024,52(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),2024,52(10):64-75. | |
| 16 | 胡若晖,张敏,许文鑫 .基于DCGAN和DANN网络的滚动轴承跨域故障诊断[J].振动与冲击,2022,41(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 Shock,2022,41(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 Research,2021,140: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 Intelligence,2021,44(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].Neurocomputing,2021,439: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 Processing,2021,155: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 Transactions,2022,120: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 Processing,2014,96: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 Informatics,2020,16(10):6263-6271. |
/
| 〈 |
|
〉 |