Test-Time Training for Deformable Multi-Scale Image Registration
arXiv:2103.13578
Abstract
Registration is a fundamental task in medical robotics and is often a crucial step for many downstream tasks such as motion analysis, intra-operative tracking and image segmentation. Popular registration methods such as ANTs and NiftyReg optimize objective functions for each pair of images from scratch, which are time-consuming for 3D and sequential images with complex deformations. Recently, deep learning-based registration approaches such as VoxelMorph have been emerging and achieve competitive performance. In this work, we construct a test-time training for deep deformable image registration to improve the generalization ability of conventional learning-based registration model. We design multi-scale deep networks to consecutively model the residual deformations, which is effective for high variational deformations. Extensive experiments validate the effectiveness of multi-scale deep registration with test-time training based on Dice coefficient for image segmentation and mean square error (MSE), normalized local cross-correlation (NLCC) for tissue dense tracking tasks. Two videos are in https://www.youtube.com/watch?v=NvLrCaqCiAE and https://www.youtube.com/watch?v=pEA6ZmtTNuQ
arXiv commentICRA 2021; 8 pages, 4 figures, 2 big tables
References in corpus (8)
- A Deep Learning Framework for Unsupervised Affine and Deformable Image Registration
- A large annotated medical image dataset for the development and evaluation of segmentation algorithms
- AnatomyNet: Deep Learning for Fast and Fully Automated Whole-volume Segmentation of Head and Neck Anatomy
- End-to-End Unsupervised Deformable Image Registration with a Convolutional Neural Network
- Unsupervised Learning for Fast Probabilistic Diffeomorphic Registration
- Recursive Cascaded Networks for Unsupervised Medical Image Registration
- AIRNet: Self-Supervised Affine Registration for 3D Medical Images using Neural Networks
- Cycle-Consistent Adversarial Autoencoders for Unsupervised Text Style Transfer