Schrödinger--Föllmer Actor--Critic: Diffusion Policy Improvement with Finite-Sample Analysis
On the map: Diffusion models → Diffusion and flow policies for robotics
also in 1 more
Abstract (arXiv)
Diffusion policies represent multimodal action distributions, but an advantage-weighted update does not specify how to sample from the resulting target distribution. We propose Schr"odinger--F"ollmer Actor--Critic (SFAC), an offline-to-online reinforcement learning (RL) method for Kullback--Leibler (KL)-regularized policy improvement through conditional diffusion. A minimax Bellman critic estimates the advantage function, which defines an exponentially tilted target policy. A Doob h-transform expresses this update as a correction to the reference diffusion drift. We derive a posterior-mean representation of the correction and estimate it using paired self-normalized importance sampling (SNIS). Supervised regression on these drift targets updates the neural actor without critic action gradients. In the small-update regime, the KL-regularized update follows the natural policy-gradient direction, and the Doob correction represents the same local change in the space of diffusion drifts. Under suitable conditions, we derive finite-sample bounds that separate the effects of critic estimation, neural drift regression, finite-sample SNIS, diffusion discretization, and inherited actor error on expected average policy suboptimality. Synthetic experiments assess the accuracy of approximation to prescribed advantage-tilted targets and sensitivity to sampling budgets. On six offline-to-online continuous-control tasks, a reference-anchored implementation achieves higher final-window returns than those of its corresponding offline initialization.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 16 of its 72 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
16 of 72 references.
Closest ancestors · 5
Same direction, newest first
- 2026Are we really tilting? The mechanics of reward guidance in flow and diffusion models1 Jun
- 2025Is RL fine-tuning harder than regression? A PDE learning approach for diffusion models2 Sep
- 2024Fine-Tuning of Continuous-Time Diffusion Models as Entropy-Regularized Control23 Feb
- 2023IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policies20 Apr
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar
Other references · 11
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,920
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,427
- 2019Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learningcited by 782
- 2020AWAC: Accelerating Online Reinforcement Learning with Offline Datasetscited by 671
- 2022Offline Reinforcement Learning via High-Fidelity Generative Behavior Modelingcited by 182
- 2025Diffusion Guidance Is a Controllable Policy Improvement Operatorcited by 58
- 2021Finite Sample Analysis of Minimax Offline Reinforcement Learning: Completeness, Fast Rates and First-Order Efficiencycited by 56
- 2023Finetuning from Offline Reinforcement Learning: Challenges, Trade-offs and Practical Solutionscited by 26
- 2024Non-Asymptotic Analysis for Single-Loop (Natural) Actor-Critic with Compatible Function Approximationcited by 11
- 2026Inference-Time Alignment for Diffusion Models via Variationally Stable Doob's Matchingcited by 5
- 2024Closing the Gap: Achieving Global Convergence (Last Iterate) of Actor-Critic under Markovian Sampling with Neural Network Parametrizationcited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024Diffusion Actor-Critic: Formulating Constrained Policy Iteration as Diffusion Noise Regression for Offline Reinforcement Learning
- 2026Explicit Critic Guidance for Aligning Diffusion Models
- 2026Refined Analysis of Entropy-Regularized Actor-Critic
- 2026Drift, Then Spread: Learning One-Step Generative Policies for Online Reinforcement Learning
- 2024Diffusion Actor-Critic with Entropy Regulator
- 2026Noisy-Space Policy Gradient for Diffusion Policies in Offline Reinforcement Learning
- 2025D2 Actor Critic: Diffusion Actor Meets Distributional Critic
- 2026Stabilizing the Q-Gradient Field for Policy Smoothness in Actor-Critic Methods