Spotlight: Synergizing Seed Exploration and Spot GPUs for DiT RL Post-Training
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Reinforcement learning (RL) post-training of Diffusion Transformers (DiTs) is prohibitively expensive, requiring thousands of high-end GPUs. Existing works explore two directions to reduce cost: seed exploration improves training convergence by selecting high-contrast samples, yet adds compute to the critical path; spot GPUs offer 69--77% lower cost, yet sit idle during training because DiT rollouts finish nearly simultaneously, which prevents LLM-style pipelining of rollout with training. Spot preemptions further break Sequence Parallelism (SP) groups, fragmenting GPU topology. We present Spotlight, the first system that harvests spot GPUs for DiT RL post-training. Spotlight rests on two key insights we devise: (1)~we show that exploration can tolerate stale model weights because exploration that uses the model weights from the previous iteration preserves the relative ranking of random seeds, allowing exploration to run on idle spot GPUs during training. (2)~SP reconfiguration can reuse on-node state, reducing group recovery from minutes to sub-second launches. Built on these insights, Spotlight introduces three techniques: a bandit-based exploration planner that maximizes reward variance within the training time budget, elastic sequence parallelism that reconfigures SP groups on the fly via persistent schedulers and intra-node weight copying, and a preemption-aware pull-based request scheduler that balances load and commits in-flight state upon preemption. We implement Spotlight on the open-source RL platform ROLL and evaluate it on Qwen-Image post-training. Spotlight reaches the same target validation score 4× faster than baselines, reducing total cost by 1.4-6.4× while achieving superior image quality on DeepSeek-OCR and Geneval datasets with resolution 512×512 and 1280×1280.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 26 of its 57 references that are in the map. First posted on arXiv 17 Jun 2026.
Builds on
26 of 57 references.
References · 26
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 413
- 2024Timestep Embedding Tells: It's Time to Cache for Video Diffusion Modelcited by 269
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 250
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2025Step-Video-T2V Technical Report: The Practice, Challenges, and Future of Video Foundation Modelcited by 168
- 2025Open-Sora 2.0: Training a Commercial-Level Video Generation Model in 200kcited by 163
- 2025Emu3.5: Native Multimodal Models are World Learnerscited by 132
- 2025Reinforcement Learning Optimization for Large-Scale Learning: An Efficient and User-Friendly Scaling Librarycited by 101
- 2023SpotServe: Serving Generative Large Language Models on Preemptible Instancescited by 87
14 more
- 2025StreamRL: Scalable, Heterogeneous, and Elastic RL for LLMs with Disaggregated Stream Generationcited by 61
- 2025BranchGRPO: Stable and Efficient GRPO with Structured Branching in Diffusion Modelscited by 61
- 2025Laminar: A Scalable Asynchronous RL Post-Training Frameworkcited by 37
- 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Trainingcited by 35
- 2025Seer: Online Context Learning for Fast Synchronous LLM Reinforcement Learningcited by 25
- 2024SkyServe: Serving AI Models across Regions and Clouds with Spot Instancescited by 20
- 2024Parcae: Proactive, Liveput-Optimized DNN Training on Preemptible Instancescited by 17
- 2025Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Draftercited by 14
- 2025RollArt: Disaggregated Multi-Task Agentic RL Training at Scalecited by 14
- 2025RLBoost: Harvesting Preemptible Resources for Cost-Efficient Reinforcement Learning on LLMscited by 11
- 2023Generative AI Beyond LLMs: System Implications of Multi-Modal Generationcited by 6
- 2025RollMux: Phase-Level Multiplexing for Disaggregated RL Post-Trainingcited by 3
- 2026FP4 Explore, BF16 Train: Diffusion Reinforcement Learning via Efficient Rollout Scalingcited by 3
- 2026RLHFless: Serverless Computing for Efficient RLHFcited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Grouping First, Attending Smartly: Training-Free Acceleration for Diffusion Transformers
- 2026Accelerating Disaggregated RL for Visual Generative LLMs with Diffusion-Based Parallelism and Trainer-Assisted Generation
- 2025Staggered Environment Resets Improve Massively Parallel On-Policy Reinforcement Learning
- 2026GF-DiT: Scheduling Parallelism for Diffusion Transformer Serving
- 2026SwiftFusion: Scalable Sequence Parallelism for Distributed Inference of Diffusion Transformers on GPUs
- 2025Chipmunk: Training-Free Acceleration of Diffusion Transformers with Dynamic Column-Sparse Deltas
- 2025Sprint: Sparse-Dense Residual Fusion for Efficient Diffusion Transformers
- 2026Fine-Tuning GPT-5 for GPU Kernel Generation