UDRSNet: An unsupervised deformable registration module based on image structure similarity.

ABSTRACT

BACKGROUND: Image registration is a challenging problem in many clinical tasks, but deep learning has made significant progress in this area over the past few years. Real-time and robust registration has been made possible by supervised transformation estimation. However, the quality of registrations using this framework depends on the quality of ground truth labels such as displacement field. PURPOSE: To propose a simple and reliable method for registering medical images based on image structure similarity in a completely unsupervised manner. METHODS: We proposed a deep cascade unsupervised deformable registration approach to align images without reliable clinical data labels. Our basic network was composed of a displacement estimation module (ResUnet) and a deformation module (spatial transformer layers). We adopted l 2 $l_2$ -norm to regularize the deformation field instead of the traditional l 1 $l_1$ -norm regularization. Additionally, we utilized structural similarity (ssim) estimation during the training stage to enhance the structural consistency between the deformed images and the reference images. RESULTS: Experiments results indicated that by incorporating ssim loss, our cascaded methods not only achieved higher dice score of 0.9873, ssim score of 0.9559, normalized cross-correlation (NCC) score of 0.9950, and lower relative sum of squared difference (SSD) error of 0.0313 on CT images, but also outperformed the comparative methods on ultrasound dataset. The statistical t $t$ -test results also proved that these improvements of our method have statistical significance. CONCLUSIONS: In this study, the promising results based on diverse evaluation metrics have demonstrated that our model is simple and effective in deformable image registration (DIR). The generalization ability of the model was also verified through experiments on liver CT images and cardiac ultrasound images.