References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Multi-scale Neural ODEs for 3D Medical Image Registration
(Submitted on 16 Jun 2021 (v1), last revised 17 Jun 2021 (this version, v2))
Abstract: Image registration plays an important role in medical image analysis. Conventional optimization based methods provide an accurate estimation due to the iterative process at the cost of expensive computation. Deep learning methods such as learn-to-map are much faster but either iterative or coarse-to-fine approach is required to improve accuracy for handling large motions. In this work, we proposed to learn a registration optimizer via a multi-scale neural ODE model. The inference consists of iterative gradient updates similar to a conventional gradient descent optimizer but in a much faster way, because the neural ODE learns from the training data to adapt the gradient efficiently at each iteration. Furthermore, we proposed to learn a modal-independent similarity metric to address image appearance variations across different image contrasts. We performed evaluations through extensive experiments in the context of multi-contrast 3D MR images from both public and private data sources and demonstrate the superior performance of our proposed methods.
Submission history
From: Junshen Xu [view email][v1] Wed, 16 Jun 2021 00:26:53 GMT (799kb,D)
[v2] Thu, 17 Jun 2021 21:05:42 GMT (799kb,D)
Link back to: arXiv, form interface, contact.