Atrial fibrillation classification based on convolutional neural networks

Kwang Sig Lee, Sunghoon Jung, Yeongjoon Gil, Ho Sung Son

Research output: Contribution to journalArticle

Abstract

Background: The global age-adjusted mortality rate related to atrial fibrillation (AF) registered a rapid growth in the last four decades, i.e., from 0.8 to 1.6 and 0.9 to 1.7 per 100,000 for men and women during 1990-2010, respectively. In this context, this study uses convolutional neural networks for classifying (diagnosing) AF, employing electrocardiogram data in a general hospital. Methods: Data came from Anam Hospital in Seoul, Korea, with 20,000 unique patients (10,000 normal sinus rhythm and 10,000 AF). 30 convolutional neural networks were applied and compared for the diagnosis of the normal sinus rhythm vs. AF condition: 6 Alex networks with 5 convolutional layers, 3 fully connected layers and the number of kernels changing from 3 to 256; and 24 residual networks with the number of residuals blocks (or kernels) varying from 8 to 2 (or 64 to 2). Results: In terms of the accuracy, the best Alex network was one with 24 initial kernels (i.e., kernels in the first layer), 5,268,818 parameters and the training time of 89 s (0.997), while the best residual network was one with 6 residual blocks, 32 initial kernels, 248,418 parameters and the training time of 253 s (0.999). In general, the performance of the residual network improved as the number of its residual blocks (its depth) increased. Conclusion: For AF diagnosis, the residual network might be a good model with higher accuracy and fewer parameters than its Alex-network counterparts.

Original languageEnglish
Article number206
JournalBMC Medical Informatics and Decision Making
Volume19
Issue number1
DOIs
Publication statusPublished - 2019 Oct 29

Keywords

  • Alex networks
  • Atrial fibrillation
  • Convolutional neural networks
  • Residual networks

ASJC Scopus subject areas

  • Health Policy
  • Health Informatics

Fingerprint Dive into the research topics of 'Atrial fibrillation classification based on convolutional neural networks'. Together they form a unique fingerprint.

  • Cite this