arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.06533cs.LGstat.ML
收藏

S2RMs:空间结构递归模

S2RMs: Spatially Structured Recurrent Modules

Max Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所);Mila;University of Tübingen(图宾根大学)

Nasim Rahaman, Anirudh Goyal, Muhammad Waleed Gondal, Manuel Wuthrich, Stefan Bauer, Yash Sharma, Yoshua Bengio, Bernhard Schölkopf

摘要:

通过适当的归纳偏差来捕捉数据生成过程的结构,可以帮助学习模型,这些模型具有良好的泛化能力,并且对输入分布的变化具有鲁棒性。虽然利用空间和时间结构的方法得到了广泛的应用,但最近的工作已经证明了利用稀疏和模块化结构的模型的潜力,这些模型使用了一系列相互作用很少的模块。在这项工作中,我们采取了一个步骤,能够同时利用模块化和时空结构的动态模型。我们通过将建模的动态系统抽象为自治但稀疏交互的子系统的集合来实现这一点。子系统根据学习的拓扑进行交互,但也由底层现实世界系统的空间结构告知。这导致一类模型非常适合于对系统的动态进行建模,这些模型仅提供对其状态的局部视图以及这些视图的相应空间位置。在具有挑战性的Starcraft2领域中,从裁剪帧进行视频预测和从部分观察进行多智能体世界建模的任务中,我们发现我们的模型对可用视图的数量更鲁棒,并且能够更好地泛化到新任务而无需额外的训练,即使与在训练分布上表现同样好或更好的强基线相比。

英文摘要:

Capturing the structure of a data-generating process by means of appropriate inductive biases can help in learning models that generalize well and are robust to changes in the input distribution. While methods that harness spatial and temporal structures find broad application, recent work has demonstrated the potential of models that leverage sparse and modular structure using an ensemble of sparingly interacting modules. In this work, we take a step towards dynamic models that are capable of simultaneously exploiting both modular and spatiotemporal structures. We accomplish this by abstracting the modeled dynamical system as a collection of autonomous but sparsely interacting sub-systems. The sub-systems interact according to a topology that is learned, but also informed by the spatial structure of the underlying real-world system. This results in a class of models that are well suited for modeling the dynamics of systems that only offer local views into their state, along with corresponding spatial locations of those views. On the tasks of video prediction from cropped frames and multi-agent world modeling from partial observations in the challenging Starcraft2 domain, we find our models to be more robust to the number of available views and better capable of generalization to novel tasks without additional training, even when compared against strong baselines that perform equally well or better on the training distribution.

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

↑