Deep Learning in Medical Image Registration: Magic or Mirage?
Rohit Jena, Deeksha Sethi, Pratik Chaudhari, James C. Gee
Abstract
Classical optimization and learning-based methods are the two reigning paradigms in deformable image registration. While optimization-based methods boast generalizability across modalities and robust performance, learning-based methods promise peak performance, incorporating weak supervision and amortized optimization. However, the exact conditions for either paradigm to perform well over the other are shrouded and not explicitly outlined in the existing literature. In this paper, we make an explicit correspondence between the mutual information of the distribution of per-pixel intensity and labels, and the performance of classical registration methods. This strong correlation hints to the fact that architectural designs in learning-based methods is unlikely to affect this correlation, and therefore, the performance of learning-based methods. This hypothesis is thoroughly validated with state-of-the-art classical and learning-based methods. However, learning-based methods with weak supervision can perform high-fidelity intensity and label registration, which is not possible with classical methods. Next, we show that this high-fidelity feature learning does not translate to invariance to domain shift, and learning-based methods are sensitive to such changes in the data distribution. Finally, we propose a general recipe to choose the best paradigm for a given registration problem, based on these observations.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 678ba89e-438c-43c6-aaaa-b888650ff4dfCited by top-tier papers1
Ask how each one uses itBuilds on10
- Recursive Cascaded Networks for Unsupervised Medical Image RegistrationShengyu Zhao, Yue Dong, Eric I-Chao Chang, Yan XuICCV 2019 · 289 citations
- Coupling-based Invertible Neural Networks Are Universal Diffeomorphism ApproximatorsTakeshi Teshima, Isao Ishikawa, Koichi Tojo, Kenta Oono et al.NeurIPS 2020 · 129 citations
- Affine Medical Image Registration with Coarse-to-Fine Vision TransformerTony C. W. Mok, Albert C. S. ChungCVPR 2022 · 94 citations
- CorticalFlow: A Diffeomorphic Mesh Transformer Network for Cortical Surface ReconstructionLéo Lebrat, Rodrigo Santa Cruz, Frédéric de Gournay, Darren Fu et al.NeurIPS 2021 · 46 citations
- Deep Equilibrium Optical Flow EstimationShaojie Bai, Zhengyang Geng, Yash Savani, J. Zico KolterCVPR 2022 · 45 citations
Related papers
- Fast Symmetric Diffeomorphic Image Registration with Convolutional Neural NetworksTony C. W. Mok, Albert C. S. ChungCVPR 2020
- Correlation-aware Coarse-to-fine MLPs for Deformable Medical Image RegistrationMingyuan Meng, Dagan Feng, Lei Bi, Jinman KimCVPR 2024 · 47 citations
- PointNetLK RevisitedXueqian Li, Jhony Kaesemodel Pontes, Simon LuceyCVPR 2021
- A Plug-and-Play Image Registration NetworkJunhao Hu, Weijie Gan, Zhixin Sun, Hongyu An et al.ICLR 2024 · 12 citations
- Rethinking Box Supervision: Bias-Free Weakly Supervised Medical SegmentationJun Wei, Hui HuangCVPR 2026
