菜单

关于 🐙 GitHub
arXiv 提交日期: 2026-05-20
📄 Abstract - \textit{Stochastic} MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent

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.

顶级标签: reinforcement learning machine learning
详细标签: off-policy generative policies mirror descent entropy regularization muoco benchmarks 或 搜索:

随机平均流策略:基于熵镜像下降的单步生成式控制方法 / \textit{Stochastic} MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent


1️⃣ 一句话总结

本文提出一种名为随机平均流策略(SMFP)的新型生成式策略,它通过单步映射将高斯噪声转化为动作,在保留单步推理效率的同时,解决了传统高斯策略无法处理多峰动作分布、而生成式策略迭代慢且熵计算困难的问题,并在多个MuJoCo基准任务中取得了优于现有方法的性能。

源自 arXiv: 2605.21282