揭示心电图模型规模效应:自监督学习+残差网络更高效
How Do Electrocardiogram Models Scale?

- 对比残差网络与注意力机制,研究不同架构和预训练方式的扩展规律
- 自监督学习在数据效率上比监督学习高16倍,跨任务迁移效率提升7.6倍
- 小模型用残差网络+自监督,大模型用自监督变压器,效果最优
尽管缩放定律为自然语言处理中的基础模型提供了基本框架,但其在心电图(ECG)模型中的适用性仍不明确。近期研究并未始终显示随着模型或预训练数据集规模增大而带来一致的下游性能提升,模型架构归纳偏置、预训练范式及其规模带来的预期改进尚不清晰。本文系统研究了心电图领域的神经网络与损失-损失缩放定律。通过对120个模型(参数量从2万到2亿)在大规模CODE数据集(230万条记录)上进行预训练,解耦了模型架构(残差网络与变换器)和预训练范式(监督学习与自监督学习)的影响。发现:(i)监督学习模型在分布内受数据瓶颈制约,而自监督学习模型在模型和数据规模上均能稳健扩展;(ii)在分布外泛化方面,残差网络比变换器参数效率高1.3至2.5倍,自监督学习在数据效率上最高达监督学习的16倍,且在未见临床任务上的迁移效率高达监督学习的7.6倍;(iii)在观测尺度下,残差网络模型通常取得最低的分布外损失,自监督学习在未见临床任务中占优,而自监督变换器在极大规模时超越其他方法。结果表明,构建高效的心电图基础模型的关键在于架构与范式的战略匹配,而非单纯扩大规模。
原文摘要 · Abstract (English)
While scaling laws have established a fundamental framework for foundation models in natural language processing, their applicability to electrocardiogram (ECG) models remains poorly characterized. Indeed, recent studies do not always yield consistent downstream gains as one increases the model size or pre-training dataset size of ECG models, leaving the exact roles of architectural inductive biases, pre-training paradigms, and expected improvements with size largely unanswered. In this work, we systematically investigate neural and loss-to-loss scaling laws within the ECG domain. By pre-training over $120$ models (ranging from $20$K to $200$M parameters) on the large-scale CODE dataset ($2.3$M records), we decouple the effects of model architecture (ResNet vs. Transformer) and pre-training paradigm, namely supervised learning (SL) versus self-supervised learning (SSL). We found that (i) SL models are data-bottlenecked in-distribution, whereas SSL models scale robustly across both model and data sizes; (ii) for out-of-distribution (OOD) generalization, ResNets are $1.3$ to $2.5$ times more parameter-efficient than Transformers, while SSL is up to $16$ times more data-efficient and achieves up to $7.6$ times higher transfer efficiency than SL on unseen clinical tasks; (iii) across the observed scales, ResNet-based models generally achieve the lowest OOD loss, with SSL dominating on unseen clinical tasks and self-supervised Transformers overtaking at very large model sizes. Our results suggest that the path to effective ECG foundation models lies in the strategic alignment of architecture and paradigm rather than brute-force scaling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。