A CNN model embedded with local feature knowledge and its application to time-varying signal classification

Abstract

A novel convolutional neural network is proposed for local prior feature embedding and imbalanced dataset modeling for multi-channel time-varying signal classification. This model consists of a single-channel signal feature parallel extraction unit, a multi-channel signal feature integration unit, a local feature embedding and feature similarity measurement unit, a full connection layer, and a Softmax classifier. An algorithm combining dynamic clustering and sliding window was used to select segments signals with typical local features in each pattern class, forming a typical local feature set. The one-dimensional CNNs were used to extract features from the single-channel signal in parallel, a comprehensive feature matrix of the multi-channel signal and the local feature matrix templates were produced. Using the method of external embedding, based on the sliding window and dynamic time warping (DTW) algorithm, the local feature similarities between the local feature template of each pattern class and the comprehensive feature sub-matrix of the input signal were measured, and the maximum values were selected to construct a local feature similarity vector in order. The information fusion was realized through a full connection layer. The proposed methodology can extract and represent both global and local signals features, strengthen the role of prior local feature in classification and improve the modeling properties of imbalanced datasets. A comprehensive learning algorithm is presented in this paper. The classification diagnosis of cardiovascular disease based on 12-lead ECG signals was used as a verification experiment. Results showed that the accuracy and generalization for the proposed technique were significantly improved.