让大模型学会配合多种推理控制器,提升实际部署的泛化能力。
Training Language Models to Cooperate with Inference-Time Controllers

- 将推理控制器嵌入训练过程,通过多任务强化学习优化模型
- 在未见过的控制器组合上表现更好,超越单一控制器优化
- 适合需要灵活推理流程的复杂应用开发者
大型语言模型性能不仅依赖基础模型,还受推理时控制器的影响。现有后训练方法通常针对单一固定交互模式优化,而真实场景中会使用链式思考、自一致性、辩论、规划和验证等多种控制器。这导致训练与部署不匹配,限制了新工作流的迁移能力。本文提出CALM(控制器感知语言模型),一种将控制器纳入训练循环的后训练框架。将控制器感知的后训练建模为控制器诱导交互协议上的多任务强化学习,其中控制器由可复用的局部推理模块组成。该结构也实现了在回合级GRPO目标下对混合控制器训练的模块级分解,支持系统性研究控制器与模块感知的训练策略。在未见过的控制器组合和更广泛的控制器迁移任务上评估表明,控制器感知的后训练能显著提升模型在不同推理工作流间的泛化能力,优于仅针对单个控制器优化的方法。
原文摘要 · Abstract (English)
Large language model (LLM) performance increasingly depends not only on the base model, but also on the inference-time controller used to organize reasoning. Existing post-training methods, however, typically optimize for a single fixed interaction pattern, despite real deployments relying on diverse controllers such as Chain-of-Thought, self-consistency, debate, planning, and verification pipelines. This creates a training--deployment mismatch and limits transfer to new workflows. We introduce CALM (Controller-Aware Language Models), a post-training framework that explicitly places controllers in the training loop. We formulate controller-aware post-training as multi-task reinforcement learning over controller-induced interaction protocols, where controllers are compositions of reusable local reasoning modules. This structure also induces a module-level decomposition of mixed-controller training under a turn-level GRPO objective, enabling a systematic study of controller and module-aware training strategies. We evaluate CALM on held-out controller compositions and broader controller shifts, showing that controller-aware post-training improves generalization across inference-time workflows beyond single-controller optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。