Relax让大模型在多模态下实现高效异步强化学习训练。
Relax: An Asynchronous Reinforcement Learning Engine for Omni-Modal Post-Training at Scale
- 构建全栈多模态支持的异步训练架构,避免文本中心管道的局限。
- 在Qwen3-Omni-30B上实现2.00倍加速,且收敛奖励水平一致。
- 适合需要大规模多模态强化学习的开发者和研究者使用。
强化学习(RL)后训练已被证明能有效激发大语言模型的推理、自省和工具使用能力。随着模型扩展至多模态输入和代理式多轮工作流,RL训练系统面临三大相互依赖的挑战:异构数据流、规模化下的操作鲁棒性,以及延迟与吞吐量的权衡。我们提出 extbf{Relax}(Reinforcement Engine Leveraging Agentic X-modality),一个开源的强化学习训练引擎,通过三个协同设计的架构层解决这些问题。首先,采用 extit{多模态原生架构},从数据预处理到模态感知并行计算再到推理生成,全栈集成多模态支持,而非对文本中心管道进行改造。其次,每个RL角色作为独立、故障隔离的服务运行,可独立扩展、恢复和升级,无需全局协调。第三,服务级解耦使异步训练通过TransferQueue数据总线实现,仅用一个延迟参数即可平滑切换于在线、准在线与完全异步执行模式之间。Relax在Qwen3-4B上的在线训练相比veRL实现1.20倍端到端加速。其完全异步模式在Qwen3-4B上较colocate提升1.76倍,在Qwen3-Omni-30B上达2.00倍,所有模式均收敛至相同奖励水平。在支持MoE模型的R3机制下,仅引入1.9%开销,而veRL在同一配置下有32%性能下降。此外,Relax在Qwen3-Omni上实现图像、文本、音频的稳定多模态强化学习收敛,视频任务持续超过2000步无退化。Relax已开源:https://github.com/rednote-ai/Relax。
原文摘要 · Abstract (English)
Reinforcement learning (RL) post-training has proven effective at unlocking reasoning, self-reflection, and tool-use capabilities in large language models. As models extend to omni-modal inputs and agentic multi-turn workflows, RL training systems face three interdependent challenges: heterogeneous data flows, operational robustness at scale, and the staleness -- throughput tradeoff. We present \textbf{Relax} (Reinforcement Engine Leveraging Agentic X-modality), an open-source RL training engine that addresses these challenges through three co-designed architectural layers. First, an \emph{omni-native architecture} builds multimodal support into the full stack -- from data preprocessing and modality-aware parallelism to inference generation -- rather than retrofitting it onto a text-centric pipeline. Second, each RL role runs as an independent, fault-isolated service that can be scaled, recovered, and upgraded without global coordination. Third, service-level decoupling enables asynchronous training via the TransferQueue data bus, where a single staleness parameter smoothly interpolates among on-policy, near-on-policy, and fully asynchronous execution. Relax achieves a 1.20$\times$ end-to-end speedup over veRL on Qwen3-4B on-policy training. Its fully async mode delivers a 1.76$\times$ speedup over colocate on Qwen3-4B and a 2.00$\times$ speedup on Qwen3-Omni-30B, while all modes converge to the same reward level. Relax supports R3 (Rollout Routing Replay)~\cite{ma2025r3} for MoE models with only 1.9\% overhead, compared to 32\% degradation in veRL under the same configuration. It further demonstrates stable omni-modal RL convergence on Qwen3-Omni across image, text, and audio, sustaining over 2{,}000 steps on video without degradation. Relax is available at https://github.com/rednote-ai/Relax.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。