Source-linked AI summary

Grounding Image Matching in 3D with MASt3R

Vincent Leroy, Yohann Cabon, Jérôme Revaud

arXiv:2406.09756v1cs.CV

TL;DR

Image matching is fundamental to 3D vision but is usually formulated in 2D despite its connection to scene geometry and camera pose. MASt3R extends DUSt3R with dense matching features and efficient reciprocal matching, improving accuracy while retaining robustness. It significantly outperforms prior methods across multiple benchmarks, including a 30% absolute VCRE AUC gain on Map-free localization.

  • Problem

    Image matching is widely used in 3D vision yet is typically treated as a 2D problem, despite corresponding pixels observing the same 3D point and relating to camera pose.

  • Method

    MASt3R augments DUSt3R with dense local feature maps trained with an InfoNCE matching loss and uses coarse-to-fine matching with a fast reciprocal matcher.

  • Results

    MASt3R significantly outperforms state-of-the-art methods on multiple matching and localization benchmarks, including a 30% absolute improvement in VCRE AUC on Map-free localization.

  • Takeaways & Limitations

    MASt3R combines DUSt3R’s viewpoint robustness with more accurate matching and efficient processing, including performance in few-view regimes.

  • Takeaways & Limitations

    MASt3R handles images up to 512 pixels in their largest dimension, so higher-resolution images must be downscaled and correspondences upscaled, potentially degrading localization or reconstruction quality.

Abstract

from arXiv · show

Image Matching is a core component of all best-performing algorithms and pipelines in 3D vision. Yet despite matching being fundamentally a 3D problem, intrinsically linked to camera pose and scene geometry, it is typically treated as a 2D problem. This makes sense as the goal of matching is to establish correspondences between 2D pixel fields, but also seems like a potentially hazardous choice. In this work, we take a different stance and propose to cast matching as a 3D task with DUSt3R, a recent and powerful 3D reconstruction framework based on Transformers. Based on pointmaps regression, this method displayed impressive robustness in matching views with extreme viewpoint changes, yet with limited accuracy. We aim here to improve the matching capabilities of such an approach while preserving its robustness. We thus propose to augment the DUSt3R network with a new head that outputs dense local features, trained with an additional matching loss. We further address the issue of quadratic complexity of dense matching, which becomes prohibitively slow for downstream applications if not carefully treated. We introduce a fast reciprocal matching scheme that not only accelerates matching by orders of magnitude, but also comes with theoretical guarantees and, lastly, yields improved results. Extensive experiments show that our approach, coined MASt3R, significantly outperforms the state of the art on multiple matching tasks. In particular, it beats the best published methods by 30% (absolute improvement) in VCRE AUC on the extremely challenging Map-free localization dataset.

1. Introduction

Image matching is central to 3D vision but is commonly treated as a 2D problem despite its intrinsic connection to scene geometry and camera pose. MASt3R instead grounds matching in 3D, improving DUSt3R’s accuracy and efficiency while preserving robustness to difficult viewpoints.

  • Motivation: Image matching establishes pixel correspondences used across mapping, localization, navigation, photogrammetry, and autonomous robotics.The paper targets accurate, dense matches that remain robust to viewpoint and illumination changes.
  • Motivation: Traditional matching extracts sparse keypoints, describes them with locally invariant features, and pairs them in feature space.Sparse keypoints make matching computationally tractable and precise under low-to-moderate viewpoint and illumination changes.
  • Motivation: Keypoint methods discard global geometric context, making repetitive patterns and low-texture regions especially difficult for local descriptors.Global optimization methods such as SuperGlue address part of this issue during pairing, but the paper argues that geometric context should enter earlier.
  • Motivation: Matching is intrinsically a 3D problem because corresponding pixels observe the same 3D point and relate to camera pose through the epipolar matrix.The paper highlights DUSt3R as evidence that a reconstruction-oriented 3D method can be highly robust to viewpoint changes.
  • Contributions: MASt3R augments DUSt3R with dense local feature maps trained using an InfoNCE loss, then combines coarse-to-fine matching with a faster reciprocal matcher.The reciprocal matching algorithm is almost two orders of magnitude faster and improves pose estimation quality.
  • Contributions: MASt3R significantly outperforms state-of-the-art methods across several absolute and relative pose localization benchmarks.The authors present this as a multi-benchmark improvement in both matching and localization performance.

2. Related works

Prior work spans keypoint-based, dense, and geometry-aware matching approaches. The paper positions 3D grounding as important for challenging viewpoint and illumination conditions where classical 2D matching can fail.

  • Keypoint-based matching: Keypoint-based matching detects keypoints, computes locally invariant descriptors, and searches for nearest neighbors in descriptor space.The paradigm has progressed from handcrafted methods such as SIFT toward learned detection and description.
  • Dense matching: Semi-dense and dense methods consider many or all possible pixel associations and commonly use coarse-to-fine schemes to reduce computational complexity.These approaches resemble optical-flow methods and pursue a more global view of correspondence estimation.
  • Pose estimation: Camera-pose estimation methods rely fundamentally on pixel matching, motivating benchmarks with strong viewpoint or illumination changes, including Map-free.The paper identifies Map-free as the most challenging of the cited benchmarks.
  • 3D-aware matching: 3D grounding is presented as necessary in challenging conditions, while prior geometry-aware methods largely leverage epipolar constraints without fundamentally changing matching.The paper contrasts these approaches with grounding correspondence estimation in physical scene properties.

3. Method

MASt3R jointly reconstructs 3D scene structure and matches pixels by augmenting DUSt3R with dense local features and a matching objective. It also introduces fast reciprocal and coarse-to-fine schemes to make dense matching practical while retaining robustness.

  • 3.1. The DUSt3R framework: MASt3R jointly performs 3D scene reconstruction and image matching from two input images using the DUSt3R framework.The method processes binocular image pairs and predicts dense scene representations for matching.
  • 3.1. The DUSt3R framework: Each input pixel receives a 3D point, confidence value, and local feature, which can be matched through a reciprocal nearest-neighbor procedure.The network produces dense outputs for both images, enabling correspondences from either pointmaps or local features.
  • 3.2. Matching prediction head and loss: A second prediction head outputs dense local feature maps because DUSt3R pointmap matching is robust to viewpoint changes but limited in precision.The added head is implemented as a two-layer MLP with GELU activations and unit-normalized features.
  • 3.2. Matching prediction head and loss: The InfoNCE matching loss rewards the correct pixel correspondence rather than a nearby pixel, directly encouraging high-precision matching.It is applied over ground-truth correspondences and combined with the regression loss.
  • 3.4. Coarse-to-fine matching: Coarse-to-fine matching addresses MASt3R’s 512-pixel maximum working resolution by matching downscaled images before refining correspondences at higher resolution.The procedure begins with coarse correspondences obtained from downscaled image pairs.

4. Experimental results

Experiments evaluate MASt3R across localization, relative pose estimation, and dense multi-view stereo tasks, using diverse training data and fast reciprocal matching. Across these benchmarks, MASt3R improves accuracy and robustness over DUSt3R and prior methods, especially in challenging viewpoint and zero-shot settings.

  • Experimental setup: MASt3R is evaluated across Map-free, CO3Dv2, RealEstate10k, Aachen, InLoc, and DTU, covering pose estimation, visual localization, and dense MVS.The experiments include multiple-view and pairwise pose estimation, retrieved-image localization, and zero-shot reconstruction.
  • Experimental setup: 650k image pairs are sampled per epoch from 14 diverse datasets, including indoor, outdoor, synthetic, real-world, and object-centric scenes.Ten of the training datasets provide metric ground truth.
  • Matching efficiency: 64× faster matching with k=3000 also significantly improves performance, motivating this subsampling choice for subsequent experiments.The paper reports that intermediate subsampling values can improve both speed and accuracy.
  • Ablations: MASt3R descriptors outperform matching 3D points, while combining 3D and matching losses improves pose accuracy over the matching-only objective.Median rotation error is 3.0° with both losses versus 10.8° with the matching-only variant.
  • Map-free localization: More than 93% VCRE AUC and a 30% absolute gain over LoFTR+KBR establish MASt3R as the leading Map-free test-set method.Median translation error decreases to 36cm, compared with approximately 2m for state-of-the-art methods.

5. Conclusion

MASt3R improves DUSt3R-based matching by combining robustness with enhanced accuracy, efficient reciprocal matching, and coarse-to-fine processing. It achieves strong performance across camera-pose and localization tasks, including few-view settings.

  • MASt3R improves DUSt3R matching while preserving robustness and attaining or surpassing pixel-matching performance.The method combines enhanced matching accuracy with DUSt3R’s robustness to viewpoint changes.
  • Fast reciprocal matching and coarse-to-fine processing let users balance matching accuracy and speed.
  • MASt3R performs in few-view regimes, including top-1 settings, increasing localization versatility.

Appendix

The appendix adds qualitative results, proves convergence of fast reciprocal matching, studies its performance gains, and evaluates coarse-to-fine matching through ablation.

  • The appendix presents additional qualitative examples across multiple datasets and the Map-free benchmark.
  • It provides a convergence proof for the fast reciprocal matching algorithm.
  • It analyzes performance gains associated with fast reciprocal matching.
  • It includes an ablation study of coarse-to-fine matching.

A. Additional Qualitative Results

Additional results show that MASt3R produces precise, dense reconstructions and robust correspondences across challenging viewpoints, textures, scales, patterns, and illumination conditions.

  • The qualitative examples cover the Map-free, InLoc, Aachen Day-Night, and DTU datasets.
  • MVS on DTU: MASt3R produces sharp and dense DTU point clouds by directly triangulating coarse-to-fine matches.The reconstructions remain complete in low-contrast homogeneous regions without using ground-truth epipolar constraints.
  • MVS on DTU: MASt3R remains robust to varied textures, materials, and Lambertian-assumption violations such as specularities.
  • Qualitative matching results: MASt3R provides approximately correct correspondences under extreme viewpoint changes, including opposing views.
  • Qualitative matching results: The method handles large scale differences, repetitive or ambiguous patterns, and environmental or day-night illumination changes.
  • Qualitative matching results: Correspondence accuracy degrades gracefully as viewpoint baseline increases, while approximate relative camera poses remain recoverable in extreme cases.

B.1. Theoretical study

Fast Reciprocal Matching operates on nearest-neighbor cycles in a directed bipartite graph, guaranteeing convergence while recovering a sparse subset of reciprocal correspondences efficiently.

  • Algorithm: FRM maps sampled pixels between images through nearest-neighbor links and repeatedly checks whether the mappings form reciprocal cycles.
  • Graph structure: FRM operates on a directed bipartite nearest-neighbor graph whose subgraphs contain at most one cycle.
  • Convergence: Similarity scores increase monotonically along graph walks until they reach a maximal-similarity cycle.
  • Convergence: Every starting point is guaranteed to converge to a reciprocal match through a path leading to the root cycle.
  • Efficiency: The number of active samples quickly drops to zero after six iterations, producing a significant computational speed-up.
  • Sampling: Starting from k ≪ HW samples, FRM recovers a reciprocal subset M_k with cardinality |M_k| = j ≤ k.

B.2. Performance improves with fast matching

FRM improves reciprocal-match sampling by favoring matches with larger convergence basins, while naive random subsampling can cause catastrophic performance drops.

  • FRM biases sampling toward reciprocal matches with larger convergence basins.More initial samples fall onto large basins than small basins.
  • Random subsampling of the full reciprocal correspondence set causes catastrophic performance drops.
  • Basin-biased subsampling performs similarly to fast reciprocal matching and can increase performance relative to naive subsampling.

C. Coarse-to-Fine

Coarse-to-fine matching improves downstream localization and reconstruction compared with coarse-only matching, while naive subsampling performs poorly on Map-free localization.

  • Coarse-to-Fine: Coarse-to-fine matching is compared with coarse-only matching on visual localization and multi-view stereo reconstruction.
  • Visual localization on Aachen Day-Night: Up to 15% top1 performance drops occur on the Night split when using coarse matching only.Aachen Day-Night localization is evaluated at three pose-error thresholds.
  • Map-free benchmark: Naive subsampling leads to catastrophic Map-free localization results, whereas fast and basin-weighted strategies perform similarly.
  • MVS: Coarse-only matching nearly doubles reconstruction errors across all reported DTU metrics relative to coarse-to-fine matching.The metrics are accuracy, completeness, and Chamfer distance.

D. Detailed experimental settings

Training uses fixed weights for confidence and matching losses, a 24-dimensional local feature representation, and an InfoNCE temperature of 0.07.

  • MASt3R training uses α=0.2, β=1, local feature dimension d=24, and InfoNCE temperature τ=0.07.The detailed hyper-parameter settings are reported in Table 6.
Loading 2406.09756v1…