Selective Adversarial Augmentation Network for Bearing Fault Diagnosis with Partial Domain Adaptation

Condition monitoring of rotating machinery is critical for ensuring industrial safety and operational reliability. As a core component of intelligent diagnostic systems, domain adaptation methods have achieved notable progress in mechanical fault diagnosis. However, most existing approaches presume a fully shared label space between source and target domains, limiting their effectiveness under partial domain adaptation scenarios commonly encountered in industrial practice. In addition, they often struggle with classification uncertainty near decision boundaries. To address these challenges, this paper proposes a Selective Adversarial Augmentation Network (SAAN) for cross-domain rolling bearing fault diagnosis with partial label space alignment. The proposed framework designs a multi-level feature extraction module to enhance transferable feature representation and a Balanced Augmentation Selective Adversarial Module (BASAM) to dynamically balance class distributions and selectively filter irrelevant source classes, thereby mitigating negative transfer and achieving fine-grained class alignment. Furthermore, an uncertainty suppression mechanism is put forth to reinforce classifier boundaries by minimizing the impact of ambiguous samples. Comprehensive experiments conducted on public and proprietary bearing datasets demonstrate that SAAN consistently surpasses state-of-the-art benchmarks in diagnostic accuracy and robustness, providing an effective solution for practical applications under class-imbalanced and variable operating conditions.