用机器学习精准识别心电图心律失常,助力远程诊疗
Electrocardiogram (ECG) Based Cardiac Arrhythmia Detection and Classification using Machine Learning Algorithms
- 用优化的双向LSTM做正常与房颤二分类,效果出色
- 升级为CNN模型,五类心律失常分类准确率更高
- 开发在线平台,适合医疗人员和数据科学家协作使用
人工智能,尤其是机器学习(ML)与深度学习(DL),为医学诊断、预后和治疗带来了新机遇。本文致力于构建高精度的机器学习模型,用于心律失常心电图(ECG)信号的分类。研究采用PhysioNet和MIT-BIH数据库中的心电图数据集。初期开展二分类任务,通过优化的双向长短期记忆网络(Bi-LSTM)在区分正常与房颤信号方面表现优异。研究还对医疗专业人员进行调查,验证了基于AI的心电图分类器的实用性,并指出需提升准确率及扩充心律失常类型。据此,开发出可识别五种不同类型心电图信号的先进卷积神经网络(CNN)系统,其性能通过严格的分层5折交叉验证确保。同时搭建了网页门户,实现模型的实时分类应用。该研究展示了此类模型在远程健康监测、预测性医疗、辅助诊断工具以及教育模拟环境中的潜力,促进数据科学与医疗人员的跨学科协作。
原文摘要 · Abstract (English)
The rapid advancements in Artificial Intelligence, specifically Machine Learning (ML) and Deep Learning (DL), have opened new prospects in medical sciences for improved diagnosis, prognosis, and treatment of severe health conditions. This paper focuses on the development of an ML model with high predictive accuracy to classify arrhythmic electrocardiogram (ECG) signals. The ECG signals datasets utilized in this study were sourced from the PhysioNet and MIT-BIH databases. The research commenced with binary classification, where an optimized Bidirectional Long Short-Term Memory (Bi-LSTM) model yielded excellent results in differentiating normal and atrial fibrillation signals. A pivotal aspect of this research was a survey among medical professionals, which not only validated the practicality of AI-based ECG classifiers but also identified areas for improvement, including accuracy and the inclusion of more arrhythmia types. These insights drove the development of an advanced Convolutional Neural Network (CNN) system capable of classifying five different types of ECG signals with better accuracy and precision. The CNN model's robust performance was ensured through rigorous stratified 5-fold cross validation. A web portal was also developed to demonstrate real-world utility, offering access to the trained model for real-time classification. This study highlights the potential applications of such models in remote health monitoring, predictive healthcare, assistive diagnostic tools, and simulated environments for educational training and interdisciplinary collaboration between data scientists and medical personnel.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。