Self-recursive Contextual Network for Unsupervised 3D Medical Image Registration

Bo Hu,Shenglong Zhou,Zhiwei Xiong,Feng Wu
DOI: https://doi.org/10.1007/978-3-030-59861-7_7
2020-01-01
Abstract:In this paper, we propose a self-recursive contextual network for unsupervised 3D medical image registration. Current learning-based registration methods refine an initial deformation field either through cascaded stages or in a coarse-to-fine manner, which improve results at the cost of a rapidly increased number of parameters. Aiming to achieve both elevation of performance and reduction of parameters, we design a novel pyramid structure with a self-recursive scheme and contextual components. Specifically, we adopt a weight-sharing generator to refine the deformation fields recursively across different levels of the pyramid. Meanwhile, we introduce a spatial pyramid pooling module in the feature extractor to capture richer contextual information, as well as a dilated receptive module for the post-processing of the deformation field. Evaluated on two benchmark datasets for 3D medical image registration, our method outperforms the state-of-the-art pyramid network with 39% less parameters, and competes to multi-stage cascaded registration networks with significantly less parameters and faster running speed.
What problem does this paper attempt to address?