arXiv:2412.19139cs.CVcs.AI2024-12AAAI被引 11

用大模型提升视频任务规划能力,支持新动作和开放词表。

PlanLLM: Video Procedure Planning with Refinable Large Language Models

  • 引入大模型生成自由形式动作序列,突破固定词表限制。
  • 通过视觉与常识信息对齐,提升特定场景下的规划准确性。
  • 在三个基准上表现更优,适合需要灵活规划的机器人应用。

视频任务规划(Video procedure planning)是指根据起始和目标状态的视频帧,规划出一系列动作步骤,是具身智能的关键能力。现有方法虽引入大语言模型(LLM)生成丰富动作描述以指导解码,但通常将动作步骤解码为封闭集的一热向量,限制了模型对新动作或任务的泛化能力。此外,基于世界级常识的固定动作描述在特定视觉状态下可能引入噪声。本文提出PlanLLM,一种结合大模型的跨模态联合学习框架。我们设计了增强型规划模块,充分利用大模型的泛化能力,生成自由形式的规划输出并增强动作解码;同时提出互信息最大化模块,连接动作描述的世界级常识与样本特定的视觉信息,使大模型能结合推理生成步骤序列。在大模型辅助下,该方法可同时处理封闭集与开放词汇的规划任务。PlanLLM在三个基准上均取得优异性能,验证了设计的有效性。

原文摘要 · Abstract (English)

Video procedure planning, i.e., planning a sequence of action steps given the video frames of start and goal states, is an essential ability for embodied AI. Recent works utilize Large Language Models (LLMs) to generate enriched action step description texts to guide action step decoding. Although LLMs are introduced, these methods decode the action steps into a closed-set of one-hot vectors, limiting the model's capability of generalizing to new steps or tasks. Additionally, fixed action step descriptions based on world-level commonsense may contain noise in specific instances of visual states. In this paper, we propose PlanLLM, a cross-modal joint learning framework with LLMs for video procedure planning. We propose an LLM-Enhanced Planning module which fully uses the generalization ability of LLMs to produce free-form planning output and to enhance action step decoding. We also propose Mutual Information Maximization module to connect world-level commonsense of step descriptions and sample-specific information of visual states, enabling LLMs to employ the reasoning ability to generate step sequences. With the assistance of LLMs, our method can both closed-set and open vocabulary procedure planning tasks. Our PlanLLM achieves superior performance on three benchmarks, demonstrating the effectiveness of our designs.

视频规划大模型具身智能

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