LOCI: Spatial Linear Memory for Streaming World Models

1

Key idea of LOCI

Figure 1: softmax-only memory versus LOCI's hybrid spatial memory at a camera revisit.

Figure 1.Projectively conditioned recurrent memory integrates historical context, while retained KV preserves observation-level detail. Recurrent readouts update the features used by subsequent historical attention, connecting the two representations through the feature stream.

2

Setup

  1. GT history as memory. The model is first given a recorded clip of the scene (the GT history) as its memory context, then generates the continuation on its own along the recorded camera path.
  2. Each figure shows three separate videos: the GT history the model received, LOCI's generated continuation, and the recorded continuation for reference. Use Play all to start the three videos together.
  3. Full history keeps the entire past in memory. Sparse memory keeps a fixed budget of past context plus the most recent frames, so memory cost stays bounded as the video grows.
3

Results

3.1

GT history as memory · full history

9 clips

LOCI conditioned on the complete GT history and everything it has generated so far.

GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 2.

Forest.

13.8 s GT history as memory → 26.2 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 3.

City.

18.8 s GT history as memory → 21.2 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 4.

Suburb, daytime.

11.3 s GT history as memory → 28.6 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 5.

Suburban street.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 6.

Temple plaza.

18.8 s GT history as memory → 21.1 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 7.

Temple courtyard.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 8.

Palace courtyard.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 9.

Container harbor at sunset.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, full history
GT continuation (reference)
Figure 10.

Mountain meadow.

5.1 s GT history as memory → 14.8 s generated.
3.2

GT history as memory · sparse memory

13 clips

The same setup with LOCI's bounded sparse memory: only a fixed budget of past context is retained, independent of video length.

GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 11.

Suburb, daytime.

11.3 s GT history as memory → 28.6 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 12.

Suburban street.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 13.

Temple plaza.

18.8 s GT history as memory → 21.1 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 14.

Temple courtyard.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 15.

Palace courtyard.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 16.

Container harbor at sunset.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 17.

Mountain meadow.

5.1 s GT history as memory → 14.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 18.

Snowy village.

11.3 s GT history as memory → 48.6 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 19.

Grassy hills.

6.3 s GT history as memory → 53.6 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 20.

Desert temple.

20.1 s GT history as memory → 39.8 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 21.

College campus.

17.6 s GT history as memory → 42.3 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 22.

Tokyo street.

11.3 s GT history as memory → 48.6 s generated.
GT history (memory)
LOCI, sparse memory
GT continuation (reference)
Figure 23.

Downtown at dusk.

13.8 s GT history as memory → 46.1 s generated.
4

Clip sources

All clips are rendered game-engine scenes. Held-out clips (not used for training): Forest, City and Temple plaza. Clips from the training data: all other scenes (Suburb, Suburban street, Temple courtyard, Palace courtyard, Container harbor, Mountain meadow, Snowy village, Grassy hills, Desert temple, College campus, Tokyo street, Downtown). Dark night-time clips are not shown.