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

Mem3R: Streaming 3D Reconstruction with Hybrid Memory via Test-Time Training

About

Streaming 3D perception is well suited to robotics and augmented reality, where long visual streams must be processed efficiently and consistently. Recent recurrent models offer a promising solution by maintaining fixed-size states and enabling linear-time inference, but they often suffer from drift accumulation and temporal forgetting over long sequences due to the limited capacity of compressed latent memories. We propose Mem3R, a streaming 3D reconstruction model with a hybrid memory design that decouples camera tracking from geometric mapping to improve temporal consistency over long sequences. For camera tracking, Mem3R employs an implicit fast-weight memory implemented as a lightweight Multi-Layer Perceptron updated via Test-Time Training. For geometric mapping, Mem3R maintains an explicit token-based fixed-size state. Compared with CUT3R, this design not only significantly improves long-sequence performance but also reduces the model size from 793M to 644M parameters. Mem3R supports existing improved plug-and-play state update strategies developed for CUT3R. Specifically, integrating it with TTT3R decreases Absolute Trajectory Error by up to 39% over the base implementation on 500 to 1000 frame sequences. The resulting improvements also extend to other downstream tasks, including video depth estimation and 3D reconstruction, while preserving constant GPU memory usage and comparable inference throughput. Project page: https://lck666666.github.io/Mem3R/

Changkun Liu, Jiezhi Yang, Zeman Li, Yuan Deng, Jiancong Guo, Luca Ballan• 2026

Related benchmarks

TaskDatasetResultRank
Camera pose estimationTUM-dynamic
ATE0.011
163
Camera pose estimationScanNet--
119
3D Reconstruction7 Scenes--
94
Depth EstimationBONN
Abs Rel0.09
56
Depth EstimationSintel ~50 frames
AbsRel0.413
47
Depth EstimationKITTI 110 frames
AbsRel10.9
46
3D ReconstructionNRGBD
Chamfer Distance0.032
44
Video Depth EstimationBonn 110 frames
AbsRel6.5
40
Scale-Invariant Depth EstimationBONN
Abs Rel7.3
30
Camera pose estimationSintel ~50 frames
ATE0.18
30
Showing 10 of 20 rows

Other info

Follow for update