Fault Diagnosis of Gear Based on Multi-Channel Feature Fusion and Dropkey-Vision Transformer

To solve the problem that it is single channel vibration signals not being able to fully express fault feature information and diagnostic networks not being able to fully capture its information resulting in low diagnostic accuracy, a new gear fault diagnosis method is proposed. Firstly, Subtraction Average Based Optimizer (SABO) as an optimization algorithm is introduced to optimize the parameters of Variational Mode Decomposition (VMD) quickly and with high quality to conduct signal pre-processing. Next, the noisy signals in each channel can be quickly and effectively processed to obtain clean one-dimensional and prominent vibration characteristics signals from multi-channel. Then, multi-channel information is fused to obtain image datasets for diagnosis based on Symmetric Dot Pattern (SDP) to realize clear signals transformed into images. A diagnostic model is proposed based on DropKey added for Vision Transformer (DVit) to enhance the diagnostic network’s ability to comprehensively capture multi-channel feature information. Finally, the proposed method is validated through three datasets from gear fault diagnosis experiments with the average accuracy in fault diagnosis reaching more than 99.5% whether it is the degree or type of fault diagnosis. The average accuracy has increased by at least 0.5% compared to before improvement and it has increased about 2%-7% compared with other methods. The results with visualization form verify the effectiveness and superiority of the proposed method.