Atrial fibrillation classification using QRS complex features and LSTM

Abstract

Classification of Atrial Fibrillation from diverse electrocardiographic (ECG) signals is the challenging objective of the 2017 Physionet Challenge. We suggest a Long Short Term Memory (LSTM) network, which learns patterns directly from pre-computed QRS complex features that classifies ECG signals. Although our architecture is considered deep, it only consists of 1791 parameters. The result is an accurate, lightweight solution that classifies ECG records as Normal, Atrial fibrillation, Other or Too noisy with final challenge score of 0.78.