Atrial Fibrillation Detection Using Weight-Pruned, Log-Quantised Convolutional Neural Networks

Deep neural networks (DNN) are a promising tool in medical applications. However, the implementation of complex DNNs on battery-powered devices is challenging due to high energy costs for communication. In this work, a convolutional neural network model is developed for detecting atrial fibrillation...

Full description

Saved in:
Bibliographic Details
Published in:arXiv.org 2022-06
Main Authors: Xiu Qi Chang, Ann Feng Chew, Benjamin Chen Ming Choong, Wang, Shuhui, Han, Rui, Wang, He, Li, Xiaolin, Panicker, Rajesh C, Deepu, John
Format: Article
Language:English
Subjects:
Artificial neural networks
Complexity
Compression ratio
Electrocardiography
Energy costs
Fibrillation
Model accuracy
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Deep neural networks (DNN) are a promising tool in medical applications. However, the implementation of complex DNNs on battery-powered devices is challenging due to high energy costs for communication. In this work, a convolutional neural network model is developed for detecting atrial fibrillation from electrocardiogram (ECG) signals. The model demonstrates high performance despite being trained on limited, variable-length input data. Weight pruning and logarithmic quantisation are combined to introduce sparsity and reduce model size, which can be exploited for reduced data movement and lower computational complexity. The final model achieved a 91.1% model compression ratio while maintaining high model accuracy of 91.7% and less than 1% loss.
ISSN:2331-8422