arXiv:2410.03450cs.LG2024-10ICLR被引 11

用交互数据训练多模态大模型,让机器人更懂任务有效性地检索经验。

MLLM as Retriever: Interactively Learning Multimodal Retrieval for Embodied Agents

  • 通过偏好学习微调多模态大模型,使其关注轨迹的任务有效性。
  • 在未见场景中任务成功率显著提升,优于基线方法。
  • 适合需要动态决策的智能体研究者或机器人开发者。

多模态大模型(MLLM)代理在复杂具身任务中展现出潜力,可通过检索与任务相关的多模态轨迹数据实现。然而,现有检索方法主要依赖轨迹的文本或视觉线索的表面相似性,忽略了其对特定任务的实际有效性。为此,我们提出一种新方法——MLLM As ReTriever(MART),利用交互数据对一个通用的MLLM检索器进行微调,使其充分考虑轨迹的任务有效性,并优先选择适用于未见任务的轨迹。我们还引入轨迹抽象机制,借助MLLM的摘要能力,以更少的标记表示轨迹并保留关键信息,帮助智能体更好地理解轨迹中的里程碑。在多种环境下的实验结果表明,该方法在未见场景中的任务成功率显著优于基线方法。本工作为具身智能体中的多模态检索提供了一种新范式,即通过微调通用的MLLM作为检索器,以评估轨迹的有效性。所有基准任务代码、模拟器修改及MLLM检索器均已公开于 https://github.com/PKU-RL/MART。

原文摘要 · Abstract (English)

MLLM agents demonstrate potential for complex embodied tasks by retrieving multimodal task-relevant trajectory data. However, current retrieval methods primarily focus on surface-level similarities of textual or visual cues in trajectories, neglecting their effectiveness for the specific task at hand. To address this issue, we propose a novel method, MLLM As ReTriever (MART), which enhances the performance of embodied agents by utilizing interaction data to fine-tune an MLLM retriever based on preference learning, such that the retriever fully considers the effectiveness of trajectories and prioritizes them for unseen tasks. We also introduce Trajectory Abstraction, a mechanism that leverages MLLMs' summarization capabilities to represent trajectories with fewer tokens while preserving key information, enabling agents to better comprehend milestones in the trajectory. Experimental results across various environments demonstrate our method significantly improves task success rates in unseen scenes compared to baseline methods. This work presents a new paradigm for multimodal retrieval in embodied agents, by fine-tuning a general-purpose MLLM as the retriever to assess trajectory effectiveness. All the code for benchmark tasks, simulator modifications, and the MLLM retriever is available at https://github.com/PKU-RL/MART.

具身智能多模态检索大模型应用

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