WorldKV
Official implementation of "WorldKV: Efficient World Memory with World Retrieval and Compression"
The official code repository for the WorldKV research paper, which proposes combining retrieval and compression to reduce memory costs in AI world models used for video generation. The code is not yet released -- only the paper exists right now.
WorldKV is the official code repository for a research paper titled "WorldKV: Efficient World Memory with World Retrieval and Compression," authored by researchers at KAIST AI and Naver AI. The paper addresses a problem in AI world models, which are systems trained to simulate how a visual environment changes over time. These models power applications like video generation and game-playing agents that need to predict what comes next in a scene. As these models run, they accumulate memory in the form of a key-value cache, and that cache grows with every frame or step, making long-horizon generation increasingly expensive.
The KV in WorldKV refers to this key-value cache. The paper proposes combining retrieval and compression to keep only the most relevant memory at each step rather than accumulating everything. The claim is that this makes world models more practical for longer sequences without sacrificing much quality.
The repository is a placeholder at the time of writing. The README states that the authors are cleaning up the code and plan to release it in early June. Three specific implementations are listed as pending checkboxes, targeting three different world model frameworks: Lingbot-World-Fast, Inspatio-World, and Matrix-Game-2.0. None are checked off, and no setup, training, or inference instructions are present yet.
If you are looking for runnable code, the project is not ready. The paper is the current deliverable; the code is described as coming soon.
Where it fits
- Read the WorldKV paper to understand how retrieval-and-compression reduces memory cost in AI world models for video generation and game agents.
- Use the pending code release to reproduce WorldKV results on Lingbot-World-Fast, Inspatio-World, or Matrix-Game-2.0 once the authors publish it.
- Apply the WorldKV memory strategy to your own transformer-based world model to reduce inference cost on long sequences.