A multiview feature fusion model for heartbeat classification

Abstract

Objective. An electrocardiogram (ECG) is one of the most common means to diagnose arrhythmia according to different waveforms clinically. Although there are advanced classification methods such as deep learning, the single view feature cannot meet the demand of classification accuracy for new individuals. To this end, a classification model based on multiview fusion was proposed. Approach. First, handcrafted view features were extracted from heartbeats and then deep view features were obtained from the deep learning model. The features of two different perspectives were fused in the fully connected layer, and the random forest classifier was used instead of the Softmax classifier for classification. Notably, Bayesian optimization was utilized in the hyper-parameter tuning of the classifier. The proposed method employed the MIT-BIH database to classify five classes: normal heartbeat (N), left bundle branch block heartbeat (LB), right bundle branch block heartbeat (RB), atrial premature contraction (APC) and premature ventricular contraction (PVC). Main results. The experimental results achieved a higher average accuracy of 98.93%, average precision of 96.92%, average sensitivity of 96.46%, and average specificity of 99.33% in five types of heartbeat classification for inter-patient. Significance. The proposed framework improves the performance of ECG detection for new individuals. And it provides an feasible algorithmic model for single-lead wearable devices with multiview fusion.