We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Beat by Beat: Classifying Cardiac Arrhythmias with Recurrent Neural Networks

Abstract: With tens of thousands of electrocardiogram (ECG) records processed by mobile cardiac event recorders every day, heart rhythm classification algorithms are an important tool for the continuous monitoring of patients at risk. We utilise an annotated dataset of 12,186 single-lead ECG recordings to build a diverse ensemble of recurrent neural networks (RNNs) that is able to distinguish between normal sinus rhythms, atrial fibrillation, other types of arrhythmia and signals that are too noisy to interpret. In order to ease learning over the temporal dimension, we introduce a novel task formulation that harnesses the natural segmentation of ECG signals into heartbeats to drastically reduce the number of time steps per sequence. Additionally, we extend our RNNs with an attention mechanism that enables us to reason about which heartbeats our RNNs focus on to make their decisions. Through the use of attention, our model maintains a high degree of interpretability, while also achieving state-of-the-art classification performance with an average F1 score of 0.79 on an unseen test set (n=3,658).
Comments: Accepted at Computing in Cardiology (CinC) 2017
Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV); Signal Processing (eess.SP)
DOI: 10.22489/CinC.2017.363-223
Cite as: arXiv:1710.06319 [cs.LG]
  (or arXiv:1710.06319v2 [cs.LG] for this version)

Submission history

From: Patrick Schwab [view email]
[v1] Tue, 17 Oct 2017 14:39:17 GMT (536kb)
[v2] Tue, 24 Oct 2017 09:51:04 GMT (536kb)

Link back to: arXiv, form interface, contact.