Lightweight Shufflenet Based CNN for Arrhythmia Classification

Huruy Tesfai, Hani Saleh, Mahmoud Al-Qutayri, Moath B. Mohammad, Temesghen Tekeste, Ahsan Khandoker, Baker Mohammad

Research output: Contribution to journalArticlepeer-review

10 Scopus citations

Abstract

Recent advances in artificial intelligence (AI) and continuous monitoring of patients using wearable devices have enhanced the accuracy of diagnosing various arrhythmias, from the captured Electrocardiogram (ECG) signals. Achieving high accuracy when using Deep Neural Network (DNN) for ECG classification is accomplished at the cost of compute and memory intensive operations, thus limiting its deployment to devices with high computing capabilities, and makes it unsuitable for wearable edge devices. To facilitate the deployment of deep neural networks on wearable mobile edge devices with limited resources, a lightweight Convolution Neural Network (CNN) model based on the ShuffleNet architecture is proposed and implemented as a solution in this paper. A sliding window of variable stride is used to increase the number of under-represented classes in the database. Moreover, a novel encoding scheme is employed for labelling and training test set samples, allowing the model to detect multiple classes in one ECG segment. A loss function (Focal loss) that proved to be effective when applied for DNN training on an imbalanced dataset was also explored in this work. The proposed model outperformed traditional CNN with 9x less trainable parameters and improved the F1-score by 2%.

Original languageBritish English
Pages (from-to)111842-111854
Number of pages13
JournalIEEE Access
Volume10
DOIs
StatePublished - 2022

Keywords

  • AI
  • CNN
  • ECG
  • health care
  • wearable electronics

Fingerprint

Dive into the research topics of 'Lightweight Shufflenet Based CNN for Arrhythmia Classification'. Together they form a unique fingerprint.

Cite this