构建首个重症监护室房颤检测数据集与基准,验证大模型效果最佳。
A Dataset and Benchmarks for Atrial Fibrillation Detection from Electrocardiograms of Intensive Care Unit Patients
- 对比特征、深度学习与心电图基础模型三类AI方法
- 基于迁移学习的心电图基础模型F1达0.89
- 适合心脏病学、AI医疗研究者参考
房颤是重症监护室患者最常见的心律失常,可能引发不良健康后果。本研究发布了一个标注的重症监护室心电图数据集及房颤检测基准。通过对比三种数据驱动的人工智能方法——基于特征的分类器、深度学习(DL)和心电图基础模型(FMs),填补了文献中的关键空白,旨在确定最适用于准确房颤检测的AI方法。实验使用加拿大重症监护室数据及2021年PhysioNet/Computing in Cardiology挑战赛数据。测试了从零样本推理到迁移学习等多种训练配置。结果显示,平均而言,在两个数据集上,心电图基础模型表现最佳,其次为深度学习,最后是基于特征的分类器。在本研究的重症监护室测试集中,通过迁移学习策略达到最高F1分数的模型为心电图基础模型(F1=0.89)。研究证明了利用AI构建自动患者监测系统的前景。通过发布标注的重症监护室数据集(链接待补充)和性能基准,本工作使研究社区能够持续推动重症监护室房颤检测的前沿发展。
原文摘要 · Abstract (English)
Objective: Atrial fibrillation (AF) is the most common cardiac arrhythmia experienced by intensive care unit (ICU) patients and can cause adverse health effects. In this study, we publish a labelled ICU dataset and benchmarks for AF detection. Methods: We compared machine learning models across three data-driven artificial intelligence (AI) approaches: feature-based classifiers, deep learning (DL), and ECG foundation models (FMs). This comparison addresses a critical gap in the literature and aims to pinpoint which AI approach is best for accurate AF detection. Electrocardiograms (ECGs) from a Canadian ICU and the 2021 PhysioNet/Computing in Cardiology Challenge were used to conduct the experiments. Multiple training configurations were tested, ranging from zero-shot inference to transfer learning. Results: On average and across both datasets, ECG FMs performed best, followed by DL, then feature-based classifiers. The model that achieved the top F1 score on our ICU test set was ECG-FM through a transfer learning strategy (F1=0.89). Conclusion: This study demonstrates promising potential for using AI to build an automatic patient monitoring system. Significance: By publishing our labelled ICU dataset (LinkToBeAdded) and performance benchmarks, this work enables the research community to continue advancing the state-of-the-art in AF detection in the ICU.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。