提出一种高效生成式策略,兼顾探索与稳定更新。
Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent

- 用均值流变换将高斯噪声转为动作,实现单步生成
- 在7个MuJoCo任务中超越高斯与生成基线表现
- 适合需要稳定探索的离线策略学习场景
在线离线强化学习受策略类别与更新规则双重影响。高斯策略虽计算快且熵可解析,但难以处理多模态动作分布;生成式策略表达能力强,却常需迭代采样或缺乏可解析熵估计。从优化角度看,SAC风格软策略改进与镜面下降(MD)分别最小化不同KL散度:前者使策略趋近于价值诱导的玻尔兹曼分布,后者在每步更新中以先前策略为正则。结合熵正则与MD约束具有吸引力,可在保持探索性的同时稳定策略优化;然而目标分布可能多模态,与单峰高斯策略不匹配。本文提出随机均值流策略(SMFP),通过均值流变换将高斯噪声映射为动作,实现可解析熵近似,并可在统一目标下进行离线镜面下降训练,支持探索性强且稳定的策略改进。在7个MuJoCo基准测试中,SMFP优于高斯与生成基线,同时保持单步推理效率。
原文摘要 · Abstract (English)
Online off-policy reinforcement learning (RL) is shaped by two coupled choices: the policy class and the update rule. Gaussian policies are fast and have tractable entropy, but struggle with multimodal action distributions. Generative policies are more expressive, but often require iterative sampling or lack tractable entropy estimates. On the optimisation side, SAC-style soft policy improvement and mirror descent (MD) can be viewed as minimising different KL divergences: the former moves the policy towards a value-induced Boltzmann distribution, while the latter regularises each update against the previous policy. Combining entropy regularisation with an MD constraint is therefore attractive, as it supports exploration while stabilising policy improvement; however, the resulting target can be multimodal and is poorly matched by unimodal Gaussian policies. We propose Stochastic MeanFlow Policies (SMFP), a one-step generative policy class that maps Gaussian noise to actions through a MeanFlow transformation. This stochastic reparameterisation yields a tractable entropy surrogate and allows MeanFlow policies to be trained within off-policy mirror descent under a unified objective for exploratory yet stable improvement. Across seven MuJoCo benchmarks, SMFP improves over Gaussian and generative baselines while retaining single-step inference efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。