An Unsupervised Learning Model for Deformable Medical Image Registration
Author(s)
Balakrishnan, Guha; Zhao, Amy; Sabuncu, Mert R.; Guttag, John; Dalca, Adrian V.
DownloadSubmitted version (3.019Mb)
Terms of use
Metadata
Show full item recordAbstract
© 2018 IEEE. We present a fast learning-based algorithm for deformable, pairwise 3D medical image registration. Current registration methods optimize an objective function independently for each pair of images, which can be time-consuming for large data. We define registration as a parametric function, and optimize its parameters given a set of images from a collection of interest. Given a new pair of scans, we can quickly compute a registration field by directly evaluating the function using the learned parameters. We model this function using a CNN, and use a spatial transform layer to reconstruct one image from another while imposing smoothness constraints on the registration field. The proposed method does not require supervised information such as ground truth registration fields or anatomical landmarks. We demonstrate registration accuracy comparable to state-of-the-art 3D image registration, while operating orders of magnitude faster in practice. Our method promises to significantly speed up medical image analysis and processing pipelines, while facilitating novel directions in learning-based registration and its applications. Our code is available at https://github.com/balakg/voxelmorph.
Date issued
2018-06Department
Massachusetts Institute of Technology. Computer Science and Artificial Intelligence LaboratoryPublisher
IEEE
Citation
Balakrishnan, Guha, Zhao, Amy, Sabuncu, Mert R., Guttag, John and Dalca, Adrian V. 2018. "An Unsupervised Learning Model for Deformable Medical Image Registration."
Version: Original manuscript