基于评分的单步平均流策略优化 / Score-Based One-step MeanFlow Policy Optimization
1️⃣ 一句话总结
本文提出了一种名为SOM的强化学习算法,它通过从价值函数中直接构造目标速度场,使得原本需要多次计算的扩散模型策略在线上任务中也能仅用一次网络计算完成动作生成,从而大幅提升训练和推理速度,并在运动控制任务上取得了领先性能。
Diffusion and flow matching have emerged as expressive policy classes in reinforcement learning, but their reliance on multi-step denoising imposes substantial computational overhead at inference time, which is particularly problematic in online RL. MeanFlow offers a promising alternative by learning an average velocity field that maps noise to data in a single network evaluation. However, MeanFlow typically requires samples from the target distribution to construct its target velocity field, which are unavailable in online RL. We propose Score-Based One-step MeanFlow Policy Optimization (SOM), an actor-critic algorithm that resolves this by constructing the target velocity field directly from the Q-function via score estimation and a probability flow ODE, thereby concentrating probability mass on high-value modes. In the fully online RL setting, SOM achieves state-of-the-art performance on locomotion tasks with a single generation step, while substantially reducing both training and inference time compared to prior diffusion- and flow-matching-based policies.
基于评分的单步平均流策略优化 / Score-Based One-step MeanFlow Policy Optimization
本文提出了一种名为SOM的强化学习算法,它通过从价值函数中直接构造目标速度场,使得原本需要多次计算的扩散模型策略在线上任务中也能仅用一次网络计算完成动作生成,从而大幅提升训练和推理速度,并在运动控制任务上取得了领先性能。
源自 arXiv: 2605.23365