Deep Learning Based Registration Using Spatial Gradients and Noisy Segmentation Labels

Estienne, Théo; Vakalopoulou, Maria; Battistella, Enzo; Carré, Alexandre; Henry, Théophraste; Lerousseau, Marvin; Robert, Charlotte; Paragios, Nikos; Deutsch, Eric

Image registration is one of the most challenging problems in medical image analysis. In the recent years, deep learning based approaches became quite popular, providing fast and performing registration strategies. In this short paper, we summarise our work presented on Learn2Reg challenge 2020. The main contributions of our work rely on (i) a symmetric formulation, predicting the transformations from source to target and from target to source simultaneously, enforcing the trained representations to be similar and (ii) integration of variety of publicly available datasets used both for pretraining and for augmenting segmentation labels. Our method reports a mean dice of 0.64 for task 3 and 0.85 for task 4 on the test sets, taking third place on the challenge. Our code and models are publicly available at https://github.com/TheoEst/abdominal_registration and https://github.com/TheoEst/hippocampus_registration.