HLA-WM: Hybrid Linear Attention for Long-Horizon Video World Models
cs.CV, cs.AI
Submitted: 2026-10-05
Updated: 2026-10-05
Project page: https://caesarhhh.github.io/hla-wm
License: http://creativecommons.org/licenses/by/4.0/
The gist: Long-horizon video world models require persistent memory to preserve scene consistency over extended rollouts.
Terminology
Abstract
Long-horizon video world models require persistent memory to preserve scene consistency over extended rollouts. Softmax attention retains the full generation history through a growing KV cache, whereas recurrent linear attention compresses history into fixed-size states with substantially lower memory cost. However, we identify severe long-range forgetting in Gated DeltaNet (GDN), where information from distant but relevant scenes is progressively attenuated by subsequent state updates. To address this limitation, we propose HLA-WM, a training-free hybrid linear-attention framework that combines coarse-grained geometry-guided retrieval with fine-grained recurrent linear-state computation. HLA-WM exploits the affine structure of GDN to cache compact chunk-wise transition summaries, retrieve scene-relevant historical chunks using camera geometry, and recompose them into query-specific recurrent states. On the 60-second SANA-WM-Bench, HLA-WM improves all six aggregate revisit-consistency and camera-control metrics of the base autoregressive generator without additional training, including a 0.74 dB PSNR gain and a 28.5% reduction in rotation error. The improvements persist after downstream refinement and generalize to MBench-A, where HLA-WM consistently improves all three revisit-consistency metrics across all four subsets and all evaluated inference modes over 547 samples. At a 60-second context, HLA-WM reduces historical-state memory by 12 times relative to full KV caching while incurring at most a 1.6% reduction in inference throughput. These results demonstrate that selectively addressable recurrent memory can improve long-range scene recall while preserving the efficiency advantages of GDN. Project page: https://caesarhhh.github.io/hla-wm/
Sources
- World Models
- Matrix-Game: Interactive World Foundation Model
- Matrix-game 2.0: An open-source, real-time, and streaming interactive world model
- Yume: An Interactive World Generation Model
- SANA-WM: Efficient Minute-Scale World Modeling with Hybrid Linear Diffusion Transformer
- RELIC: Interactive Video World Model with Long-Horizon Memory
- Matrix-Game 3.0: Real-Time and Streaming Interactive World Model with Long-Horizon Memory
- VideoMLA: Low-Rank Latent KV Cache for Minute-Scale Autoregressive Video Diffusion
- Attend Locally, Remember Linearly: Linear Attention as Cross-Frame Memory for Autoregressive Video Diffusion
- MBench: A Comprehensive Benchmark on Memory Capability for Video World Models
- Fast Autoregressive Video Diffusion and World Models with Temporal Cache Compression and Sparse Attention
- Addressable Memory for Video World Models
- Geometry-Aware Implicit Memory for Video World Models
- AlayaWorld: Interactive Long-Horizon World Modeling -- Full Technical Report
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models