FOUND-AF:用于房颤检测的ECG基础模型基准测试

FOUND-AF: Benchmarking ECG Foundation Models for Atrial Fibrillation Detection

精选理由

房颤检测选哪个ECG基础模型?FOUND-AF用4个数据集统一测了9个模型,ECGFounder综合最强。

AI 摘要

FOUND-AF基准测试在AFDB、CinC2017、CPSC2021和LTAFDB四个ECG数据集上,以冻结特征提取器加XGBoost分类器的统一流程,评估了来自五个家族的九个公开基础模型,包括HuBERT-ECG、CLEF、ST-MEM、ECG-JEPA和ECGFounder。ECGFounder在准确率、模型大小、推理时间和内存占用之间取得最佳平衡。该框架采用记录级分组交叉验证和配对bootstrap统计检验,确保结果可比。

原文 · arXiv cs.LG

FOUND-AF: Benchmarking ECG Foundation Models for Atrial Fibrillation Detection

Atrial fibrillation (AF) is the most common sustained cardiac arrhythmia and is associated with increased risks of stroke, heart failure, and mortality. Recent ECG foundation models offer transferable representations for automated AF detection. However, their relative effectiveness remains unclear because existing studies use different datasets, preprocessing procedures, classifiers, and validation protocols. This study presents FOUND-AF, a unified, leakage-controlled, and deployment-oriented benchmarking framework that evaluates the quality of pretrained ECG representations under identical experimental conditions. Nine publicly available foundation models from five families, including HuBERT-ECG, CLEF, ST-MEM, ECG-JEPA, and ECGFounder, were evaluated across four heterogeneous ECG datasets, namely AFDB, CinC2017, CPSC2021, and LTAFDB. All models were used as frozen feature extractors with standardized preprocessing, model-native resampling, a fixed XGBoost classifier, and recording-level grouped cross-validation. The evaluation included classification metrics, receiver operating characteristic analysis, paired recording-level bootstrap comparisons with Holm correction, embedding-space visualization, and computational efficiency profiling. The ECGFounder model consistently achieved the strongest overall performance across datasets while offering a favorable trade-off between accuracy, model size, inference time, and memory usage. FOUND-AF therefore provides a reproducible framework for selecting ECG foundation models and demonstrates that compact, clinically pretrained encoders can support robust and computationally efficient AF detection across heterogeneous acquisition settings.