Fault Diagnosis Method for Imbalanced Data Based on Adaptive Diffusion Models and Generative Adversarial Networks

17 Pages Posted: 14 Dec 2024

See all articles by xueyi li

xueyi li

Northeast Forestry University

Xudong Wu

Northeast Forestry University

Tianyang Wang

Tsinghua University

yining xie

Northeast Forestry University

Fulei Chu

Tsinghua University

Abstract

In engineering practice, the challenge of collecting fault samples often leads to data imbalance, significantly affecting the performance of fault diagnosis. Data augmentation methods offer an effective solution to address this issue by supplementing fault data. However, these methods must meet the requirements of generating high-quality samples, achieving comprehensive mode coverage, and ensuring fast sample generation. Traditional Generative Adversarial Networks are prone to mode collapse, while diffusion models suffer from slow generation speeds, making it difficult for either to fully satisfy these demands. To overcome these challenges, this paper proposes a data augmentation method for fault diagnosis based on an adaptive diffusion model integrated with Generative Adversarial Networks. By introducing a length-adaptive forward diffusion chain to generate Gaussian mixture noise, this method not only ensures smoother and more stable gradients but also avoids computational redundancy and gradient vanishing problems. At each diffusion timestep, the discriminator learns to distinguish real and generated data across different noise ratios and timesteps, enhancing the diversity of fault samples and effectively mitigating mode collapse. Experimental results on two datasets demonstrate that the proposed method outperforms other data augmentation techniques in terms of generation efficiency and stability, effectively addressing the data imbalance problem and significantly improving fault diagnosis performance.

Keywords: fault diagnosis, Diffusion Model, Generative Adversarial Network, Data imbalance

Suggested Citation

li, xueyi and Wu, Xudong and Wang, Tianyang and xie, yining and Chu, Fulei, Fault Diagnosis Method for Imbalanced Data Based on Adaptive Diffusion Models and Generative Adversarial Networks. Available at SSRN: https://ssrn.com/abstract=5055225 or http://dx.doi.org/10.2139/ssrn.5055225

Xueyi Li (Contact Author)

Northeast Forestry University ( email )

Harbin
China

Xudong Wu

Northeast Forestry University ( email )

Harbin
China

Tianyang Wang

Tsinghua University ( email )

Beijing, 100084
China

Yining Xie

Northeast Forestry University ( email )

Harbin
China

Fulei Chu

Tsinghua University ( email )

Beijing, 100084
China

Do you have a job opening that you would like to promote on SSRN?

Paper statistics

Downloads
30
Abstract Views
166
PlumX Metrics