Rolling bearing fault diagnosis method using time-frequency information integration and multi-scale TransFusion network

Abstract

Advances in deep learning methods have demonstrated remarkable development in diagnosing faults of rotating machinery. The currently popular deep neural networks suffer from design flaws in their network structure, leading to issues of long-term dependencies in fault diagnosis models built upon conventional deep neural networks. Consequently, such models exhibit insufficient global perceptual capabilities towards fault features. Furthermore, how accurately pre-trained models can diagnose faults is hugely impacted by changes in bearings’ working conditions. To tackle the aforementioned issues, this study puts forth a multi-scale TransFusion (MSTF) model for diagnosing faults in rolling bearings under multiple operating conditions. Firstly, a time-frequency symmetric dot pattern transformation technique is designed to transform the original vibration signals into two-dimensional representations. This method can effectively highlight the distinctions between different fault types. Secondly, a multi-scale feature fusion module is established, which fully extracts low-level features from the time-frequency signals and reduces the complexity of the subsequent attention calculations. Meanwhile, relying on the advantages of the Transformer model in capturing global dependencies, the long-range periodic fault information is deeply mined. Finally, multi-head and multi-layer attention are visualized to enhance the interpretability of the model. After analyzing two case studies with both public and experimental datasets, the examination demonstrated that the developed model outperformed other state-of-the-art models. The diagnostic model developed in this study exhibits the ability to accurately diagnose bearing faults across multiple operating conditions while maintaining high robustness to signals contaminated with noise.