arXiv:2510.09913cs.CL2025-10被引 10

让预训练和对齐模型轮流生成,发挥各自优势提升表现

Don't Throw Away Your Pretrained Model

  • 用切换器模型决定每段由哪个版本的模型生成
  • 在18个数据集上平均比基线高12.9%性能
  • 可复用训练中被丢弃的模型产物,适合资源有限场景

对齐训练存在权衡:虽提升语言模型的推理与指令遵循能力,但可能损失创造力和校准等技能,而未对齐的基础模型表现更优。本文通过模型协作实现优势互补,提出Switch Generation方法——让预训练与对齐模型在生成序列中交替“发言”。具体地,通过学习在不同查询和上下文中选择不同模型生成下一子段的效果,训练一个切换器语言模型。推理时,该切换器动态引导各模型检查点在最需要其优势时生成相应段落。在8种协作基线和18个数据集上的实验表明:1)模型协作在18项任务中优于单个模型,有16项胜出;2)Switch Generation相较基线平均提升12.9%。进一步分析显示,该方法能发现组合式技能解决个体模型难以应对的问题,并泛化至未见过的模型与任务,重用原本被丢弃的昂贵训练过程副产品。

原文摘要 · Abstract (English)

Alignment training has tradeoffs: it helps language models (LMs) gain in reasoning and instruction following but might lose out on skills such as creativity and calibration, where unaligned base models are better at. We aim to make the best of both worlds through model collaboration, where different models in the training pipeline collaborate and complement each other. Since LM responses feature interleaving skills that favor different models, we propose Switch Generation, where pretrained and aligned model versions take turns to ``speak'' in a response sequence. Specifically, we train a switcher LM by learning from outcomes of choosing different models to generate the next segment across diverse queries and contexts. At inference time, the switcher LM guides different model checkpoints to dynamically generate the next segment where their strengths are most needed. Extensive experiments with 8 model collaboration baselines and 18 datasets show that 1) model collaboration consistently outperforms individual models on 16 out of 18 tasks, and 2) Switch Generation further outperforms baselines by 12.9% on average. Further analysis reveals that Switch Generation discovers compositional skills to solve problems where individual models struggle and generalizes to unseen models and tasks, reusing and repurposing by-products in expensive model training pipelines that are otherwise discarded.

模型协作提示工程高效训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。