arXiv:2510.26374cs.AI2025-10中稿 · ICLR被引 15

用贝叶斯方法动态选任务,让大模型强化学习更省力高效。

BOTS: A Unified Framework for Bayesian Online Task Selection in LLM Reinforcement Finetuning

  • 基于贝叶斯推断在线更新任务难度,自适应选择最有价值的任务。
  • 在多个数据集和模型规模上,相比基线提升数据效率与性能。
  • 无需额外推理即可估算未选任务难度,计算开销极低,适合实际部署。

强化微调(RFT)是使大语言模型对齐人类偏好、提升推理能力的关键技术,但其效果高度依赖训练中探索的任务。均匀采样效率低下,浪费算力于简单或无法解决的任务;现有任务选择方法常面临高回放成本、适应性差或证据不全的问题。我们提出 BOTS,一个面向大模型强化微调的贝叶斯在线任务选择统一框架。基于贝叶斯推断,BOTS 随模型演进动态维护任务难度的后验估计,同时融合选定任务的显式评估证据与未选任务的隐式推断证据,通过汤普森采样实现探索与利用的合理平衡。为实用化隐式证据,我们引入一种超轻量插件式插值方法,无需额外回放即可估算任务难度,开销可忽略。实验表明,无论在何种领域或模型规模下,BOTS 均持续优于基线与消融方案,为 RFT 中的动态任务选择提供了一种实用且可扩展的解决方案。代码已公开于 https://github.com/agentscope-ai/Trinity-RFT/tree/main/examples/bots。

原文摘要 · Abstract (English)

Reinforcement finetuning (RFT) is a key technique for aligning Large Language Models (LLMs) with human preferences and enhancing reasoning, yet its effectiveness is highly sensitive to which tasks are explored during training. Uniform task sampling is inefficient, wasting computation on tasks that are either trivial or unsolvable, while existing task selection methods often suffer from high rollout costs, poor adaptivity, or incomplete evidence. We introduce BOTS, a unified framework for Bayesian Online Task Selection in LLM reinforcement finetuning. Grounded in Bayesian inference, BOTS adaptively maintains posterior estimates of task difficulty as the model evolves. It jointly incorporates explicit evidence from direct evaluations of selected tasks and implicit evidence inferred from these evaluations for unselected tasks, with Thompson sampling ensuring a principled balance between exploration and exploitation for task selection. To make implicit evidence practical, we instantiate it with an ultra-light interpolation-based plug-in that estimates difficulties of tasks without extra rollouts, adding negligible overhead. Empirically, across diverse domains and LLM scales, BOTS consistently improves data efficiency and performance over baselines and ablations, providing a practical and extensible solution for dynamic task selection in RFT. Code is available at https://github.com/agentscope-ai/Trinity-RFT/tree/main/examples/bots.

强化微调任务选择贝叶斯方法

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