A novel interpretable multilevel wavelet decomposition deep network for actual heartbeat classification

YanRui Jin,ZhiYuan Li,YuanYuan Tian,XiaoYang Wei,ChengLiang Liu
DOI: https://doi.org/10.1007/s11431-023-2639-9
2024-05-30
Science China Technological Sciences
Abstract:Arrhythmias may lead to sudden cardiac death if not detected and treated in time. A supraventricular premature beat (SPB) and premature ventricular contraction (PVC) are important categories of arrhythmia disease. Recently, deep learning methods have been applied to the PVC/SPB heartbeats detection. However, most researchers have focused on time-domain information of the electrocardiogram and there has been a lack of exploration of the interpretability of the model. In this study, we design an interpretable and accurate PVC/SPB recognition algorithm, called the interpretable multilevel wavelet decomposition deep network (IMWDDN). Wavelet decomposition is introduced into the deep network and the squeeze and excitation (SE)-Residual block is designed for extracting time-domain and frequency-domain features. Additionally, inspired by the idea of residual learning, we construct a novel loss function for the constant updating of the multilevel wavelet decomposition parameters. Finally, the IMWDDN is evaluated on the Third China Physiological Signal Challenge Dataset and the MIT-BIH Arrhythmia database. The comparison results show IMWDDN has better detection performance with 98.51% accuracy and a 93.75% F1-macro on average, and its areas of concern are similar to those of an expert diagnosis to a certain extent. Generally, the IMWDDN has good application value in the clinical screening of PVC/SPB heartbeats.
materials science, multidisciplinary,engineering
What problem does this paper attempt to address?