SEFormer:A Lightweight CNN-Transformer Based on Separable Multiscale Depthwise Convolution and Efficient Self-Attention for Rotating Machinery Fault Diagnosis  

作　　者：Hongxing Wang Xilai Ju Hua Zhu Huafeng Li 

机构地区：[1]State Key Laboratory of Mechanics and Control for Aerospace Structures,Nanjing University of Aeronautics and Astronautics,Nanjing,210016,China [2]School of Computer Science and Engineering,Nanyang Technological University,Singapore,639798,Singapore

出　　处：《Computers, Materials & Continua》2025年第1期1417-1437,共21页计算机、材料和连续体（英文）

基　　金：supported by the National Natural Science Foundation of China(No.52277055).

摘　　要：Traditional data-driven fault diagnosis methods depend on expert experience to manually extract effective fault features of signals,which has certain limitations.Conversely,deep learning techniques have gained prominence as a central focus of research in the field of fault diagnosis by strong fault feature extraction ability and end-to-end fault diagnosis efficiency.Recently,utilizing the respective advantages of convolution neural network(CNN)and Transformer in local and global feature extraction,research on cooperating the two have demonstrated promise in the field of fault diagnosis.However,the cross-channel convolution mechanism in CNN and the self-attention calculations in Transformer contribute to excessive complexity in the cooperative model.This complexity results in high computational costs and limited industrial applicability.To tackle the above challenges,this paper proposes a lightweight CNN-Transformer named as SEFormer for rotating machinery fault diagnosis.First,a separable multiscale depthwise convolution block is designed to extract and integrate multiscale feature information from different channel dimensions of vibration signals.Then,an efficient self-attention block is developed to capture critical fine-grained features of the signal from a global perspective.Finally,experimental results on the planetary gearbox dataset and themotor roller bearing dataset prove that the proposed framework can balance the advantages of robustness,generalization and lightweight compared to recent state-of-the-art fault diagnosis models based on CNN and Transformer.This study presents a feasible strategy for developing a lightweight rotating machinery fault diagnosis framework aimed at economical deployment.

关 键 词：CNN-Transformer separable multiscale depthwise convolution efficient self-attention fault diagnosis 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]