Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

MINA: Multilevel Knowledge-Guided Attention for Modeling Electrocardiography Signals

About

Electrocardiography (ECG) signals are commonly used to diagnose various cardiac abnormalities. Recently, deep learning models showed initial success on modeling ECG data, however they are mostly black-box, thus lack interpretability needed for clinical usage. In this work, we propose MultIlevel kNowledge-guided Attention networks (MINA) that predict heart diseases from ECG signals with intuitive explanation aligned with medical knowledge. By extracting multilevel (beat-, rhythm- and frequency-level) domain knowledge features separately, MINA combines the medical knowledge and ECG data via a multilevel attention model, making the learned models highly interpretable. Our experiments showed MINA achieved PR-AUC 0.9436 (outperforming the best baseline by 5.51%) in real world ECG dataset. Finally, MINA also demonstrated robust performance and strong interpretability against signal distortion and noise contamination.

Shenda Hong, Cao Xiao, Tengfei Ma, Hongyan Li, Jimeng Sun• 2019

Related benchmarks

TaskDatasetResultRank
Atrial Fibrillation PredictionPhysioNet 2017 (test)
ROC AUC94.88
6
ECG ClassificationPhysioNet dataset 2017
Average Accuracy85.97
4
Showing 2 of 2 rows

Other info

Code

Follow for update