CLSTM-Transformer: inter-patient ECG classification for ventricular arrhythmia

Tong Liu,Qingjun Wang,Mujun Zang,Jingxuan Wang
DOI: https://doi.org/10.1117/12.2678623
2023-04-14
Abstract:Ventricular arrhythmia is a common arrhythmia diseases, which poses a severe threaten to human health. Electrocardiogram (ECG) carries plenty of pathological information about the heart activities of a patient and has been widely used for diagnosing arrhythmia. Despite some models that have been proposed so far for the automatic classification of arrhythmia by the features of the ECG, their performance will be limited, because the extracted features are relatively monotonous and simple. In order to improve the performance of arrhythmia classification, a Convolutional Neural Networks (CNN), Transformer and Long Short-Term Memory (LSTM) based assemble neural network framework named CLSTM-Transformer is proposed for automatic heartbeat classification under the inter-patient paradigm. It's worth noting that we use the transformer, which is rarely used in medical signaling to pay attention to the important heartbeats. Compared to most current single network models, CLSTM-Transformer extracts feature from three different levels, which can extract more hidden information in the heartbeats and makes the model more advantageous. The experiment results show that our model achieves a 98.56% accuracy, a 93.45% sensitivity, a 93.45% specificity, and a 98.54% positive predictive value, respectively. Compared with other models, this model has better classification performance, which makes it more applicable to the diagnosis of arrhythmia diseases.
Computer Science,Medicine,Engineering
What problem does this paper attempt to address?