Elsevier

Medical Image Analysis

Volume 72, August 2021, 102139
Medical Image Analysis

CNN-based lung CT registration with multiple anatomical constraints

https://doi.org/10.1016/j.media.2021.102139Get rights and content
Under a Creative Commons license
open access

Highlights

  • We present a deep-learning-based method for lung registration.

  • We introduce a novel constraining method to control volume change and therefore avoid foldings inside the deformation field.

  • We integrate keypoints correspondence into the loss function to increase the alignment of airways and vessels.

Abstract

Deep-learning-based registration methods emerged as a fast alternative to conventional registration methods. However, these methods often still cannot achieve the same performance as conventional registration methods because they are either limited to small deformation or they fail to handle a superposition of large and small deformations without producing implausible deformation fields with foldings inside.

In this paper, we identify important strategies of conventional registration methods for lung registration and successfully developed the deep-learning counterpart. We employ a Gaussian-pyramid-based multilevel framework that can solve the image registration optimization in a coarse-to-fine fashion. Furthermore, we prevent foldings of the deformation field and restrict the determinant of the Jacobian to physiologically meaningful values by combining a volume change penalty with a curvature regularizer in the loss function. Keypoint correspondences are integrated to focus on the alignment of smaller structures.

We perform an extensive evaluation to assess the accuracy, the robustness, the plausibility of the estimated deformation fields, and the transferability of our registration approach. We show that it achieves state-of-the-art results on the COPDGene dataset compared to conventional registration method with much shorter execution time. In our experiments on the DIRLab exhale to inhale lung registration, we demonstrate substantial improvements (TRE below 1.2 mm) over other deep learning methods. Our algorithm is publicly available at https://grand-challenge.org/algorithms/deep-learning-based-ct-lung-registration/.

Keywords

Image registration
Lung CT
Deep learning
Volume change control
Keypoints
Multilevel

Cited by (0)