arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.02192cs.LGcs.DC
收藏

ECHO-2: 一种面向经济高效强化学习的大规模分布式推演框架

ECHO-2: A Large-Scale Distributed Rollout Framework for Cost-Efficient Reinforcement Learning

The University of Hong Kong(香港大学);Fudan University(复旦大学);Gradient;University of Edinburgh(爱丁堡大学);Soochow University(苏州大学);Technical University of Darmstadt(达姆施塔特技术大学);University of the Chinese Academy of Sciences(中国科学院大学)
Jingwei Song, Meng Chen, Jie Xiao, Qingnan Ren, Jiaqi Huang, Yangshen Deng, Chris Tong, Wanyi Chen, Suli Wang, Zhisheng Chen, Ziqian Bi, Shuo Lu, Yiqun Duan, Xu… 展开作者

Jingwei Song, Meng Chen, Jie Xiao, Qingnan Ren, Jiaqi Huang, Yangshen Deng, Chris Tong, Wanyi Chen, Suli Wang, Zhisheng Chen, Ziqian Bi, Shuo Lu, Yiqun Duan, Xu Wang, Rymon Yu, Lynn Ai, Eric Yang, Tianyu Shi

AI总结:

提出ECHO-2分布式强化学习框架,通过重叠推演生成、传播与训练,结合对等辅助流水线广播和成本感知异构工作节点激活,在保持奖励性能的同时显著提升成本效率。

摘要:

强化学习(RL)是大语言模型(LLM)后训练的关键阶段,涉及推演生成、奖励评估和集中学习之间的反复交互。分布式推演执行提供了利用更具成本效益的推理资源的机会,但引入了广域协调和策略传播方面的挑战。我们提出了ECHO-2,一个用于后训练的分布式RL框架,使用远程推理工作节点且传播延迟不可忽略。ECHO-2将集中学习与分布式推演相结合,将有界策略过时性视为用户可控参数,使得推演生成、传播和训练能够重叠。我们引入了一个基于重叠的容量模型,关联训练时间、传播延迟和推演吞吐量,得出了一个维持学习器利用率的实用配置规则。为了缓解传播瓶颈并降低成本,ECHO-2采用了对等辅助流水线广播和成本感知的异构工作节点激活。在真实广域网带宽条件下,对4B到32B参数规模的LLM进行GRPO后训练的实验表明,ECHO-2在保持与强基线相当的RL奖励的同时,显著提高了成本效率。

英文摘要:

Reinforcement learning (RL) is a critical stage in post-training large language models (LLMs), involving repeated interaction between rollout generation, reward evaluation, and centralized learning. Distributing rollout execution offers opportunities to leverage more cost-efficient inference resources, but introduces challenges in wide-area coordination and policy dissemination. We present ECHO-2, a distributed RL framework for post-training with remote inference workers and non-negligible dissemination latency. ECHO-2 combines centralized learning with distributed rollouts and treats bounded policy staleness as a user-controlled parameter, enabling rollout generation, dissemination, and training to overlap. We introduce an overlap-based capacity model that relates training time, dissemination latency, and rollout throughput, yielding a practical provisioning rule for sustaining learner utilization. To mitigate dissemination bottlenecks and lower cost, ECHO-2 employs peer-assisted pipelined broadcast and cost-aware activation of heterogeneous workers. Experiments on GRPO post-training of LLMs ranging from 4B to 32B parameters under real wide-area bandwidth regimes show that ECHO-2 significantly improves cost efficiency while preserving RL reward comparable to strong baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑