Diffusion model-assisted cross-domain fault diagnosis for rotating machinery under limited data

Yongchao Zhang, Zhiyuan Wang, Caizi Fan, Zeyu Jiang, Kun Yu, Zhaohui Ren and Ke Feng

Reliability Engineering and System Safety, 2025, vol. 264, issue PB

Abstract: In industrial scenarios, cross-domain fault diagnosis faces the challenge of data scarcity due to the difficulty of data acquisition and the high cost of labeling. To overcome this issue, this paper proposes a diffusion model-assisted data generation method to enhance the model’s cross-domain diagnostic capability by generating target domain data. Specifically, this paper establishes a diffusion model-assisted cross-domain fault diagnosis method, where a diffusion model is first constructed to augment the target domain data, and then a deep learning model is jointly trained using source domain data, a small amount of target domain data, and the generated target domain data to learn and transfer diagnostic knowledge. To align the global feature distributions, the maximum mean discrepancy loss is first employed to align the source domain data with both the target domain data and the generated target domain data. Additionally, a cross-domain triplet loss is established to achieve category alignment and separation, ensuring similar categories are aligned while different categories are distinguished. Finally, the deep consistency regularization is designed to enforce consistency across target domain data and its augmented versions, enhancing the model’s robustness. Extensive experiments on two rotating machinery systems demonstrate the effectiveness of the proposed method in addressing limited-data cross-domain fault diagnosis, highlighting its potential for practical applications in intelligent health monitoring of rotating machinery.

Keywords: Fault diagnosis; Diffusion model; Domain adaptation; Rotating machinery; Maximum mean discrepancy (search for similar items in EconPapers)
Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832025005733
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:264:y:2025:i:pb:s0951832025005733

DOI: 10.1016/j.ress.2025.111372

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().