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

槽位结构化世界模型

Slot Structured World Models

Leiden Institute of Advanced Computer Science(莱顿高级计算机科学研究所);Leiden University(莱顿大学)

Jonathan Collu, Riccardo Majellaro, Aske Plaat, Thomas M. Moerland

AI总结:

本文提出槽位结构化世界模型,结合槽位注意力编码器与潜在图动力学模型,解决前馈编码器无法提取物体中心表示及分离相似物体的问题,在Spriteworld基准多步预测任务中优于基线。

摘要:

感知和推理单个物体及其相互作用的能力是构建智能人工系统所要实现的目标。最先进的方法使用前馈编码器提取物体嵌入,并使用潜在图神经网络来建模这些物体嵌入之间的相互作用。然而,前馈编码器无法提取以物体为中心的表示,也无法分离外观相似的多个物体。为解决这些问题,我们引入了槽位结构化世界模型(SSWM),这是一类将基于槽位注意力的以物体为中心的编码器与基于潜在图的动力学模型相结合的世界模型。我们在具有简单物理交互规则的Spriteworld基准上评估了我们的方法,在带动作条件物体交互的一系列(多步)预测任务中,槽位结构化世界模型始终优于基线方法。复现论文实验的所有代码可从 https://github.com/JonathanCollu/Slot-Structured-World-Models 获取。

英文摘要:

The ability to perceive and reason about individual objects and their interactions is a goal to be achieved for building intelligent artificial systems. State-of-the-art approaches use a feedforward encoder to extract object embeddings and a latent graph neural network to model the interaction between these object embeddings. However, the feedforward encoder can not extract {\it object-centric} representations, nor can it disentangle multiple objects with similar appearance. To solve these issues, we introduce {\it Slot Structured World Models} (SSWM), a class of world models that combines an {\it object-centric} encoder (based on Slot Attention) with a latent graph-based dynamics model. We evaluate our method in the Spriteworld benchmark with simple rules of physical interaction, where Slot Structured World Models consistently outperform baselines on a range of (multi-step) prediction tasks with action-conditional object interactions. All code to reproduce paper experiments is available from \url{https://github.com/JonathanCollu/Slot-Structured-World-Models}.

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

↑