DrivenMorph: Bridging Attention Mechanism and Variational Image Registration via Difference Modeling
This work provides an explainable and efficient solution for learning-based medical image registration, addressing the lack of physical explainability and fine-grained deformation control in existing approaches.
DrivenMorph introduces a physically inspired inductive bias via difference modeling to bridge attention mechanisms with variational image registration, achieving superior performance over state-of-the-art methods on multiple 3D brain MRI datasets.
Medical image registration benefits significantly from deep learning, yet existing approaches often lack physical explainability and fine-grained deformation control. Motivated by Demons algorithms, we propose a novel DrivenMorph framework that bridges attention mechanisms with variational image registration by incorporating difference modeling as a physically inspired inductive bias. The resulting driving force, computed from local differences in the latent feature space, provides explicit semantic guidance throughout the registration process. It directly drives the registration process through a neural Demons layer that simulates force-displacement interactions to generate smooth and anatomically consistent deformation. Unlike previous methods, our approach not only integrates traditional registration principles with popular deep networks, providing an explainable and efficient solution for learning-based medical image registration, but also separates difference modeling from deformation, improving modularity and explainability. Extensive experiments on multiple 3D brain MRI datasets demonstrate superior performance over state of-the-art learning-based and optimization-based methods. Furthermore, visualizations and statistical analyses confirm that the learned driving force aligns closely with actual deformation patterns, supporting its explanatory value.