BrainTaskonomy: Learning How to Pretrain and What to Transfer in fMRI Foundation Models

ABSTRACT

中文摘要

fMRI 基础模型越来越多地汇集不同脑状态、队列和采集条件下的异质数据，但现有预训练通常把所有数据域视为平坦混合，下游任务也彼此独立地适配。作者研究能否在不修改主干模型的前提下，用实际测得的学习关系来组织预训练和迁移两个阶段。预训练阶段使用轻量 Brain-DiT 代理模型，在十个 fMRI 数据域之间估计学习难度和有方向的促进关系，据此形成按优先级累积加入数据域的课程，并与从高噪声到低噪声的扩散时间步调度及联合巩固结合。适配阶段则在十五项任务之间执行受控的一阶与高阶迁移，构建有向任务谱系；随后用预算约束整数规划选择需要直接监督的源任务，以及通向不同目标任务的特定迁移路径。与在数据域和扩散时间步上均匀采样相比，联合的优先域课程与高到低噪声课程分别把体素归一化均方误差、功率谱密度归一化均方误差和功能连接均方误差降低 6.5%、16.3% 和 10.5%，并在六项域内及域外任务上取得较强表现。任务谱系进一步显示，迁移关系具有明显不对称性并依赖具体目标；探索性的封闭测试还表明，当允许更高阶路径时，整数规划策略比匹配随机策略获得更大的描述性增益。结果支持依据测得的学习关系来安排 fMRI 预训练域与任务适配，而不是把它们当作彼此独立的平坦集合。