Schrödinger--Föllmer Actor--Critic: Diffusion Policy Improvement with Finite-Sample Analysis
Organizations: School of Artificial Intelligence, and Hubei Key Laboratory of Computational Science, Wuhan University, Wuhan, 430072, China. · Institute for Math and AI, Hubei Key Laboratory of Computational Science, and School of Artificial Intelligence, Wuhan University, Wuhan, 430072, China. · School of Mathematics and Statistics, and Hubei Key Laboratory of Computational Science, Wuhan University, Wuhan, 430072, China. · School of Mathematics and Statistics, Wuhan University, Wuhan, 430072, China.
Abstract
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ödinger--Föllmer 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 -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.
Figures & tables
| Method | Update object | Refit | O2O | Characterization / analysis | |
|---|---|---|---|---|---|
| Diffusion-QL ( Wang et al., 2023 ) | Diffusion actor parameters | Denoising and action-value objective | |||
| IDQL ( Hansen-Estruch et al., 2023 ) | Candidate reweighting | Optional | Implicit-policy characterization | ||
| CEP/QGPO ( Lu et al., 2023a ) | Intermediate energy guidance | Population guidance consistency | |||
| DAC ( Fang et al., 2025 ) | Q-guided noise predictor | KL/noise-regression correspondence | |||
| CFGRL ( Frans et al., 2025 ) | Guided sampling distribution | Product-policy improvement | |||
| SFAC (ours) | Posterior-mean drift correction | Critic, actor, Monte Carlo, and discretization errors in average performance |
| baselines | SFAC | |||||||
|---|---|---|---|---|---|---|---|---|
| Suite | Dataset | IQL | AWAC | Cal-QL | DQL | IDQL | Offline init. | Final window |
| MuJoCo | HC-M | |||||||
| Hop-M | ||||||||
| Wal-M | ||||||||
| HC-MR | ||||||||
| Hop-MR | ||||||||
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
Diffusion Policy
Diffusion policies generate multimodal action sequences by iteratively denoising noise, enabling imitation, planning, and control in robotics.
See all 206 papers