解决大模型智能体强化学习中长尾轨迹导致的性能瓶颈
Heddle: A Distributed Orchestration System for Agentic RL Rollout
- 以轨迹为中心设计调度、部署与资源管理机制
- 实测提升2.5倍端到端回放吞吐量,显著缓解长尾延迟
- 适合大规模智能体强化学习系统开发者参考
智能体强化学习(Agentic RL)通过交替执行数据收集回放和策略训练阶段,使大语言模型能够完成复杂任务。在回放阶段,智能体生成轨迹——即大模型与外部工具的多步交互序列。然而频繁的工具调用引发长尾轨迹生成,成为回放瓶颈。根源在于原有以步骤为中心的设计忽视了轨迹上下文,导致队列延迟、干扰开销和每标记时间增加三大问题。本文提出Heddle,一种以轨迹为中心的分布式编排系统,优化智能体回放执行的时机、位置与方式。Heddle集成三项核心机制:基于运行时预测和渐进优先级的轨迹级调度,最小化累积队列延迟;通过预排序动态规划与空闲工具调用时段的主动迁移实现轨迹感知部署,降低干扰;以及自适应资源管理器,动态调整模型并行度,在加速长尾轨迹每标记处理时间的同时保持短轨迹高吞吐。在多种智能体强化学习工作负载下的评估表明,Heddle有效消除长尾瓶颈,相比现有最优基线,端到端回放吞吐量最高提升2.5倍。
原文摘要 · Abstract (English)
Agentic Reinforcement Learning (RL) enables LLMs to solve complex tasks by alternating between a data-collection rollout phase and a policy training phase. During rollout, the agent generates trajectories, i.e., multi-step interactions between LLMs and external tools. Yet, frequent tool calls induce long-tailed trajectory generation that bottlenecks rollouts. This stems from step-centric designs that ignore trajectory context, triggering three system problems for long-tail trajectory generation: queueing delays, interference overhead, and inflated per-token time. We propose Heddle, a trajectory-centric system to optimize the when, where, and how of agentic rollout execution. Heddle integrates three core mechanisms: trajectory-level scheduling using runtime prediction and progressive priority to minimize cumulative queueing; trajectory-aware placement via presorted dynamic programming and opportunistic migration during idle tool call intervals to minimize interference; and trajectory-adaptive resource manager that dynamically tunes model parallelism to accelerate the per-token time of long-tail trajectories while maintaining high throughput for short trajectories. Evaluations across diverse agentic RL workloads demonstrate that Heddle effectively neutralizes the long-tail bottleneck, achieving up to 2.5$\times$ higher end-to-end rollout throughput compared to state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。