Reinforcement Learning for Flow-Matching Policies
On the map: Diffusion models → Diffusion and flow policies for robotics
Abstract (arXiv)
Flow-matching policies have emerged as a powerful paradigm for generalist robotics. These models are trained to imitate an action chunk, conditioned on sensor observations and textual instructions. Often, training demonstrations are generated by a suboptimal policy, such as a human operator. This work explores training flow-matching policies via reinforcement learning to surpass the original demonstration policy performance. We particularly note minimum-time control as a key application and present a simple scheme for variable-horizon flow-matching planning. We then introduce two families of approaches: a simple Reward-Weighted Flow Matching (RWFM) scheme and a Group Relative Policy Optimization (GRPO) approach with a learned reward surrogate. Our policies are trained on an illustrative suite of simulated unicycle dynamics tasks, and we show that both approaches dramatically improve upon the suboptimal demonstrator performance, with the GRPO approach in particular generally incurring between 50% and 85% less cost than a naive Imitation Learning Flow Matching (ILFM) approach.
Key facts. Cited by 14 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 39 of its 40 references that are in the map. First posted on arXiv 20 Jul 2025.
Builds on
39 of 40 references.
Closest ancestors · 13
Same direction, newest first
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Generative Trajectory Stitching through Diffusion Composition7 Mar
- 2025Online Reward-Weighted Fine-Tuning of Flow Matching with Wasserstein Regularization9 Feb
- 2024Adjoint Matching: Fine-tuning Flow and Diffusion Generative Models with Memoryless Stochastic Optimal Control13 Sep
- 2024Diffusion Policy Policy Optimization1 Sep
- 2024Hierarchical Diffusion Policy for Kinematics-Aware Multi-Task Robotic Manipulation6 Mar
- 2024Stitching Sub-Trajectories with Conditional Diffusion Model for Goal-Conditioned Offline RL11 Feb
- 2024Simple Hierarchical Planning with Diffusion5 Jan
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Refining Diffusion Planner for Reliable Behavior Synthesis by Automatic Detection of Infeasible Plans30 Oct
- 2023DPOK: Reinforcement Learning for Fine-tuning Text-to-Image Diffusion Models25 May
- 2023Training Diffusion Models with Reinforcement Learning22 May
Other references · 26
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,777
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2023PaLM-E: An Embodied Multimodal Language Modelcited by 2,037
- 2024Octo: An Open-Source Generalist Robot Policycited by 1,427
- 2022TimesNet: Temporal 2D-Variation Modeling for General Time Series Analysiscited by 1,030
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,023
14 more
- 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignmentcited by 672
- 2024Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusioncited by 634
- 2025SmolVLA: A Vision-Language-Action Model for Affordable and Efficient Roboticscited by 507
- 2025Improving Video Generation with Human Feedbackcited by 267
- 2024Understanding the performance gap between online and offline alignment algorithmscited by 110
- 2025Improving Vision-Language-Action Model with Online Reinforcement Learningcited by 107
- 2022Discriminator-Weighted Offline Imitation Learning from Suboptimal Demonstrationscited by 89
- 2024Vision Language Models are In-Context Value Learnerscited by 84
- 2024Sequence-Augmented SE(3)-Flow Matching For Conditional Protein Backbone Generationcited by 36
- 2025Grounding Multimodal LLMs to Embodied Agents that Ask for Help with Reinforcement Learningcited by 21
- 2024A Self-Correcting Vision-Language-Action Model for Fast and Slow System Manipulationcited by 20
- 2025Refined Policy Distillation: From VLA Generalists to RL Expertscited by 19
- 2025CHD: Coupled Hierarchical Diffusion for Long-Horizon Taskscited by 6
- 2025CHATS: Combining Human-Aligned Optimization and Test-Time Sampling for Text-to-Image Generationcited by 4
Built on by
Citing papers per year: 2025: 4; 2026: 10.
14 papers.
Sort:
By area:
- 2025FlowRL: Matching Reward Distributions for LLM Reasoningcited by 38
- 2025One-Step Flow Policy Mirror Descentcited by 18
- 2026HapticVLA: Contact-Rich Manipulation via Vision-Language-Action Model without Inference-Time Tactile Sensingcited by 9
- 2026From Prior to Pro: Efficient Skill Mastery via Distribution Contractive RL Finetuningcited by 8
- 2025Reinforcement Fine-Tuning of Flow-Matching Policies for Vision-Language-Action Modelscited by 7
- 2026π-StepNFT: Wider Space Needs Finer Steps in Online RL for Flow-based VLAscited by 5
- 2026Test-Time Gradient Guidance of Flow Policies in Reinforcement Learningcited by 5
- 2026SA-VLA: Spatially-Aware Flow-Matching for Vision-Language-Action Reinforcement Learningcited by 3
- 2026Contrastive Conceptor Activation Steering (COAST): Unlocking Vision-Language-Action Models through Hidden Statescited by 3
- 2026Preserving Foundational Capabilities in Flow-Matching VLAs through Conservative SFTcited by 1
- 2026FlowMPC: Improving Flow Matching policies with World Modelscited by 1
- 2025RLFR: Extending Reinforcement Learning for LLMs with Flow Environmentcited by 0
- 2026DSDyn-VLA: A Dual-Stream Dynamic Manipulation Framework with Motion Perception, Future Awareness, and Realtime Correction30 Sep 2026
- 2026FlowMPC: Improving Flow Matching policies with World Models15 Jun 2026
- 2026Test-Time Gradient Guidance of Flow Policies in Reinforcement Learning9 Jun 2026
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transport7 Jun 2026
- 2026Contrastive Conceptor Activation Steering (COAST): Unlocking Vision-Language-Action Models through Hidden States16 May 2026
- 2026Preserving Foundational Capabilities in Flow-Matching VLAs through Conservative SFT9 May 2026
- 2026HapticVLA: Contact-Rich Manipulation via Vision-Language-Action Model without Inference-Time Tactile Sensing16 Mar 2026
- 2026From Prior to Pro: Efficient Skill Mastery via Distribution Contractive RL Finetuning10 Mar 2026
- 2026π-StepNFT: Wider Space Needs Finer Steps in Online RL for Flow-based VLAs2 Mar 2026
- 2026SA-VLA: Spatially-Aware Flow-Matching for Vision-Language-Action Reinforcement Learning31 Jan 2026
- 2025RLFR: Extending Reinforcement Learning for LLMs with Flow Environment11 Oct 2025
- 2025Reinforcement Fine-Tuning of Flow-Matching Policies for Vision-Language-Action Models11 Oct 2025
2 more
- 2025FlowRL: Matching Reward Distributions for LLM Reasoning18 Sep 2025
- 2025One-Step Flow Policy Mirror Descent31 Jul 2025
- 2026HapticVLA: Contact-Rich Manipulation via Vision-Language-Action Model without Inference-Time Tactile Sensingcited by 9
- 2025Reinforcement Fine-Tuning of Flow-Matching Policies for Vision-Language-Action Modelscited by 7
- 2026π-StepNFT: Wider Space Needs Finer Steps in Online RL for Flow-based VLAscited by 5
- 2026SA-VLA: Spatially-Aware Flow-Matching for Vision-Language-Action Reinforcement Learningcited by 3
- 2026Preserving Foundational Capabilities in Flow-Matching VLAs through Conservative SFTcited by 1
- 2026DSDyn-VLA: A Dual-Stream Dynamic Manipulation Framework with Motion Perception, Future Awareness, and Realtime Correctioncited by 0
- 2026DSDyn-VLA: A Dual-Stream Dynamic Manipulation Framework with Motion Perception, Future Awareness, and Realtime Correction30 Sep 2026
- 2026Preserving Foundational Capabilities in Flow-Matching VLAs through Conservative SFT9 May 2026
- 2026HapticVLA: Contact-Rich Manipulation via Vision-Language-Action Model without Inference-Time Tactile Sensing16 Mar 2026
- 2026π-StepNFT: Wider Space Needs Finer Steps in Online RL for Flow-based VLAs2 Mar 2026
- 2026SA-VLA: Spatially-Aware Flow-Matching for Vision-Language-Action Reinforcement Learning31 Jan 2026
- 2025Reinforcement Fine-Tuning of Flow-Matching Policies for Vision-Language-Action Models11 Oct 2025
- 2025One-Step Flow Policy Mirror Descentcited by 18
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transportcited by 0
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transport7 Jun 2026
- 2025One-Step Flow Policy Mirror Descent31 Jul 2025
- 2026FlowMPC: Improving Flow Matching policies with World Models15 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026FlowDPG: Deterministic Policy Gradient on Flow Matching Policies for Real-World Manipulation
- 2025Extremum Flow Matching for Offline Goal Conditioned Reinforcement Learning
- 2025Flow Matching Policy Gradients
- 2025Guided Flow Policy: Learning from High-Value Actions in Offline Reinforcement Learning
- 2026ScoRe-Flow: Complete Distributional Control via Score-Based Reinforcement Learning for Flow Matching
- 2025Dense-Jump Flow Matching with Non-Uniform Time Scheduling for Robotic Policies: Mitigating Multi-Step Inference Degradation
- 2025FM-IRL: Flow-Matching for Reward Modeling and Policy Regularization in Reinforcement Learning
- 2025ReinFlow: Fine-tuning Flow Matching Policy with Online Reinforcement Learning