Drift, Then Spread: Learning One-Step Generative Policies for Online Reinforcement Learning
On the map: Deep reinforcement learning methods → RL with diffusion and flow models also fits: RL theory and sample complexity
Abstract (arXiv)
Generative policies offer expressive action distributions for online reinforcement learning (RL), but many diffusion- and flow-based policies require costly iterative sampling. We propose DriftSpread, a one-step generative policy based on Drifting for online maximum-entropy (MaxEnt) RL. To incorporate entropy without a tractable policy density, we build on an energy-based view of Drifting and derive critic-guided Drift and entropy-induced Spread from the negative MaxEnt objective. Our method learns these updates sequentially in the same actor, using action regression for value improvement and distribution matching to capture the effect of entropy. The actor updates require neither direct samples from the critic-defined Boltzmann target nor policy-density or score estimation. Under exact fitting with a fixed critic, we prove convergence to the Boltzmann target in the small-step limit followed by the long-time limit. On MuJoCo, DriftSpread achieves the highest aggregate normalized return among all evaluated methods, while outperforming the diffusion- and flow-based baselines in both training and inference efficiency.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in RL with diffusion and flow models, one of 15 directions in Deep reinforcement learning methods. Builds on 24 of its 30 references that are in the map. First posted on arXiv 11 Oct 2026.
Builds on
24 of 30 references.
Closest ancestors · 13
Same direction, newest first
- 2026Drift Q-Learning29 May
- 2026Score-Based One-step MeanFlow Policy Optimization22 May
- 2026Mean Flow Policy Optimization16 Apr
- 2026Reverse Flow Matching: A Unified Framework for Online Reinforcement Learning with Diffusion and Flow Policies13 Jan
- 2025SAC Flow: Sample-Efficient Reinforcement Learning of Flow-Based Policies via Velocity-Reparameterized Sequential Modeling30 Sep
- 2025One-Step Flow Policy Mirror Descent31 Jul
- 2025Flow-Based Policy for Online Reinforcement Learning15 Jun
- 2025Maximum Entropy Reinforcement Learning with Diffusion Policy17 Feb
- 2025DIME:Diffusion-Based Maximum Entropy Reinforcement Learning4 Feb
- 2025Efficient Online Reinforcement Learning for Diffusion Policy1 Feb
- 2024Sampling from Energy-based Policies using Diffusion2 Oct
- 2024Diffusion-based Reinforcement Learning via Q-weighted Variational Policy Optimization25 May
1 more
Other references · 11
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2018Addressing Function Approximation Error in Actor-Critic Methodscited by 2,417
- 2025Mean Flows for One-step Generative Modelingcited by 561
- 2015Generative Moment Matching Networkscited by 257
- 2026Generative Modeling via Driftingcited by 116
- 2019A Characteristic Function Approach to Deep Implicit Generative Modelingcited by 26
- 2026Sinkhorn-Drifting Generative Modelscited by 24
- 2026One-Step Generative Modeling via Wasserstein Gradient Flowscited by 24
- 2026On the Wasserstein Gradient Flow Interpretation of Drifting Modelscited by 11
- 2026Wasserstein Proximal Policy Gradientcited by 4
- 2026Drifting Field Policy: A One-Step Generative Policy via Wasserstein Gradient Flowcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Positive-Only Drifting Policy Optimization
- 2025EXPO: Stable Reinforcement Learning with Expressive Policies
- 2026Truncated Rectified Flow Policy for Reinforcement Learning with One-Step Sampling
- 2025Training Diffusion Policies via Prior-Mapping Co-Evolution
- 2026Scalable Maximum Entropy Reinforcement Learning for Diffusion Policies via Adjoint Matching
- 2026Drift-Based Policy Optimization: Native One-Step Policy Learning for Online Robot Control
- 2026Diffusing to Coordinate: Efficient Online Multi-Agent Diffusion Policies
- 2026Generative Actor-Critic with Soft Bridge Policies