A Method for Domain Generalization in Rolling Bearing Fault Diagnosis Based on Mamba and Causal Prototype Decoupling Constraints
DOI:
https://doi.org/10.63808/aepm.v2i3.424Keywords:
Rolling bearings, Fault diagnosis, Domain generalization, Causal prototypes, Feature decouplingAbstract
To address the insufficient generalization capability of rolling bearing fault diagnosis models under complex operating conditions such as variable rotational speed, variable load, and variable radial force, this paper proposes a domain generalization fault diagnosis method based on Mamba feature extraction and causal generalization loss. First, the raw vibration signals are standardized and segmented using a sliding window strategy. Then, the selective state space model Mamba is employed to model long-range temporal dependencies and local dynamic variations in fault impact signals. Subsequently, from the perspective of causal invariance, generalization constraints are constructed by treating stable representations related to fault categories as causal features, while regarding amplitude variations, noise disturbances, and speed fluctuations induced by operating-condition changes as non-causal factors. A causal generalization loss consisting of class-conditional causal prototype consistency loss and feature decoupling regularization is designed to enhance the model’s adaptability to unseen operating conditions. Experiments are conducted on the Paderborn University (PU) and JNU datasets, where four operating conditions are regarded as four domains to construct cross-condition diagnosis tasks under a leave-one-condition-out evaluation protocol. The proposed method is compared with support vector machine (SVM), one-dimensional convolutional neural network (1D-CNN), one-dimensional residual network (ResNet1D), temporal convolutional network (TCN), correlation alignment (CORAL), domain-adversarial neural network (DANN), maximum mean discrepancy (MMD), and vanilla Mamba. Experimental results show that the proposed method achieves an average accuracy of 93.56% on the PU dataset and 96.28% on the JNU dataset, outperforming all comparison methods and further demonstrating its effectiveness.
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] Chen, X., Lei, Y., Li, Y.-F., Parkinson, S., Li, X., Liu, J., Lu, F., Wang, H., Wang, Z., Yang, B., Ye, S., & Zhao, Z. (2025). Large models for machine monitoring and fault diagnostics: Opportunities, challenges, and future direction. Journal of Dynamics, Monitoring and Diagnostics, 4(2), 76–90. https://doi.org/10.37965/jdmd.2025.832
[4] 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
[5] Gu, A., & Dao, T. (2023). Mamba: Linear-time sequence modeling with selective state spaces [Preprint]. arXiv. https://arxiv.org/abs/2312.00752
[6] Gu, A., Goel, K., & Ré, C. (2022). Efficiently modeling long sequences with structured state spaces [Preprint]. arXiv. https://arxiv.org/abs/2111.00396
[7] Hasan, M., Deng, Z., Redonnet, S., & Sanusi, B. M. (2025). Aerodynamic optimization of box-wing planform through machine learning integration. Transactions of Nanjing University of Aeronautics and Astronautics, 42(6), 789–800. https://doi.org/10.16356/j.1005-1120.2025.06.006
[8] 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
[9] Janssens, O., Slavkovikj, V., Vervisch, B., Stockman, K., Loccufier, M., Verstockt, S., Van de Walle, R., & Van Hoecke, S. (2016). Convolutional neural network based fault detection for rotating machinery. Journal of Sound and Vibration, 377, 331–345. https://doi.org/10.1016/j.jsv.2016.05.027
[10] 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
[11] 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
[12] Lu, Q., Cheng, L., Zhu, D., & Li, M. (2026). Fault prediction method towards gearbox based on digital twin and deep transfer learning. Engineering Research Express, 8(8), Article 085506. https://doi.org/10.1088/2631-8695/ae5b41
[13] Lu, Q., Li, M., & Huang, X. (2025). Digital twin and data-driven remaining useful life prediction of gearbox. IEEE Access, 13, 111614–111627. https://doi.org/10.1109/ACCESS.2025.3583313
[14] Lv, F., Liang, J., Li, S., Zang, B., Liu, C. H., Wang, Z., & Liu, D. (2022). Causality inspired representation learning for domain generalization. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 8046–8056). IEEE. https://doi.org/10.1109/CVPR52688.2022.00788
[15] Mahajan, D., Tople, S., & Sharma, A. (2021). Domain generalization using causal matching. In Proceedings of the 38th International Conference on Machine Learning (pp. 7313–7324). PMLR. https://proceedings.mlr.press/v139/mahajan21a.html
[16] Paderborn University, Chair of Design and Drive Technology. (n.d.). Bearing DataCenter. Retrieved July 29, 2026, from https://mb.uni-paderborn.de/kat/forschung/bearing-datacenter
[17] Pan, S. J., & Yang, Q. (2010). A survey on transfer learning. IEEE Transactions on Knowledge and Data Engineering, 22(10), 1345–1359. https://doi.org/10.1109/TKDE.2009.191
[18] Pearl, J. (2009). Causality: Models, reasoning, and inference (2nd ed.). Cambridge University Press.
[19] Peters, J., Bühlmann, P., & Meinshausen, N. (2016). Causal inference using invariant prediction: Identification and confidence intervals. Journal of the Royal Statistical Society: Series B (Statistical Methodology), 78(5), 947–1012. https://doi.org/10.1111/rssb.12167
[20] Randall, R. B., & Antoni, J. (2011). Rolling element bearing diagnostics—A tutorial. Mechanical Systems and Signal Processing, 25(2), 485–520. https://doi.org/10.1016/j.ymssp.2010.07.017
[21] 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
[22] Wang, B., Lei, Y., Li, N., & Li, N. (2020). A hybrid prognostics approach for estimating remaining useful life of rolling element bearings. IEEE Transactions on Reliability, 69(1), 401–412. https://doi.org/10.1109/TR.2018.2882682
[23] Wang, C. (n.d.). JNU bearing dataset [Data set]. GitHub. Retrieved July 29, 2026, from https://github.com/ClarkGableWang/JNU-Bearing-Dataset
[24] Wang, J., Lan, C., Liu, C., Ouyang, Y., Qin, T., Wang, L., Chen, Y., Zeng, W., & Yu, P. S. (2023). Generalizing to unseen domains: A survey on domain generalization. IEEE Transactions on Knowledge and Data Engineering, 35(8), 8052–8072. https://doi.org/10.1109/TKDE.2022.3178128
[25] Zhang, S., Zhang, S., Wang, B., & Habetler, T. G. (2020). Deep learning algorithms for bearing fault diagnostics—A comprehensive review. IEEE Access, 8, 29857–29881. https://doi.org/10.1109/ACCESS.2020.2972859
[26] 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
[27] Zhao, C., Zio, E., & Shen, W. (2024). Domain generalization for cross-domain fault diagnosis: An application-oriented perspective and a benchmark study. Reliability Engineering & System Safety, 245, Article 109964. https://doi.org/10.1016/j.ress.2024.109964
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.