Adversarial learning for mono- or multi-modal registration

被引:118
作者
Fan, Jingfan [1 ,2 ]
Cao, Xiaohuan [1 ,2 ]
Wang, Qian [3 ]
Yap, Pew-Thian [1 ,2 ]
Shen, Dinggang [1 ,2 ,4 ]
机构
[1] Univ N Carolina, Dept Radiol, Chapel Hill, NC 27515 USA
[2] Univ N Carolina, BRIC, Chapel Hill, NC 27515 USA
[3] Shanghai Jiao Tong Univ, Inst Med Imaging Technol, Sch Biomed Engn, Shanghai, Peoples R China
[4] Korea Univ, Dept Brain & Cognit Engn, Seoul 02841, South Korea
关键词
Deformable image registration; Fully convolutional neural network; Generative adversarial network; DIFFEOMORPHIC IMAGE REGISTRATION; ALIGNMENT; ROBUST; DEMONS;
D O I
10.1016/j.media.2019.101545
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper introduces an unsupervised adversarial similarity network for image registration. Unlike existing deep learning registration methods, our approach can train a deformable registration network without the need of ground-truth deformations and specific similarity metrics. We connect a registration network and a discrimination network with a deformable transformation layer. The registration network is trained with the feedback from the discrimination network, which is designed to judge whether a pair of registered images are sufficiently similar. Using adversarial training, the registration network is trained to predict deformations that are accurate enough to fool the discrimination network. The proposed method is thus a general registration framework, which can be applied for both mono-modal and multi-modal image registration. Experiments on four brain MRI datasets and a multi-modal pelvic image dataset indicate that our method yields promising registration performance in accuracy, efficiency and generalizability compared with state-of-the-art registration methods, including those based on deep learning. (C) 2019 Elsevier B.V. All rights reserved.
引用
收藏
页数:10
相关论文
共 65 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], INT C MED IM COMP CO
[3]  
[Anonymous], 2017, ICCV
[4]  
[Anonymous], ARXIV180410735
[5]  
[Anonymous], 2015, PROC CVPR IEEE
[6]  
[Anonymous], 2015, IEEE I CONF COMP VIS, DOI DOI 10.1109/ICCV.2015.123
[7]  
[Anonymous], 2015, PROC 28 INT C NEURAL
[8]   A fast diffeomorphic image registration algorithm [J].
Ashburner, John .
NEUROIMAGE, 2007, 38 (01) :95-113
[9]   Symmetric diffeomorphic image registration with cross-correlation: Evaluating automated labeling of elderly and neurodegenerative brain [J].
Avants, B. B. ;
Epstein, C. L. ;
Grossman, M. ;
Gee, J. C. .
MEDICAL IMAGE ANALYSIS, 2008, 12 (01) :26-41
[10]   An Unsupervised Learning Model for Deformable Medical Image Registration [J].
Balakrishnan, Guha ;
Zhao, Amy ;
Sabuncu, Mert R. ;
Guttag, John ;
Dalca, Adrian V. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :9252-9260