Broad-UNet-diff: Diffeomorphic Deformable Medical Image Registration based on Multi-Scale Feature Learning

Abstract

Introduction: To propose a medical image registration method with significant performance improvement. The spatial transformation obtained by the traditional deformable image registration technology is not smooth enough, and the calculation amount is too large to solve the optimization parameters. The network model proposed based on deep learning medical image registration technology has some limitations, which cannot guarantee the registration of topological structures, resulting in the loss of spatial features. It makes the model have topological conservation and transform reversibility, has the ability to learn more multi-scale features and complex image structures, and captures finer changes while clearly encoding global correlation. Method: Based on the core UNet model, a deformable image registration method with a new architecture Broad-UNet-diff is proposed. The model is equipped with asymmetric parallel convolution and uses diffeomorphism mapping. Result: Compared with the seven classical registration methods under the brain MRI datasets, the proposed method has significantly improved the registration performance. In particular, compared with the advanced TransMorph-diff registration method, the Dice score can be improved by 12 %, but only the 1/10 parameters are needed. Conclusion: This method confirms its convincing effectiveness and accuracy.