An Xiang, Shi Shiwen, Wang Qian, Yu Yansuo, Liu Qiang
Academy of Artificial Intelligence, Beijing Institute of Petrochemical Technology, Beijing 102617, China.
Sensors (Basel). 2024 Dec 10;24(24):7896. doi: 10.3390/s24247896.
Arrhythmias are among the diseases with high mortality rates worldwide, causing millions of deaths each year. This underscores the importance of real-time electrocardiogram (ECG) monitoring for timely heart disease diagnosis and intervention. Deep learning models, trained on ECG signals across twelve or more leads, are the predominant approach for automated arrhythmia detection in the AI-assisted medical field. While these multi-lead ECG-based models perform well in automatic arrhythmia detection, their complexity often restricts their use on resource-constrained devices. In this paper, we propose an efficient, lightweight arrhythmia classification model using a knowledge distillation technique to train a student model from a teacher model, tailored for embedded intelligence in wearable devices. The results show that the student model achieves 96.32% accuracy, which is comparable to the teacher model, with a remarkable compression ratio that is 1242.58 times smaller, outperforming other lightweight models. Enabled by the proposed model, we developed a wearable ECG monitoring system based on the STM32F429 Discovery kit and ADS1292R chip, achieving real-time arrhythmia detection on small wearable devices.
心律失常是全球死亡率较高的疾病之一,每年导致数百万人死亡。这凸显了实时心电图(ECG)监测对于及时诊断和干预心脏病的重要性。在十二导联或更多导联的心电图信号上训练的深度学习模型,是人工智能辅助医疗领域中自动心律失常检测的主要方法。虽然这些基于多导联心电图的模型在自动心律失常检测方面表现良好,但其复杂性常常限制了它们在资源受限设备上的应用。在本文中,我们提出了一种高效、轻量级的心律失常分类模型,该模型使用知识蒸馏技术从教师模型训练学生模型,专为可穿戴设备中的嵌入式智能量身定制。结果表明,学生模型的准确率达到了96.32%,与教师模型相当,其压缩比显著小1242.58倍,性能优于其他轻量级模型。基于所提出的模型,我们开发了一种基于STM32F429 Discovery套件和ADS1292R芯片组的可穿戴式心电图监测系统,可在小型可穿戴设备上实现心律失常的实时检测。