Diffeomorphic Lung Registration Using Deep CNNs and Reinforced Learning

Image registration is a well-known problem in the field of medical imaging. In this paper, we focus on the registration of chest inspiratory and expiratory computed tomography (CT) scans from the same patient. Our method recovers the diffeomorphic elastic displacement vector field (DVF) by jointly regressing the direct and the inverse transformation. Our architecture is based on the RegNet network but we implement a reinforced learning strategy that can accommodate a large training dataset. Our results show that our method performs with a lower estimation error for the same number of epochs than the RegNet approach.

[1]  Brian B. Avants,et al.  Evaluation of Registration Methods on Thoracic CT: The EMPIRE10 Challenge , 2011, IEEE Transactions on Medical Imaging.

[2]  Jan Rühaak,et al.  Highly accurate fast lung CT registration , 2013, Medical Imaging.

[3]  E. Regan,et al.  Genetic Epidemiology of COPD (COPDGene) Study Design , 2011, COPD.

[4]  Brian B. Avants,et al.  Symmetric diffeomorphic image registration with cross-correlation: Evaluating automated labeling of elderly and neurodegenerative brain , 2008, Medical Image Anal..

[5]  Bram van Ginneken,et al.  A survey on deep learning in medical image analysis , 2017, Medical Image Anal..

[6]  Thomas Guerrero,et al.  A reference dataset for deformable image registration spatial accuracy evaluation using the COPDgene study archive , 2013, Physics in medicine and biology.

[7]  Daniel Rueckert,et al.  Nonrigid registration using free-form deformations: application to breast MR images , 1999, IEEE Transactions on Medical Imaging.

[8]  Boudewijn P. F. Lelieveldt,et al.  Nonrigid Image Registration Using Multi-scale 3D Convolutional Neural Networks , 2017, MICCAI.

[9]  Andrew W. Moore,et al.  Reinforcement Learning: A Survey , 1996, J. Artif. Intell. Res..

[10]  Josien P. W. Pluim,et al.  Supervised local error estimation for nonlinear image registration using convolutional neural networks , 2017, Medical Imaging.

[11]  Jasper Snoek,et al.  Practical Bayesian Optimization of Machine Learning Algorithms , 2012, NIPS.

[12]  Carl-Fredrik Westin,et al.  Lung Extraction, Lobe Segmentation and Hierarchical Region Assessment for Quantitative Analysis on High Resolution Computed Tomography Images , 2009, MICCAI.

[13]  Z. Jane Wang,et al.  A CNN Regression Approach for Real-Time 2D/3D Registration , 2016, IEEE Transactions on Medical Imaging.