用手工特征比图像方法更准,99%准确率识别心电图心跳类型。
Heart Rate Classification in ECG Signals Using Machine Learning and Deep Learning
- 提取心率变异性等时序特征,用LightGBM分类
- 手工特征模型准确率达99%,F1为0.94
- 适合医疗健康领域做心律分析的工程师
本研究通过两种方法对心电图信号中的心跳进行分类:传统机器学习使用手工提取特征,深度学习则将心电信号转换为图像。数据预处理包括降采样、滤波和归一化。第一种方法提取心率变异性(HRV)、均值、方差和RR间隔等特征,训练SVM、随机森林、AdaBoost、LSTM、双向LSTM及LightGBM等分类器。第二种方法采用格拉米安角场(GAF)、马尔可夫转移场(MTF)和递归图(RP)将信号转为图像,再用VGG和Inception等卷积神经网络分类。实验结果表明,LightGBM模型表现最佳,准确率为99%,F1分数达0.94,优于图像类方法(F1为0.85)。SVM和AdaBoost表现较差,说明其不适用于该任务。研究显示,手工特征更能捕捉心电信号的时间与形态变化。未来可结合多导联心电图和连续心跳间的时序依赖提升性能。
原文摘要 · Abstract (English)
This study addresses the classification of heartbeats from ECG signals through two distinct approaches: traditional machine learning utilizing hand-crafted features and deep learning via transformed images of ECG beats. The dataset underwent preprocessing steps, including downsampling, filtering, and normalization, to ensure consistency and relevance for subsequent analysis. In the first approach, features such as heart rate variability (HRV), mean, variance, and RR intervals were extracted to train various classifiers, including SVM, Random Forest, AdaBoost, LSTM, Bi-directional LSTM, and LightGBM. The second approach involved transforming ECG signals into images using Gramian Angular Field (GAF), Markov Transition Field (MTF), and Recurrence Plots (RP), with these images subsequently classified using CNN architectures like VGG and Inception. Experimental results demonstrate that the LightGBM model achieved the highest performance, with an accuracy of 99% and an F1 score of 0.94, outperforming the image-based CNN approach (F1 score of 0.85). Models such as SVM and AdaBoost yielded significantly lower scores, indicating limited suitability for this task. The findings underscore the superior ability of hand-crafted features to capture temporal and morphological variations in ECG signals compared to image-based representations of individual beats. Future investigations may benefit from incorporating multi-lead ECG signals and temporal dependencies across successive beats to enhance classification accuracy further.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。