arXiv:2606.03287cs.CV2026-06

用迭代Transformer模拟经典光束法平差,提升3D重建精度与一致性

BA-T: An Iterative Transformer for Two-View Bundle Adjustment

  • 借鉴光束法平差思想,设计可重复的轻量级迭代更新模块
  • 仅用16%参数即超越更大模型,多视角一致性显著提升
  • 适合追求高效高精度3D重建的视觉算法研究者

基于前馈模型的3D重建方法虽借助深度跨视图注意力取得良好性能,但通常依赖复杂的解码器堆叠,且缺乏结构化的几何优化机制,导致多视图一致性差。本文受经典光束法平差(BA)启发,将其视为姿态与局部几何间迭代信息传播过程,提出BA-T:一种在隐式标记空间中实现BA式结构化更新的迭代Transformer。与依赖深层注意力堆栈不同,BA-T通过单个轻量层基于潜在残差进行预测优化。实验表明,BA-T在迭代过程中逐步提升姿态与重建精度,相比传统解码器具备更强的跨视图一致性,且仅使用16%的解码器参数即可达到甚至超越更大模型的性能。BA-T为深度注意力提供了一种紧凑、高效且结构化的替代方案,可在轻量架构中实现高精度3D重建。

原文摘要 · Abstract (English)

Feed-forward models for 3D reconstruction have achieved strong performance using deep cross-view attention to exchange information across images. However, these approaches often depend on heavy decoder stacks and lack a structured mechanism for geometry refinement, resulting in poor multi-view consistency. We address this by drawing inspiration from classical bundle adjustment (BA), which can be viewed as an iterative information propagation process between poses and local geometry. Inspired by BA, we propose BA-T, an iterative Transformer that implements BA-style structured updates as a repeatable layer in implicit token space. Instead of relying on deep attention stacks, BA-T refines predictions based on latent residual by a single lightweight layer. Experiments demonstrate that BA-T progressively improves pose and reconstruction accuracy across iterations, achieves stronger cross-view consistency than conventional decoders, and matches or surpasses substantially larger models while using only 16% of their decoder parameters. BA-T provides a compact, efficient, and structural alternative to depth-heavy attention, enabling accurate 3D reconstruction within a lightweight architecture. The code will be made publicly at https://github.com/zhangganlin/BA-T.

3D重建Transformer光束法平差

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