A Study on Domain Generalization Methods for Rolling Bearing Fault Diagnosis Based on Mean Differences and Domain Adversarial Learning
DOI:
https://doi.org/10.63808/aepm.v2i3.425Keywords:
Rolling bearings, Fault diagnosis, Domain generalization, Mean difference, Domain adversarial learningAbstract
To address the issue that intelligent fault diagnosis models for rolling bearings are susceptible to the effects of load and speed variations, as well as shifts in data distribution, when applied across different operating conditions, this paper proposes a fault diagnosis domain generalization method that combines mean difference constraints with domain adversarial learning. Taking raw one-dimensional vibration signals as input, this method first employs a multiscale one-dimensional convolutional network to extract fault impact features and periodic features across different time scales; Subsequently, mean difference constraints between source domains are introduced in the feature space to reduce the offset in the centers of feature distributions across different source operating conditions; simultaneously, a domain-adversarial discriminator based on gradient inversion layers is constructed to weaken the operating condition discriminative information in the feature representations, thereby prompting the model to learn domain-invariant features that possess both fault class discriminability and operating condition insensitivity. Experiments were conducted using two rolling bearing datasets from Case Western Reserve University (CWRU) and Paderborn University (PU) to establish a leave-one-out cross-condition diagnosis task. The results show that the proposed method achieves an average accuracy of 99.08% ± 0.19% on the CWRU dataset, and an average accuracy of 91.86% ± 0.34% on the PU dataset, both outperforming comparison methods such as SVM, 1D-CNN, ResNet1D, TCN, CORAL, MMD, and DANN. Furthermore, a Welch’s t-test based on summary statistics indicates that the improvement over the best baseline method is statistically significant. Ablation experiments and parameter sensitivity analysis further validate the effectiveness of multiscale feature extraction, mean difference constraints, and domain adversarial learning in enhancing the model’s generalization capability across different operating conditions.
References
[1] Arjovsky, M., Bottou, L., Gulrajani, I., & Lopez-Paz, D. (2019). Invariant risk minimization [Preprint]. arXiv. https://arxiv.org/abs/1907.02893
[2] Bai, S., Kolter, J. Z., & Koltun, V. (2018). An empirical evaluation of generic convolutional and recurrent networks for sequence modeling [Preprint]. arXiv. https://arxiv.org/abs/1803.01271
[3] Bhatti, S., Shaikh, A. A., Mansoor, A., & Hussain, M. (2024). New approaches of stochastic models to examine the vibration features in roller bearings. Applied Sciences, 14(4), Article 1616. https://doi.org/10.3390/app14041616
[4] Bhatti, S., Shaikh, F., Mansoor, A., & Shaikh, A. A. (2025). A unique statistical framework to predict the health of a machine by utilizing the vibration features of rolling element bearing data. Mehran University Research Journal of Engineering and Technology, 44(3), 37–41. https://doi.org/10.22581/muet1982.0166
[5] Case Western Reserve University Bearing Data Center. (n.d.). Bearing Data Center dataset [Data set]. Case Western Reserve University. https://engineering.case.edu/bearingdatacenter
[6] Chen, L., Li, Q., Shen, C., Zhu, J., Wang, D., & Xia, M. (2022). Adversarial domain-invariant generalization: A generic domain-regressive framework for bearing fault diagnosis under unseen conditions. IEEE Transactions on Industrial Informatics, 18(3), 1790–1800. https://doi.org/10.1109/TII.2021.3078712
[7] Cortes, C., & Vapnik, V. (1995). Support-vector networks. Machine Learning, 20(3), 273–297. https://doi.org/10.1007/BF00994018
[8] Ganin, Y., Ustinova, E., Ajakan, H., Germain, P., Larochelle, H., Laviolette, F., Marchand, M., & Lempitsky, V. (2016). Domain-adversarial training of neural networks. Journal of Machine Learning Research, 17(59), 1–35. https://www.jmlr.org/papers/v17/15-239.html
[9] Gretton, A., Borgwardt, K. M., Rasch, M. J., Schölkopf, B., & Smola, A. (2012). A kernel two-sample test. Journal of Machine Learning Research, 13, 723–773. https://www.jmlr.org/papers/v13/gretton12a.html
[10] Guo, Y., Ju, G., & Zhang, J. (2024). Generalization study of intelligent fault diagnosis under various working conditions and its application to planetary gearboxes. Scientific Reports, 14, Article 25447. https://doi.org/10.1038/s41598-024-75088-8
[11] He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. In 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (pp. 770–778). IEEE. https://doi.org/10.1109/CVPR.2016.90
[12] Kim, I., Kim, S. W., Kim, J., Huh, H., Jeong, I., Choi, T., Kim, J., & Lee, S. (2024). Single domain generalizable and physically interpretable bearing fault diagnosis for unseen working conditions. Expert Systems with Applications, 241, Article 122455. https://doi.org/10.1016/j.eswa.2023.122455
[13] Lei, Y., Yang, B., Jiang, X., Jia, F., Li, N., & Nandi, A. K. (2020). Applications of machine learning to machine fault diagnosis: A review and roadmap. Mechanical Systems and Signal Processing, 138, Article 106587. https://doi.org/10.1016/j.ymssp.2019.106587
[14] Lessmeier, C., Kimotho, J. K., Zimmer, D., & Sextro, W. (2016). Condition monitoring of bearing damage in electromechanical drive systems by using motor current signals of electric motors: A benchmark data set for data-driven classification. PHM Society European Conference, 3(1). https://doi.org/10.36001/phme.2016.v3i1.1577
[15] Li, D., Yang, Y., Song, Y.-Z., & Hospedales, T. M. (2017). Deeper, broader and artier domain generalization. In 2017 IEEE International Conference on Computer Vision (ICCV) (pp. 5542–5550). IEEE. https://doi.org/10.1109/ICCV.2017.593
[16] Muandet, K., Balduzzi, D., & Schölkopf, B. (2013). Domain generalization via invariant feature representation. In S. Dasgupta & D. McAllester (Eds.), Proceedings of the 30th International Conference on Machine Learning (Vol. 28, pp. 10–18). PMLR. https://proceedings.mlr.press/v28/muandet13.html
[17] Ren, H., Wang, J., Huang, W., Jiang, X., & Zhu, Z. (2023). Domain-invariant feature fusion networks for semi-supervised generalization fault diagnosis. Engineering Applications of Artificial Intelligence, 126, Article 107117. https://doi.org/10.1016/j.engappai.2023.107117
[18] Sun, B., & Saenko, K. (2016). Deep CORAL: Correlation alignment for deep domain adaptation. In G. Hua & H. Jégou (Eds.), Computer vision—ECCV 2016 workshops (pp. 443–450). Springer. https://doi.org/10.1007/978-3-319-49409-8_35
[19] Zhang, S., Su, L., Gu, J., Li, K., Zhou, L., & Pecht, M. (2023). Rotating machinery fault detection and diagnosis based on deep domain adaptation: A survey. Chinese Journal of Aeronautics, 36(1), 45–74. https://doi.org/10.1016/j.cja.2021.10.006
[20] Zhang, W., Li, C., Peng, G., Chen, Y., & Zhang, Z. (2018). A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load. Mechanical Systems and Signal Processing, 100, 439–453. https://doi.org/10.1016/j.ymssp.2017.06.022
[21] Zhang, W., Peng, G., Li, C., Chen, Y., & Zhang, Z. (2017). A new deep learning model for fault diagnosis with good anti-noise and domain adaptation ability on raw vibration signals. Sensors, 17(2), Article 425. https://doi.org/10.3390/s17020425
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Jiabing Zhou, Xiang Gu, Bo Zhang, Xinrui Yu, Yining Xu
This work is licensed under a Creative Commons Attribution 4.0 International License.