arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.12072cs.CV

世界模型自蒸馏:训练世界模型以解决通用任务

World Model Self-Distillation: Training World Models to Solve General Tasks

Department of Computer Science(计算机科学系)

Sebastian Stapf, Pablo Acuaviva Huertos, Aram Davtyan, Paolo Favaro

AI总结:

提出结合自蒸馏与强化学习的框架,从预训练视频生成器中提取任务解决能力,无需配对任务视频,在基准测试中超越原始模型。

摘要:

预训练视频生成器是有前景的视觉世界模型,展现出涌现的任务解决能力;然而,它们对详细文本描述的依赖限制了其在规划和决策中的直接使用。现有方法要么将这种推理外包给语言或视觉-语言模型,要么依赖带有配对任务执行视频的监督微调,后者收集成本高且难以扩展。我们提出一个可扩展的框架,通过结合自蒸馏与强化学习来激发此类模型的任务解决能力。给定一张无标注场景图像,视觉-语言模型生成候选任务和详细的逐步解决方案。该解决方案条件化一个预训练视频扩散模型(演示者);我们将其行为蒸馏到一个仅以图像和简短任务提示为条件的执行者中。这将执行知识从字幕引导生成转移到指令条件任务解决,无需精心策划的任务视频监督。我们进一步通过来自VLM反馈的强化学习改进执行者,利用判断采样视频是否满足任务与生成解决方案之间的不对称性。在我们提出的WorldTasks-Benchmark和DreamGen机器人基准上的实验表明,在我们基于VLM的评估协议下,执行者超越了演示者,并具有竞争力地迁移到机器人任务。

英文摘要:

Pretrained video generators are promising visual world models that exhibit emergent task-solving abilities; however, their reliance on detailed textual descriptions limits their direct use for planning and decision-making. Existing approaches either outsource this reasoning to language or vision-language models, or rely on supervised fine-tuning with paired task-execution videos, which are costly to collect and difficult to scale. We propose a scalable framework that elicits task-solving ability in such models by combining self-distillation with reinforcement learning. Given an unlabeled scene image, a vision-language model generates a candidate task and a detailed step-by-step solution. The solution conditions a pretrained video diffusion model, the Demonstrator; we distill its behavior into an Executor conditioned only on the image and a short task prompt. This transfers execution knowledge from caption-guided generation to instruction-conditioned task solving without curated task-video supervision. We further improve the Executor with reinforcement learning from VLM feedback, exploiting the asymmetry between judging whether a sampled video satisfies a task and generating the solution. Experiments on our proposed WorldTasks-Benchmark and the DreamGen robotics benchmark show that the Executor surpasses the Demonstrator under our VLM-based evaluation protocol and transfers competitively to robotic tasks.

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

↑