arXiv:2507.01663cs.LGcs.AI2025-07被引 38

AsyncFlow通过异步流式架构提升大模型强化学习训练效率

AsyncFlow: An Asynchronous Streaming RL Framework for Efficient LLM Post-Training

  • 采用异步生产者-消费者模型,实现任务流水线自动重叠与动态负载均衡
  • 相比顶尖基线,平均吞吐提升1.59倍,有效减少计算空闲时间
  • 架构解耦底层引擎,支持自定义训练/推理系统,适合定制化部署

强化学习已成为大语言模型后训练的关键技术。传统任务共置框架存在显著可扩展性瓶颈,而任务分离框架则面临复杂数据流导致的资源空闲和负载不均问题。此外,多数现有框架与大模型训练或推理引擎紧密耦合,难以支持自定义引擎。为此,我们提出AsyncFlow,一种用于高效后训练的异步流式强化学习框架。具体而言,引入分布式数据存储与传输模块,实现全流式统一数据管理与细粒度调度;该架构天然支持RL任务间自动流水线重叠与动态负载均衡。同时,设计基于生产者-消费者模式的异步工作流,通过在容忍延迟阈值内延迟参数更新,最大限度减少计算空闲。此外,AsyncFlow的核心能力在架构上与底层训练和推理引擎解耦,并通过面向服务的用户接口封装,提供模块化与可定制化体验。大量实验表明,其平均吞吐相较当前最优基线提升1.59倍。本文提出的架构为下一代强化学习训练系统设计提供了切实可行的思路。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a pivotal technology in the post-training phase of large language models (LLMs). Traditional task-colocated RL frameworks suffer from significant scalability bottlenecks, while task-separated RL frameworks face challenges in complex dataflows and the corresponding resource idling and workload imbalance. Moreover, most existing frameworks are tightly coupled with LLM training or inference engines, making it difficult to support custom-designed engines. To address these challenges, we propose AsyncFlow, an asynchronous streaming RL framework for efficient post-training. Specifically, we introduce a distributed data storage and transfer module that provides a unified data management and fine-grained scheduling capability in a fully streamed manner. This architecture inherently facilitates automated pipeline overlapping among RL tasks and dynamic load balancing. Moreover, we propose a producer-consumer-based asynchronous workflow engineered to minimize computational idleness by strategically deferring parameter update process within staleness thresholds. Finally, the core capability of AsynFlow is architecturally decoupled from underlying training and inference engines and encapsulated by service-oriented user interfaces, offering a modular and customizable user experience. Extensive experiments demonstrate an average of 1.59 throughput improvement compared with state-of-the-art baseline. The presented architecture in this work provides actionable insights for next-generation RL training system designs.

强化学习大模型训练异步框架流式处理

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