HyperPredict: Estimating Hyperparameter Effects for Instance-Specific Regularization in Deformable Image Registration

Aisha Lawal Shuaibu10000-0002-6061-2231, Ivor J. A. Simpson10000-0001-5605-6626
1: Informatics, University of Sussex
Publication date: 2024/03/15
https://doi.org/10.59275/j.melba.2024-d434
PDF · Code

Abstract

Methods for medical image registration infer geometric transformations that align pairs, or groups, of images by maximising an image similarity metric. This problem is ill-posed as several solutions may have equivalent likelihoods, also optimising purely for image similarity can yield implausible deformable transformations. For these reasons regularization terms are essential to obtain meaningful registration results. However, this requires the introduction of at least one hyperparameter, often termed λ, which serves as a trade-off between loss terms. In some approaches and situations, the quality of the estimated transformation greatly depends on hyperparameter choice, and different choices may be required depending on the characteristics of the data. Analyzing the effect of these hyperparameters requires labelled data, which is not commonly available at test-time. In this paper, we propose a novel method for evaluating the influence of hyperparameters and subsequently selecting an optimal value for given pair of images. Our approach, which we call HyperPredict, implements a Multi-Layer Perceptron that learns the effect of selecting particular hyperparameters for registering an image pair by predicting the resulting segmentation overlap and measures of deformation smoothness. This approach enables us to select optimal hyperparameters at test time without requiring labelled data, removing the need for a one-size-fits-all cross-validation approach. Furthermore, the criteria used to define optimal hyperparameter is flexible post-training, allowing us to efficiently choose specific properties (e.g. overlap of specific anatomical regions of interest, smoothness/plausibility of the final displacement field). We evaluate our proposed method on the OASIS brain MR standard benchmark dataset using a recent deep learning approach (cLapIRN) and an algorithmic method (Niftyreg). Our results demonstrate good performance in predicting the effects of regularization hyperparameters and highlight the benefits of our image-pair specific approach to hyperparameter selection.

Keywords

Deformable Image Registration · Hyperparameter Selection · Regularization

Bibtex @article{melba:2024:005:shuaibu, title = "HyperPredict: Estimating Hyperparameter Effects for Instance-Specific Regularization in Deformable Image Registration", author = "Shuaibu, Aisha Lawal and Simpson, Ivor J. A.", journal = "Machine Learning for Biomedical Imaging", volume = "2", issue = "Special Issue on Image Registration", year = "2024", pages = "686--716", issn = "2766-905X", doi = "https://doi.org/10.59275/j.melba.2024-d434", url = "https://melba-journal.org/2024:005" }
RISTY - JOUR AU - Shuaibu, Aisha Lawal AU - Simpson, Ivor J. A. PY - 2024 TI - HyperPredict: Estimating Hyperparameter Effects for Instance-Specific Regularization in Deformable Image Registration T2 - Machine Learning for Biomedical Imaging VL - 2 IS - Special Issue on Image Registration SP - 686 EP - 716 SN - 2766-905X DO - https://doi.org/10.59275/j.melba.2024-d434 UR - https://melba-journal.org/2024:005 ER -

2024:005 cover