An Enhanced Partial Transfer Fault Diagnosis Network Aided by Dual-Force Boundary Refinement and Inverse-Forward Adaptive Alignment

Yannan Yu,Zhuyun Chen,Jaspreet Singh Dhupia,Lihua Tang,Zhiping Liu,Jiawei Xiang
DOI: https://doi.org/10.1109/jsen.2024.3505792
IF: 4.3
2024-01-01
IEEE Sensors Journal
Abstract:Partial domain adaptation (PDA) is better suited to address the practical challenges of transfer fault diagnosis in rotating machinery. However, existing PDA methods often suffer from two key issues: the propagation of uncertainty caused by high prediction scores for incorrect source domain classes, and instability in label distribution alignment due to relying on only low-weight filtering by the classifier. To overcome these issues, this paper proposes an enhanced partial transfer fault diagnosis network with dual-force boundary refinement (DFBR) and inverse-forward adaptive alignment (IFAA). Firstly, the one-dimensional (1D) vibration signals are converted into 2D representations using the Gram Angle Field (GAF). ResNet50, embedded with Coordinate Attention (CA), is employed as a feature extractor to capture high-dimensional, fine-grained features for knowledge transfer. Secondly, the developed loss fusion enhancement strategy increases classifier discrimination and stabilize label distribution alignment. DFBR refines decision boundaries by promoting high-confidence predictions for correct classes while penalizing incorrect ones. IFAA steadily balances cross-domain label distribution by combining forward filtering with inverse progressive supplementation, extending the outlier samples filtering strategy by gradually enriching the target domain with source domain features. Finally, the superiority of the proposed method is validated through extensive comparisons with state-of-the-art models on PDA mechanical fault diagnosis tasks.
What problem does this paper attempt to address?