A CNN Feature Extraction and BKA-Optimized LSSVM Classification Method for Small-Sample Rolling Bearing Fault Diagnosis
TL;DR
Experiments indicate that the proposed fault diagnosis approach that combines Continuous Wavelet Transform, Convolutional Neural Network, CNN, Black-winged Kite Algorithm, and Least Squares Support Vector Machine outperforms CNN, CNN-SVM, and CNN-BiGRU under small-sample conditions.
Abstract
Rolling bearings are indispensable elements in mechanical equipment, and their condition is closely related to system reliability and operational safety. To enhance diagnostic performance with limited samples and reduce the dependence on manual parameter selection, this study proposes a fault diagnosis approach that combines Continuous Wavelet Transform (CWT), Convolutional Neural Network (CNN), Black-winged Kite Algorithm (BKA), and Least Squares Support Vector Machine (LSSVM). The original 1-D vibration signals are first processed by CWT to obtain 2-D time–frequency representations. CNN is then employed to learn deep fault-sensitive features, which are subsequently fed into LSSVM for state classification. To further improve classification performance, BKA is used to automatically search for the optimal LSSVM parameters, with validation accuracy adopted as the fitness criterion. Experiments conducted on three public bearing datasets, namely CWRU, JNU, and SEU, indicate that the proposed method outperforms CNN, CNN-SVM, and CNN-BiGRU under small-sample conditions. In addition, t-SNE results show more distinct feature clusters, while BKA exhibits faster convergence and better global search capability than Particle Swarm Optimization (PSO) and Genetic Algorithm (GA).