arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2607.06262cs.RO
收藏

用于流策略引导和加速的最优传输Q学习

Optimal Transport Q-Learning for Flow Policy Steering and Acceleration

University of Edinburgh(爱丁堡大学);Honda Research Institute Europe(本田欧洲研究院)

Andreas Sochopoulos, Esmeralda S. Whitammer, Nikolaos Tsagkas, João Moura, Michael Gienger, Sethu Vijayakumar

AI总结:

针对流策略性能优化难题,提出最优传输Q学习(OTQL),利用机器人经验,通过优势加权条件最优传输流匹配微调加速流策略,提升单任务和VLA策略成功率,减少推理步骤数。

摘要:

扩散和流策略最近在机器人应用中通过准确捕捉多模态机器人轨迹分布展现出卓越性能,特别是在视觉语言动作(VLA)模型中。然而,高质量策略性能还需快速推理和高质量演示,这往往难以实现,导致策略行为次优及分布转移时失败。本文通过强化学习后训练,利用机器人经验解决微调并加速次优流策略的问题。引入最优传输Q学习(OTQL),通过优势加权条件最优传输流匹配微调流策略。OTQL能在50 - 60轮交互预算下微调并加速流,避免模拟和实际机器人任务中计算昂贵的蒸馏。结果表明,OTQL利用机器人自身经验对流策略进行后训练,将单任务策略平均成功率从36%提高到86%,预训练VLA从38%提高到76%,同时将每个动作生成的推理步骤数减少70%。

英文摘要:

Diffusion and flow policies have recently demonstrated remarkable performance in robotic applications by accurately capturing multimodal robot trajectory distributions, especially in the context of vision language action (VLA) models. However, high quality policy performance also requires fast inference and high quality demonstrations, which are often hard to get. Lack of these leads to suboptimal policy behaviors and failure under distribution shifts. In this work we address the problem of fine-tuning and accelerating suboptimal flow-based policies using the robot's experience through RL post-training. We introduce Optimal Transport Q-Learning (OTQL), a new method for finetuning flow policies using advantage weighted conditional optimal transport flow matching. OTQL can finetune and accelerate flows with an interaction budget of 50-60 episodes while avoiding computationally expensive distillation in simulation and real-world robot tasks. Our results show that OTQL post-trains flow policies using the robot's own experience, increasing average success percentage of single-task policies from 36% to 86% and of a pre-trained VLA from 38% to 76% while reducing the number of inference steps per action generation by 70%.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑