胡含颖,王细洋. 基于连续小波变换与卷积神经网络−视觉Transformer的齿轮故障诊断[J]. 失效分析与预防,2026,21(3):239-248. doi: 10.3969/j.issn.1673-6214.2026.03.007
    引用本文: 胡含颖,王细洋. 基于连续小波变换与卷积神经网络−视觉Transformer的齿轮故障诊断[J]. 失效分析与预防,2026,21(3):239-248. doi: 10.3969/j.issn.1673-6214.2026.03.007
    HU Hanying,WANG Xiyang. Gear fault diagnosis based on continuous wavelet transform and convolutional neural network-vision transformer[J]. Failure analysis and prevention,2026,21(3):239-248. doi: 10.3969/j.issn.1673-6214.2026.03.007
    Citation: HU Hanying,WANG Xiyang. Gear fault diagnosis based on continuous wavelet transform and convolutional neural network-vision transformer[J]. Failure analysis and prevention,2026,21(3):239-248. doi: 10.3969/j.issn.1673-6214.2026.03.007

    基于连续小波变换与卷积神经网络−视觉Transformer的齿轮故障诊断

    Gear Fault Diagnosis Based on Continuous Wavelet Transform and Convolutional Neural Network-Vision Transformer

    • 摘要: 为解决非平稳性强及特征融合不足问题,本文提出一种基于连续小波变换(CWT)与卷积神经网络−视觉Transformer(CNN-ViT)的齿轮故障诊断方法。首先,利用CWT将一维振动信号转换为时频图,以提取信号的非平稳特征;然后,构建一个CNN-ViT双分支架构,其中CNN分支提取局部特征,ViT分支建模全局依赖关系;最后,设计了交叉注意力融合模块,通过交叉注意力机制实现ViT特征、CNN特征与统计特征的深度交互,并引入门控机制自适应调节不同模态特征的融合权重。研究结果表明,与单一模态模型相比,该方法在齿轮故障分类任务中具有更高的识别精度与泛化能力,能够有效提升复杂工况下的诊断鲁棒性。

       

      Abstract: To address the challenges of strong non-stationarity and insufficient feature fusion in gear fault diagnosis, this study proposes a novel method based on continuous wavelet transform (CWT) and a hybrid convolutional neural network-vision transformer (CNN-ViT) architecture. First, one-dimensional vibration signals are converted into time-frequency representation via CWT to extract their non-stationary characteristics. A dual-branch framework is then constructed, wherein a CNN branch captures local features and a vision transformer (ViT) branch models long-range contextual dependencies. Furthermore, a cross-attention fusion module is designed to enable deep interaction among the ViT, CNN, and statistical features. A gating mechanism is introduced to adaptively weight the contributions from these different feature modalities. Experimental results demonstrate that the proposed method outperforms single-modality models in gear fault classification, achieving higher recognition accuracy and superior generalization capability. The framework effectively enhances diagnostic robustness under complex operating conditions.

       

    /

    返回文章
    返回