Reinforcement Learning via Fixed-point Flow Matching
cs.LG, cs.RO
Submitted: 2026-10-09
Updated: 2026-10-14
Code: https://github.com/google-deepmind/mujoco_playground
License: http://creativecommons.org/licenses/by/4.0/
The gist: In maximum entropy and behavior-constrained reinforcement learning (RL), the optimal policy is given by a Boltzmann distribution.
Terminology
Abstract
In maximum entropy and behavior-constrained reinforcement learning (RL), the optimal policy is given by a Boltzmann distribution. Generating samples from this distribution can be challenging, as it is often complex and multimodal. While flow policies trained with flow matching provide a powerful framework for representing such distributions, they traditionally rely on access to training data, which is generally unavailable in online RL. To address this limitation, we propose fixed-point flow matching (FFM), a data-free procedure for training and fine-tuning flow policies in RL. FFM leverages the target flow identity to formulate the actor update as a fixed-point iteration, utilizing a flow-matching-style regression objective where the critic's action gradients serve as regression targets. We show that the optimal policy is a fixed point of this iteration. Through extensive experiments, we demonstrate that FFM effectively fine-tunes pretrained flow policies in the offline-to-online setting and enables efficient and accurate training of flow policies from scratch in online RL. These results establish fixed-point flow matching as a principled approach for learning expressive flow-based policies without requiring training data.
Sources
- Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching
- One-Step Flow Policy Mirror Descent
- Target Score Matching
- Energy-Weighted Flow Matching: Unlocking Continuous Normalizing Flows for Efficient and Scalable Boltzmann Sampling
- Adjoint Matching through the Lens of the Stochastic Maximum Principle in Optimal Control
- A unified perspective on fine-tuning and sampling with diffusion and flow models
- Trust Region Q Adjoint Matching
- Q-Flow: Stable and Expressive Reinforcement Learning with Flow-Based Policy
- Flow Matching Policy Optimization with Mirror Descent and Entropy Constraints
- Entropy-Regularized Adjoint Matching for Offline Reinforcement Learning
- Improved techniques for fine-tuning flow models via adjoint matching: a deterministic control pipeline
- IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policies
- FLAG: Flow Policy MaxEnt-RL by Latent Augmented Guidance
- Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Review
- Reverse Flow Matching: A Unified Framework for Online Reinforcement Learning with Diffusion and Flow Policies
- Flow-GRPO: Training Flow Matching Models via Online RL
- AWAC: Accelerating Online Reinforcement Learning with Offline Datasets
- Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning
- Tilt Matching for Scalable Sampling and Fine-Tuning
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks