用17个特征工程+简单神经网络,97.36%准确率实现心电图分类。
Compact Neural Network Algorithm for Electrocardiogram Classification
- 采用17个手工设计特征+轻量ANN,避免深度学习依赖
- 在MIT-BIH和St. Petersburg数据库上达97.36%准确率
- 适合算力有限的临床实时诊断场景
本文提出一种高效紧凑的心电图(ECG)分类算法,用于心律失常诊断,旨在减少对深度学习和卷积神经网络(CNN)的依赖。当前方法常需大量计算资源和标注数据,本工作通过简单结构的神经网络结合先进特征工程,提取17个关键特征,利用数学变换、信号处理与数据提取算法,高效捕捉心电图的形态与生理特征。该模型在识别房颤、窦性心动过速、窦性心动过缓和心室扑动四种心律失常任务中,性能接近主流先进模型。在MIT-BIH与圣彼得堡INCART心律失常数据库上,准确率达97.36%。该方法为资源受限环境中的实时心脏疾病诊断提供了可行方案。
原文摘要 · Abstract (English)
In this paper, we present a powerful, compact electrocardiogram (ECG) classification algorithm for cardiac arrhythmia diagnosis that addresses the current reliance on deep learning and convolutional neural networks (CNNs) in ECG analysis. This work aims to reduce the demand for deep learning, which often requires extensive computational resources and large labeled datasets. Our approach introduces an artificial neural network (ANN) with a simple architecture combined with advanced feature engineering techniques. A key contribution of this work is the incorporation of 17 engineered features that enable the extraction of critical patterns from raw ECG signals. By integrating mathematical transformations, signal processing methods, and data extraction algorithms, our model captures the morphological and physiological characteristics of ECG signals with high efficiency, without requiring deep learning. Our method demonstrates a similar performance to other state-of-the-art models in classifying 4 types of arrhythmias, including atrial fibrillation, sinus tachycardia, sinus bradycardia, and ventricular flutter. Our algorithm achieved an accuracy of 97.36% on the MIT-BIH and St. Petersburg INCART arrhythmia databases. Our approach offers a practical and feasible solution for real-time diagnosis of cardiac disorders in medical applications, particularly in resource-limited environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。