arXiv:2605.22537cs.LG2026-05中稿 · ICML

让不同模型协作训练大模型,还能保持效果不降反而提升。

F-TIS: Harnessing Diverse Models in Collaborative GRPO

  • 用过滤截断重要性采样处理异构模型生成的离策略数据。
  • 在多种异构设置下,模型最终性能与同策略训练相当,部分任务提升12%。
  • 适合分布式、资源不均的协同训练场景,通信效率高。

强化学习方法如GRPO在大语言模型后训练中广受欢迎。在GRPO中,模型对一组提示生成补全,获得奖励后,策略向高奖励补全方向更新。由于模型的自回归特性,此类训练的生成阶段极为耗时。为解决此问题,先前工作尝试将推理步骤分布到多个节点并行处理,但主要假设训练中模型同质,以保证样本尽可能接近同策略。这一假设在去中心化系统中不切实际,因各方计算能力与偏好各异,难以统一。因此,去中心化训练需支持异构模型协作。然而,这导致训练中出现大量离策略样本,而此前研究指出离策略样本会损害GRPO收敛。为此,我们提出过滤截断重要性采样(F-TIS),一种可利用离策略样本提升本地模型学习的GRPO式训练范式。该框架允许多种模型在同一强化学习训练中协作,且通信高效。我们在多种异构设置下广泛评估F-TIS,结果表明其最终模型收敛性能与纯同策略训练相当;此外,在某些设置下,对分布外任务的泛化能力优于同策略训练,性能最高提升12%。

原文摘要 · Abstract (English)

Reinforcement learning methods such as GRPO have seen great popularity in LLM post-training. In GRPO, models produce completions to a set of prompts, which are rewarded, and the policy is updated towards the relatively high reward completions. Due to the auto-regressive nature of models, the generation phase of such style of training can be extremely time consuming. As a solution, prior work has sought to distribute the inference step across many nodes, working parallel. These works assume primarily homogeneous models in the training in order to keep samples as close to on-policy as possible. This assumption may be impractical in decentralized systems, where parties with various computes and preferences may wish to collaborate on the same task. Thus, decentralized training requires an approach that can handle heterogeneous models - different models collaborating on the same tasks. However, this leads to highly off-policy samples presented during training, which prior work has identified that off-policy samples can hurt GRPO convergence. To enable heterogeneity, we propose Filtered Truncated Importance Sampling (F-TIS) - a GRPO-style training paradigm that can use off-policy samples to improve local model's learning. Our framework allows various models to collaborate in the same RL training run while being communication efficient. We extensively evaluate F-TIS in various heterogeneous setups and we show that it exhibits identical final model convergence to purely on-sample training. Furthermore, we observe in some setups better generalization on out-of-distribution tasks than on-policy training, increasing model's performance by up to 12\%.

强化学习大模型训练异构协作GRPO

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