XRefine: Attention-Guided Keypoint Match Refinement
About
Sparse keypoint matching is crucial for 3D vision tasks, yet current keypoint detectors often produce spatially inaccurate matches. Existing refinement methods mitigate this issue through alignment of matched keypoint locations, but they are typically detector-specific, requiring retraining for each keypoint detector. We introduce XRefine, a novel, detector-agnostic approach for sub-pixel keypoint refinement that operates solely on image patches centered at matched keypoints. Our cross-attention-based architecture learns to predict refined keypoint coordinates without relying on internal detector representations, enabling generalization across detectors. Furthermore, XRefine can be extended to handle multi-view feature tracks. Experiments on MegaDepth, KITTI, and ScanNet demonstrate that the approach consistently improves geometric estimation accuracy, achieving superior performance compared to existing refinement methods while maintaining runtime efficiency. Our code and trained models can be found at https://github.com/boschresearch/xrefine.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Pose Estimation | KITTI odometry | AUC584.7 | 51 | |
| Pose Estimation | ScanNet | AUC @ 5 deg22.49 | 41 | |
| Pose Estimation | MegaDepth | AUC563.07 | 31 | |
| 3D Triangulation | ETH3D | Accuracy (1cm)84.31 | 9 | |
| Keypoint Refinement | NVIDIA RTX A5000 | Runtime (ms)3.61 | 5 | |
| Relative Pose Estimation | MegaDepth | Avg Improvement AUC515.99 | 4 | |
| Relative Pose Estimation | KITTI | Avg Improvement AUC51.18 | 4 | |
| Relative Pose Estimation | ScanNet | Average AUC5 Improvement17.52 | 4 | |
| Point cloud triangulation | ETH3D Outdoor | Accuracy (1cm)62.83 | 3 |