Interpretable End-to-End heart sound classification

被引:3
作者
Li, Shuaizhong [1 ]
Sun, Jing [1 ]
Yang, Hongbo [2 ]
Pan, Jiahua [2 ]
Guo, Tao [2 ]
Wang, Weilian [1 ]
机构
[1] Yunnan Univ, Sch Informat Sci & Technol, South East Outer Ring Rd, Kunming 650504, Peoples R China
[2] Fuwai Yunnan Cardiovasc Hosp, Kunming 650102, Peoples R China
基金
中国国家自然科学基金;
关键词
Heart sound; Congenital heart disease (CHD); End-to-End; Muti-head self-attention; Interpretability; SEGMENTATION;
D O I
10.1016/j.measurement.2024.115113
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Heart sound analysis is a non-invasive and economical technique that can aid in diagnosing cardiovascular disease. A novel End-to-End heart sound classification method was proposed in this paper, in which a combination of multi-scale dense network and multi-head recurrent neural network technology was used. It can be used to diagnose congenital heart disease (CHD) without using the manual extraction of features. An F-beta score of 94.33% and an accuracy of 94.41% were achieved by the method on dataset A, which consisted of 1,000 individuals and 5,000 signals. Similarly, the widely used dataset B (Physio Net/CinC 2016 dataset), comprising 764 individuals and 3,240 signals, resulted in an F-beta score of 93.75% and an accuracy of 92.97%. The results show the proposed method had a significant potential to assist in diagnosing CHD. The SHAP algorithm which is a kind of Interpretable method was applied in this study to interpret the prediction results of model. It was shown that the model's prediction process is similar to a doctor's diagnosing mode.
引用
收藏
页数:8
相关论文
共 33 条
[1]   Screening of Heart Sounds using Hidden Markov and Gammatone Filterbank Models [J].
Alexander, Ben ;
Nallathambi, Gabriel ;
Selvaraj, Nandakumar .
2018 17TH IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND APPLICATIONS (ICMLA), 2018, :1460-1465
[2]  
Asmare MH, 2020, IEEE ENG MED BIO, P168, DOI [10.1109/embc44109.2020.9176544, 10.1109/EMBC44109.2020.9176544]
[3]   Window Size Impact in Human Activity Recognition [J].
Banos, Oresti ;
Galvez, Juan-Manuel ;
Damas, Miguel ;
Pomares, Hector ;
Rojas, Ignacio .
SENSORS, 2014, 14 (04) :6474-6499
[4]   The Effect of Signal Duration on the Classification of Heart Sounds: A Deep Learning Approach [J].
Bao, Xinqi ;
Xu, Yujia ;
Kamavuako, Ernest Nlandu .
SENSORS, 2022, 22 (06)
[5]  
Chen H., 2021, Explainable AI in Healthcare and Medicine: Building a Culture of Transparency and Accountability, V914, P261, DOI [DOI 10.1007/978-3-030-53352-624, 10.1007/978-3-030-53352-6_24]
[6]   End-to-end heart sound segmentation using deep convolutional recurrent network [J].
Chen, Yao ;
Sun, Yanan ;
Lv, Jiancheng ;
Jia, Bijue ;
Huang, Xiaoming .
COMPLEX & INTELLIGENT SYSTEMS, 2021, 7 (04) :2103-2117
[7]   Comparison of envelope extraction algorithms for cardiac sound signal segmentation [J].
Choi, Samjin ;
Jiang, Zhongwei .
EXPERT SYSTEMS WITH APPLICATIONS, 2008, 34 (02) :1056-1069
[8]   Time-Frequency Analysis, Denoising, Compression, Segmentation, and Classification of PCG Signals [J].
Chowdhury, Md Tanzil Hoque ;
Poudel, Khem Narayan ;
Hu, Yating .
IEEE ACCESS, 2020, 8 :160882-160890
[9]   SpectroCardioNet: An Attention-Based Deep Learning Network Using Triple-Spectrograms of PCG Signal for Heart Valve Disease Detection [J].
Chowdhury, Sakib ;
Morshed, Monjur ;
Fattah, Shaikh Anowarul .
IEEE SENSORS JOURNAL, 2022, 22 (23) :22799-22807
[10]   Heart sound classification based on improved MFCC features and convolutional recurrent neural networks [J].
Deng, Muqing ;
Meng, Tingting ;
Cao, Jiuwen ;
Wang, Shimin ;
Zhang, Jing ;
Fan, Huijie .
NEURAL NETWORKS, 2020, 130 :22-32