Domain-Shift Immunity in Deep Deformable Registration via Local Feature Representations
By: Mingzhen Shao, Sarang Joshi
Potential Business Impact:
Makes medical scans match perfectly, even different types.
Deep learning has advanced deformable image registration, surpassing traditional optimization-based methods in both accuracy and efficiency. However, learning-based models are widely believed to be sensitive to domain shift, with robustness typically pursued through large and diverse training datasets, without explaining the underlying mechanisms. In this work, we show that domain-shift immunity is an inherent property of deep deformable registration models, arising from their reliance on local feature representations rather than global appearance for deformation estimation. To isolate and validate this mechanism, we introduce UniReg, a universal registration framework that decouples feature extraction from deformation estimation using fixed, pre-trained feature extractors and a UNet-based deformation network. Despite training on a single dataset, UniReg exhibits robust cross-domain and multi-modal performance comparable to optimization-based methods. Our analysis further reveals that failures of conventional CNN-based models under modality shift originate from dataset-induced biases in early convolutional layers. These findings identify local feature consistency as the key driver of robustness in learning-based deformable registration and motivate backbone designs that preserve domain-invariant local features.
Similar Papers
Strategies for Robust Deep Learning Based Deformable Registration
CV and Pattern Recognition
Makes brain scans match better, even with different pictures.
Disentangling Progress in Medical Image Registration: Beyond Trend-Driven Architectures towards Domain-Specific Strategies
Image and Video Processing
Improves medical scans by focusing on what matters.
Surrogate Supervision for Robust and Generalizable Deformable Image Registration
CV and Pattern Recognition
Makes medical scans match perfectly, even if messy.