Compressed Deep Learning to Classify Arrhythmia in an Embedded Wearable Device

  • Kwang Sig Lee
  • , Hyun Joon Park
  • , Ji Eon Kim
  • , Hee Jung Kim
  • , Sangil Chon
  • , Sangkyu Kim
  • , Jaesung Jang
  • , Jin Kook Kim
  • , Seongbin Jang
  • , Yeongjoon Gil
  • , Ho Sung Son*
  • *Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

The importance of an embedded wearable device with automatic detection and alarming cannot be overstated, given that 15–30% of patients with atrial fibrillation are reported to be asymp-tomatic. These asymptomatic patients do not seek medical care, hence traditional diagnostic tools including Holter are not effective for the further prevention of associated stroke or heart failure. This is likely to be more so in the era of COVID-19, in which patients become more reluctant on hospitalization and checkups. However, little literature is available on this important topic. For this reason, this study developed efficient deep learning with model compression, which is designed to use ECG data and classify arrhythmia in an embedded wearable device. ECG-signal data came from Korea University Anam Hospital in Seoul, Korea, with 28,308 unique patients (15,412 normal and 12,896 arrhythmia). Resnets and Mobilenets with model compression (TensorFlow Lite) were applied and compared for the diagnosis of arrhythmia in an embedded wearable device. The weight size of the compressed model registered a remarkable decrease from 743 MB to 76 KB (1/10000), whereas its performance was almost the same as its original counterpart. Resnet and Mobilenet were similar in terms of accuracy, i.e., Resnet-50 Hz (97.3) vs. Mo-bilenet-50 Hz (97.2), Resnet-100 Hz (98.2) vs. Mobilenet-100 Hz (97.9). Here, 50 Hz/100 Hz denotes the down-sampling rate. However, Resnets took more flash memory and longer inference time than did Mobilenets. In conclusion, Mobilenet would be a more efficient model than Resnet to classify arrhythmia in an embedded wearable device.

Original languageEnglish
Article number1776
JournalSensors
Volume22
Issue number5
DOIs
Publication statusPublished - 2022 Mar 1

Bibliographical note

Publisher Copyright:
© 2022 by the authors. Licensee MDPI, Basel, Switzerland.

Keywords

  • Arrhythmia
  • Compressed deep learning
  • Embedded wearable device
  • Mobilenet
  • Resnet

ASJC Scopus subject areas

  • Analytical Chemistry
  • Information Systems
  • Atomic and Molecular Physics, and Optics
  • Biochemistry
  • Instrumentation
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Compressed Deep Learning to Classify Arrhythmia in an Embedded Wearable Device'. Together they form a unique fingerprint.

Cite this