SuperFlow: Training Flow Matching Models with RL on the Fly
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
Recent progress in flow-based generative models and reinforcement learning (RL) has improved text-image alignment and visual quality. However, current RL training for flow models still has two main problems: (i) GRPO-style fixed per-prompt group sizes ignore variation in sampling importance across prompts, which leads to inefficient sampling and slower training; and (ii) trajectory-level advantages are reused as per-step estimates, which biases credit assignment along the flow. We propose SuperFlow, an RL training framework for flow-based models that adjusts group sizes with variance-aware sampling and computes step-level advantages in a way that is consistent with continuous-time flow dynamics. Empirically, SuperFlow reaches promising performance while using only 5.4% to 56.3% of the original training steps and reduces training time by 5.2% to 16.7% without any architectural changes. On standard text-to-image (T2I) tasks, including text rendering, compositional image generation, and human preference alignment, SuperFlow improves over SD3.5-M by 4.6% to 47.2%, and over Flow-GRPO by 1.7% to 16.0%.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 63 of its 77 references that are in the map. First posted on arXiv 17 Dec 2025.
Builds on
63 of 77 references.
Closest ancestors · 11
Same direction, newest first
- 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Models6 Aug
- 2025MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDE29 Jul
- 2025Towards Self-Improvement of Diffusion Models via Group Preference Optimization16 May
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2025Diffusion Model as a Noise-Aware Latent Reward Model for Step-Level Preference Optimization3 Feb
- 2024Diffusion Policy Policy Optimization1 Sep
- 2023Learning a Diffusion Model Policy from Rewards via Q-Score Matching18 Dec
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models25 May
- 2023Training Diffusion Models with Reinforcement Learning22 May
Other references · 52
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2016Perceptual Losses for Real-Time Style Transfer and Super-Resolutioncited by 2,678
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2024DeepSeek-V2: A Strong, Economical, and Efficient Mixture-of-Experts Language Modelcited by 1,093
38 more
- 2014Deep Visual-Semantic Alignments for Generating Image Descriptionscited by 966
- 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 960
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 946
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 891
- 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 655
- 2024Autoregressive Image Generation without Vector Quantizationcited by 644
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 573
- 2025Mean Flows for One-step Generative Modelingcited by 524
- 2025Process Reinforcement through Implicit Rewardscited by 478
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 463
- 2025BLIP3-o: A Family of Fully Open Unified Multimodal Models-Architecture, Training and Datasetcited by 363
- 2025VL-Rethinker: Incentivizing Self-Reflection of Vision-Language Models with Reinforcement Learningcited by 315
- 2025Transfer between Modalities with MetaQueriescited by 231
- 2023Any-to-Any Generation via Composable Diffusioncited by 203
- 2024Free Process Rewards without Process Labelscited by 162
- 2025The Unreasonable Effectiveness of Entropy Minimization in LLM Reasoningcited by 160
- 2024Fluid: Scaling Autoregressive Text-to-image Generative Models with Continuous Tokenscited by 147
- 2024ReNO: Enhancing One-step Text-to-Image Models through Reward-based Noise Optimizationcited by 115
- 2024Golden Noise for Diffusion Models: A Learning Frameworkcited by 91
- 2025What Makes a Reward Model a Good Teacher? An Optimization Perspectivecited by 70
- 2025GPT-IMAGE-EDIT-1.5M: A Million-Scale, GPT-Generated Image Datasetcited by 68
- 2025Diffuse and Disperse: Image Generation with Representation Regularizationcited by 67
- 2025Seedream 2.0: A Native Chinese-English Bilingual Image Generation Foundation Modelcited by 59
- 2025Meta CLIP 2: A Worldwide Scaling Recipecited by 59
- 2025BLIP3o-NEXT: Next Frontier of Native Image Generationcited by 50
- 2025Reconstruction Alignment Improves Unified Multimodal Modelscited by 48
- 2025No Prompt Left Behind: Exploiting Zero-Variance Prompts in LLM Reinforcement Learning via Entropy-Guided Advantage Shapingcited by 48
- 2025Fractal Generative Modelscited by 44
- 2025Is Noise Conditioning Necessary for Denoising Generative Models?cited by 42
- 2025ImageGen-CoT: Enhancing Text-to-Image In-context Learning with Chain-of-Thought Reasoningcited by 37
- 2025Noise Hypernetworks: Amortizing Test-Time Compute in Diffusion Modelscited by 35
- 2025R2I-Bench: Benchmarking Reasoning-Driven Text-to-Image Generationcited by 31
- 2025Test-Time Scaling of Diffusion Models via Noise Trajectory Searchcited by 28
- 2025Single-stream Policy Optimizationcited by 27
- 2025Denoising Hamiltonian Network for Physical Reasoningcited by 10
- 2025UniGen-1.5: Enhancing Image Generation and Editing through Reward Unification in Reinforcement Learningcited by 8
- 2024Can DPO Learn Diverse Human Values? A Theoretical Scaling Lawcited by 7
50 most cited of 52 All 52 (JSON)
Built on by
Citing papers per year: 2026: 10.
10 papers.
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPOcited by 6
- 2026Diffusion Reinforcement Learning via Centered Reward Distillationcited by 3
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shapingcited by 3
- 2026When Rubrics Fail: Error Enumeration as Reward in Reference-Free RL Post-Training for Virtual Try-Oncited by 1
- 2026Advances in GRPO for Generation Models: A Surveycited by 1
- 2026Adaptive Multi-Stage Patent Claim Generation with Unified Quality Assessmentcited by 0
- 2026ProHiFlo: Hierarchical Flow Matching with Functional Guidance for De Novo Protein Generationcited by 0
- 2026FinInvest-GTCN: Explainable Graph-Temporal-Causal Modeling for Risk-Aware Investment Decision Optimizationcited by 0
- 2026EVLA: An Electro-Aware Multimodal Assistant for Physically-Grounded Driving Reasoning and Controlcited by 0
- 2026A3M: Adaptive, Adversarial and Multi-Objective Learning for Strategic Bidding in Repeated Auctionscited by 0
- 2026A3M: Adaptive, Adversarial and Multi-Objective Learning for Strategic Bidding in Repeated Auctions27 Jun 2026
- 2026EVLA: An Electro-Aware Multimodal Assistant for Physically-Grounded Driving Reasoning and Control27 Jun 2026
- 2026FinInvest-GTCN: Explainable Graph-Temporal-Causal Modeling for Risk-Aware Investment Decision Optimization27 Jun 2026
- 2026ProHiFlo: Hierarchical Flow Matching with Functional Guidance for De Novo Protein Generation3 Jun 2026
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPO14 May 2026
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping11 May 2026
- 2026Diffusion Reinforcement Learning via Centered Reward Distillation14 Mar 2026
- 2026When Rubrics Fail: Error Enumeration as Reward in Reference-Free RL Post-Training for Virtual Try-On5 Mar 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2026Adaptive Multi-Stage Patent Claim Generation with Unified Quality Assessment14 Jan 2026
Similar, not cited
Close in wording, no citation either way
- 2026Faster and Better Alignment for Flow Matching Models via Step-aware Advantages
- 2026PromptRL: Prompt Matters in RL for Flow-Based Image Generation
- 2025Smart-GRPO: Smartly Sampling Noise for Efficient RL of Flow-Matching Models
- 2025Fine-tuning Flow Matching Generative Models with Intermediate Feedback
- 2026Finite Difference Flow Optimization for RL Post-Training of Text-to-Image Models
- 2025Principled RL for Flow Matching Emerges from the Chunk-level Policy Optimization
- 2026LeapAlign: Post-Training Flow Matching Models at Any Generation Step by Building Two-Step Trajectories
- 2025Value Gradient Guidance for Flow Matching Alignment