arXiv:2605.09644cs.CV2026-05被引 1

不训练即可实现长序列3D重建,用相似度自动选关键帧

Attention Itself Could Retrieve.RetrieveVGGT: Training-Free Long Context Streaming 3D Reconstruction via Query-Key Similarity Retrieval

论文配图:Attention Itself Could Retrieve.RetrieveVGGT: Training-Free Long Context Streaming 3D Reconstruction via Query-Key Similarity Retrieval
图 1 · 摘自论文原文
  • 用查询与缓存键的相似度直接选相关帧,无需额外学习评分
  • 每步只保留固定数量关键帧,内存恒定且优于现有方法
  • 适合需要长时间视频3D重建的研究者和工业应用

视觉几何定位变压器(VGGT)通过可扩展的Transformer架构推动3D重建发展,但全局注意力的二次复杂度限制了长上下文应用。StreamVGGT采用因果注意力实现实时流处理,但其键值缓存随帧数线性增长,导致内存溢出和质量下降。本文提出RetrieveVGGT,一种无需训练的框架,将VGGT的上下文构建转化为检索问题。通过每步检索固定数量的相关帧,VGGT可维持可控内存预算,接近其训练上下文长度。有趣的是,我们发现VGGT首个全局注意力层中当前帧查询与历史帧键的相似度已是强相关性指标,无需额外学习评分。为提升信息多样性,类似推荐系统,提出分段采样策略,使检索覆盖不同相关片段而非单一高相似区域。设计姿态感知空间记忆机制,根据已估计相机位姿组织历史帧,实现位置感知检索。大量实验表明,RetrieveVGGT在性能上超越StreamVGGT、TTT3R和InfiniteVGGT,同时保持序列长度无关的恒定内存消耗。代码已开源。

原文摘要 · Abstract (English)

Visual Geometry Grounded Transformer (VGGT) advances 3D reconstruction via scalable Transformer architecture, but the quadratic complexity of global attention prevents long context application. StreamVGGT enables streaming with causal attention, yet its KV cache grows linearly with frames, causing memory overflow and quality degradation. We present RetrieveVGGT, a training-free framework, which formulates context construction for VGGT as a retrieval problem. By retrieving a fixed number of relevant frames at each step, VGGT maintains a controllable memory budget, which is close to its training context length. Interestingly, we find that the similarity between current frame queries and cached history frame keys at the first global attention layer of VGGT is already a strong indicator of relevance, eliminating the need for additional learned scoring. To enhance information diversity similar to a recommender system, we propose Segment Sampling so that the retrieval spans distinct relevant segments rather than a single high-similarity region. We design a pose-aware spatial memory mechanism that organizes history frames according to their already estimated camera poses, enabling location-aware retrieval. Extensive experiments demonstrate that RetrieveVGGT achieves state-of-the-art performance, outperforming StreamVGGT, TTT3R, and InfiniteVGGT while maintaining constant memory usage regardless of sequence length. Code is available at https://github.com/zzctmd/RetrieveVGGT.

3D重建长序列记忆机制检索

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。