A bearing fault data augmentation method based on hybrid-diversity loss diffusion model and parameter transfer

Yuan Wei, Zhijun Xiao, Xiangyan Chen, Xiaohui Gu and Schrö der, Kai-Uwe

Reliability Engineering and System Safety, 2025, vol. 253, issue C

Abstract: The fault diagnosis of mechanical equipment can prevent potential mechanical failures, avoid property damage and personal injury, and ensure the stable and safe operation of mechanical equipment. Data driven is an important aspect of intelligent fault diagnosis. When data is scarce, it can seriously affect the accuracy of fault diagnosis and make it difficult to ensure the smooth and safe operation of machinery. Faced with this challenge, a classifier-free guidance diffusion model combining hybrid loss and diversity loss (CFGDMHD) is proposed for data augmentation of fault samples. This new data augmentation method generates samples with the same data distribution as real samples from random noise through diffusion process. CFGDMHD can generate multi-class samples simultaneously without the need for additional classifier guidance in the joint training of unconditional diffusion models and conditional diffusion models. This work proposes diversity loss to improve the diversity of generated samples. We conducted experiments using a bearing dataset. The results indicate that the sample quality and diversity generated by this method are excellent, which can help improve the accuracy of fault diagnosis and ensure the safe operation of mechanical systems.

Keywords: Data augmentation; Diffusion model; CFGDMHD; Fault diagnosis (search for similar items in EconPapers)
Date: 2025
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832024006392
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:253:y:2025:i:c:s0951832024006392

DOI: 10.1016/j.ress.2024.110567

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 ().