Multi-target Domain Adaptation Intelligent Diagnosis Method for Rotating Machinery Based on Multi-Source Attention Mechanism and Mixup Feature Augmentation

Mengyu Liu,Zhe Cheng,Yu Yang,Niaoqing Hu,Yi Yang
DOI: https://doi.org/10.1016/j.ress.2024.110298
IF: 7.247
2024-01-01
Reliability Engineering & System Safety
Abstract:Intelligent diagnostic methods for identifying faults in rotating machinery, based on domain adaptation, have garnered significant attention. However, most current domain adaptation approaches are primarily designed for single-source domain and single-target domain (SSST) applications. There is a dearth of domain adaptation approaches tailored for single-source to multi-target domains (SSMT). In contrast to SSST, SSMT takes a more comprehensive approach by considering relationships across multiple target domains. This approach offers increased versatility and a broader range of potential applications. To address this, an end-to-end multi-target adversarial subdomain adaptation method is proposed that leverages attention mechanism data fusion and mixup feature augmentation. Firstly, the attention mechanism is used to fuse data from different sensors in both channel and spatial dimensions. Subsequently, a mixup-based feature augmentation method is proposed for multi-target domain adaptation. The method is combined with subdomain adaptation and domain discrimination to further reduce the distributional differences between the source and various target domains while relieving the overfitting problem during domain adaptation. Finally, with the above approach, a robust and stable model for multiple target domain fault diagnosis can be trained. Our experimental results illustrate that our approach has higher accuracy and robustness compared to several popular domain adaptation methods.
What problem does this paper attempt to address?