Inter-patient ECG Arrhythmia Classification with LGNs and LUTNs

TL;DR

This paper introduces novel deep logic gate and lookup table networks for ECG arrhythmia classification, achieving high accuracy with significantly reduced computational complexity and power consumption, suitable for wearable and implantable devices.

Contribution

It presents new methods for training LGNs and LUTNs, including a novel LUT training approach and rate coding, and benchmarks these models on the inter-patient MIT-BIH dataset with superior results.

Findings

Achieved up to 94.28% accuracy on MIT-BIH dataset.

Models use 3-6 orders of magnitude fewer FLOPs than SOTA.

Low power consumption of 5-7 mW on FPGA.

Abstract

Deep Differentiable Logic Gate Networks (LGNs) and Lookup Table Networks (LUTNs) are demonstrated to be suitable for the automatic classification of electrocardiograms (ECGs) using the inter-patient paradigm. The methods are benchmarked using the MIT-BIH arrhythmia data set, achieving up to 94.28% accuracy and a $j\kappa$ index of 0.683 on a four-class classification problem. Our models use between 2.89k and 6.17k FLOPs, including preprocessing and readout, which is three to six orders of magnitude less compared to SOTA methods. A novel preprocessing method is utilized that attains superior performance compared to existing methods for both the mixed-patient and inter-patient paradigms. In addition, a novel method for training the Lookup Tables (LUTs) in LUTNs is devised that uses the Boolean equation of a multiplexer (MUX). Additionally, rate coding was utilized for the first time in these LGNs and LUTNs, enhancing the performance of LGNs. Furthermore, it is the first time that LGNs and LUTNs have been benchmarked on the MIT-BIH arrhythmia dataset using the inter-patient paradigm. Using an Artix 7 FPGA, between 2000 and 2990 LUTs were needed, and between 5 to 7 mW (i.e. 50 pJ to 70 pJ per inference) was estimated for running these models. The performance in terms of both accuracy and $j\kappa$-index is significantly higher compared to previous LGN results. These positive results suggest that one can utilize LGNs and LUTNs for the detection of arrhythmias at extremely low power and high speeds in heart implants or wearable devices, even for patients not included in the training set.