arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.09644cs.CV

注意力本身即可检索。RetrieveVGGT:通过查询-键相似性检索实现免训练长上下文流式3D重建

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

Zichen Zou, Xiaosong Jia, Zuxuan Wu, Yu-Gang Jiang

AI总结:

RetrieveVGGT通过将VGGT的上下文构建问题转化为检索问题,利用查询-键相似性实现可控内存预算,提升流式3D重建性能。

摘要:

视觉几何基础变换器(VGGT)通过可扩展的Transformer架构推进3D重建,但全局注意力的二次复杂度限制了长上下文应用。StreamVGGT通过因果注意力实现流式处理,但KV缓存随帧数线性增长导致内存溢出和质量下降。本文提出RetrieveVGGT,一种无需训练的框架,将VGGT的上下文构建视为检索问题。通过在每一步检索固定数量的相关帧,VGGT保持可控的内存预算,接近其训练上下文长度。有趣的是,我们发现VGGT中第一个全局注意力层中当前帧查询与缓存历史帧键之间的相似性已是一个强相关指示器,无需额外学习评分。为增强信息多样性,我们提出Segment Sampling,使检索跨越不同的相关段而非单一高相似性区域。我们设计了一种姿态感知的空间记忆机制,根据已估计的相机姿态组织历史帧,实现位置感知检索。大量实验表明,RetrieveVGGT在性能上达到最先进的水平,优于StreamVGGT、TTT3R和InfiniteVGGT,同时保持内存使用恒定,无论序列长度如何。代码可在https://github.com/zzctmd/RetrieveVGGT获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑