论文标题

解释单铅心电图心律失常分类的深神网络

Interpreting Deep Neural Networks for Single-Lead ECG Arrhythmia Classification

论文作者

Vijayarangan, Sricharan, Murugesan, Balamurali, R, Vignesh, SP, Preejith, Joseph, Jayaraj, Sivaprakasam, Mohansankar

论文摘要

心律不齐是心脏病中发病率和死亡率的普遍原因。早期诊断对于为患有心律不齐的患者提供干预至关重要。传统上,诊断是通过心脏病专家检查心电图(ECG)来进行的。由于缺乏对专家心脏病专家的可访问性,这种诊断方法受到了阻碍。在相当长的一段时间内,信号处理方法已被用于自动化心律不齐的诊断。但是,这些传统方法需要专家知识,并且无法建模广泛的心律不齐。最近,深度学习方法为进行心律不齐的诊断提供了解决方案。但是,这些模型的黑盒性质禁止心律不齐的临床解释。迫切需要将所获得的模型输出与ECG的相应段相关联。为此,提出了两种方法来为模型提供解释性。第一种方法是梯度加权类激活图(GRAD-CAM)的新颖应用,用于可视化CNN模型的显着性。在第二种方法中,通过学习LSTM模型的输入缺失掩码来得出显着性。可视化是在通过与基准的比较来确定能力的模型上提供的。模型显着性的结果不仅可以深入了解模型的预测能力,而且还与心律不齐的医学文献一致。

Cardiac arrhythmia is a prevalent and significant cause of morbidity and mortality among cardiac ailments. Early diagnosis is crucial in providing intervention for patients suffering from cardiac arrhythmia. Traditionally, diagnosis is performed by examination of the Electrocardiogram (ECG) by a cardiologist. This method of diagnosis is hampered by the lack of accessibility to expert cardiologists. For quite some time, signal processing methods had been used to automate arrhythmia diagnosis. However, these traditional methods require expert knowledge and are unable to model a wide range of arrhythmia. Recently, Deep Learning methods have provided solutions to performing arrhythmia diagnosis at scale. However, the black-box nature of these models prohibit clinical interpretation of cardiac arrhythmia. There is a dire need to correlate the obtained model outputs to the corresponding segments of the ECG. To this end, two methods are proposed to provide interpretability to the models. The first method is a novel application of Gradient-weighted Class Activation Map (Grad-CAM) for visualizing the saliency of the CNN model. In the second approach, saliency is derived by learning the input deletion mask for the LSTM model. The visualizations are provided on a model whose competence is established by comparisons against baselines. The results of model saliency not only provide insight into the prediction capability of the model but also aligns with the medical literature for the classification of cardiac arrhythmia.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源