Medical image registration is a typical two-image task which requires specialized feature representation networks for deep-learning-based methods (The existing methods and their limitations have been evaluated in our papers). Therefore, we designed a X-shape feature representation backbone which combines the relationship-aware capacity of Transformer and the traits of two-image tasks which foucus not only on structure information of each image but also on cross correspondence between the image pair. The overall structure of our network is following:
This repository provides the official implementation of XMorpher and its application under two different strategies in the following paper:
XMorpher: Full Transformer for Deformable Medical Image Registration via Cross Attention
Jiacheng Shi1, Yuting He1, Youyong Kong1,2,3,
Jean-Louis Coatrieux1,2,3, Huazhong Shu1,2,3, Guanyu Yang1,2,3, and Shuo Li4
1 LIST, Key Laboratory of Computer Network and Information Integration (Southeast University), Ministry of Education, Nanjing, China
2 Jiangsu Provincial Joint International Research Laboratory of Medical Information Processing
3 Centre de Recherche en Information Biomédicale Sino-Français (CRIBs)
4 Dept. of Medical Biophysics, University of Western Ontario, London, ON, Canada
International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI), 2022
paper | code | poster | video
If you use this code or use our pre-trained weights for your research, please cite our papers:
@inproceedings{shi2022xmorpher,
title={XMorpher: Full Transformer for Deformable Medical Image Registration via Cross Attention},
author={Shi, Jiacheng and He, Yuting and Kong, Youyong and Coatrieux, Jean-Louis and Shu, Huazhong and Yang, Guanyu and Li, Shuo},
booktitle={International Conference on Medical Image Computing and Computer-Assisted Intervention},
pages={217--226},
year={2022},
organization={Springer}
}
- MindSpore/ (updating)
- Pytorch/
★ Notes: implemented under two training strategies VoxelMorph and PC-Reg and the detailed corresponding main functions are Unsup-train.py and Semi-train.py respectively (Pytorch)
- XMorpher has the best DSC score and Jacobian score under both strategies
- XMorpher has visual superiority on some detailed structures
This work was supported in part by the National Natural Science Foundation under grants (62171125, 61828101), CAAI-Huawei MindSpore Open Fund, CANN(Compute Architecture for Neural Networks), Ascend AI Processor, and Big Data Computing Center of Southeast University.