EAAI Journal 2025 Journal Article
A global information-guided denoising diffusion probabilistic model for fault diagnosis with imbalanced data
- Han Yang
- Yan Song
- Daichao Wang
- Jinghao Xing
- Yibin Li
Data imbalance is a significant challenge in intelligent fault diagnosis. Diffusion models offer potential solutions to data imbalance by generating samples from limited data. However, their limited global inference capability can create discrepancies between the generated and real data. To address this issue, this paper introduces a global information-guided denoising diffusion probabilistic model (GI-DDPM) for intelligent fault diagnosis. GI-DDPM employs a pretrained Vision Transformer (ViT) as a guide for global information. It extracts the global information from the data through ViT and aligns it with features extracted by the diffusion model. The cosine similarity between the two sets of features is calculated and used as a regularization term to guide the diffusion model to pay more attention to the comprehensive features in the data. This approach encourages the model to generate fault data with more comprehensive representations. The proposed method is tested on two datasets with different balance ratios for imbalanced fault diagnosis experiments, achieving average accuracy rates of 96. 95%, 97. 18% and 96. 06%. These results outperform other intelligent methods, demonstrating significant potential in handling imbalanced fault diagnosis.