A novel unsupervised learning model for diffeomorphic image registration

Abstract

Deformable medical image registration is vital for doctor's diagnosis and quantitative analysis. In this paper, we propose a novel unsupervised learning model (denoted as BSADM) for 3D diffeomorphic medical image registration. Inspired by spatial attention module, we propose a new network architecture BSAU-Net by introducing a novel Binary Spatial Attention Module (BSAM) into skip connection, which can take full advantages of the spatial information extracted from the encoding path and corresponding decoding path. In addition, from variational method in differential geometry, monitor function f is used to control the Jacobian determinant (JD) of registration field ɸ. So, we also propose a novel orientation-consistent regularization loss to penalize the local regions with negative Jacobian determinant, which further encourages the diffeomorphic property of the transformations. We verify our method on two datasets including ADNI and PPMI dataset, and obtain excellent improvement on magnetic resonance (MR) image registration with higher average Dice scores and better diffeomorphic registration.