The Contextual Correspondence Problem

Anonymous Author(s)

Under review. Please do not distribute.

Contextual correspondence overview

Abstract

Image correspondence is a fundamental problem in vision and a useful spatial interface for robotics, where correspondences can ground reconstruction, contact selection, and action transfer. It is often formulated as context-free, where correspondences are determined solely by “visual similarity,” which is inherently ambiguous. In embodied settings, the correct match depends on the intended use: the same rim point on a mug may correspond to a teapot’s spout when pouring liquid out, or to its top opening when receiving liquid.

In this paper, we introduce contextual correspondence, in which the correspondence-matching task is conditioned on contextual information, such as language or visual cues. Under this view, classical correspondence is a special case where the context is empty, while task-level functional correspondence is a special case where the context is a discrete label (e.g., “pouring from”). We instantiate this formulation using off-the-shelf promptable foundation models and propose an evaluation suite consisting of 2D functional keypoint matching, 3D articulated object reconstruction, and manipulation trajectory transfer in both simulation and on a real robot arm.

Without additional training, explicit context conditioning consistently improves correspondence accuracy and downstream task success, with the largest gains occurring in settings where visual similarity alone is ambiguous. We also show that even strong foundation models remain unreliable in many cases, leaving contextual correspondence as an open problem for embodied perception and action.

Video

Supplementary video.

Results

We instantiate contextual correspondence with two off-the-shelf promptable foundation models, Nano-Banana 2 and Gemini-ER 1.6, and test it across four applications. The core comparison is always the same model and pipeline, with versus without the task context, against context-free feature baselines (DINOv2, DINOv3, DiFT, TLFR, CLIP). Across all four, adding context consistently helps, with the largest gains exactly where visual similarity alone is ambiguous. In every table, blue marks the best result in a column.

1. Functional Point Matching (2D)

Prompt-based models already have strong generic priors, but explicit context still improves matching accuracy (PCK at three thresholds). The gains are largest on Contextual FunKPoint, where the intended interaction determines the correct match.

Qualitative comparison on FunKPoint
Qualitative comparison on FunKPoint (top) and Contextual FunKPoint (bottom). Adding context (w/ ctx) shifts predictions toward the task-relevant target; dots mark ground truth.
Method FunKPoint Random FunKPoint Selected Contextual FunKPoint
@0.05@0.10@0.15 @0.05@0.10@0.15 @0.05@0.10@0.15
Nano-Banana 20.3170.4840.5860.2900.4380.5480.4600.6240.699
w/o context0.2900.4290.5410.2170.3390.4390.2590.4190.511
Gemini-ER 1.60.2610.4800.6240.2480.4620.5790.4360.6220.685
w/o context0.2580.4510.5840.2130.3770.4870.3540.4830.564
DINOv20.1460.3240.4230.1180.2540.3710.2390.4360.503
DINOv30.1230.2750.4040.1140.2410.3370.2640.4600.515
DiFT0.1350.2590.3500.1050.2010.3090.3070.4110.521
TLFR0.1330.2460.3300.1130.2180.2870.2090.3190.387
CLIP0.0540.0700.1400.0440.1150.1850.0740.1600.227

PCK on FunKPoint and Contextual FunKPoint (higher is better).

2. 3D Articulated Object Reconstruction

Predicted correspondences feed an RGB-only articulation-estimation pipeline. Context helps the model place keypoints on the moving part rather than a visually similar static region, avoiding degenerate axis estimates. The pipeline is also lightweight: under 15 s per instance versus 8+ min for the point-cloud-based iTACO reference, which it approaches in several categories.

3D articulation reconstruction
RGB-only pipeline (left) and qualitative comparison on Microwave (top) and Table (bottom). Lines are the projected joint axes; adding context promotes points that better represent the motion.
Method Dishwasher Laptop Microwave Table (P.) Washing Mach.
ori°pos ori°pos ori°pos ori° ori°pos
Nano-Banana 227.330.2737.660.0586.550.0398.6140.240.166
w/o context33.600.3629.920.0576.900.04211.2043.900.216
Gemini-ER 1.620.890.36310.030.0844.230.04015.2049.540.369
w/o context21.750.38910.450.1135.490.04215.9649.040.268
DINOv323.640.28123.640.0907.890.02610.0857.720.300
DINOv224.80*0.470*13.700.18921.300.09010.48*58.700.210
TLFR29.670.30411.180.20628.680.15623.7752.78*0.210*
iTACO†14.070.13021.250.0907.270.10020.257.530.040

Per-category articulation estimation (lower is better); orientation in degrees, position in meters. iTACO† uses point-cloud reconstruction and is excluded from ranking; * marks degenerate runs.

3. Simulated Manipulation (Trajectory Transfer)

We transfer a demonstrated contact configuration to novel objects on Scooping, Hanging, and Hooking, building on MAGIC and swapping only the correspondence module. Context substantially raises success rates; most strikingly, on Hooking the context-free prompt-based variants fail completely.

Simulation manipulation transfer
Pipeline (left) and qualitative results on Hooking (top) and Hanging (bottom). Without context, contact points land on visually plausible but functionally wrong locations, causing the transferred grasp to fail.
MethodScoopingHangingHooking
Gemini-ER 1.60.3330.7270.200
w/o context0.1670.3640.000
Nano-Banana 20.6670.6970.550
w/o context0.5000.2730.000
DINO w/o rot.0.5000.0000.100
DINO w/ rot.†0.6670.6360.400

Success rate on MAGIC simulation manipulation (higher is better).

4. Real-World Robot Manipulation

On a real arm (Dropping, Lifting, Hooking), explicit context improves every task. Lifting rises from 0.200 to 0.567 and Hooking from 0.433 to 0.733 for Gemini-ER. The reference image is essential: removing it (ER w/o ref.) drops Lifting to 0.067. Even so, the best success rates stay well below saturation, so contextual correspondence remains an open problem in physical deployment.

Real-world manipulation transfer
Pipeline (left) and qualitative results on Lifting (top) and Hooking (bottom). Without context, predicted keypoints miss the task-relevant geometry.
MethodDroppingLiftingHooking
Gemini-ER 1.60.7670.5670.733
w/o context0.5670.2000.433
Nano-Banana 20.5330.4000.500
w/o context0.2670.0670.233
DINOv30.1000.0000.067
DINOv20.2330.1330.033
DiFT0.4000.2000.367
ER w/o ref.0.3670.0670.400

Real-world manipulation success rate, averaged over 3 rounds of 10 trials (higher is better).

Demo Rollouts

Successful rollouts on a real robot arm, grouped by task. Each model (Gemini ER, Nano Banana 2) is shown on two target tools.

Dropping

Gemini ERTool 1
Gemini ERTool 2
Nano Banana 2Tool 1
Nano Banana 2Tool 2

Lifting

Gemini ERTool 1
Gemini ERTool 2
Nano Banana 2Tool 1
Nano Banana 2Tool 2

Hooking

Gemini ERTool 1
Gemini ERTool 2
Nano Banana 2Tool 1
Nano Banana 2Tool 2

Conclusion

We formulate contextual correspondence as a context-conditioned mapping between image-level queries, recovering classical and discrete-label functional correspondence as special cases. Across four embodied reasoning and manipulation settings, prompting off-the-shelf foundation models with explicit context substantially improves correspondence and downstream success over context-free baselines. Yet even strong foundation models remain unreliable, leaving contextual correspondence an open problem for embodied perception and action.