因果语言模型的短暂改道:改进编码器的领域继续预训练 A Causal Language Modeling Detour Improves Encoder Continued Pretraining
编码器继续预训练中插入临时 CLM 阶段,显著提升生物医学领域表现。
前置知识
Masked Language Modeling (MLM)
掩码语言建模,编码器(如 BERT)的标准预训练目标:随机遮盖输入中约 15% 的 token,让模型根据双向上下文预测被遮盖的 token。训练时注意力是双向的,每个位置都能看到所有其他位置。
本文所有基线和 decay 阶段都使用 MLM,理解 MLM 的双向注意力和稀疏监督(只有被 mask 的位置计算损失)是看懂 CLM detour 效果差异的前提。
Causal Language Modeling (CLM)
因果语言建模,也叫自回归语言建模,GPT 系列的标准训练目标:用因果(单向)注意力遮罩,模型只能看到当前位置之前的 token,通过下一 token 预测来训练。每个位置都参与损失计算,监督信号密集。
本文核心创新就是临时把 MLM 换成 CLM,必须理解 CLM 的因果遮罩和密集监督这两个关键差异,才能明白为什么切换目标会改变低层表征。
Centered Kernel Alignment (CKA)
一种衡量两组神经网络表征相似度的方法,对表征做线性变换后计算相似度,输出 0–1 的标量(1 表示完全相同)。本文使用其变体 linear CKA,按层比较两个模型的中间表征,并用 1-CKA 作为发散度(divergence)指标。
全文分析依赖 CKA 来量化 CLM 阶段对各层表征的影响程度,是支撑'CLM 主要改变低层'这一核心论断的实验工具。
ModernBERT 架构
Warner et al. 2025 提出的现代编码器架构,结合 FlashAttention、旋转位置编码(RoPE)、局部-全局交替注意力以及 unpadding,支持 8,192 token 的长上下文。Base 大小约 150M 参数,Large 约 350M 参数。
本文所有模型都基于 ModernBERT,长上下文能力是其在生物医学长文档(出院小结、肿瘤报告)上有效的基础;French 版本叫 ModernCamemBERT。
领域自适应继续预训练 (Domain-Adaptive CPT)
在通用语料预训练好的模型基础上,再用目标领域(如生物医学)的无标注文本继续训练。Gururangan et al. 2020 证明这种策略在领域差距大时收益最大。本文在 ModernBERT 之上用生物医学语料做继续预训练。
CLM detour 是对继续预训练流程的改进,必须理解'继续预训练+领域适配'的范式才能评估这个改进的实际价值。
研究动机
把通用编码器适配到生物医学等领域时,标准做法是继续用 MLM 在领域语料上训练,但这一策略面临两个关键瓶颈。第一,长上下文缺失:BioBERT、PubMedBERT、CamemBERT-bio 等主流生物医学编码器都受限于 BERT 原始的 512 token 上下文窗口,而临床文档(出院小结、肿瘤报告、电子病历)常常超过数千 token,被截断后会丢失关键诊断与编码信息。第二,领域差距与训练目标的张力:ModernBERT 已经在网页、代码、含 PubMed 的科学文献上预训练过,其低层已部分编码生物医学特征,继续 MLM 难以进一步重塑低层表征;现代 CamemBERT 则完全没用生物医学语料预训练,领域差距巨大但缺乏针对性机制。两种情况下,研究者只能机械地堆叠 MLM 训练步骤,无法解释为什么这样做有效,也无法回答'该不该继续 MLM''应该改哪些层'等基本问题。
本文的目标是本文提出 CLM detour 配方:把继续预训练拆成两阶段——Phase 1 用因果注意力和下一 token 预测(CLM)在 10B 级别领域语料上训练,Phase 2 恢复双向注意力和 MLM 做一段 10% 长度的 decay,目标是在不改变模型架构的前提下显著提升下游表现,并在英语和法语生物医学任务上同时刷新 SOTA,最终在 Base 和 Large 两个规模上发布开源的 ModernBERT-bio 与 ModernCamemBERT-bio。同时,本文希望系统回答'为什么一个临时的目标切换能留下持久收益',把训练目标序列对最终模型的影响讲清楚。
与已有工作不同的是,现有继续预训练工作(BioBERT、PubMedBERT、BioClinical-ModernBERT、CamemBERT-bio 等)几乎全部沿用纯 MLM 路线,AntLM、Gisserot-Boukhlef et al. 等少数工作虽然探索了 CLM/MLM 混合目标,但要么只在小规模(10M 词)实验,要么只从零预训练,且都没有回答'为什么切换目标有效'。本文的独特切入角度是:(1) 在 10B–50B token 规模、8192 token 长上下文的真实生物医学继续预训练场景中验证 CLM detour;(2) 用 CKA 把'CLM 对各层的影响程度'量化出来,发现低层 0–7 的 CKA 发散度是 seed noise 的 5–44 倍;(3) 用 freeze 干预给出因果证据——冻结低层让收益消失,冻结中层则保留收益;(4) 与 layer transplantation 配合,证明收益在训练中是必要的,但训练后无法被局部定位。
核心方法
直觉上,MLM 只在 15% 的位置产生监督信号,模型可以'偷懒'只学好这些位置的上下文;CLM 则在每个位置都产生损失,迫使每个位置都学会有意义的表征,对低层的影响也更彻底。基于这个观察,作者提出 CLM detour:把继续预训练切成 Phase 1(CLM,causal mask,next-token prediction)和 Phase 2(MLM,bidirectional mask,15% masking,1B 级别短 decay)两段,整体架构、词表、优化器状态都保持不变,只改注意力遮罩和损失函数。在 Phase 1 的密集监督下,低层表征被深度重塑;Phase 2 用一段短 MLM 把模型'接回'双向注意力的工作模式,并不会擦除这些低层变化。整条管线和纯 MLM 基线在数据、算力、超参数上完全对齐,唯一的差别就是 Phase 1 的目标。
核心创新是'目标切换 + 短 decay'的范式,区别于已有方法的本质在三点。第一,不是从零预训练(区别于 Gisserot-Boukhlef et al. 的 biphasic 方案),而是把已预训练好的 ModernBERT 编码器接进 CLM 阶段,所以可以低成本继承通用语义。第二,不是简单交替 epoch(区别于 AntLM),而是固定顺序的两阶段:先用 CLM 强制重塑,再用一段 MLM 衰减稳定成双向编码器。第三,配合 CKA 表征分析和 freeze 干预,把'为什么有效'从黑箱变成可验证的因果机制——CLM 主要改写低层(layers 0–7),中层(8–14)不必要,decay 不会擦除低层印记。这把'训练目标序列影响最终表征'从经验现象升级为有理论支撑的设计原则。
方法步骤详情
完整方法分四步。步骤 1:初始化与数据准备。Base 模型从 ModernBERT(英语)或 ModernCamemBERT(法语)开始,Large 同理;法语用 10B token(7B 法语生物医学文献 + 2B 合成医疗 QA + 400M E3C 临床病例 + 600M EMA 药品说明书),英语用 10B 或 50B token(Biomed-Enriched 60–80% + 医疗指令 20% + MIMIC-III 20%),数据按 FineWeb-Edu 风格用 Qwen3-235B 打分并上采样高质量段落。步骤 2:Phase 1 CLM 训练。把双向 attention mask 替换为 causal mask,损失换成 next-token prediction(100% 位置都有监督),训练 10B(Base 法语/英语)或 25B(Large 法语)或 50B(Large 英语)token,peak lr $2\times10^{-4}$,AdamW $\beta_1=0.9, \beta_2=0.98$,weight decay $10^{-5}$,global batch 384 条序列(约 3.1M token),bf16 混合精度,4×H100。步骤 3:Phase 2 MLM decay。恢复 bidirectional attention,mask 率 15%,训练 token 量是 Phase 1 的 10%,学习率按 $1-\sqrt{t/T}$ 调度从峰值降到峰值的 10%,无 warmup,优化器状态保留。步骤 4:评估与分析。8 个法语任务(FrACCO-30/100、CANTEMIST、DISTEMIST、MedDialog、DiaMED、EMEA NER、Medline NER)和 11 个英语任务(ChemProt、Phenotype、COS、Social History、DEID、AnatEM、BC5CDR、JNLPBA、NCBI、GAD、HoC)上做 9(法语)/5(英语)seed 的 macro-F1 评估;用 500 篇 held-out 文本按层计算 linear CKA 发散度 $1-\text{CKA}$,与 seed-noise baseline 对比;做 3 个 freeze 干预实验和 layer transplantation 验证因果机制。
技术新颖性
技术新颖性集中在四点。其一,'临时 CLM detour + 短 MLM decay'作为通用继续预训练范式,区别于从零预训练的 biphasic 方案和交替 epoch 方案,证明了目标切换不一定要贯穿整个训练。其二,在 10B–50B 规模、8192 token 长上下文、跨语种(英/法)的真实生物医学场景中系统验证有效性,并刷新多个 SOTA。第三,CKA 与 freeze 干预的组合实验设计:先通过 CKA 定位'CLM 在低层影响最大',再通过 freeze 反向验证'冻结低层则收益消失、冻结中层则收益保留',把观察性结论升级为因果证据。其四,layer transplantation 揭示了'训练中必要 ≠ 训练后可定位',低层 CLM 印记必须和后续层的协同训练一起存在才能发挥作用,这与 transfer learning 中'lower layers carry reusable features'的直觉形成对比。
实验结果
实验覆盖 19 个生物医学任务和 4 个模型规模/语种组合,CLM detour 在所有组合下都优于匹配算力的 MLM 基线。法语 Base(150M)平均 macro-F1 61.6% vs MLM 58.9%,提升 +2.8pp,是所有组合中最大的收益,且在 8 个任务上全部获胜;法语 Large(350M)64.2% vs 63.0%,+1.2pp。英语 Base 10B 77.8% vs 77.3%,+0.5pp;英语 Base 50B 78.0% vs 77.7%,+0.3pp;英语 Large 50B 78.7% vs 77.9%,+0.8pp,CLM 在 11 个任务中赢 7 个。英语提升小于法语的根本原因已被实验确认:ModernBERT 预训练语料已包含 PubMed 等生物医学文本,低层已部分编码领域特征,可重塑空间小;ModernCamemBERT 完全没用生物医学语料,领域差距大,CLM 收益也大。作者由此推测对低资源生物医学语言(意大利、西班牙、葡萄牙语)会有更大收益。Decay 比例扫参表明 10% 的 CLM 长度是甜点(2.5–4% 短 1.0pp,20–50% 长无额外收益)。Needle-in-haystack 实验显示 CLM 62.2% vs MLM 51.6%,+10.7pp,CLM 在所有上下文长度(512–8192)和所有 needle 位置(start/middle/end)都更好,且对长文档退化的鲁棒性更强。CKA 分析揭示了机制:CLM 在低层 0–7 的发散度是 seed-noise baseline 的 5–44 倍,中深层则接近 1×;冻结低层(0–7)期间跑 CLM 把 F1 从 61.6% 砸到 59.3%(与 MLM 基线 58.9% 统计不可区分,p=0.25),冻结中层(8–14)则保留 61.2% 的收益,冻结低层只发生在 decay 阶段则影响极小(-0.4pp)。Decay 阶段把低层 CLM 印记牢牢保留住,CKA 发散度 1.5B token 后稳定在 56.5%,再训 8B token 也只波动 0.6pp。规模缩放方面,Large(350M)encoder 的 CKA 发散度 67.2% ± 0.3% 高于 Base(150M)56.5% ± 0.7%,CLM 印记随容量放大;decoder 控制实验用 Gemma-3 270M 反向做 MLM→CLM,发散度仅 26.3% ± 1.0%,约为同规模 encoder 的 2.6× 之差,提示 encoder 架构对 CLM 印记更友好。Layer transplantation 实验显示低/中/后段层组单独移植最多只能恢复 DiaMED 7.1pp gap 中的 35%,'训练中必要 ≠ 训练后可定位'是 CLM 印记的关键特征。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| French biomedical (Base 150M, 8 tasks avg) | macro-F1 (avg over 8 tasks, 9 seeds) | 61.6% | 58.9% (MLM baseline) | +2.8pp |
| French biomedical (Large 350M, 8 tasks avg) | macro-F1 (avg over 8 tasks, 9 seeds) | 64.2% | 63.0% (MLM baseline) | +1.2pp |
| English biomedical (Base 150M @ 10B, 11 tasks avg) | macro-F1 (avg over 11 tasks, 5 seeds) | 77.8% | 77.3% (MLM baseline) | +0.5pp |
| English biomedical (Base 150M @ 50B, 11 tasks avg) | macro-F1 (avg over 11 tasks, 5 seeds) | 78.0% | 77.7% (MLM baseline) | +0.3pp |
| English biomedical (Large 350M @ 50B, 11 tasks avg) | macro-F1 (avg over 11 tasks, 5 seeds) | 78.7% | 77.9% (MLM baseline) | +0.8pp |
| Needle-in-haystack (CLM vs MLM, 512-8192 tokens) | binary accuracy (avg over positions and lengths) | 62.2% | 51.6% (MLM) | +10.7pp |
| FrACCO-30 (French, long-context oncology multilabel) | macro-F1 (9 seeds) | 74.8% | 69.9% (MLM baseline) | +4.9pp |
| CANTEMIST (French, multilabel classification) | macro-F1 (9 seeds) | 71.0% | 64.9% (MLM baseline) | +6.1pp |
| Freeze low layers (0-7) during CLM, French Base | macro-F1 (avg over 8 tasks, 9 seeds) | 59.3% | 61.6% (full CLM detour) | -2.3pp (drops to MLM-baseline 58.9% level, p=0.25) |
| Freeze mid layers (8-14) during CLM, French Base | macro-F1 (avg over 8 tasks, 9 seeds) | 61.2% | 61.6% (full CLM detour) | -0.4pp (benefit preserved) |
| DiaMED layer transplant (best group: mid 8-14) | macro-F1 (linear probing, 3 seeds) | 35.3% | 32.8% (MLM) / 39.9% (CLM full) | recovers +35% of the 7.1pp gap |
局限与改进
作者明确指出的局限有两点。第一,encoder-decoder 架构不对称是探索性的:decoder 控制实验用 Gemma-3 270M 而 encoder 用 ModernCamemBERT 不同家族,且语种、数据、tokenizer 都不同,'decoder 发散度只有 encoder 39%'这一结论可能受混淆因素影响,作者明确说需要 encoder 和 decoder 的同家族对比才能确认。第二,英语增益较小(+0.3-0.8pp),且作者没在英语上做 freeze 干预或 layer transplantation,因此英语低层是否也有'CLM 印记 -> 必要'的因果结构并未直接验证;CKA 在英语上的一致性只是辅助证据。第三,本文只验证了生物医学一个领域,作者推测低资源语言和领域差距大的场景会有更大收益,但还没在法律、金融、代码等领域做迁移实验。第四,2.6× 的 encoder/decoder 差异在文中被解释为'CLM decay 100% 监督 vs MLM decay 15% 监督',但 decay 阶段的监督密度差异不是唯一可能解释,模型容量、训练数据量、词表、tokenizer 都可能参与。第五,训练算力门槛不低:法语 Large 25B+2.5B token 在 4×H100 上跑,加上 9 个 seed × 8 个任务的微调,复现成本对一般实验室是显著负担。第六,作者承认论文只测试了 ModernBERT 架构,没有验证 CLM detour 是否对 RoBERTa、DeBERTa、xlm-R 等其他编码器同样有效。
独立分析的弱点
独立看有三个明显弱点。第一,论文把成功归因于'低层被重塑',但没有进一步打开黑箱:CKA 给出的是表征相似度,不是功能解释。低层 0-7 具体编码了什么新特征、为什么这些特征对生物医学有用,作者没有做 probing 或 attention head 分析,只停留在'低层变得不一样'。改进方向是对低层做 token-level probing,看 CLM 后低层是否学到了新的词级领域特征或子词结构。第二,layer transplantation 给出'训练中必要 ≠ 训练后可定位'的有趣结果,但只测了 3 个层组、1 个任务,没在更多任务/更细粒度(如单层、attention vs MLP)上做系统化扫参,结论比较粗。改进方向是按单层和子模块(attention-only、MLP-only)扫描移植效果,并加上 task-specific 微调后的移植对比。第三,最佳 decay 比例 10% 的扫参只测了 3 个法语任务,没有在英语或更长训练上验证,且 1.0pp 的差异统计显著性未给出。改进方向是跨语种、跨规模、跨任务族复现 10% sweet spot,并给出 confidence interval。第四,CLM 阶段的 mask 设计、超参数(mask rate、sequence length、batch size)是不是最优,本文没有 ablation,因为沿用了 ModernBERT 的原设置但加了 causal mask,理论上 causal mask 下某些超参应该有不同最优值。
未来方向
作者明确提出的方向有三条:(1) 验证 CLM detour 在低资源生物医学语言(意大利、西班牙、葡萄牙语)上的扩展性,预测这些语言因 ModernBERT 同源预训练语料覆盖不足会有更大收益;(2) 探索 CLM detour 在非生物医学领域(法律、金融、代码)的有效性,验证'领域差距越大收益越大'这一假设的普适性;(3) 在 encoder/decoder 同家族架构上做受控对比,确认 2.6× 架构不对称性的真实性。基于成果可延伸的方向至少有四:(a) 把 CLM detour 推广到目标检测/视觉编码器,测试这一范式是否跨模态通用;(b) 研究 CLM detour 与 LoRA、adapter 等参数高效方法的兼容性,看低层重塑能否通过局部参数实现;(c) 探索更长或更复杂的训练目标序列(如 MLM -> CLM -> span corruption -> MLM)是否叠加收益;(d) 用 mechanistic interpretability 工具分析低层具体学到了什么,建立'训练目标 -> 表征变化 -> 下游提升'的完整因果链。
复现评估
复现评估整体偏积极但成本不低。完全开源的部分:作者明确承诺发布 ModernCamemBERT-bio 和 ModernBERT-bio(Base + Large,4 个模型权重),模型 checkpoint 即可直接用于下游任务;论文 3.2 节给出了详细超参数(peak lr $2\times10^{-4}$,AdamW $\beta_1=0.9, \beta_2=0.98$,weight decay $10^{-5}$,batch 384 sequences ≈ 3.1M tokens,bf16,4×H100,linear warmup 100M tokens);3.3 节给出了 freeze 干预的实现细节(forward 之后 zeroing gradients)。半开源的部分:训练数据中 Biomed-Enriched 已经在前作 Touchent et al. 2025 中发布,但 7B token 的法语生物医学主语料和 2B 合成医疗 QA 是否完全公开未在本文中明确说明;E3C 临床病例和 EMA 说明书分别来自 Magnini et al. 2020 和欧盟公开数据,理论上可获取;CKA 计算脚本未单独发布。算力门槛:法语 Base 10B+1B token + 9 seeds × 8 tasks 微调,法语 Large 25B+2.5B token + 9 seeds × 8 tasks 微调,英语 Large 50B+5B token + 5 seeds × 11 tasks 微调,加上 3 个 freeze 干预、layer transplantation 和 CKA 实验,总算力估计需要数十到上百 H100·天,门槛偏高。复现难度:方法本身简单(只改 mask + 损失),但要拿到论文中具体的 F1 数字需要复现整个数据筛选(Qwen3-235B 打分、上采样)和训练流程。综合评价:模型权重可立即使用,但从头复现训练结果需要较大算力和对数据流水线的精确复刻。
论文图表