arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.15312cs.LG

面向可穿戴设备的内存高效心跳分类:多特征融合与压缩Bi-LSTM方法

Multi-Feature Fusion and Compressed Bi-LSTM for Memory-Efficient Heartbeat Classification on Wearable Devices

School of Electrical and Electronic Engineering, University College Dublin(都柏林大学学院电气与电子工程学院);School of Electrical and Electronic Engineering, California Institute of Technology(加州理工学院电气与电子工程学院)

Reza Nikandish, Jiayu He, Benyamin Haghi

AI总结:

本文提出结合多特征融合与压缩Bi-LSTM的资源高效ECG心跳分类方法,通过预处理降噪、融合时间间隔与抗噪曲线下面积特征,在MIT-BIH数据集上显著提升难分类别准确率,并开发多规模模型实现全类别高准确率,适配可穿戴设备。

摘要:

本文提出了一种资源高效的基于心电图(ECG)的心跳分类方法,该方法结合了多特征融合与双向长短期记忆网络(Bi-LSTM)。实验所用数据集包含MIT-BIH Arrhythmia Database的5类原始心跳:正常(N)、左束支传导阻滞(LBBB)、右束支传导阻滞(RBBB)、室性早搏(PVC)和起搏搏动(PB)。研究采用离散小波变换和双移动平均窗口等预处理方法,降低原始ECG信号中的噪声与伪影,并提取ECG波形的关键特征点(PQRST)。多特征融合通过将时间间隔与本文提出的曲线下面积作为输入特征实现,这类特征本身具备较强的抗噪声能力。仿真实验结果表明,加入曲线下面积特征后,分类难度较高的RBBB和LBBB类别的分类准确率得到显著提升:RBBB从31.4%提升至84.3%,LBBB从69.6%提升至87.0%。与传统LSTM网络相比,采用Bi-LSTM网络在RBBB类别上实现了更高的准确率(33.8%对比21.8%),同时所需网络参数减少了28%。研究还开发了多种参数规模的神经网络模型,包括微型(84k)、小型(150k)、中型(478k)和大型(1.25M)模型,以实现“所有类别均具备高准确率”的目标——这是比整体分类准确率更关键、更具挑战性的目标。

英文摘要:

In this article, we present a resource-efficient approach for electrocardiogram (ECG) based heartbeat classification using multi-feature fusion and bidirectional long short-term memory (Bi-LSTM). The dataset comprises five original classes from the MIT-BIH Arrhythmia Database: Normal (N), Left Bundle Branch Block (LBBB), Right Bundle Branch Block (RBBB), Premature Ventricular Contraction (PVC), and Paced Beat (PB). Preprocessing methods including the discrete wavelet transform and dual moving average windows are used to reduce noise and artifacts in the raw ECG signal, and extract the main points (PQRST) of the ECG waveform. Multi-feature fusion is achieved by utilizing time intervals and the proposed under-the-curve areas, which are inherently robust against noise, as input features. Simulations demonstrated that incorporating under-the-curve area features improved the classification accuracy for the challenging RBBB and LBBB classes from 31.4\% to 84.3\% for RBBB, and from 69.6\% to 87.0\% for LBBB. Using a Bi-LSTM network, rather than a conventional LSTM network, resulted in higher accuracy (33.8\% vs 21.8\%) with a 28\% reduction in required network parameters for the RBBB class. Multiple neural network models with varying parameter sizes, including tiny (84k), small (150k), medium (478k), and large (1.25M) models, are developed to achieve high accuracy \textit{across all classes}, a more crucial and challenging goal than overall classification accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑