The Right Prior for the Right Deformation: Rethinking Continuous Deformable Image Registration
Organizations: Department of Radiation Oncology, University of California, San Francisco, San Francisco, CA, USA · Biomedical Imaging Research Institute, Cedars-Sinai Medical Center, Los Angeles, CA, USA · Department of Bioengineering, University of California, Los Angeles, Los Angeles, CA, USA · Department of Radiation Oncology, University of California, Los Angeles, Los Angeles, CA, USA
Abstract
Deformable image registration models implicitly encode deformation priors through their parametrization and optimization. In this work, we conduct a validation study on continuous registration methods to examine how these implicit priors affect performance across different registration tasks. Classic B-Spline transformations impose locality, smoothness, and scale through their control-point structure, whereas recent INR-based methods impose different priors through neural parameterization and optimization. We compare INR-Dense (IDIR), which directly models a dense displacement field using a SIREN-based INR; INR-BSCP (SINR), which predicts B-Spline control points with an INR; D-BSCP, which directly optimizes single-scale B-Spline control points; and MR-D-BSCP, which adds a multiresolution coarse-to-fine scheme. Experiments on inter-subject brain MR registration (OASIS) and intra-subject exhale-to-inhale lung CT registration (DIR-LAB 4DCT) reveal different behavior across deformation regimes. On OASIS, where deformations are moderate but locally complex, D-BSCP matches or slightly outperforms INR-BSCP, suggesting that the B-Spline parameterization accounts for much of INR-BSCP's effectiveness. On DIR-LAB 4DCT, where respiratory motion is larger and more coherent, single-scale B-Spline methods (D-BSCP and INR-BSCP) are less suitable, while INR-Dense and MR-D-BSCP are more effective. Across both tasks, MR-D-BSCP achieves the best performance among the tested continuous parameterizations. These findings highlight that registration accuracy depends strongly on matching the induced deformation prior to the target motion pattern, and support prior-deformation matching as a practical design principle for medical image registration. Our code will be available at https://github.com/HengjieLiu/RightPriorDIR.
Figures & tables
| Method | Dice | Steps | Cold (s) | Warm (s) | Peak GPU (GB) |
|---|---|---|---|---|---|
| base fast | base fast | base fast | base fast | - | |
| MR-D-BSCP (8-4-2) | 0.810 0.809 | 2500 440 | 152.0 101.1 | 62.4 11.0 | 3.93 |
| D-BSCP (cps2) | 0.805 0.805 | 2500 431 | 135.4 76.9 | 80.9 14.4 | 2.16 |
| D-BSCP (cps4) | 0.799 0.797 | 2500 477 | 126.3 78.6 | 80.9 15.3 | 1.83 |
| INR-BSCP (cps2) | 0.799 0.793 | 2500 799 | 370.9 171.4 | 277.9 90.2 | 7.72 |
| INR-BSCP (cps4) | 0.796 0.786 | 2500 644 | 175.0 90.4 | 107.8 29.2 | 2.58 |
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