用轻量逻辑网络实现跨患者心电图心律失常高精度分类。
Inter-patient ECG Arrhythmia Classification with LGNs and LUTNs
- 基于逻辑门与查表网络,设计低功耗可微分模型。
- 在MIT-BIH数据集上达94.28%准确率,jκ指数0.683。
- 适合植入式或可穿戴设备,支持未训练患者泛化。
深度可微分逻辑门网络(LGNs)和查表网络(LUTNs)被证明适用于跨患者心电图(ECG)自动分类。在MIT-BIH心律失常数据集上进行基准测试,四类分类任务中最高准确率达94.28%,jκ指数为0.683。模型总计算量仅2.89k至6.17k FLOPs(含预处理与读出),比当前最优方法低三到六数量级。提出一种新型预处理方法,在混合患者与跨患者范式下均表现更优。同时,设计了一种基于多路选择器(MUX)布尔方程的LUT训练新方法,并首次在LGNs与LUTNs中引入速率编码,提升性能。首次在跨患者范式下对LGNs与LUTNs于MIT-BIH数据集进行基准测试。在Artix 7 FPGA上运行时,需2000至2990个LUTs,功耗5至7 mW(即每推理50至70 pJ),表明其在极低功耗与高速度下具备部署潜力,适用于未参与训练患者的实时心律失常检测。
原文摘要 · Abstract (English)
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κ$ 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κ$-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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。