SAC Flow: Sample-Efficient Reinforcement Learning of Flow-Based Policies via Velocity-Reparameterized Sequential Modeling
Not on the map. Why?
Abstract (arXiv)
Training expressive flow-based policies with off-policy reinforcement learning is notoriously unstable due to gradient pathologies in the multi-step action sampling process. We trace this instability to a fundamental connection: the flow rollout is algebraically equivalent to a residual recurrent computation, making it susceptible to the same vanishing and exploding gradients as RNNs. To address this, we reparameterize the velocity network using principles from modern sequential models, introducing two stable architectures: Flow-G, which incorporates a gated velocity, and Flow-T, which utilizes a decoded velocity. We then develop a practical SAC-based algorithm, enabled by a noise-augmented rollout, that facilitates direct end-to-end training of these policies. Our approach supports both from-scratch and offline-to-online learning and achieves state-of-the-art performance on continuous control and robotic manipulation benchmarks, eliminating the need for common workarounds like policy distillation or surrogate objectives.
Key facts. Cited by 27 papers in the map (data through 7 Oct 2026). Builds on 32 of its 36 references that are in the map. First posted on arXiv 30 Sep 2025.
Builds on
32 of 36 references.
References · 32
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,730
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2018Soft Actor-Critic Algorithms and Applicationscited by 1,179
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 699
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 543
- 2020dm_control: Software and Tasks for Continuous Controlcited by 368
- 2024Diffusion Policy Policy Optimizationcited by 255
- 2024OGBench: Benchmarking Offline Goal-Conditioned RLcited by 216
- 2025Flow Q-Learningcited by 162
20 more
- 2023A Survey of Imitation Learning: Algorithms, Recent Developments, and Challengescited by 145
- 2023Learning a Diffusion Model Policy from Rewards via Q-Score Matchingcited by 126
- 2024Diffusion-based Reinforcement Learning via Q-weighted Variational Policy Optimizationcited by 99
- 2025Reinforcement Learning with Action Chunkingcited by 97
- 2025Being-H0: Vision-Language-Action Pretraining from Large-Scale Human Videoscited by 94
- 2025ReinFlow: Fine-tuning Flow Matching Policy with Online Reinforcement Learningcited by 83
- 2024From Imitation to Refinement -- Residual RL for Precise Assemblycited by 80
- 2023Data Quality in Imitation Learningcited by 72
- 2024Policy Agnostic RL: Offline RL and Online RL Fine-Tuning of Any Class and Backbonecited by 71
- 2025DIME:Diffusion-Based Maximum Entropy Reinforcement Learningcited by 52
- 2024Learning Multimodal Behaviors from Scratch with Diffusion Policy Gradientcited by 49
- 2024Towards Generalizable Vision-Language Robotic Manipulation: A Benchmark and LLM-guided 3D Policycited by 37
- 2025Streaming Flow Policy: Simplifying diffusion/flow-matching policies by treating action trajectories as flow trajectoriescited by 34
- 2025Flow-Based Policy for Online Reinforcement Learningcited by 33
- 2025Maximum Entropy Reinforcement Learning with Diffusion Policycited by 29
- 2017R2N2: Residual Recurrent Neural Networks for Multivariate Time Series Forecastingcited by 10
- 2025D3P: Dynamic Denoising Diffusion Policy via Reinforcement Learningcited by 10
- 2023Wasserstein Gradient Flows for Optimizing Gaussian Mixture Policiescited by 8
- 2025NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flowscited by 5
- 2024When Do Off-Policy and On-Policy Policy Gradient Methods Align?cited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 26.
27 papers.
Sort:
By area:
- 2026Beyond Imitation: Reinforcement Learning-Based Sim-Real Co-Training for VLA Modelscited by 8
- 2026ALOE: Action-Level Off-Policy Evaluation for Vision-Language-Action Model Post-Trainingcited by 7
- 2026DeFlow: Decoupling Manifold Modeling and Value Maximization for Offline Policy Extractioncited by 5
- 2026FLAC: Maximum Entropy RL via Kinetic Energy Regularized Bridge Matchingcited by 5
- 2026SEAR: Sample Efficient Action Chunking Reinforcement Learningcited by 5
- 2026Reparameterization Flow Policy Optimizationcited by 4
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matchingcited by 4
- 2025Training Diffusion Policies via Prior-Mapping Co-Evolutioncited by 3
- 2026Trust Region Q Adjoint Matchingcited by 3
- 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Samplingcited by 2
- 2026Reversal Q-Learningcited by 2
- 2026Causal Flow Q-Learning for Robust Offline Reinforcement Learningcited by 1
15 more
- 2026GeMPO: Generalized Measure Matching for Online Diffusion Reinforcement Learningcited by 1
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraintscited by 1
- 2026AcceRL: A Distributed Asynchronous Reinforcement Learning and World Model Framework for Vision-Language-Action Modelscited by 1
- 2026Mean Flow Policy Optimizationcited by 1
- 2026Decentralized Diffusion Policy Learning for Enhanced Exploration in Cooperative Multi-agent Reinforcement Learningcited by 1
- 2026Generative Actor-Critic with Soft Bridge Policiescited by 1
- 2026Dual-Flow Reinforcement Learning with State-Aware Explorationcited by 1
- 2026Path-Space Mirror Descent for On-Policy Reinforcement Learning under the Generalized Schr"odinger Bridgecited by 0
- 2026Flow-based Policy With Distributional Reinforcement Learning in Trajectory Optimizationcited by 0
- 2026Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descentcited by 0
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transportcited by 0
- 2026VINE: Taming Generative Control Policies for Reinforcement Learningcited by 0
- 2026Critic-Free Pretraining for Efficient Online Reinforcement Learning Fine-Tuningcited by 0
- 2026Safe Score Matching: Diffusion Policies with Hamilton-Jacobi Reachability for Online Safe Reinforcement Learningcited by 0
- 2026Flowing Faster to Coordinate: One-Step Online Multi-Agent Flow Policiescited by 0
- 2026Flowing Faster to Coordinate: One-Step Online Multi-Agent Flow Policies1 Oct 2026
- 2026Safe Score Matching: Diffusion Policies with Hamilton-Jacobi Reachability for Online Safe Reinforcement Learning27 Sep 2026
- 2026Critic-Free Pretraining for Efficient Online Reinforcement Learning Fine-Tuning11 Aug 2026
- 2026VINE: Taming Generative Control Policies for Reinforcement Learning11 Jul 2026
- 2026Dual-Flow Reinforcement Learning with State-Aware Exploration29 Jun 2026
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching21 Jun 2026
- 2026Reversal Q-Learning16 Jun 2026
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transport7 Jun 2026
- 2026Trust Region Q Adjoint Matching26 May 2026
- 2026Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent20 May 2026
- 2026Generative Actor-Critic with Soft Bridge Policies9 May 2026
- 2026Decentralized Diffusion Policy Learning for Enhanced Exploration in Cooperative Multi-agent Reinforcement Learning8 May 2026
15 more
- 2026Mean Flow Policy Optimization16 Apr 2026
- 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Sampling10 Apr 2026
- 2026Flow-based Policy With Distributional Reinforcement Learning in Trajectory Optimization1 Apr 2026
- 2026Path-Space Mirror Descent for On-Policy Reinforcement Learning under the Generalized Schr"odinger Bridge23 Mar 2026
- 2026AcceRL: A Distributed Asynchronous Reinforcement Learning and World Model Framework for Vision-Language-Action Models19 Mar 2026
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints18 Mar 2026
- 2026GeMPO: Generalized Measure Matching for Online Diffusion Reinforcement Learning10 Mar 2026
- 2026SEAR: Sample Efficient Action Chunking Reinforcement Learning2 Mar 2026
- 2026FLAC: Maximum Entropy RL via Kinetic Energy Regularized Bridge Matching13 Feb 2026
- 2026ALOE: Action-Level Off-Policy Evaluation for Vision-Language-Action Model Post-Training13 Feb 2026
- 2026Beyond Imitation: Reinforcement Learning-Based Sim-Real Co-Training for VLA Models13 Feb 2026
- 2026Reparameterization Flow Policy Optimization3 Feb 2026
- 2026Causal Flow Q-Learning for Robust Offline Reinforcement Learning2 Feb 2026
- 2026DeFlow: Decoupling Manifold Modeling and Value Maximization for Offline Policy Extraction15 Jan 2026
- 2025Training Diffusion Policies via Prior-Mapping Co-Evolution2 Dec 2025
- 2026DeFlow: Decoupling Manifold Modeling and Value Maximization for Offline Policy Extractioncited by 5
- 2026FLAC: Maximum Entropy RL via Kinetic Energy Regularized Bridge Matchingcited by 5
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matchingcited by 4
- 2025Training Diffusion Policies via Prior-Mapping Co-Evolutioncited by 3
- 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Samplingcited by 2
- 2026Reversal Q-Learningcited by 2
- 2026Causal Flow Q-Learning for Robust Offline Reinforcement Learningcited by 1
- 2026GeMPO: Generalized Measure Matching for Online Diffusion Reinforcement Learningcited by 1
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraintscited by 1
- 2026Mean Flow Policy Optimizationcited by 1
- 2026Decentralized Diffusion Policy Learning for Enhanced Exploration in Cooperative Multi-agent Reinforcement Learningcited by 1
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transportcited by 0
12 most cited of 14 All 14 (JSON)
- 2026Safe Score Matching: Diffusion Policies with Hamilton-Jacobi Reachability for Online Safe Reinforcement Learning27 Sep 2026
- 2026VINE: Taming Generative Control Policies for Reinforcement Learning11 Jul 2026
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching21 Jun 2026
- 2026Reversal Q-Learning16 Jun 2026
- 2026Reinforcement Learning for Flow-Matching Policies with Density Transport7 Jun 2026
- 2026Decentralized Diffusion Policy Learning for Enhanced Exploration in Cooperative Multi-agent Reinforcement Learning8 May 2026
- 2026Mean Flow Policy Optimization16 Apr 2026
- 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Sampling10 Apr 2026
- 2026Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints18 Mar 2026
- 2026GeMPO: Generalized Measure Matching for Online Diffusion Reinforcement Learning10 Mar 2026
- 2026FLAC: Maximum Entropy RL via Kinetic Energy Regularized Bridge Matching13 Feb 2026
- 2026Causal Flow Q-Learning for Robust Offline Reinforcement Learning2 Feb 2026
12 newest of 14 All 14 (JSON)
- 2026Trust Region Q Adjoint Matchingcited by 3
- 2026Critic-Free Pretraining for Efficient Online Reinforcement Learning Fine-Tuningcited by 0
- 2026Critic-Free Pretraining for Efficient Online Reinforcement Learning Fine-Tuning11 Aug 2026
- 2026Trust Region Q Adjoint Matching26 May 2026
- 2026SEAR: Sample Efficient Action Chunking Reinforcement Learningcited by 5
- 2026Reparameterization Flow Policy Optimizationcited by 4
- 2026Generative Actor-Critic with Soft Bridge Policiescited by 1
- 2026Dual-Flow Reinforcement Learning with State-Aware Explorationcited by 1
- 2026Path-Space Mirror Descent for On-Policy Reinforcement Learning under the Generalized Schr"odinger Bridgecited by 0
- 2026Flow-based Policy With Distributional Reinforcement Learning in Trajectory Optimizationcited by 0
- 2026Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descentcited by 0
- 2026Flowing Faster to Coordinate: One-Step Online Multi-Agent Flow Policiescited by 0
- 2026Flowing Faster to Coordinate: One-Step Online Multi-Agent Flow Policies1 Oct 2026
- 2026Dual-Flow Reinforcement Learning with State-Aware Exploration29 Jun 2026
- 2026Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent20 May 2026
- 2026Generative Actor-Critic with Soft Bridge Policies9 May 2026
- 2026Flow-based Policy With Distributional Reinforcement Learning in Trajectory Optimization1 Apr 2026
- 2026Path-Space Mirror Descent for On-Policy Reinforcement Learning under the Generalized Schr"odinger Bridge23 Mar 2026
- 2026SEAR: Sample Efficient Action Chunking Reinforcement Learning2 Mar 2026
- 2026Reparameterization Flow Policy Optimization3 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2026RoMAN-Flow: Taming Autoregressive Normalizing Flows for Offline Reinforcement Learning in Robotic Manipulation
- 2026FlowDPG: Deterministic Policy Gradient on Flow Matching Policies for Real-World Manipulation
- 2026RFPO: Rectified Flow Policy Optimization for Embodied Control
- 2026Refinement-based Flow Policy Optimization
- 2026VGFM: Expressive Robot Policies via Dense Value Guidance in Flow Matching
- 2025Guided Flow Policy: Learning from High-Value Actions in Offline Reinforcement Learning
- 2025FlowQ: Energy-Guided Flow Policies for Offline Reinforcement Learning
- 2025Controllable Flow Matching for Online Reinforcement Learning