RIPE: Reinforcement Learning on Unlabeled Image Pairs for Robust Keypoint Extraction
About
We introduce RIPE, an innovative reinforcement learning-based framework for weakly-supervised training of a keypoint extractor that excels in both detection and description tasks. In contrast to conventional training regimes that depend heavily on artificial transformations, pre-generated models, or 3D data, RIPE requires only a binary label indicating whether paired images represent the same scene. This minimal supervision significantly expands the pool of training data, enabling the creation of a highly generalized and robust keypoint extractor. RIPE utilizes the encoder's intermediate layers for the description of the keypoints with a hyper-column approach to integrate information from different scales. Additionally, we propose an auxiliary loss to enhance the discriminative capability of the learned descriptors. Comprehensive evaluations on standard benchmarks demonstrate that RIPE simplifies data preparation while achieving competitive performance compared to state-of-the-art techniques, marking a significant advancement in robust keypoint extraction and description. To support further research, we have made our code publicly available at https://github.com/fraunhoferhhi/RIPE.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Pose Estimation | Graz4K (test) | AUC@554.5 | 29 | |
| Stereo Pose Recovery | MD1500 | AUC@543.1 | 22 | |
| Relative Pose Estimation | MegaDepth 1500 (test) | AUC@5°45.4 | 20 | |
| Sparse 3D Reconstruction | ETH Local Feature Benchmark Madrid Metropolis v1.0 | nReg644 | 17 | |
| 3D Reconstruction | ETH local feature benchmark Gendarmenmarkt | Image Count1.07e+3 | 16 | |
| 3D Reconstruction | ETH local feature benchmark Tower of London | Image Count823 | 16 | |
| Local Feature Matching | HPatches (108 scenes) | MMA @1px38.3 | 11 | |
| Relative Pose Estimation | ScanNet (test) | AUC@5°9.4 | 10 | |
| Visual Odometry | KITTI Odometry Benchmark Seq-03 (test) | ATE3.9 | 5 | |
| Visual Odometry | KITTI Odometry Benchmark Seq-01 (test) | ATE43.7 | 5 |