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

• Main Authors: Chang, Xiu Qi, Chew, Ann Feng, Choong, Benjamin Chen Ming, Wang, Shuhui, Han, Rui, He, Wang, Xiaolin, Li, Panicker, Rajesh C, John, Deepu
• Format: Journal Article
• Language: English
• Published: 14.06.2022
• Subjects: Computer Science - Learning
• DOI: 10.48550/arxiv.2206.07649

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.