Unsupervised End-to-end Learning for Deformable Medical Image Registration

Abstract

We propose a registration algorithm for 2D CT/MRI medical images with a new unsupervised end-to-end strategy using convolutional neural networks. The contributions of our algorithm are threefold: (1) We transplant traditional image registration algorithms to an end-to-end convolutional neural network framework, while maintaining the unsupervised nature of image registration problems. The image-to-image integrated framework can simultaneously learn both image features and transformation matrix for registration. (2) Training with additional data without any label can further improve the registration performance by approximately 10 %. (3) The registration speed is 100x faster than traditional methods. The proposed network is easy to implement and can be trained efficiently. Experiments demonstrate that our system achieves state-of-the-art results on 2D brain registration and achieves comparable results on 2D liver registration. It can be extended to register other organs beyond liver and brain such as kidney, lung, and heart.

References in corpus (1)

• Very Deep Convolutional Networks for Large-Scale Image Recognition

Cited by in corpus (9)

• Deep Learning in Medical Image Registration: A Review
• TorchIO: A Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learning
• Inverse-Consistent Deep Networks for Unsupervised Deformable Image Registration
• AIRNet: Self-Supervised Affine Registration for 3D Medical Images using Neural Networks
• When Deep Learning Meets Data Alignment: A Review on Deep Registration Networks (DRNs)
• Applying Transfer Learning To Deep Learned Models For EEG Analysis
• FAIM -- A ConvNet Method for Unsupervised 3D Medical Image Registration
• On Reducing Negative Jacobian Determinant of the Deformation Predicted by Deep Registration Networks
• A Unified Generation-Registration Framework for Improved MR-based CT Synthesis in Proton Therapy