通用密集奖励框架VLLR

Arxiv cs.RO2026-04-02🔗 查看原文
提出VLLR,一种结合外在与内在密集奖励的框架:用LLM将长时序任务分解为可验证子任务,利用VLM估计进度并在短期内初始化值函数以暖启动训练;训练过程以策略自信度作为内在奖励,持续指导PPO微调。无需人工奖励设计,在CHORES基准上对预训练策略最高提升56个百分点,相较最先进RL微调同分布提升最多5%、分布外最多10%,VLM初始化提高完成效率,自信度奖励提升成功率。
原文内容
arXiv:2604.00055v1 Announce Type: new
Abstract: Existing robotic foundation policies are trained primarily via large-scale imitation learning. While such models demonstrate strong capabilities, they often struggle with long-horizon tasks due to distribution shift and error accumulation. While reinforcement learning (RL) can finetune these models, it cannot work well across diverse tasks without manual reward engineering. We propose VLLR, a dense reward framework combining (1) an extrinsic reward from Large Language Models (LLMs) and Vision-Language Models (VLMs) for task progress recognition, and (2) an intrinsic reward based on policy self-certainty. VLLR uses LLMs to decompose tasks into verifiable subtasks and then VLMs to estimate progress to initialize the value function for a brief warm-up phase, avoiding prohibitive inference cost during full training; and self-certainty provides per-step intrinsic guidance throughout PPO finetuning. Ablation studies reveal complementary benefits: VLM-based value initialization primarily improves task completion efficiency, while self-certainty primarily enhances success rates, particularly on out-of-distribution tasks. On the CHORES benchmark covering mobile manipulation and navigation, VLLR achieves up to 56% absolute success rate gains over the pretrained policy, up to 5% gains over state-of-the-art RL finetuning methods on in-distribution tasks, and up to $10\%$ gains on out-of-distribution tasks, all without manual reward engineering. Additional visualizations can be found in https://silongyong.github.io/vllr_project_page/