Source-linked AI summary
ReDFeat: Recoupling Detection and Description for Multimodal Feature Learning
Yuxin Deng, Jiayi Ma
TL;DR
Joint detection-and-description feature learning faces unstable training, particularly for multimodal images with extreme imaging variation. ReDFeat recouples detection and description constraints with detached mutual weights and a Super Detector. The authors report that it surpasses prior state-of-the-art features on their cross-modal benchmark and can be trained from scratch.
Problem
Naively coupling detection scores to descriptor matching risk can lead to a zero detection map and neglect hard descriptors, with both problems magnified by cross-modal imaging variance.
Source: A. Background of Coupled Constraints · p. 3Method
ReDFeat uses detached mutual weights to couple detection and description constraints, alongside a large-receptive-field Super Detector with learnable local non-maximum suppression.
Source: §I · p. 2, D. Network Architecture · p. 5, §VI · p. 10Results
ReDFeat surpasses previous state-of-the-art methods on the authors’ cross-modal benchmark, and its proposed components are reported effective.
Source: §I · p. 2, B. Ablation Study · p. 7, §VI · p. 10Takeaways & Limitations
The benchmark covers cross-visible, near-infrared, infrared, and synthetic-aperture-radar pairs for feature matching and image registration evaluation.
Source: §I · p. 2, §VI · p. 10
Abstract
from arXiv · showhide
Deep-learning-based local feature extraction algorithms that combine detection and description have made significant progress in visible image matching. However, the end-to-end training of such frameworks is notoriously unstable due to the lack of strong supervision of detection and the inappropriate coupling between detection and description. The problem is magnified in cross-modal scenarios, in which most methods heavily rely on the pre-training. In this paper, we recouple independent constraints of detection and description of multimodal feature learning with a mutual weighting strategy, in which the detected probabilities of robust features are forced to peak and repeat, while features with high detection scores are emphasized during optimization. Different from previous works, those weights are detached from back propagation so that the detected probability of indistinct features would not be directly suppressed and the training would be more stable. Moreover, we propose the Super Detector, a detector that possesses a large receptive field and is equipped with learnable non-maximum suppression layers, to fulfill the harsh terms of detection. Finally, we build a benchmark that contains cross visible, infrared, near-infrared and synthetic aperture radar image pairs for evaluating the performance of features in feature matching and image registration tasks. Extensive experiments demonstrate that features trained with the recoulped detection and description, named ReDFeat, surpass previous state-of-the-arts in the benchmark, while the model can be readily trained from scratch.
I. INTRODUCTION
Multimodal feature extraction lacks a general robust solution, while jointly trained detection and description can be unstable when detection is weakly supervised or coupled inappropriately. ReDFeat recouples their constraints, adds a Super Detector, and is evaluated on a cross-modal benchmark.
I. INTRODUCTION
- Multimodal images still lack a general robust feature solution, despite complementary information across modalities and progress in visible-image matching.Joint detection and description can be unstable without a proper definition of detection; naive suppression of hard-to-match features can drive detection probabilities toward zero.Source: §I · p. 1
- ReDFeat uses detached mutual weights to encourage reliable features to peak and repeat while prioritizing high-detection-score descriptors during optimization.The descriptor loss is weighted by detection scores; detection losses are weighted by descriptor matching risk and similarity, with the weights detached from backpropagation.Source: §I · p. 2, Figure 1 · p. 2
- Super Detector combines a large-receptive-field convolutional branch with learnable local non-maximum suppression layers to improve detection.The detector models prior and conditional keypoint probabilities with a fully connected network and a deep convolutional network, respectively.Source: §I · p. 2
- The authors build a benchmark of cross-visible, near-infrared, infrared, and synthetic-aperture-radar image pairs and report that ReDFeat is superior on it.The benchmark evaluates feature matching and image registration; the method is described as trainable from scratch.Source: §I · p. 2
II. RELATED WORKS
Prior work spans handcrafted features, two-stage deep pipelines, and jointly trained one-stage methods. The paper positions its work against limited multimodal joint methods and their suitability for accurate image matching.
Handcrafted Methods
- Handcrafted multimodal features remain prominent amid scarce well-registered data, and their focus on corners or edges motivates injecting edge-based priors into learning.Examples include MFP, SR-SIFT, PCSD, OS-SIFT, RIFT [18], and KAZE-SAR.Source: §II · p. 2
Two-stage Deep Methods
- Two-stage deep methods use classical detectors before learning patch descriptors, limiting detector modifications; their independent constraints also provide a basis for joint frameworks.Key.Net is noted as an early effort to learn keypoint extraction with repeatability as a constraint.Source: §II · p. 2
One-stage Deep Methods
- One-stage methods jointly learn dense detection scores and descriptors, but their coupling strategies vary across synthetic supervision, non-maximum suppression, reliability masks, and reinforcement learning.The paper notes that RL may be infeasible in multimodal scenarios because it is computation- and data-intensive.Source: §II · pp. 2–3
- Besides CMMNet, MAP-Net is identified as the only other joint method customized for multimodal images, and its retrieval-oriented pipeline differs from accurate matching.The authors therefore focus on more feasible joint detection-and-description methods for cross-modal image matching.Source: §II · p. 3
A. Background of Coupled Constraints
Naively coupling detection scores to descriptor matching risk can create a zero-score local minimum and neglect hard descriptors, especially under cross-modal imaging variation. The section sets out basic descriptor and detector constraints for addressing that problem.
A. Background of Coupled Constraints
- Naively weighting descriptor matching risk by detection scores can make the zero detection map a local minimum and deprive hard descriptors of attention, a problem magnified by cross-modal imaging variance.Prior normalization approaches may break detection convexity and make balancing detection and description difficult, while reliability loss can retain the zero-score problem.Source: A. Background of Coupled Constraints · p. 3
B. Basic Constraints
- For cross-modal descriptor learning, the basic contrastive objective minimizes distance to the correspondence while maximizing distance to intra-modal and selected inter-modal non-matching neighbors.The formulation maximizes the larger of the two inter-modal neighbor distances rather than maximizing both simultaneously; angular distance and quadratic matching risk are used to balance pairs and emphasize hard samples.Source: B. Basic Constraints · pp. 3–4
- Detection is constrained by repeatability across warped image pairs and local peaking so that salient, unique points can be extracted.The basic losses use 17×17 patches with stride 8 for repeatability and 17×17 pooling with stride 1 for peaking, following R2D2 empirically.Source: B. Basic Constraints · p. 4
C. Recoupled Constraints
ReDFeat recouples detection and description through mutual weighting: descriptor matching risk guides detection, while detection scores guide descriptor optimization. Detached weights and an edge prior shape the constraints without directly suppressing hard descriptors.
C. Recoupled Constraints
- Descriptor matching risk weights the detection peaking loss, while an edge-based prior discourages interest points on smooth areas.The matching-risk weight is negatively correlated with descriptor risk and is detached from backpropagation.Source: C. Recoupled Constraints · p. 4
- Detached recoupling weights constrain detection peaking without directly suppressing hard descriptors or affecting description learning.The weights are dynamically normalized by their expectation so they remain nonzero and functional.Source: C. Recoupled Constraints · p. 4
- Descriptor similarity between corresponding patches weights repeatability because some multimodal regions do not warrant repeatability.The local similarity is estimated from the average cosine similarity of the corresponding descriptors.Source: C. Recoupled Constraints · p. 4
- Detection scores weight matching risk in descriptor optimization, and the mutual-weighting strategy is intended to stabilize and boost feature learning.The total loss combines descriptor, peaking, and repeatability terms with one hyperparameter.Source: C. Recoupled Constraints · pp. 4–5
D. Network Architecture
The architecture adapts the R2D2 encoder and detector, then replaces the raw linear detector with a Super Detector designed to capture broader context and produce locally peaked detections.
D. Network Architecture
- The raw architecture adopts R2D2’s nine-layer encoder and linear detector, with six shallow adapter layers unshared across modalities and three later convolutional layers shared.The raw detector outputs 128-dimensional dense descriptors and a score map.Source: D. Network Architecture · p. 5
- The Super Detector addresses the raw linear detector’s limited receptive field with a two-branch design for keypoint-prior and conditional detection probabilities.The conditional-probability branch uses stacked convolutional layers to incorporate global information.Source: D. Network Architecture · p. 5
- Learnable non-maximum suppression layers transform features, apply local softmax, and use batch and instance normalization with ReLU to identify maxima.The conditional branch outputs two channels; the first channel after softmax gives conditional detection probability, which is multiplied by keypoint-prior probability for the approximate posterior score.Source: D. Network Architecture · p. 5
IV. BENCHMARK
The paper builds a benchmark spanning three cross-modal image-pair types for evaluating feature matching and image registration.
IV. BENCHMARK
- The benchmark contains RGB-NIR, VIS-IR, and VIS-SAR image pairs for evaluating features in matching and registration pipelines.The authors identify the lack of a multimodal feature-learning benchmark as a major reason for slow development in the area.Source: §IV · p. 5
A. Dataset
The benchmark combines three modality-specific datasets, with distinct scene coverage and evaluation conditions across near-infrared, long-wave infrared, and SAR pairs.
A. Dataset
- VIS-NIR contains 345 training pairs and 128 test pairs drawn from nine scenes, with test ground truths manually validated and filtered.Pairs average 983×686 pixels and cover settings including country, field, forest, indoor, mountain, old building, street, urban, and water.Source: A. Dataset · p. 5
- VIS-IR contains 265 roughly registered visible and long-wave infrared image pairs averaging 533 × 321; 44 static RGB-LWIR pairs and 221 RoadScene video-frame pairs, with 47 images randomly reserved for testing and the rest for training.Source: A. Dataset · p. 5
- VIS-SAR uses aligned optical-SAR pairs split into 2,011 training and 424 test pairs, though SAR fuzziness makes ground-truth validation or landmark labeling difficult.The pairs are 512×512 and cover field and town scenes.Source: A. Dataset · p. 6
B. Evaluation Protocol
The protocol evaluates transformed image pairs through feature matching and image registration, using landmark-based reprojection error where the standard error metric is not indicative for VIS-IR.
- Random perspective distortion, rotation from −10° to 10°, and scaling from 0.8 to 1.0 generate warped test pairs from aligned raw images.Source: B. Evaluation Protocol · p. 6
- Feature matching reports correspondences, repeatable rate, correct matches, and matching score using bidirectional nearest-neighbor matching.Repeatable rate is correspondences divided by detected keypoints; matching score is correct matches divided by detected keypoints in the overlap.Source: B. Evaluation Protocol · p. 6
- Table II covers training stability for joint detection-and-description methods using 1024 keypoints on VIS-IR.Source: Table II · p. 6
- For registration, matched features estimate a homography with RANSAC, while landmark reprojection error is used because the standard metric is not indicative for VIS-IR.Registration success is counted below an error threshold and summarized as a successful registration rate for each subset.Source: B. Evaluation Protocol · p. 6
V. EXPERIMENTS
The experiments describe ReDFeat’s implementation and ablations, testing scratch training, the repeatability-loss weight, and successive component additions. The reported ablations show gains from the proposed components, with the final Super Detector configuration achieving state-of-the-art results.
A. Implementation
- The implementation trains on 192 × 192 crops for about 10,000 iterations with batch size 2, and compares ReDFeat with handcrafted and learned feature methods.Training uses PyTorch and Kornia, Adam, and an NVIDIA RTX3090 GPU.Source: A. Implementation · p. 6
B. Ablation Study
- ReDFeat trains readily from scratch, while CMMNet and R2D2 fail to learn discriminative features without pretraining.ReDFeat obtains only a tiny improvement from self-supervised pretraining; subsequent ReDFeat experiments therefore use scratch training.Source: B. Ablation Study · pp. 6–7
- Registration performance peaks at λ=8, although increasing λ beyond 8 is reported to improve registration while reducing correct matches.The authors retain λ=8 in subsequent experiments.Source: B. Ablation Study · p. 7, Table III · p. 7
- The basic loss improves R2D2 across all metrics, recoupling further improves feature matching, and adding Super Detector achieves state-of-the-art results.Registration with recoupled constraints is comparable to the basic loss.Source: B. Ablation Study · p. 7, Table IV · p. 7
- Detection-score maps compare R2D2 with proposed component configurations; the Super Detector’s learnable local NMS blocks are described as discretizing the scores.The authors also describe edge-based priors as concentrating detections in texture-rich areas.Source: B. Ablation Study · p. 7, Figure 4 · p. 7
C. Feature Matching Performance
ReDFeat leads the reported feature-matching comparisons and registers more image pairs across the tested cases, with particularly strong results on VIS-SAR. Its learned detector has less precise localization than edge- or corner-based traditional detectors, reflecting limits in the training correspondences.
C. Feature Matching Performance
- ReDFeat achieves state-of-the-art repeatable rate and matching score at all thresholds on three subsets, with matching-score margins especially large on VIS-SAR.R2D2, despite pretrained initialization, still fails to optimize description on VIS-SAR.Source: C. Feature Matching Performance · p. 7, Figure 5 · p. 8
- ReDFeat keypoints appear more densely distributed in textured areas, while traditional detectors produce more interpretable points attached to edges or corners.The authors attribute deep-detector localization weakness to training data that cannot provide strict correspondences, leaving keypoints less precisely located.Source: C. Feature Matching Performance · pp. 8–9
- Except for correspondence counts on VIS-IR and VIS-SAR, ReDFeat achieves the best scores on all reported metrics at 3px across 1024, 2048, and 4096 keypoints.Handcrafted local-maximum-search detectors may fail to extract large numbers of keypoints for some image pairs.Source: Figure 6 · p. 8, C. Feature Matching Performance · p. 8
D. Image Registration Performance
- ReDFeat registers more image pairs in every reported case, with a more prominent margin on the challenging VIS-SAR subset.The study uses two projection-error measures across the three subsets according to ground-truth quality.Source: Figure 8 · p. 9, D. Image Registration Performance · p. 9
- On the illustrated examples, only ReDFeat succeeds; other methods fail to obtain even rough registration on VIS-SAR samples.Some alternatives estimate approximate transforms on VIS-NIR and VIS-IR, but their registration accuracy does not meet expectations.Source: D. Image Registration Performance · p. 9
- Except on VIS-NIR with 2048 and 4096 keypoints, ReDFeat achieves the most successful registrations and lowest mean reprojection error.On VIS-SAR, it boosts successful registration rate while obtaining the most precise registration performance.Source: Figure 9 · p. 9, D. Image Registration Performance · p. 9
E. Runtime
ReDFeat improves feature performance at the cost of longer extraction time, attributed to Super Detector’s complex operations. The conclusion presents recoupled detection and description, Super Detector, and a cross-modal benchmark as the paper’s contributions.
E. Runtime
- ReDFeat takes longer to extract features because of Super Detector’s complex operations, while R2D2 has the shortest per-image runtime.The authors characterize the performance gains as coming with few extra costs; SIFT has runtime of the same order of magnitude as R2D2 and ReDFeat.Source: E. Runtime · pp. 9–10
VI. CONCLUSION
- The paper concludes that ReDFeat can be trained from scratch and applied to cross-modal image registration using recoupled constraints and Super Detector.Its benchmark comprises visible–near-infrared, infrared, and synthetic aperture radar image pairs, and experiments support the proposed components.Source: §VI · p. 10