Unsupervised Deformable Image Registration Revisited: Enhancing Performance with Registration-Specific Designs

Hengjie Liu, Dan Ruan, Ke Sheng

Keywords: Deformable Image Registration, Unsupervised Learning, Benchmark, Ablation Study

TL;DR: We pinpoint the true contributors to good unsupervised learning-based deformable image registration.

Abstract: Deformable image registration (DIR) is ill-posed. Many registration-specific designs and regularizations, whose rationale carries across classic optimization methods to deep-learning-based (DL) frameworks, are crucial to registration performance. This paper presents a comprehensive “ablation” type study to pinpoint the key modules for unsupervised mono-modal DL-DIR. Our findings highlight the value of incorporating multi-resolution pyramids, local correlation, and inverse consistency constraints, and show that even simple network architectures can be highly effective. We conducted controlled experiments and benchmarked performance against state-of-the-art methods. The code will be publicly available at: https://github.com/HengjieLiu/Unsupervised-DL-DIR-Revisited.

Submission Number: 21