arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.04693cs.CVcs.AI

Unsupervised Object-Based Transition Models for 3D Partially Observable Environments

DeepMind(深度思维(DeepMind))

Antonia Creswell, Rishabh Kabra, Chris Burgess, Murray Shanahan

英文摘要:

We present a slot-wise, object-based transition model that decomposes a scene into objects, aligns them (with respect to a slot-wise object memory) to maintain a consistent order across time, and predicts how those objects evolve over successive frames. The model is trained end-to-end without supervision using losses at the level of the object-structured representation rather than pixels. Thanks to its alignment module, the model deals properly with two issues that are not handled satisfactorily by other transition models, namely object persistence and object identity. We show that the combination of an object-level loss and correct object alignment over time enables the model to outperform a state-of-the-art baseline, and allows it to deal well with object occlusion and re-appearance in partially observable environments.

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

↑