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 条
[21]  
Sharma LN, 2015, INT CONF ICT KNOWL, P33, DOI 10.1109/ICTKE.2015.7368467
[22]   Short unsegmented PCG classification based on ensemble classifier [J].
Sinam, Sinam Ajitkumar ;
Majumder, Swanirbhar .
TURKISH JOURNAL OF ELECTRICAL ENGINEERING AND COMPUTER SCIENCES, 2020, 28 (02) :875-889
[23]   Logistic Regression-HSMM-Based Heart Sound Segmentation [J].
Springer, David B. ;
Tarassenko, Lionel ;
Clifford, Gari D. .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2016, 63 (04) :822-832
[24]   Global, regional, and national time trends in mortality for congenital heart disease, 1990-2019: An age-period-cohort analysis for the Global Burden of Disease 2019 study [J].
Su, Zhanhao ;
Zou, Zhiyong ;
Hay, Simon I. ;
Liu, Yiwei ;
Li, Shoujun ;
Chen, Huiwen ;
Naghavi, Mohsen ;
Zimmerman, Meghan S. ;
Martin, Gerard R. ;
Wilner, Lauren B. ;
Sable, Craig A. ;
Murray, Christopher J. L. ;
Kassebaum, Nicholas J. ;
Patton, George C. ;
Zhang, Hao .
ECLINICALMEDICINE, 2022, 43
[25]   Automatic moment segmentation and peak detection analysis of heart sound pattern via short-time modified Hilbert transform [J].
Sun, Shuping ;
Jiang, Zhongwei ;
Wang, Haibin ;
Fang, Yu .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2014, 114 (03) :219-230
[26]  
Syed Z.H., 2003, Ph.D. thesis
[27]  
Szegedy C, 2015, PROC CVPR IEEE, P1, DOI 10.1109/CVPR.2015.7298594
[28]  
Vaswani A, 2017, ADV NEUR IN, V30
[29]  
Wang Yuanlin, 2022, Sheng Wu Yi Xue Gong Cheng Xue Za Zhi, V39, P1140, DOI 10.7507/1001-5515.202111059
[30]  
[肖斌 Xiao Bin], 2022, [电子学报, Acta Electronica Sinica], V50, P2425