arXiv:2606.08602cs.LGcs.AI2026-06

用密度传输方法提升流匹配策略的强化学习性能。

Reinforcement Learning for Flow-Matching Policies with Density Transport

论文配图:Reinforcement Learning for Flow-Matching Policies with Density Transport
图 1 · 摘自论文原文
  • 将强化学习视为动作密度向高奖励区域传输,利用斯坦因变分梯度下降构建传输场。
  • 在多种连续控制任务中,相比基线方法提升奖励质量并加快收敛速度。
  • 适合需要高效训练和多模态动作建模的机器人长时序操控场景。

我们提出一种在线强化学习算法,用于微调连续控制问题中的流匹配策略。核心思想是将强化学习策略改进视为动作密度向高奖励区域的传输,这与流匹配模型的传输形式天然契合。以往方法或近似当前/最优策略分布,或依赖蒸馏,导致梯度偏差或牺牲多模态建模能力。本文提出的RLDT方法,基于最大熵强化学习目标,使用斯坦因变分梯度下降(SVGD)构建传输场,并微调预训练的流匹配策略以对齐该场。由于流匹配策略通过多步去噪过程生成动作,直接梯度优化困难。为此,我们采用期望目标估计法,从中间去噪步骤近似策略动作,使传输场更新能稳定传播至网络参数,避免不稳定的时序反向传播。实验表明,RLDT在奖励质量与收敛速度上均优于多个基线方法,适用于包含密集与稀疏奖励、状态与视觉输入的长时序机器人操控任务。

原文摘要 · Abstract (English)

We present an online reinforcement learning (RL) algorithm for fine-tuning flow-matching policies in continuous-control problems. Our key insight is to view RL-based policy improvement as a transport of action densities towards regions of high reward, which naturally aligns with the transport formulation of flow matching models. Prior methods either approximate the current or optimal policy distribution or resort to distillation, which introduces biased gradients or sacrifices multimodal modeling capacity. In contrast, our approach for RL with Density Transport, which we name \emph{RLDT}, constructs a transport field from a maximum-entropy RL objective using Stein Variational Gradient Descent (SVGD). Then, it finetunes a pretrained flow matching policy to align with this field. Training with this alignment objective is nontrivial because flow-matching policies generate actions via a multi-step process, making direct gradient-based optimization challenging. To overcome this challenge and stabilize training, we approximate policy actions from intermediate denoising steps via expected-target estimation. This allows the transport-field update to propagate into the network parameters without unstable backpropagation through time. Experimental results demonstrate that RLDT outperforms competitive baselines in reward quality and convergence speed. This performance holds across diverse continuous-control tasks, encompassing both dense and sparse rewards, as well as state- and vision-based long-horizon robot manipulation. The project webpage is \href{https://rpfey.github.io/rldt/}{https://rpfey.github.io/rldt/}.

强化学习流匹配机器人控制

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