Computer Science ›› 2025, Vol. 52 ›› Issue (11): 213-222.doi: 10.11896/jsjkx.250300117

• Artificial Intelligence • Previous Articles     Next Articles

Multi-source Domain Generalization Fault Diagnosis Method Based on Instance-level PromptGeneration

LI Shugang1, LI Mingjia1, YUAN Longhui1, QI Guangpeng2,3, LIU Chi1   

  1. 1 School of Computer Science and Technology,Beijing Institute of Technology,Beijing 100081,China
    2 INSPUR Group Co.,Ltd.,Jinan 250101,China
    3 INSPUR Yunzhou Industrial Internet Co.,Ltd.,Jinan 250098,China
  • Received:2025-03-24 Revised:2025-05-14 Online:2025-11-15 Published:2025-11-06
  • About author:LI Shugang,born in 1996,Ph.D candidate.His main research interests include fault diagnosis,transfer learning and object detection.
    LIU Chi,born in 1984,Ph.D,professor,Ph.D.supervisor,is a distinguished member of CCF(No.33627D).His main research interest is big data processing for Internet of Things.

Abstract: This paper proposes multi-source domain generalization fault diagnosis method based on instance-level prompt generation to enhance the model's fault recognition capability in cross-domain environments.This method employs a cross-frequency aligned prompt generator to dynamically generate instance-level prompts,enabling refined modeling of local features across diffe-rent samples.It incorporates a semantic consistency enhancement module to ensure the semantic validity of instance-level prompts.Furthermore,to improve the model's stability and adaptability in cross-domain tasks,a memory bank-enhanced contrastive learning module is introduced,which fully utilizes cross-domain positive and negative samples.By storing and dynamically updating feature representations of training samples,this module expands the diversity of positive and negative sample distributions and enhances the effectiveness of cross-domain feature learning.Additionally,a FourierMix module is adopted to perform frequency-domain feature mixing of samples from different source domains,dynamically generating simulated samples to strengthen the model's adaptability on unseen target domains.Experimental results on CWRU and Paderborn datasets demonstrate that the proposed method outperforms existing approaches across multiple unseen target domains,achieving average classification accuracies of 93.54%(1.52% improvement over state-of-the-art) on CWRU dataset and 90.52% (1.30% improvement) on Paderborn dataset.Experimental results prove its effectiveness and robustness in industrial fault diagnosis tasks.

Key words: Fault diagnosis, Prompt learning, Multi-source domain generalization, Transfer learning, Contrastive learning

CLC Number: 

  • TP181
[1]FANG W,ZHOU B,WANG K,et al.A Fault Diagnosis and Fault-Tolerant Control Method for Open-Switch Faults of Converters in DSEM Drive System [J].IEEE Transactions on Industrial Electronics,2024,71(8):8459-8470.
[2]JIN H,ZUO Z,WANG Y,et al.Small Fault Diagnosis With Gap Metric [J].IEEE Transactions on Systems,Man,and Cyberne-tics:Systems,2023,53(9):5715-5728.
[3]LYU K,WANG D,HUANG W,et al.Research on Fault Indicator for Integrated Fault Diagnosis System of PMSM Based on Stator Tooth Flux [J].IEEE Journal of Emerging and Selected Topics in Power Electronics,2024,12(1):985-996.
[4]LI S,BU R,LI S,et al.Principal Properties Attention Matching for Partial Domain Adaptation in Fault Diagnosis [J].IEEE Transactions on Instrumentation and Measurement,2024,73:1-12.
[5]ZHU Z,ZHAI W,LIU H,et al.Juggler-ResNet:A Flexible and High-Speed ResNet Optimization Method for Intrusion Detection System in Software-Defined Industrial Networks [J].IEEE Transactions on Industrial Informatics,2022,18(6):4224-4233.
[6]JIAO J,LIANG K,DING C,et al.Towards Prediction Con-straints:A Novel Domain Adaptation Method for Machine Fault Diagnosis [J].IEEE Transactions on Industrial Informatics,2022,18(10):7198-7207.
[7]XIAO G,PENG S,XIANG W,et al.CMFT:Contrastive Memory Feature Transfer for Nonshared-and-Imbalanced Unsupervised Domain Adaption [J].IEEE Transactions on Industrial Informatics,2023,19(8):9227-9238.
[8]SONG Y,LI Y,JIA L,et al.Retraining Strategy-Based Domain Adaption Network for Intelligent Fault Diagnosis [J].IEEE Transactions on Industrial Informatics,2020,16(9):6163-6171.
[9]LI S,XUAN J,WANG Z,et al.Noisy Open Set Adversarial Domain Adaption for Bearing Fault Diagnosis Based on Optimized Divergence [J].IEEE Transactions on Instrumentation and Measurement,2024,73:1-17.
[10]CHEN L,LI Q,SHEN C,et al.Adversarial Domain-InvariantGeneralization:A Generic Domain-Regressive Framework for Bearing Fault Diagnosis Under Unseen Conditions [J].IEEE Transactions on Industrial Informatics,2022,18(3):1790-1800.
[11]LI Y,TIAN X,GONG M,et al.Deep domain generalization via conditional invariant adversarial networks[C]//Proceedings of the European Conference on Computer Vision.2018:624-639.
[12]LU F,TONG Q,JIANG X,et al.Prior knowledge embedding convolutional autoencoder:A single-source domain generalized fault diagnosis framework under small samples [J].Computers in Industry,2025,164:104169.
[13]RAGAB M,CHEN Z,ZHANG W,et al.Conditional contrastive domain generalization for fault diagnosis [J].IEEE Transactions on Instrumentation and Measurement,2022,71:1-12.
[14]FAN Z,XU Q,JIANG C,et al.Deep Mixed Domain Generalization Network for Intelligent Fault Diagnosis Under Unseen Conditions [J].IEEE Transactions on Industrial Electronics,2024,71(1):965-974.
[15]HAN T,LI Y,QIAN M,et al.A Hybrid Generalization Network for Intelligent Fault Diagnosis of Rotating Machinery Under Unseen Working Conditions [J].IEEE Transactions on Instrumentation and Measurement,2021,70:1-11.
[16]QIAN Q,ZHOU J,QIN Y,et al.Relationship Transfer Domain Generalization Network for Rotating Machinery Fault Diagnosis Under Different Working Conditions [J].IEEE Transactions on Industrial Informatics,2023,19(9):9898-9908.
[17]LI J,WANG Y,ZI Y,et al.Causal Consistency Network:A Collaborative Multimachine Generalization Method for Bearing Fault Diagnosis [J].IEEE Transactions on Industrial Informa-tics,2023,19(4):5915-5924.
[18]XIE S,ZHENG Z,CHEN L,et al.Learning semantic representations for unsupervised domain adaptation[C]//Proceedings of the International Conference on Machine Learning.PMLR,2018:5423-5432.
[19]PEI Z,CAO Z,LONG M,et al.Multi-adversarial domain adaptation[C]//Proceedings of the AAAI Conference on Artificial Intelligence.AAAI,2018.
[20]WANG Z,RAO Y,YU X,et al.Point-to-Pixel Prompting for Point Cloud Analysis With Pre-Trained Image Models [J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2024,46(6):4381-4397.
[21]YANG L,LI X,WANG Y,et al.Fine-Grained Visual TextPrompting [J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2025,47(3):1594-1609.
[22]LI S,LI B,SUN B,et al.Towards Visual-Prompt Temporal Answer Grounding in Instructional Video [J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2024,46(12):8836-8853.
[23]ZHU P,WANG X,ZHU L,et al.Prompt-Based Learning forUnpaired Image Captioning [J].IEEE Transactions on Multimedia,2024,26:379-393.
[24]FANG H,XIONG P,XU L,et al.Transferring Image-CLIP to Video-Text Retrieval via Temporal Relations [J].IEEE Transactions on Multimedia,2023,25:7772-7785.
[25]ZHOU K,YANG J,LOY C C,et al.Learning to prompt forvision-language models [J].International Journal of Computer Vision,2022,130(9):2337-2348.
[26]VASWANI A,SHAZEER N,PARMAR N,et al.Attention isall you need[C]//Proceedings of the 31st International Confe-rence on Neural Information Processing Systems.2017:6000-6010.
[27]LI H,PAN S J,WANG S,et al.Domain generalization with adversarial feature learning[C]//Proceedings of the IEEE Confe-rence on Computer Vision and Pattern Recognition.IEEE,2018:5400-5409.
[28]LYU F,LIANG J,LI S,et al.Improving generalization with domain convex game[C]//Proceedings of the IEEE/CVF Confe-rence on Computer Vision and Pattern Recognition.IEEE,2023:24315-24324.
[29]VAN DER MAATEN L,HINTON G.Visualizing data usingt-SNE [J].Journal of Machine Learning Research,2008,9(86):2579-2605.
[1] ZENG Lili, XIA Jianan, LI Shaowen, JING Maike, ZHAO Huihui, ZHOU Xuezhong. M2T-Net:Cross-task Transfer Learning Tongue Diagnosis Method Based on Multi-source Data [J]. Computer Science, 2025, 52(9): 47-53.
[2] HUANG Chao, CHENG Chunling, WANG Youkang. Source-free Domain Adaptation Method Based on Pseudo Label Uncertainty Estimation [J]. Computer Science, 2025, 52(9): 212-219.
[3] CAI Qihang, XU Bin, DONG Xiaodi. Knowledge Graph Completion Model Using Semantically Enhanced Prompts and Structural Information [J]. Computer Science, 2025, 52(9): 282-293.
[4] CHENG Zhangtao, HUANG Haoran, XUE He, LIU Leyuan, ZHONG Ting, ZHOU Fan. Event Causality Identification Model Based on Prompt Learning and Hypergraph [J]. Computer Science, 2025, 52(9): 303-312.
[5] ZHANG Shiju, GUO Chaoyang, WU Chengliang, WU Lingjun, YANG Fengyu. Text Clustering Approach Based on Key Semantic Driven and Contrastive Learning [J]. Computer Science, 2025, 52(8): 171-179.
[6] LI Maolin, LIN Jiajie, YANG Zhenguo. Confidence-guided Prompt Learning for Multimodal Aspect-level Sentiment Analysis [J]. Computer Science, 2025, 52(7): 241-247.
[7] ZHANG Taotao, XIE Jun, QIAO Pingjuan. Specific Emitter Identification Based on Progressive Self-training Open Set Domain Adaptation [J]. Computer Science, 2025, 52(7): 279-286.
[8] LI Bo, MO Xian. Application of Large Language Models in Recommendation System [J]. Computer Science, 2025, 52(6A): 240400097-7.
[9] CHEN Qirui, WANG Baohui, DAI Chencheng. Research on Electrocardiogram Classification and Recognition Algorithm Based on Transfer Learning [J]. Computer Science, 2025, 52(6A): 240900073-8.
[10] LI Mingjie, HU Yi, YI Zhengming. Flame Image Enhancement with Few Samples Based on Style Weight Modulation Technique [J]. Computer Science, 2025, 52(6A): 240500129-7.
[11] LI Jianghui, DING Haiyan, LI Weihua. Prediction of Influenza A Antigenicity Based on Few-shot Contrastive Learning [J]. Computer Science, 2025, 52(6A): 240800053-6.
[12] YE Jiale, PU Yuanyuan, ZHAO Zhengpeng, FENG Jue, ZHOU Lianmin, GU Jinjing. Multi-view CLIP and Hybrid Contrastive Learning for Multimodal Image-Text Sentiment Analysis [J]. Computer Science, 2025, 52(6A): 240700060-7.
[13] FU Shufan, WANG Zhongqing, JIANG Xiaotong. Zero-shot Stance Detection in Chinese by Fusion of Emotion Lexicon and Graph ContrastiveLearning [J]. Computer Science, 2025, 52(6A): 240500051-7.
[14] HUANG Bocheng, WANG Xiaolong, AN Guocheng, ZHANG Tao. Transmission Line Fault Identification Method Based on Transfer Learning and Improved YOLOv8s [J]. Computer Science, 2025, 52(6A): 240800044-8.
[15] LIU Yuanhong, WU Yubin. Local Linear Embedding Algorithm Based on Probability Model and Information Entropy [J]. Computer Science, 2025, 52(6A): 240500021-8.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!