Convolutional recurrent neural networks for electrocardiogram classification

Authors

Martin Zihlmann, Dmytro Perekrestenko, and Michael Tschannen

Reference

2017 Computing in Cardiology (CinC), Rennes, France, pp. 1-4, Sept. 2017.

DOI: 10.22489/CinC.2017.070-060

Abstract

We propose two deep neural network architectures for classification of arbitrary-length electrocardiogram (ECG) recordings and evaluate them on the atrial fibrillation (AF) classification data set provided by the PhysioNet/CinC Challenge 2017. The first architecture is a deep convolutional neural network (CNN) with averaging-based feature aggregation across time. The second architecture combines convolutional layers for feature extraction with long-short term memory (LSTM) layers for temporal aggregation of features. As a key ingredient of our training procedure we introduce a simple data augmentation scheme for ECG data and demonstrate its effectiveness in the AF classification task at hand. The second architecture was found to outperform the first one, obtaining an F1 score of 82.1% on the hidden challenge testing set.

Keywords

electrocardiogram, convolutional recurrent neural network, classification, atrial fibrillation

Comments

Code available at https://github.com/yruffiner/ecg-classification.

Download this document:

This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. In most cases, these works may not be reposted without the explicit permission of the copyright holder.