Lightweight Semantic EEG Foundation Model for Frozen Cross-Disorder Transfer

ABSTRACT

中文摘要

大型 EEG 基础模型可以学习通用表示，但参数规模、训练成本以及下游任务是否真正能够在冻结条件下迁移仍是实际障碍。作者提出 USE-FM，一个 146 万参数的轻量 EEG 基础模型，在 Temple University EEG Corpus 上以自监督重建目标预训练，然后完全冻结编码器，仅训练简单下游分类器。在 TUAB 异常 EEG 检测与 TUEP 癫痫检测上，模型分别达到 85.24±5.61 的 AUC，以及 75.00±14.14 的灵敏度和 70.37±4.01 的 F1；作者报告其参数量约为比较模型的五分之一，并与 LUNA-Base、CBraMod 等基础模型比较。PCA、t-SNE 和聚类分析显示，冻结表示能够形成与临床状态相关的结构。结果支持以较小模型在大规模异质 EEG 上预训练，再通过冻结特征完成跨任务和跨疾病迁移。