ACorD: Adaptive Correspondence Discovery for Medical Image Registration
Organizations: University of California, Irvine
Abstract
Medical image registration is a fundamental task for aligning anatomical structures across images. Exhaustively searching for point-wise correspondences is prohibitively expensive in both computation and memory, particularly for high-resolution 3D medical images. Most recent learning-based methods directly predict correspondences in a coarse-to-fine manner, but often propagate intermediate estimates across scales without explicitly weighting correspondence evidence, which can allow coarse-scale errors to persist during refinement. To address these challenges, we propose ACorD, an adaptive correspondence discovery framework designed to mitigate error propagation with an attention-guided process. ACorD first predicts multiple potential correspondence candidates through Correlation-guided Cross-scale Offset Prediction (CCOP), which combines correlation maps as spatial matching priors with cross-scale features as global deformation context. Then, Multi-Head Deformable Aggregation (MHDA) aggregates these candidate correspondences using attention, producing a learned, non-uniform weighting over competing matches rather than treating them equally. The multi-head design further captures complementary deformation patterns, ensuring both global alignment and local precision. By integrating adaptive offset prediction with selective, attention-based aggregation of correspondence evidence, ACorD enables accurate deformation refinement. Extensive experiments on four public medical registration benchmarks demonstrate that adaptive correspondence discovery within a multi-scale refinement framework improves registration robustness across diverse anatomical structures.
Figures & tables
| Group | Method | MindBoggle (97 ROIs) | LPBA (54 ROIs) | OASIS (35 ROIs) | #params (M) | ||||||
| DSC | ASSD | NJD(%) | DSC | ASSD | NJD(%) | DSC | ASSD | NJD(%) | |||
| Traditional | Affine | 0.404 0.021 | 2.137 0.196 | 0 | 0.635 0.020 | 2.120 0.125 | 0 | 0.571 0.052 | 1.618 0.281 | 0 | – |
| SyN [ 2 ] | 0.513 0.031 | 1.692 0.211 | 0 | 0.695 0.017 | 1.780 0.109 | 0 | 0.755 0.029 | 0.837 0.123 | 0 | – | |
| Single-Scale | VoxelMorph [ 4 ] | 0.500 0.108 | 2.147 1.092 | 0.883 0.289 | 0.638 0.031 | 2.122 0.242 | 0.589 0.273 | 0.763 0.031 | 0.823 0.145 | 0.963 0.179 | 0.27 |
| ViT-V-Net [ 6 ] | 0.526 0.112 | 2.086 1.128 | 0.797 0.332 | 0.648 0.030 | 2.049 0.225 | 0.735 0.247 | 0.787 0.031 | 0.733 0.136 | 0.917 0.151 | 31.56 | |
| TransMorph [ 7 ] | 0.551 0.118 | 1.963 1.083 | 0.691 0.181 | 0.663 0.030 | 1.982 0.232 | 0.648 0.199 | 0.806 0.025 | 0.661 0.103 | 0.827 0.169 | 46.77 | |
| Group | Method | AbdomenCT (13 ROIs) | ||
|---|---|---|---|---|
| DSC | ASSD | NJD(%) | ||
| Traditional | Affine | 0.251 0.070 | 6.629 2.416 | 0 |
| SyN [ 2 ] | 0.319 0.084 | 5.780 2.443 | 0 | |
| Single-Scale | VoxelMorph [ 4 ] | 0.349 0.051 | 4.793 1.047 | 1.746 0.094 |
| ViT-V-Net [ 6 ] | 0.385 0.054 | 4.547 1.032 | 4.798 1.400 | |
| TransMorph [ 7 ] | 0.409 0.061 | 4.422 1.118 | 5.554 1.368 | |
| Components | AbdomenCT (13 ROIs) | ||||||||
|---|---|---|---|---|---|---|---|---|---|
| Exp | CG | CS | DA | MH | Agg | Offset | DSC | ASSD | NJD(%) |
| 1 | 0.549 0.054 | 2.83 0.67 | 0.105 0.08 | ||||||
| 2 | ✓ | 0.570 0.051 | 2.731 0.658 | 0.103 0.066 | |||||
| 3 | ✓ | ✓ | 0.577 0.046 | 2.626 0.646 | 0.151 0.105 | ||||
| 4 | ✓ | Attn | Learned | 0.564 0.045 | 2.649 0.598 | 0.098 0.064 | |||
| 5 | ✓ | ✓ | Attn | Learned | 0.579 0.049 | 2.585 0.651 | 0.084 0.048 | ||
| Lvl. | MindBoggle | AbdomenCT | ||
|---|---|---|---|---|
| Unif. | Attn. | Unif. | Attn. | |
| 1 | 73.89 | 74.40 | 78.35 | 78.56 |
| 2 | 65.95 | 68.07 | 74.78 | 75.84 |
| 3 | 55.55 | 58.03 | 68.79 | 70.79 |
| 4 | 49.56 | 49.64 | 59.14 | 62.07 |
| 5 | 49.19 | 48.91 | 48.29 | 47.31 |
| Transition | ACorD(%) | Exp.7(%) | ACorD(%) | Exp.8(%) |
|---|---|---|---|---|
| L4 L3 | 40.4 | 38.3 ( ) | 40.2 | 37.0 ( ) |
| L3 L2 | 23.6 | 22.6 ( ) | 23.8 | 20.6 ( ) |
| L2 L1 | 13.2 | 12.2 ( ) | 12.6 | 10.4 ( ) |
| Lvl. | AbdomenCT | LPBA | OASIS | MindBoggle | ||||
|---|---|---|---|---|---|---|---|---|
| Ratio | Frac. | Ratio | Frac. | Ratio | Frac. | Ratio | Frac. | |
| 1 | 1.46 0.02 | 0.73 0.01 | 1.45 0.07 | 0.75 0.03 | 1.07 0.01 | 0.55 0.00 | 1.05 0.02 | 0.51 0.05 |
| 2 | 1.14 0.02 | 0.55 0.01 | 1.18 0.03 | 0.68 0.02 | 0.68 0.00 | 0.11 0.00 | 1.09 0.02 | 0.58 0.01 |
| 3 | 0.93 0.02 | 0.35 0.02 | 0.95 0.01 | 0.32 0.01 | 0.79 0.01 | 0.20 0.00 | 0.97 0.01 | 0.36 0.01 |
| 4 | 1.15 0.03 | 0.50 0.02 | 0.65 0.01 | 0.09 0.01 | 0.75 0.00 | 0.20 0.00 | 0.65 0.03 | 0.13 0.03 |
| 5 | 1.54 0.04 | 0.73 0.01 | 1.10 0.01 | 0.49 0.01 | 0.97 0.01 | 0.29 0.01 | 0.91 0.00 | 0.19 0.00 |
| AbdomenCT | MindBoggle | |||||
|---|---|---|---|---|---|---|
| DSC | ASSD | NJD(%) | DSC | ASSD | NJD(%) | |
| 8 | 0.595 0.056 | 2.475 0.614 | 0.135 0.096 | 0.621 0.054 | 1.512 0.517 | 0.005 0.006 |
| 0.597 0.052 | 2.411 0.603 | 0.136 0.092 | 0.624 0.049 | 1.468 0.447 | 0.005 0.004 | |
| ACorD | ||||
|---|---|---|---|---|
| AbdomenCT | MindBoggle | |||
| DSC | NJD(%) | DSC | NJD(%) | |
| 0 | 0.590 | 8.381 | 0.625 | 0.191 |
| 20 | 0.592 | 0.317 | 0.623 | 0.011 |
| 50 | 0.597 | 0.136 | 0.624 | 0.005 |
| 80 | 0.595 | 0.086 | 0.623 | 0.003 |
| Head Allocation | DSC | ASSD | NJD(%) |
|---|---|---|---|
| default |
| Model | MACs (GMac) | Runtime (ms) | Memory (MB) |
|---|---|---|---|
| TransMorph | 714.16 | 161.15 | 5353.16 |
| TransMatch | 717 | 257 | 4768 |
| TransDCA | 4766 | 593 | 9431 |
| NICE-Trans | 313 | 415 | 6205 |
| RDPNet | 3109 | 428 | 9306 |
| Dual-PR++ | 1773 | 491 | 9461 |
| Model | L5 | L4 | L3 | L2 | L1 |
|---|---|---|---|---|---|
| Exp.8 (Default+Attn) | 0.298 | 0.403 | 0.476 | 0.534 | 0.562 |
| Exp.7 (Learned+Avg) | 0.289 | 0.403 | 0.472 | 0.554 | 0.587 |
| ACorD | 0.304 | 0.408 | 0.500 | 0.563 | 0.597 |
Medical Image Registration
Medical image registration aligns scans across time, patients, or imaging modalities to track anatomy and support diagnosis, treatment, and image-guided procedures.
See all 96 papers