arXiv:2607.10369cs.ROcs.AI2026-07

提出VINE方法,让生成式策略在强化学习中稳定训练且保持高表达力。

VINE: Taming Generative Control Policies for Reinforcement Learning

论文配图:VINE: Taming Generative Control Policies for Reinforcement Learning
图 1 · 摘自论文原文
  • 设计新采样方式,每步重构中间状态,实现稳定梯度传播
  • 在OGBench和真实机器人任务中均超越现有最优方法
  • 适合需要高复杂动作建模的强化学习场景

流匹配策略作为机器人学习中的有效策略参数化方法,通过从噪声迭代生成动作,能高度表达复杂多模态动作分布。然而,以往研究发现,将其与价值梯度强化学习(RL)结合常导致训练不稳定。现有方法归因于迭代生成过程,因而放弃端到端价值梯度优化以换取稳定性。本文指出,不稳定性实源于原始行为克隆设计的采样策略,在价值梯度RL下变得脆弱。为此,我们提出VINE,一种面向强化学习的采样方法,支持流匹配策略的端到端价值梯度优化。VINE不沿单一流轨迹推进,而是在每步去噪时重构新的插值状态,形成稳定的可微路径,同时兼容原有去噪流程。结果表明,尽管对全部十步去噪进行端到端反向传播,VINE仍实现稳定策略提升,并在OGBench离线强化学习基准及真实机器人操作任务中持续优于当前最优方法。

原文摘要 · Abstract (English)

Flow-matching policies have emerged as an effective policy parameterization for robot learning. They iteratively generate actions from noise, enabling highly expressive modeling of complex and multimodal action distributions. However, prior works observed that scaling these policies with value-gradient reinforcement learning (RL) often leads to training instability. Existing methods attribute this instability to iterative generation and therefore avoid end-to-end value-gradient optimization by sacrificing iterative generation, high expressiveness, or value-gradient optimization. Contrary to prior belief, we show the instability does not stem from iterative generation itself, but from the vanilla sampling strategy originally designed for behavior cloning, which becomes brittle under value-gradient RL. Motivated by this insight, we propose VINE, an RL-oriented sampling method that enables stable end-to-end value-gradient optimization for flow-matching policies. Instead of following a single flow trajectory, VINE reconstructs a new interpolation state at every denoising step, creating a stable differentiable path for value-gradient propagation while remaining compatible with the original flow-matching denoising process. As a result, VINE preserves the expressiveness and iterative generation of flow-matching without sacrificing end-to-end value-gradient optimization. Despite performing end-to-end backpropagation through all ten denoising steps, VINE achieves stable policy improvement and consistently outperforms state-of-the-art RL methods on the OGBench offline RL benchmark and real-world robotic manipulation task. Videos are available on our website: https://agibottech.github.io/vine.

强化学习生成策略稳定训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。