Skip to yearly menu bar Skip to main content


Poster

Slot State Space Models

Jindong Jiang · Fei Deng · Gautam Singh · Minseung Lee · Sungjin Ahn

[ ] [ Project Page ]
Fri 13 Dec 4:30 p.m. PST — 7:30 p.m. PST

Abstract:

Recent State Space Models (SSMs) such as S4, S5, and MAMBA have shown remarkable computational benefits in long-range temporal dependency modeling. However, in many sequence modeling problems, the underlying process is inherently modular and it is of interest to have inductive biases that mimic this modular structure. In this paper, we introduce SlotSSM, a novel framework for incorporating independent mechanisms into SSMs to preserve or encourage separation of information. Unlike conventional SSMs that maintain a monolithic state vector, SlotSSM maintains the state as a collection of multiple vectors called slots. Crucially, the state transitions are performed independently per slot with sparse interactions across slots implemented via the bottleneck of self-attention. In experiments, we evaluate our model in object-centric video understanding and video prediction tasks, which involve modeling multiple objects and their long-range temporal dependencies. We find that that our proposed design offers substantial performance gains over existing sequence modeling methods.

Live content is unavailable. Log in and register to view live content