\(\operatorname {cIDIR}\) : Conditioned Implicit Neural Representation for Regularized Deformable Image Registration
摘要
Regularization is essential in deformable image registration (DIR) to ensure that the estimated Deformation Vector Field (DVF) remains smooth, physically plausible, and anatomically consistent. However, fine-tuning regularization parameters in learning-based DIR frameworks is computationally expensive, often requiring multiple training iterations. To address this, we propose \(\operatorname {cIDIR}\) , a novel DIR framework based on Implicit Neural Representations (INRs) that conditions the registration process on regularization hyperparameters. Unlike conventional methods that require retraining for each regularization hyperparameter setting, \(\operatorname {cIDIR}\) is trained over a prior distribution of these hyperparameters, then optimized over the regularization hyperparameters by using the segmentations masks as an observation. Additionally, \(\operatorname {cIDIR}\) models a continuous and differentiable DVF, enabling seamless integration of advanced regularization techniques via automatic differentiation. Evaluated on the DIR-LAB [5, 6] dataset, \(\operatorname {cIDIR}\) achieves high accuracy and robustness across the dataset.