Deep Adversarial Learning for Multi-Modality Missing Data Completion

被引:114
作者
Cai, Lei [1 ]
Wang, Zhengyang [1 ]
Gao, Hongyang [1 ]
Shen, Dinggang [2 ]
Ji, Shuiwang [1 ]
机构
[1] Washington State Univ, Pullman, WA 99164 USA
[2] Univ N Carolina, Chapel Hill, NC 27515 USA
来源
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING | 2018年
基金
美国国家卫生研究院; 美国国家科学基金会;
关键词
Deep learning; adversarial loss function; missing data completion; disease diagnosis;
D O I
10.1145/3219819.3219963
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Multi-modality data are widely used in clinical applications, such as tumor detection and brain disease diagnosis. Different modalities can usually provide complementary information, which commonly leads to improved performance. However, some modalities are commonly missing for some subjects due to various technical and practical reasons. As a result, multi-modality data are usually incomplete, raising the multi-modality missing data completion problem. In this work, we formulate the problem as a conditional image generation task and propose an encoder-decoder deep neural network to tackle this problem. Specifically, the model takes the existing modality as input and generates the missing modality. By employing an auxiliary adversarial loss, our model is able to generate high-quality missing modality images. At the same time, we propose to incorporate the available category information of subjects in training to enable the model to generate more informative images. We evaluate our method on the Alzheimer's Disease Neuroimaging Initiative (ADNI) database, where positron emission tomography (PET) modalities are missing. Experimental results show that the trained network can generate high-quality PET modalities based on existing magnetic resonance imaging (MRI) modalities, and provide complementary information to improve the detection and tracking of the Alzheimer's disease. Our results also show that the proposed methods generate higher quality images than baseline methods as measured by various image quality statistics.
引用
收藏
页码:1158 / 1166
页数:9
相关论文
共 38 条
[1]   Multi-Level Canonical Correlation Analysis for Standard-Dose PET Image Estimation [J].
An, Le ;
Zhang, Pei ;
Adeli, Ehsan ;
Wang, Yan ;
Ma, Guangkai ;
Shi, Feng ;
Lalush, David S. ;
Lin, Weili ;
Shen, Dinggang .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2016, 25 (07) :3303-3315
[2]  
[Anonymous], 2016, PROC 4 INT C LEARN R
[3]  
[Anonymous], 2015, PROC CVPR IEEE
[4]  
[Anonymous], 2016, ARXIV161009585
[5]  
[Anonymous], ELECT J DIFFEREN EQU
[6]  
[Anonymous], 2017, ARXIV170506820
[7]  
[Anonymous], 2015, CoRR
[8]  
[Anonymous], 2015, P INT C MACHINE LEAR
[9]  
[Anonymous], 2017, IEEE T PATTERN ANAL, DOI DOI 10.1109/TPAMI.2016.2644615
[10]  
[Anonymous], 2017, P IEEE C COMP VIS PA