Fault detection of high-speed train wheelset bearings based on improved auxiliary classifier generative adversarial networks and VAE

Jiandong Qiu, Jiaxuan Liu, Minan Tang, Dingwang Zhang and Meng Li

PLOS ONE, 2025, vol. 20, issue 10, 1-38

Abstract: Fault detection in high-speed train wheelset bearings is paramount for ensuring operational safety. However, the scarcity of fault samples limits the accuracy of traditional detection methods. To address this challenge, this paper proposes a supervised generative model that integrates an Improved Auxiliary Classifier Generative Adversarial Network (IACGAN) with a Variational Auto-Encoder (VAE). Firstly, the method employs the VAE as the generator, introducing latent variables with prior information to optimize the encoding and generation process; Secondly, an independent classifier network is integrated into the ACGAN framework to enhance compatibility between classification and discriminative capabilities. Concurrently, a loss function incorporating Wasserstein distance and gradient penalty terms is designed to prevent gradient vanishing during training while satisfying Lipschitz constraints, thereby improving model stability. Experiments conducted on the XJTU bearing dataset validate that samples generated by the proposed method demonstrate superior quality assessment at both the data and feature levels compared to several GAN variants. Furthermore, the constructed VAE-IACGAN-CNN detection model achieves an average classification accuracy of 88.04%, representing a maximum improvement of 15.17% over comparative methods. This significantly mitigates accuracy degradation caused by sample imbalance, demonstrating the proposed approach’s efficacy in resolving low fault detection accuracy stemming from imbalanced high-speed train wheelset bearing samples.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0335368 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 35368&type=printable (application/pdf)

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:plo:pone00:0335368

DOI: 10.1371/journal.pone.0335368

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().