HEARTBEAT CLASSIFICATION ALGORITHM BASED ON ONE-DIMENSIONAL CONVOLUTION NEURAL NETWORK

Abstract

The morbidity of cardiovascular disease increasingly rises, which makes great impact upon people’s health and life. Electrocardiogram (ECG) beat classification is of great significance to clinical diagnosis of cardiovascular diseases. Traditional ECG signal classification algorithm relies heavily on the accuracy of feature extraction or increases the complexity of the calculation process by means of the correlation characteristic coefficient transformation, which results in that the ECG beat classification effect is still not satisfactory. Aimed at this problem, a novel method based on convolution neural network (CNN) is presented in this paper. First, ECG signal is preprocessed to suppress the noise and to locate the R peaks, and five kinds of ECG beat waveform data are obtained. Then taking ECG beat sampling points as input, four layers of one-dimensional CNN are constructed for feature extraction and classification. Finally, experimental verification is carried out on the data from MIT-BIH database, and the accuracy of recognition and classification of the presented method reaches 99.10%. Comparison with the methods based on artificial features, this method shows better performance, which avoids serious dependence on the accuracy of feature extraction, skips the steps of feature extraction and selection, and reduces the complexity of computational process.