Accelerating Disaggregated RL for Visual Generative LLMs with Diffusion-Based Parallelism and Trainer-Assisted Generation
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning (RL) has become a dominant post-training paradigm, driving the emergence of high-performance RL systems such as veRL for autoregressive large language models (LLMs). In parallel, diffusion-oriented RL algorithms, e.g., DanceGRPO and FlowGRPO, have rapidly expanded the scope of RL from language reasoning to diffusion-based visual and flow-based generation. However, efficient RL systems for diffusion generative LLMs remain underexplored. Existing implementations, e.g., veRL-Omni, still rely on colocated execution, which simplifies synchronization but couples rollout and training resources, limits heterogeneous deployment, and constrains independent scaling. To this end, we introduce DigenRL, a disaggregated RL framework for diffusion-based generative LLMs that supports flexible resource allocation, accommodates heterogeneous GPUs, and facilitates efficient task scheduling. To maximally reduce the execution bubbles in the disaggregated architecture, we propose: 1) a generation-axis pipeline (GAP) and time-step parallelism (TSP) in the diffusion architecture to enable finer-grained pipelining between rollout and training; 2) an elastic trainer-assisted generation (TAG) approach to enable the trainer GPU resources to dynamically assist in executing rollout generations; and 3) a tightly one-step constrained asynchronous strategy to further utilize the tail bubble in the pipeline. Extensive experiments are conducted on three hardware testbeds with 16-32 GPUs using HunyuanVideo-13B, Wan2.1-14B, FLUX.1-12B, and QwenImage-20B generative models. Experimental results show that DigenRL achieves 1.56-2.10x throughput improvements over state-of-the-art diffusion RL systems, veRL-Omni and GenRL.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 31 of its 40 references that are in the map. First posted on arXiv 23 Jun 2026.
Builds on
31 of 40 references.
References · 31
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2017Neural Discrete Representation Learningcited by 4,712
- 2025Qwen3-VL Technical Reportcited by 2,793
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2024HunyuanVideo: A Systematic Framework For Large Video Generative Modelscited by 1,660
19 more
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2025Qwen-Image Technical Reportcited by 1,044
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 413
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 187
- 2025MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDEcited by 178
- 2023DeepSpeed-Chat: Easy, Fast and Affordable RLHF Training of ChatGPT-like Models at All Scalescited by 108
- 2025Reinforcement Learning Optimization for Large-Scale Learning: An Efficient and User-Friendly Scaling Librarycited by 101
- 2025Scaling RL to Long Videoscited by 84
- 2025StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generationcited by 61
- 2025LlamaRL: A Distributed Asynchronous Reinforcement Learning Framework for Efficient Large-scale LLM Trainingcited by 42
- 2025Laminar: A Scalable Asynchronous RL Post-Training Frameworkcited by 37
- 2025RollArt: Disaggregated Multi-Task Agentic RL Training at Scalecited by 14
- 2026OrchestrRL: Dynamic Compute and Network Orchestration for Disaggregated RLcited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Sketch Then Paint: Hierarchical Reinforcement Learning for Diffusion Multi-Modal Large Language Models
- 2026Efficient Resource-Shiftable RL Post-training with Generative Rewards
- 2025History Rhymes: Accelerating LLM Reinforcement Learning with RhymeRL
- 2026GrowMTP: Can RL Grow Its Own Draft Head?
- 2025Integrating Reinforcement Learning with Visual Generative Models: Foundations and Advances
- 2025AsyncFlow: An Asynchronous Streaming RL Framework for Efficient LLM Post-Training
- 2026SortedRL: Accelerating RL Training for LLMs through Online Length-Aware Scheduling
- 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learning