Training CNNs for Image Registration from Few Samples with Model-based Data Augmentation

Hristina Uzunova, Matthias Wilms, Heinz Handels, Jan Ehrhardt

In: Maxime Descoteaux , Lena Maier-Hein , Alfred Franz , Pierre Jannin , D. Louis Collins , Simon Duchesne (editor). Medical Image Computing and Computer Assisted Intervention (MICCAI 2017). Medical Image Computing and Computer Assisted Intervention (MICCAI-2017) September 10-14 Quebec City Quebec Canada Pages 223-231 ISBN 978-3-319-66182-7 Springer International Publishing 2017.


Convolutional neural networks (CNNs) have been successfully used for fast and accurate estimation of dense correspondences between images in computer vision applications. However, much of their success is based on the availability of large training datasets with dense ground truth correspondences, which are only rarely available in medical applications. In this paper, we, therefore, address the problem of CNNs learning from few training data for medical image registration. Our contributions are threefold: (1) We present a novel approach for learning highly expressive appearance models from few training samples, (2) we show that this approach can be used to synthesize huge amounts of realistic ground truth training data for CNN-based medical image registration, and (3) we adapt the FlowNet architecture for CNN-based optical flow estimation to the medical image registration problem. This pipeline is applied to two medical data sets with less than 40 training images. We show that CNNs learned from the proposed generative model outperform those trained on random deformations or displacement fields estimated via classical image registration.

German Research Center for Artificial Intelligence
Deutsches Forschungszentrum für Künstliche Intelligenz