EAAI Journal 2026 Journal Article
A novel model-data-driven sample generation approach for bearing fault diagnosis under imbalanced data conditions
- Zhimin Wang
- Wei Yu
- Jianwu Wang
- Lizhang Cheng
- Chuanchuan Shao
Bearing fault diagnosis is vital for equipment reliability, but the scarcity of fault data in industrial settings hinders the performance and generalization of diagnostic models. To address this issue, a model-data-driven sample generation (MDDSG) framework is proposed that integrates physics-based modeling with data-driven augmentation. First, a nonlinear dynamic model is constructed to simulate vibration signals under various fault types, thereby enriching the sample space. Then, a feature spectrum-based generation criterion is introduced, which utilizes healthy signals as prior knowledge to extract device-specific features. These features are integrated with model-generated fault characteristics to guide realistic sample generation and reduce distribution discrepancies. To further enhance data diversity and authenticity, a threshold-guided Generative Adversarial Network is employed, incorporating an adaptive similarity-based training strategy. Finally, training on the generated dataset yields robust diagnostic performance, with average accuracies of 98. 35% on the synthetic dataset and 96. 95% on the real-world dataset. These results validate the ability of MDDSG to overcome data imbalance and support robust industrial fault diagnosis.