[Heartbeat-based end-to-end classification of arrhythmias]

We propose a heartbeat-based end-to-end classification of arrhythmias to improve the classification performance for supraventricular ectopic beat (SVEB) and ventricular ectopic beat (VEB).

The ECG signals were preprocessed by heartbeat segmentation and heartbeat alignment. An arrhythmia classifier was constructed based on convolutional neural network, and the proposed loss function was used to train the classifier.

The proposed algorithm was verified on MIT-BIH arrhythmia database. The AUC of the proposed loss function for SVEB and VEB reached 0.77 and 0.98, respectively. With the first 5 min segment as the local data, the diagnostic sensitivities for SVEB and VEB were 78.28% and 98.88%, respectively; when 0, 50, 100, and 150 samples were used as the local data, the diagnostic sensitivities for SVEB and VEB reached 82.25% and 93.23%, respectively.

The proposed method effectively reduces the negative impact of class-imbalance and improves the diagnostic sensitivities for SVEB and VEB, and thus provides a new solution for automatic arrhythmia classification.