2021, 43(10): 3018-3027.
doi: 10.11999/JEIT200774
Abstract:
The accurate classification of ElectroCardioGram (ECG) signals is essential for the automatic diagnosis of heart disease. In order to realize the intelligent classification of arrhythmia, an accurate classification method based on wavelet decomposition and 1D-GoogLeNet is proposed. In this method, Db6 wavelet is used to decompose the ECG signal in eight levels to obtain multi-dimensional data containing both time domain information and frequency domain information. Subsequently, Decomposed samples are used as input to 1D-GoogLeNet to train the model. In the proposed 1D-GoogLeNet model, using Inception's excellent performance in image feature extraction, the two-dimensional convolution is transformed into one-dimensional convolution to learn the features of ECG, and the structure of each Inception is simplified, and the model parameters are reduced. The deep learning classifier proposed in this paper can effectively alleviate the problems of low computational efficiency, difficulty in convergence and model degradation. In the experiment, the MIT-BIH arrhythmia dataset is used to test the performance of the proposed model. The experiment compares the detection results when different decomposition component combinations are used as input. When the input data is combined by {d2-d7}, the proposed 1D-GoogLeNet model can achieve an average accuracy of 96.58%. In addition, the performance of the model and the simple one-dimensional GoogLeNet without structural optimization on the data set is compared. The accuracy of the former is 4.7% higher than the latter, and the training efficiency is increased by 118%.