A poisson flow-based data augmentation and lightweight diagnosis framework for imbalanced rolling bearing faults
Xin Liu,
Han Wang,
Zhiyong Du,
Xu Xu and
Bo Song
PLOS ONE, 2025, vol. 20, issue 10, 1-22
Abstract:
Accurate diagnosis of rolling bearing faults is vital for the safe operation of rotating machinery. However, real-world fault datasets often suffer from severe class imbalance, which hinders the performance of deep learning models. To address this challenge, we propose PFRNet, a novel diagnostic framework integrating a Poisson Flow-based generative model with a lightweight residual network. Raw vibration signals are transformed into time-frequency representations via CWT to capture non-stationary fault features. The Poisson generative mechanism models sample evolution in high-dimensional latent space to synthesize realistic minority-class samples by learning statistical distributions of real data, mitigating imbalance. These augmented datasets are subsequently classified using an efficient residual network designed for robust feature extraction with minimal complexity. Experiments on the CWRU benchmark demonstrate that PFRNet outperforms state-of-the-art methods in diagnostic accuracy, robustness, and generalization across various imbalance scenarios. Quantitative evaluations further confirm that the generated samples closely resemble real data in both quality and diversity, supporting the effectiveness of the proposed method. The proposed approach offers a promising solution for reliable fault diagnosis under practical, imbalance-prone industrial conditions.
Date: 2025
References: Add references at CitEc
Citations:
Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0332994 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 32994&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:0332994
DOI: 10.1371/journal.pone.0332994
Access Statistics for this article
More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().