• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

计算机工程与科学

• • 上一篇    下一篇

基于时频联合建模与先验引导ViT的PMSM音频故障诊断模型

刘惠临,江 宇,刘钊汐,王 涛,周华平,顾成杰   

  1. (1.安徽理工大学 计算机科学与工程学院,安徽 淮南 232001;
    2.煤炭无人化开采数智技术全国重点实验室,安徽 淮南 232001;
    3.滁州学院无人应急装备与灾害过程数字化重建安徽省联合共建学科重点实验室,安徽 滁州 239004)

PMSM based on time–frequency joint modeling and priori guiding ViT audio fault diagnosis model

LIU Huilin,JIANG Yu,LIU Zhaoxi,WANG Tao,ZHOU Huaping   

  1. (1.School of Computer Science and Engineering, Anhui University of Science and Technology, Huainan 232001;
    (2. Unmanned Emergency Equipment and Digital Reconstruction of Disaster Process Joint Laboratory of Anhui Province, Chuzhou 239004, China)

摘要: 针对现有算法存在的时域特征缺失、高噪声环境下模型鲁棒性不足、少样本数据下永磁同步电机故障分类准确度下降等问题,本文提出了一种基于时频联合建模与先验引导注意力机制的永磁同步电机音频故障诊断模型。首先,使用卷积块注意力机制将对数梅尔谱和时频域特征进行特征融合。其次,通过时频双维度长短时记忆网络构建时频域耦合关系并计算时频权重矩阵,通过时频权重矩阵引导Vision Transformer的8×8Patch层提取关键特征,并结合跨尺度残差补偿模块避免模型在块合并过程的特征丢失。实验结果显示,在高噪声、少样本数据下的情况下,所提算法模型在平均F1指标上达到了94.32%,显著超越所对比的其它音频故障诊断算法,在退磁、转子偏心、匝间短路等类别的故障检测中显著提升了诊断精度。

关键词: 时频特征融合, 时频联合建模, 时频权重, 跨尺度残差补偿

Abstract: Aiming at the problems of the existing algorithms, such as missing time domain features, lack of model robustness in high noise environment, and decline of fault classification accuracy in small sample data, this paper proposes an audio fault diagnosis model of permanent magnet synchronous motor based on time–frequency joint modeling and prior guided attention mechanism. First, the convolutional block attention mechanism is used to fuse the logmel spectrum and time-frequency domain features. Secondly, the time-frequency domain coupling relationship is constructed and the time-frequency weight matrix is calculated through the time-frequency dual-dimension memory network. Through the time-frequency weight matrix, the 8×8Patch layer of Vision Transformer is guided to extract key features, and the cross-scale residual compensation module is combined to avoid feature loss in the image block merging process. The experimental results show that under the condition of high noise and small sample data, the proposed algorithm model reaches 94.32% in the average F1 index, significantly surpassing other audio fault diagnosis algorithms compared, and significantly improving the diagnostic accuracy in the fault detection of demagnetization, rotor eccentricity, interturn short circuit and other categories.

Key words: Time-frequency feature fusion, Time–Frequency Joint Modeling, Time-frequency weight, Cross-scale residual compensation