The Fault Diagnosis of Rolling Bearings Based on FFT-SE-TCN-SVM
Yanqiu Wu, Juying Dai, Xiaoqiang Yang, Faming Shao, Jiancheng Gong, Peng Zhang, Shaodong Liu
Abstract
Traditional fault diagnosis methods often require extracting features from raw vibration signals based on prior knowledge, which are then input into intelligent classifiers for pattern recognition. This process is prone to information loss and can be inaccurate when relying on human experience for fault identification. To address this issue, this paper proposes an intelligent fault classification and diagnosis model for rolling bearings based on Fast Fourier Transform (FFT) combined with a time convolutional network (SE-TCN) incorporating an attention mechanism, with a Support Vector Machine (SVM) used as the classifier. First, the FFT is applied to transform the collected raw time-domain data of bearing faults into the frequency domain, obtaining the sequence information in the frequency domain. Second, the frequency–domain sequence data are fed into the SE-TCN model, which uses multiple convolutional layers and a channel attention mechanism to extract deep fault features. Finally, the extracted feature vectors are input into the SVM classifier, and the Particle Swarm Optimization (PSO) algorithm is used to optimize the SVM parameters. The optimal separating hyperplane is obtained through training to classify the fault types of the rolling bearings. To verify the effectiveness and diagnostic performance of the proposed method, experiments are conducted using bearing fault datasets from Case Western Reserve University (CWRU) and a laboratory self-built fault diagnosis experimental platform. The experimental results show that the classification accuracy of the proposed method exceeds 99% on the CWRU test dataset, and it also demonstrates advantages in handling small sample data, with an accuracy of over 90%. Additionally, it exhibits good diagnostic performance on the bearing fault data collected from the laboratory self-built platform. The results validate the effectiveness of the proposed classification model in bearing a fault diagnosis.