Enhancing cardiac disease prediction with explainable bidirectional LSTM

利用可解释的双向LSTM增强心脏疾病预测

阅读:1

Abstract

Cardiovascular disorders (heart diseases) are the most prevalent cause of death on a global scale. So early detection and classification increase the likelihood of survival. In the context of machine learning techniques, there is always a need for an accurate and explainable predictive model for detecting various diseases, such as cardiac disorders. The work carried out in this paper stacks bidirectional long short-term memory with deep learning to propose two models. The first model is used to detect cardiac disease with a binary label classification, while the second one classifies cardiac disease, which is a multi-label classification problem. Bidirectional LSTM is used as an approximate algorithm for feature extraction. Deep learning is used for classification purposes. The proposed models are trained and validated over the PTB-XL dataset. The performance of these models is evaluated and compared against state-of-the-art methods. The comparison shows the proposed model outperforms other methods in terms of accuracy, precision, f1-score, and recall. SHAP is used to make these models explainable, which in turn helps to annotate different diseases on the ECG report.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。