Small sample image generation method applied to gearbox fault diagnosis
DOI:
CSTR:
Author:
Affiliation:

1.School of Artificial Intelligence, China University of Mining and Technology (Beijing), Beijing 100083, China; 2.Anhui Industrial Internet Intelligence Application and Security Engineering Research Center, Ma′anshan 243023, China

Clc Number:

TH132;TN919.8

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Planetary gearboxes are widely used as essential transmission devices in industrial applications, yet they are prone to failures under complex operating conditions and prolonged loads. Traditional fault diagnosis methods heavily rely on expert knowledge and expensive equipment, facing challenges such as data scarcity and low diagnostic efficiency. To address these limitations, the development of generative adversarial networks (GANs) has provided innovative solutions for image generation and data augmentation in recent years. However, existing GAN models often encounter issues such as semantic misalignment and artifacts when processing small-sample datasets, limiting their potential in intelligent fault diagnosis. In this context, this paper proposes a multi-scale attention and progressive feature fusion GAN (MSA-PF-GAN) model, which integrates a progressive decoder structure with multi-scale attention mechanisms to significantly improve image generation quality and fault diagnosis accuracy under small-sample conditions. Experiments conducted on two independent planetary gearbox fault datasets validate the proposed method, showing that it substantially reduces the FID score and enhances diagnostic accuracy (by 35% and 20%, respectively). Across multiple evaluation metrics, the MSA-PF-GAN outperforms other state-of-the-art methods. Further analysis demonstrates that the model, through progressive feature fusion and multi-scale attention mechanisms, excels in generating diverse and realistic images while effectively capturing complex fault features. Therefore, this technique shows promising potential and practical value in the field of planetary gearbox fault diagnosis.

    Reference
    Related
    Cited by
Get Citation
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: May 16,2025
  • Published:
Article QR Code