arXiv Computer Vision
Sep 11

Two-Parameter Flow Map Learning for Continuous-Time Diffeomorphic Image Registration

The paper introduces a new framework for learning continuous-time diffeomorphic image registration by modeling a non-autonomous ODE as a two-parameter flow map. By enforcing cocycle consistency, the method learns flow maps without time discretization or velocity integration during training, enabling efficient inference with few compositions. Experiments on nine datasets show consistent alignment improvements, including a 2.1% Dice gain on brain MRI, 12% TRE reduction on lung CT, and 2.6% Dice improvement on cardiac MRI and ultrasound.

By Mohammadjavad Matinkia, Nilanjan Ray