Cross-Domain Bearing Fault Diagnosis Under Class Imbalance: A Dynamic Maximum Triple-View Classifier Discrepancy Network
Abstract
1. Introduction
- We propose a Triple-View Classifier (TVC) Architecture based on a Sample Aggregation Mechanism and Primary–Auxiliary Fused Cooperative Loss (PAFL). To address minority sample scarcity, we introduce an auxiliary binary classification view. The Sample Aggregation Mechanism aggregates fine-grained fault sub-classes into a unified “Fault Super-class.” This fundamentally reduces the imbalance ratio and helps learn a robust “normal-fault” boundary. Driven by PAFL, this boundary acts as a hierarchical geometric constraint to suppress the primary classifier’s tendency to misclassify faults as normal. This explicitly prevents the decision boundary from encroaching upon minority regions, significantly reducing missed detections.
- We develop a dynamic weighting strategy to adaptively regulate the adversarial process. A dynamic factor is introduced to regulate the weight of the discrepancy loss. In the early training stage, a larger weight amplifies the divergence between classifiers. This forces the generator to conduct extensive feature exploration and prevents premature convergence to majority-class local optima. In the later stage, the weight is reduced to enforce consensus and promote compact alignment.
- We validate superior diagnostic robustness under extreme class imbalance. Extensive experiments on CWRU and JNU datasets confirm that even under an extreme 20:1 imbalance ratio, DMTVCD maintains superior performance, significantly outperforming state-of-the-art methods.
2. Related Work: Maximum Classifier Discrepancy (MCD)
2.1. Basic Architecture and Discrepancy Metric
2.2. Adversarial Training Steps
3. Proposed Method: DMTVCD
3.1. Problem Definition
3.2. DMTVCD Network Architecture
- Feature Generator (G): Taking inspiration from the WDCNN architecture [25], we implement a deep 1-Dimensional Convolutional Neural Network (1D-CNN) as the feature extractor to achieve efficient end-to-end learning. Compared to traditional signal processing methods (e.g., FFT or Wavelet) which heavily rely on domain prior knowledge, or 2D-CNNs which introduce significant time–frequency conversion latency, our 1D-CNN directly extracts discriminative high-dimensional feature vectors from raw vibration signals. For source domain input and target domain input , the extracted features are denoted as and , respectively. Specifically, it utilizes a wide first-layer kernel () to effectively suppress high-frequency noise and capture the periodic impulsive characteristics inherent in bearing faults, followed by smaller kernels (e.g., ) for deep nonlinear mapping. This design provides a superior balance between feature discriminability and high computational efficiency required for real-time edge deployment.
- Triple-View Classifier (TVC) Architecture: To address the issue where multi-class classifiers tend to overlook minority classes under imbalanced data distributions, we design a synergistic module containing three independent classifiers:
- Primary Classifier (): A K-class multi-classifier outputting probabilities . Its task is to identify fine-grained specific health states.
- Auxiliary Classifiers (): Two binary classifiers tasked with coarse-grained discrimination between normal and fault states. In our implementation, both auxiliary classifiers share an identical two-layer multi-layer perceptron (MLP) architecture. Specifically, this comprises a hidden layer with 256 units followed by a ReLU activation function, and a linear output layer yielding 2-dimensional predictions. To ensure they capture diverse discriminative features and establish a meaningful discrepancy space during adversarial training, and are independently initialized using Kaiming uniform initialization with different random seeds. This explicit structural design and independent initialization strategy firmly guarantee the constructing of a robust geometric decision boundary.
3.3. Dynamic Discrepancy Loss and Optimization Objectives
3.3.1. Multi-Task Classification Loss
3.3.2. Primary–Auxiliary Fused Discrepancy Loss (PAFL)
- The Hierarchical Geometric Constraint Term (first part) acts as a hard constraint enforced by the auxiliary views. Geometrically, the robust “normal-fault” decision boundary established by the auxiliary classifiers acts as an anchor. It compels the primary classifier to align its aggregated predictions with the robust binary boundary, effectively suppressing the tendency of the primary decision boundary to shift towards and misclassify fault samples as normal under extreme class imbalance.
- The Binary Boundary Reinforcement Term (second part) maximizes the discrepancy between the two auxiliary views. This explicitly mines ambiguous samples at the coarse-grained level to construct a robust “normal-fault” geometric boundary.
3.3.3. Dynamic Weighting Factor
3.4. Adversarial Training Steps
| Algorithm 1 Training procedure of DMTVCD. |
Require: Source domain , Target domain ; Batch size m, Epochs , Steps n; Params . Ensure: Optimized parameters .
|
4. Experimental Verification
4.1. Dataset Description and Imbalanced Sample Construction
4.1.1. CWRU Bearing Dataset
4.1.2. JNU Bearing Dataset
4.1.3. Imbalanced Sample Construction and Splitting
4.2. Comparison Methods and Parameter Configuration
4.2.1. Comparison Methods
4.2.2. Parameter Configuration
4.3. Analysis of Experimental Results on CWRU Dataset
4.4. Analysis of Experimental Results on JNU Dataset
4.5. Robustness to Varying Imbalance Ratios
4.6. Robustness in Complex and Strong-Noise Environments
- Additive White Gaussian Noise (AWGN): Evaluated at specific Signal-to-Noise Ratios (SNRs: 8, 4, 0, and −4 dB) [28].
4.7. Ablation Study and Training Dynamics Analysis
4.8. Visualization Analysis
5. Conclusions
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Wang, Y.; Xu, J.; Li, Z.; Zhang, Y. A comprehensive review of deep learning-based fault diagnosis approaches for rolling bearings: Advancements and challenges. AIP Adv. 2025, 15, 020702. [Google Scholar] [CrossRef] [Scilit]
- Randall, R.B.; Antoni, J. Rolling element bearing diagnostics—A tutorial. Mech. Syst. Signal Process. 2011, 25, 485–520. [Google Scholar] [CrossRef] [Scilit]
- Lei, Y.; Yang, B.; Jiang, X.; Jia, F.; Li, N.; Nandi, A.K. Applications of machine learning to machine fault diagnosis: A review and reasonable prospect. Mech. Syst. Signal Process. 2020, 142, 106805. [Google Scholar]
- Altaf, M.; Akram, T.; Khan, M.A.; Iqbal, M.; Ch, M.M.; Hsu, C.-H. A new statistical features based approach for bearing fault diagnosis using vibration signals. Sensors 2022, 22, 2012. [Google Scholar] [CrossRef] [Scilit]
- Yu, H.; Wang, K.; Li, Y.; Zhao, W. Representation learning with class level autoencoder for intelligent fault diagnosis. IEEE Signal Process. Lett. 2019, 26, 1521–1525. [Google Scholar] [CrossRef] [Scilit]
- Zheng, K.; Yao, D.; Shi, Y.; Zhang, B. An adaptive group sparse feature decomposition method in frequency domain for rolling bearing fault diagnosis. ISA Trans. 2023, 138, 562–581. [Google Scholar] [CrossRef] [Scilit] [PubMed]
- Yu, H.; Wang, K.; Li, Y. Multiscale representations fusion with joint multiple reconstructions autoencoder for intelligent fault diagnosis. IEEE Signal Process. Lett. 2019, 26, 94–98. [Google Scholar]
- Wang, J.; Mo, Z.; Zhang, H.; Miao, Q. A deep learning method for bearing fault diagnosis based on time-frequency image. IEEE Access 2019, 7, 42373–42383. [Google Scholar] [CrossRef] [Scilit]
- Zhang, W.; Li, X.; Ding, Q. Transfer Learning-Motivated Intelligent Fault Diagnosis Designs: A Survey, Insights, and Perspectives. IEEE Trans. Neural Netw. Learn. Syst. 2024, 35, 2969–2983. [Google Scholar]
- Lu, W.; Liang, B.; Cheng, Y.; Meng, D.; Yang, J.; Zhang, T. Deep model based domain adaptation for fault diagnosis. IEEE Trans. Ind. Electron. 2017, 64, 2296–2305. [Google Scholar] [CrossRef] [Scilit]
- Qian, Q.; Qin, Y.; Luo, J.; Wang, Y.; Wu, F. Deep discriminative transfer learning network for cross-machine fault diagnosis. Mech. Syst. Signal Process. 2023, 186, 109884. [Google Scholar]
- Tzeng, E.; Hoffman, J.; Zhang, N.; Saenko, K.; Darrell, T. Deep domain confusion: Maximizing for domain invariance. arXiv 2014, arXiv:1412.3474. [Google Scholar] [CrossRef] [Scilit]
- Ganin, Y.; Ustinova, E.; Ajakan, H.; Germain, P.; Larochelle, H.; Laviolette, F.; Marchand, M.; Lempitsky, V. Domain-adversarial training of neural networks. J. Mach. Learn. Res. 2016, 17, 1–35. [Google Scholar]
- Saito, K.; Watanabe, K.; Ushiku, Y.; Harada, T. Maximum classifier discrepancy for unsupervised domain adaptation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Salt Lake City, UT, USA, 18–23 June 2018; pp. 3723–3732. [Google Scholar]
- Gao, H.; Xue, Z.; Han, H.; Li, F. Class-aware multi-source domain adaptation for imbalanced fault diagnosis. In Proceedings of the 14th Asian Control Conference (ASCC), Dalian, China, 5–8 July 2024; pp. 1–6. [Google Scholar]
- Chen, Z.; Yu, W.; Wang, L.; Ding, X.; Huang, W.; Shao, Y. A dual-view style mixing network for unsupervised cross-domain fault diagnosis with imbalanced data. Knowl.-Based Syst. 2023, 278, 110918. [Google Scholar] [CrossRef] [Scilit]
- Yang, W.; Wang, K. Domain invariant feature learning based on cluster contrastive learning for intelligence fault diagnosis with limited labeled data. IEEE Signal Process. Lett. 2023, 30, 1787–1791. [Google Scholar] [CrossRef] [Scilit]
- Chen, C.; Li, X.; Shi, J.; Yue, D.; Wang, C.; Feng, L.; Chen, H.; Churakova, A.A.; Alexandrov, I.V. A triple domain adversarial neural network for bearing fault diagnosis. Mech. Syst. Signal Process. 2025, 238, 113202. [Google Scholar] [CrossRef] [Scilit]
- Tang, S.; Yuan, S.; Zhu, Y. Adversarial Deep Transfer Learning in Fault Diagnosis: Progress, Challenges, and Future Prospects. Sensors 2023, 23, 7263. [Google Scholar] [CrossRef] [Scilit]
- Jia, F.; Lei, Y.; Guo, L.; Lin, J.; Xing, S. Deep normalized convolutional neural network for imbalanced fault classification of machinery and its understanding. Mech. Syst. Signal Process. 2018, 110, 349–367. [Google Scholar] [CrossRef] [Scilit]
- Han, H.; Wang, W.-Y.; Mao, B.-H. Borderline-SMOTE: A new over-sampling method in imbalanced data sets learning. In Proceedings of the International Conference on Intelligent Computing (ICIC), Hefei, China, 23–26 August 2005; pp. 878–887. [Google Scholar]
- Lin, W.-C.; Tsai, C.-F.; Hu, Y.-H.; Jhang, J.-S. Clustering-based undersampling in class-imbalanced data. Inf. Sci. 2017, 409, 17–26. [Google Scholar] [CrossRef] [Scilit]
- Li, W.; Zhong, X.; Shao, H.; Cai, B.; Yang, X. Multi-mode data augmentation and fault diagnosis of rotating machinery using modified ACGAN designed with new framework. Adv. Eng. Inform. 2022, 52, 101552. [Google Scholar]
- Cheng, X.; Lu, Y.; Liang, Z.; Zhao, L.; Gong, Y.; Wang, M. A Bearing Fault Diagnosis Method in Scenarios of Imbalanced Samples and Insufficient Labeled Samples. Appl. Sci. 2024, 14, 8582. [Google Scholar] [CrossRef] [Scilit]
- Zhang, W.; Peng, G.; Li, C.; Chen, Y.; Zhang, Z. A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals. Sensors 2017, 17, 425. [Google Scholar] [CrossRef] [Scilit]
- Smith, W.A.; Randall, R.B. Rolling element bearing diagnostics using the Case Western Reserve University data: A benchmark study. Mech. Syst. Signal Process. 2015, 64–65, 100–131. [Google Scholar] [CrossRef] [Scilit]
- Li, K.; Ping, X.; Wang, H.; Chen, P.; Cao, Y. Sequential fuzzy diagnosis method for motor roller bearing in variable operating conditions based on vibration analysis. Sensors 2013, 13, 8013–8041. [Google Scholar] [CrossRef] [Scilit]
- Zhang, W.; Li, C.; Peng, G.; Chen, Y.; Zhang, Z. A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load. IEEE Trans. Ind. Electron. 2018, 65, 4324–4333. [Google Scholar] [CrossRef] [Scilit]
- Hebda Sobkowicz, J.; Zimroz, R.; Wyłomańska, A. Selection of the Informative Frequency Band in a Bearing Fault Diagnosis in the Presence of Non-Gaussian Noise Comparison of Recently Developed Methods. Appl. Sci. 2020, 10, 2657. [Google Scholar] [CrossRef] [Scilit]








| Health Condition | Training Source (Samples) | Training Target (Samples) | Testing Target (Samples) | Label |
|---|---|---|---|---|
| Normal | 400 | 400 | 100 | 0 |
| Inner Race (0.007″) | 20 | 20 | 100 | 1 |
| Inner Race (0.014″) | 20 | 20 | 100 | 2 |
| Inner Race (0.021″) | 20 | 20 | 100 | 3 |
| Ball (0.007″) | 20 | 20 | 100 | 4 |
| Ball (0.014″) | 20 | 20 | 100 | 5 |
| Ball (0.021″) | 20 | 20 | 100 | 6 |
| Outer Race (0.007″) | 20 | 20 | 100 | 7 |
| Outer Race (0.014″) | 20 | 20 | 100 | 8 |
| Outer Race (0.021″) | 20 | 20 | 100 | 9 |
| Health Condition | Training Source (Samples) | Training Target (Samples) | Testing Target (Samples) | Label |
|---|---|---|---|---|
| Normal | 400 | 400 | 100 | 0 |
| Inner Race Fault | 20 | 20 | 100 | 1 |
| Rolling Element Fault | 20 | 20 | 100 | 2 |
| Outer Race Fault | 20 | 20 | 100 | 3 |
| Module | Layer | Type | Configuration () | Output Shape |
|---|---|---|---|---|
| Feature Generator (G) | 1 | Conv1d + BN + ReLU MaxPool1d | ||
| 2 | Conv1d + BN + ReLU MaxPool1d | |||
| 3 | Conv1d + BN + ReLU MaxPool1d | |||
| 4 | Conv1d + BN + ReLU MaxPool1d | |||
| 5 | Conv1d + BN + ReLU MaxPool1d | |||
| Flatten Layer | 256 | |||
| Main Classifier () | 1 | Linear + ReLU | - | 256 |
| 2 | Linear | - | K | |
| Auxiliary Classifiers () | 1 | Linear + ReLU | - | 256 |
| 2 | Linear | - | 2 | |
| Task | CNN [25] | DANN [13] | MCD [14] | DVSMN [16] | CMDA [15] | DMTVCD | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | |
| 0-1 | 69.66 ± 3.85 | 6.8 | 86.90 ± 4.88 | 8.5 | 89.90 ± 1.46 | 35.2 | 76.22 ± 2.86 | 13.4 | 86.12 ± 4.39 | 9.5 | 90.36 ± 4.85 | 39.5 |
| 0-2 | 69.84 ± 3.87 | 5.7 | 89.86 ± 3.20 | 12.5 | 90.18 ± 0.99 | 41.2 | 76.12 ± 1.23 | 18.5 | 91.88 ± 3.65 | 10.8 | 99.40 ± 0.28 | 40.7 |
| 0-3 | 65.62 ± 2.17 | 5.1 | 83.02 ± 5.66 | 4.6 | 90.52 ± 5.71 | 33.0 | 69.46 ± 2.44 | 10.5 | 89.84 ± 3.72 | 7.5 | 94.62 ± 7.03 | 38.9 |
| 1-0 | 76.68 ± 2.26 | 6.8 | 86.64 ± 2.43 | 9.3 | 89.66 ± 2.33 | 31.7 | 78.18 ± 4.13 | 13.3 | 84.02 ± 3.99 | 9.6 | 96.08 ± 1.58 | 25.1 |
| 1-2 | 79.18 ± 4.40 | 5.4 | 79.28 ± 4.00 | 13.0 | 93.40 ± 3.41 | 40.4 | 85.12 ± 3.41 | 17.0 | 90.86 ± 5.67 | 8.6 | 99.30 ± 0.77 | 45.5 |
| 1-3 | 70.92 ± 5.05 | 9.4 | 73.14 ± 3.21 | 14.0 | 89.96 ± 5.56 | 41.6 | 75.78 ± 3.83 | 9.6 | 87.98 ± 3.77 | 9.7 | 99.36 ± 0.45 | 41.9 |
| 2-0 | 75.78 ± 3.01 | 9.4 | 85.18 ± 2.35 | 12.2 | 86.78 ± 3.15 | 38.7 | 75.04 ± 4.14 | 11.5 | 84.10 ± 4.20 | 5.0 | 95.54 ± 0.94 | 31.0 |
| 2-1 | 77.58 ± 2.66 | 3.5 | 79.78 ± 2.35 | 4.6 | 92.50 ± 2.67 | 30.4 | 82.20 ± 2.76 | 10.4 | 88.92 ± 3.00 | 14.2 | 97.92 ± 1.06 | 39.0 |
| 2-3 | 77.78 ± 3.91 | 8.4 | 82.30 ± 3.17 | 10.7 | 96.98 ± 3.15 | 34.1 | 86.44 ± 3.04 | 6.7 | 96.40 ± 1.45 | 5.9 | 99.44 ± 0.28 | 37.2 |
| 3-0 | 67.22 ± 3.76 | 4.3 | 74.88 ± 4.02 | 10.4 | 79.78 ± 4.17 | 40.0 | 69.60 ± 4.81 | 16.0 | 83.52 ± 3.69 | 15.0 | 87.88 ± 5.40 | 38.4 |
| 3-1 | 67.98 ± 3.02 | 7.7 | 69.52 ± 2.08 | 6.1 | 81.62 ± 4.21 | 28.4 | 75.30 ± 4.05 | 13.9 | 84.10 ± 2.42 | 5.2 | 89.98 ± 4.95 | 27.8 |
| 3-2 | 75.26 ± 4.24 | 7.9 | 81.30 ± 3.07 | 7.4 | 93.14 ± 3.71 | 28.2 | 83.70 ± 3.44 | 14.8 | 95.30 ± 1.95 | 14.1 | 99.88 ± 0.07 | 39.5 |
| Average | 72.79 | 6.7 | 80.98 | 9.4 | 89.54 | 35.2 | 77.76 | 13.0 | 88.59 | 9.6 | 95.81 | 37.0 |
| Task | CNN [25] | DANN [13] | MCD [14] | DVSMN [16] | CMDA [15] | DMTVCD | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | Acc (%) | Time (s) | |
| 0-1 | 63.23 ± 2.80 | 8.9 | 70.12 ± 3.50 | 6.4 | 73.45 ± 5.69 | 55.0 | 65.72 ± 2.12 | 17.8 | 74.78 ± 3.84 | 11.6 | 80.58 ± 2.20 | 86.8 |
| 0-2 | 56.67 ± 3.67 | 8.1 | 64.10 ± 4.12 | 14.5 | 65.33 ± 6.08 | 73.6 | 58.17 ± 2.81 | 19.2 | 67.97 ± 4.41 | 16.1 | 71.47 ± 2.88 | 59.4 |
| 1-0 | 64.05 ± 2.67 | 8.1 | 73.45 ± 2.84 | 17.8 | 81.58 ± 1.06 | 75.0 | 68.33 ± 2.38 | 19.3 | 75.67 ± 3.86 | 17.0 | 79.42 ± 2.59 | 69.6 |
| 1-2 | 62.65 ± 5.24 | 4.1 | 71.35 ± 6.07 | 13.8 | 80.35 ± 4.40 | 78.8 | 66.17 ± 5.47 | 12.7 | 74.12 ± 5.23 | 14.6 | 83.22 ± 1.75 | 61.0 |
| 2-0 | 56.12 ± 3.39 | 7.5 | 67.10 ± 1.73 | 14.6 | 75.80 ± 2.97 | 63.5 | 58.85 ± 4.90 | 18.2 | 73.20 ± 3.05 | 16.4 | 76.20 ± 2.89 | 74.9 |
| 2-1 | 60.92 ± 4.45 | 8.8 | 73.45 ± 3.18 | 10.3 | 85.53 ± 3.12 | 68.9 | 64.80 ± 5.04 | 17.7 | 76.75 ± 4.21 | 12.6 | 83.12 ± 3.47 | 67.1 |
| Average | 60.61 | 7.6 | 69.93 | 12.9 | 77.01 | 69.2 | 63.67 | 17.5 | 73.75 | 14.7 | 79.00 | 69.8 |
| Task | DMTVCD-Static () | DMTVCD (Dynamic ) |
|---|---|---|
| 0-1 | 92.66 ± 1.72 | 90.36 ± 4.85 |
| 0-2 | 96.20 ± 1.12 | 99.40 ± 0.28 |
| 0-3 | 98.86 ± 0.41 | 94.62 ± 7.03 |
| 1-0 | 93.08 ± 1.90 | 96.08 ± 1.58 |
| 1-2 | 97.10 ± 1.15 | 99.30 ± 0.77 |
| 1-3 | 95.60 ± 3.31 | 99.36 ± 0.45 |
| 2-0 | 91.24 ± 2.49 | 95.54 ± 0.94 |
| 2-1 | 94.14 ± 1.68 | 97.92 ± 1.06 |
| 2-3 | 99.52 ± 0.20 | 99.44 ± 0.28 |
| 3-0 | 78.60 ± 2.96 | 87.88 ± 5.40 |
| 3-1 | 83.86 ± 3.27 | 89.98 ± 4.95 |
| 3-2 | 96.68 ± 2.38 | 99.88 ± 0.07 |
| Average | 93.13 | 95.81 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Luo, R.; Xie, H.; Wen, H.; He, H.; Li, Y.; Wang, K. Cross-Domain Bearing Fault Diagnosis Under Class Imbalance: A Dynamic Maximum Triple-View Classifier Discrepancy Network. Algorithms 2026, 19, 228. https://doi.org/10.3390/a19030228
Luo R, Xie H, Wen H, He H, Li Y, Wang K. Cross-Domain Bearing Fault Diagnosis Under Class Imbalance: A Dynamic Maximum Triple-View Classifier Discrepancy Network. Algorithms. 2026; 19(3):228. https://doi.org/10.3390/a19030228
Chicago/Turabian StyleLuo, Rui, Huiyang Xie, Haitian Wen, Hongying He, Yitong Li, and Kai Wang. 2026. "Cross-Domain Bearing Fault Diagnosis Under Class Imbalance: A Dynamic Maximum Triple-View Classifier Discrepancy Network" Algorithms 19, no. 3: 228. https://doi.org/10.3390/a19030228
APA StyleLuo, R., Xie, H., Wen, H., He, H., Li, Y., & Wang, K. (2026). Cross-Domain Bearing Fault Diagnosis Under Class Imbalance: A Dynamic Maximum Triple-View Classifier Discrepancy Network. Algorithms, 19(3), 228. https://doi.org/10.3390/a19030228

