Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

STAC: Plug-and-Play Spatio-Temporal Aware Cache Compression for Streaming 3D Reconstruction

About

Online 3D reconstruction from streaming inputs requires both long-term temporal consistency and efficient memory usage. Although causal variants of VGGT address this challenge through a key-value (KV) cache mechanism, the cache grows linearly with the stream length, creating a major memory bottleneck. Under limited memory budgets, early cache eviction significantly degrades reconstruction quality and temporal consistency. In this work, we observe that attention in causal transformers for 3D reconstruction exhibits intrinsic spatio-temporal sparsity. Based on this insight, we propose STAC, a Spatio-Temporally Aware Cache Compression framework for streaming 3D reconstruction with large causal transformers. STAC consists of three key components: (1) a Working Temporal Token Caching mechanism that preserves long-term informative tokens using decayed cumulative attention scores; (2) a Long-term Spatial Token Caching scheme that compresses spatially redundant tokens into voxel-aligned representations for memory-efficient storage; and (3) a Chunk-based Multi-frame Optimization strategy that jointly processes consecutive frames to improve temporal coherence and GPU efficiency. Extensive experiments show that STAC achieves state-of-the-art reconstruction quality while reducing memory consumption by nearly 10x and accelerating inference by 4x, substantially improving the scalability of real-time 3D reconstruction in streaming settings.

Runze Wang, Yuxuan Song, Youcheng Cai, Ligang Liu• 2026

Related benchmarks

TaskDatasetResultRank
Camera pose estimationSintel
ATE0.198
192
Camera pose estimationTUM-dynamic
ATE0.026
163
Camera pose estimationScanNet
RPE (t)0.022
119
Point Cloud ReconstructionNRGBD stride 5
ACC Mean12.6
15
Point Cloud Reconstruction7-Scenes stride 5
Mean Accuracy5.6
15
Showing 5 of 5 rows

Other info

Follow for update