MB-ViT: MBConv vision transformer with time–frequency feature fusion for bearing fault diagnosis
摘要
Roller bearings play a crucial role in mechanical systems, where their operational condition directly impacts system performance and lifespan. However, detecting early-stage bearing faults during routine maintenance remains a challenge due to the cost and technical limitations of current fault diagnosis methods, often resulting in reduced accuracy. To address this issue, this paper proposes a bearing fault diagnosis method based on the MBConv Vision Transformer (MB-ViT) with frequency feature fusion. Specifically, we introduce a novel approach that transforms bearing fault signals into RGB images by combining Continuous Wavelet Transform (CWT) and Gramian Angular Summation Field (GASF) images, thereby enhancing feature representation and improving fault recognition. Additionally, recognizing the limitations of the traditional Vision Transformer (ViT) in capturing local features, we introduce the MB-Multi-Head Self-Attention (MB-MSA) module to overcome this challenge. Experimental results using data from Case Western Reserve University and Xi’an Jiaotong University show that feature fusion significantly improves fault diagnosis accuracy, while the MB-MSA module enhances both diagnostic precision and robustness. MB-ViT achieves an accuracy of 99.90