ZipMap: Linear-Time Stateful 3D Reconstruction via Test-Time Training
Organizations: Google DeepMind · Cornell University · Massachusetts Institute of Technology
Abstract
Feed-forward transformer models have driven rapid progress in 3D vision, but state-of-the-art methods such as VGGT and have a computational cost that scales quadratically with the number of input images, making them inefficient when applied to large image collections. Sequential-reconstruction approaches reduce this cost but sacrifice reconstruction quality. We introduce ZipMap, a stateful feed-forward model that achieves linear-time, bidirectional 3D reconstruction while matching or surpassing the accuracy of quadratic-time methods. ZipMap employs test-time training layers to zip an entire image collection into a compact hidden scene state in a single forward pass, enabling reconstruction of over 700 frames in under 10 seconds on a single H100 GPU, more than faster than state-of-the-art methods such as VGGT. Moreover, we demonstrate the benefits of having a stateful representation in real-time scene-state querying and its extension to sequential streaming reconstruction.
Figures & tables
| Method | RealEstate10K | Co3Dv2 | ||||
| AUC@5 | AUC@15 | AUC@30 | AUC@5 | AUC@15 | AUC@30 | |
| Inference Speed | ||||||
| Fast3R [ 79 ] | 22.36 | 46.71 | 61.68 | 31.05 | 59.63 | 73.43 |
| FLARE [ 85 ] | 38.47 | 67.02 | 80.01 | 23.84 | 57.78 | 73.99 |
| VGGT [ 68 ] | 38.71 | 66.46 | 78.89 | 67.84 | 83.95 | 89.99 |
| [ 75 ] | 63.10 | 80.31 | 87.40 | 57.12 | 79.86 | 87.93 |
| Method | Sintel | TUM-dynamics | ScanNet (seen) | ||||||
| ATE | RPE trans | RPE rot | ATE | RPE trans | RPE rot | ATE | RPE trans | RPE rot | |
| Inference Speed | |||||||||
| Fast3R [ 79 ] | 0.371 | 0.298 | 13.750 | 0.090 | 0.101 | 1.425 | 0.155 | 0.123 | 3.491 |
| FLARE [ 85 ] | 0.207 | 0.090 | 3.015 | 0.026 | 0.013 | 0.475 | 0.064 | 0.023 | 0.971 |
| VGGT [ 68 ] | 0.172 | 0.061 | 0.471 | 0.012 | 0.010 | 0.309 | 0.035 | 0.015 | 0.376 |
| [ 75 ] | 0.073 | 0.038 | 0.288 | 0.014 | 0.009 | 0.307 | 0.030 | 0.013 | 0.345 |
| Method | 7-Scenes | NRGBD | ||||||||||
| Acc. | Comp. | NC. | Acc. | Comp. | NC. | |||||||
| Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | |
| Sparse View | ||||||||||||
| Fast3R [ 79 ] | 0.095 | 0.065 | 0.144 | 0.089 | 0.673 | 0.759 | 0.135 | 0.091 | 0.163 | 0.104 | 0.759 | 0.877 |
| CUT3R [ 76 ] | 0.080 | 0.055 | 0.102 | 0.066 | 0.711 | 0.811 | 0.098 | 0.038 | 0.075 | 0.029 | 0.830 | 0.974 |
| TTT3R [ 13 ] | 0.098 | 0.062 | 0.159 | 0.107 | 0.681 | 0.768 | 0.101 | 0.039 | 0.076 | 0.029 | 0.826 | 0.973 |
| Method | DTU | ETH3D | ||||||||||
| Acc. | Comp. | N.C. | Acc. | Comp. | N.C. | |||||||
| Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | Mean | Med. | |
| Inference Speed | ||||||||||||
| Fast3R [ 79 ] | 3.340 | 1.919 | 2.929 | 1.125 | 0.671 | 0.755 | 0.832 | 0.691 | 0.978 | 0.683 | 0.667 | 0.766 |
| FLARE [ 85 ] | 2.541 | 1.468 | 3.174 | 1.420 | 0.684 | 0.774 | 0.464 | 0.338 | 0.664 | 0.395 | 0.744 | 0.864 |
| VGGT [ 68 ] | 1.308 | 0.761 | 1.929 | 1.015 | 0.665 | 0.751 | 0.270 | 0.174 | 0.304 | 0.180 | 0.841 | 0.942 |
| Sintel | Bonn | KITTI | |||||
| Method | Params | AbsRel | AbsRel | AbsRel | |||
| Inference Speed | |||||||
| Fast3R [ 79 ] | 648M | 0.638 | 0.422 | 0.194 | 0.772 | 0.138 | 0.834 |
| FLARE [ 85 ] | 1.40B | 0.729 | 0.336 | 0.152 | 0.790 | 0.356 | 0.570 |
| VGGT [ 68 ] | 1.26B | 0.298 | 0.643 | 0.055 | 0.971 | 0.073 | 0.965 |
| [ 75 ] | 959M | 0.228 | 0.672 | 0.051 | 0.975 | 0.038 | 0.986 |
| Method | Acc. | Comp. | N.C. | |||
| Mean | Med. | Mean | Med. | Mean | Med. | |
| Ours | 0.337 | 0.224 | 0.357 | 0.217 | 0.810 | 0.918 |
| Ours w/o gated unit | 0.354 | 0.251 | 0.381 | 0.234 | 0.802 | 0.901 |
| Ours w/o Newton–Schulz | 0.408 | 0.283 | 0.430 | 0.249 | 0.787 | 0.898 |
| Ours w/ global TTT lr (0.1) | 0.411 | 0.303 | 0.490 | 0.317 | 0.779 | 0.886 |
| Ours w/ global TTT lr (1.0) | 0.464 | 0.366 | 0.537 | 0.343 | 0.782 | 0.890 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | Complexity | Params | Time to Reconstruct Frames (seconds) | |||||||||
| 5 | 10 | 25 | 50 | 100 | 200 | 300 | 400 | 500 | 750 | |||
| VGGT [ 68 ] | 1.26B | 0.102 | 0.194 | 0.569 | 1.524 | 4.689 | 16.040 | 34.022 | 58.842 | 90.389 | 200.364 | |
| [ 75 ] | 959M | 0.087 | 0.157 | 0.450 | 1.186 | 3.604 | 12.190 | 25.765 | 44.464 | 68.255 | 151.159 | |
| CUT3R [ 76 ] | 793M | 0.206 | 0.413 | 1.018 | 2.056 | 4.088 | 8.222 | 12.430 | 16.618 | 21.025 | 31.246 | |
| TTT3R [ 13 ] | 793M | 0.206 | 0.411 | 1.033 | 2.036 | 4.128 | 8.267 | 12.435 | 16.511 | 20.767 | 31.197 | |
| Ours | 1.40B | 0.125 | 0.183 | 0.383 | 0.712 | 1.362 | 2.681 | 4.017 | 5.348 | 6.671 | 9.999 | |
| Sintel | Bonn | KITTI | NYU-v2 | |||||
| Method | AbsRel | AbsRel | AbsRel | AbsRel | ||||
| MASt3R [ 34 ] | 0.413 | 0.569 | 0.123 | 0.833 | 0.077 | 0.948 | 0.110 | 0.865 |
| MonST3R [ 84 ] | 0.402 | 0.525 | 0.069 | 0.954 | 0.098 | 0.895 | 0.094 | 0.887 |
| Fast3R [ 79 ] | 0.544 | 0.509 | 0.169 | 0.796 | 0.120 | 0.861 | 0.093 | 0.898 |
| CUT3R [ 76 ] | 0.418 | 0.520 | 0.058 | 0.967 | 0.097 | 0.914 | 0.081 | 0.914 |
| FLARE [ 85 ] | 0.606 | 0.402 | 0.130 | 0.836 | 0.312 | 0.513 | 0.089 | 0.898 |
| Sintel | Bonn | KITTI | |||||
| Method | Params | AbsRel | AbsRel | AbsRel | |||
| Inference Speed | |||||||
| Fast3R [ 79 ] | 648M | 0.518 | 0.486 | 0.196 | 0.768 | 0.139 | 0.808 |
| FLARE [ 85 ] | 1.40B | 0.791 | 0.358 | 0.142 | 0.797 | 0.357 | 0.579 |
| VGGT [ 68 ] | 1.26B | 0.226 | 0.683 | 0.049 | 0.974 | 0.059 | 0.961 |
| [ 75 ] | 959M | 0.206 | 0.735 | 0.045 | 0.976 | 0.036 | 0.986 |
| Method | Sintel | TUM-dynamics | ScanNet (seen) | ||||||
| ATE | RPE trans | RPE rot | ATE | RPE trans | RPE rot | ATE | RPE trans | RPE rot | |
| Ours w/ ref | 0.125 | 0.058 | 0.420 | 0.012 | 0.009 | 0.309 | 0.034 | 0.015 | 0.398 |
| Ours w/o ref | 0.132 | 0.066 | 0.438 | 0.012 | 0.010 | 0.310 | 0.034 | 0.015 | 0.385 |
| Sintel | Bonn | KITTI | ||||
| Method | AbsRel | AbsRel | AbsRel | |||
| CUT3R [ 76 ] | 0.432 | 0.510 | 0.072 | 0.951 | 0.152 | 0.805 |
| TTT3R [ 13 ] | 0.426 | 0.522 | 0.061 | 0.970 | 0.149 | 0.812 |
| Ours-streaming | 0.273 | 0.638 | 0.067 | 0.965 | 0.100 | 0.903 |
| Method | Sintel | Co3Dv2 | ||||
| ATE | RPE trans | RPE rot | AUC@5 | AUC@15 | AUC@30 | |
| CUT3R [ 76 ] | 0.2160 | 0.0710 | 0.6220 | 24.88 | 56.28 | 71.72 |
| TTT3R [ 13 ] | 0.2040 | 0.0850 | 0.6900 | 22.61 | 53.49 | 69.46 |
| Ours-Streaming | 0.1593 | 0.0655 | 0.7508 | 45.38 | 72.58 | 83.12 |
| Method | DTU | ETH3D | NRGBD-dense | ||||||
| Acc. | Comp. | N.C. | Acc. | Comp. | N.C. | Acc. | Comp. | N.C. | |
| CUT3R [ 76 ] | 5.045 | 6.437 | 0.666 | 0.593 | 0.747 | 0.754 | 0.065 | 0.036 | 0.812 |
| TTT3R [ 13 ] | 5.337 | 6.593 | 0.666 | 0.763 | 0.881 | 0.739 | 0.074 | 0.037 | 0.803 |
| Ours-streaming | 4.091 | 3.495 | 0.693 | 0.614 | 0.941 | 0.750 | 0.038 | 0.028 | 0.836 |
| Method | 6 Views | 16 Views | ||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |
| NoPoSplat [ 82 ] | 15.92 | 0.416 | 0.541 | 15.47 | 0.361 | 0.606 |
| Flare [ 85 ] | 15.35 | 0.407 | 0.573 | 13.21 | 0.348 | 0.695 |
| AnySplat [ 28 ] | 18.32 | 0.524 | 0.329 | 21.85 | 0.670 | 0.250 |
| Ours | 15.60 | 0.414 | 0.486 | 17.65 | 0.459 | 0.374 |