GAN-based synthetic medical image augmentation for increased CNN performance in liver lesion classification

被引:1144
作者
Frid-Adar, Maayan [1 ]
Diamant, Idit [1 ]
Klang, Eyal [2 ]
Amitai, Michal [2 ]
Goldberger, Jacob [3 ]
Greenspan, Hayit [1 ]
机构
[1] Tel Aviv Univ, Dept Biomed Engn, Tel Aviv, Israel
[2] Cha Sheba Med Ctr, Dept Diagnost Imaging, Tel Hashomer, Israel
[3] Bar Ilan Univ, Fac Engn, IL-52900 Ramat Gan, Israel
关键词
Image synthesis; Data augmentation; Convolutional neural networks; Generative adversarial network; Deep learning; Liver lesions; Lesion classification; CONVOLUTIONAL NEURAL-NETWORKS; COMPUTED-TOMOGRAPHY; HEPATIC-LESIONS; CT IMAGES; DIAGNOSIS; FEATURES;
D O I
10.1016/j.neucom.2018.09.013
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning methods, and in particular convolutional neural networks (CNNs), have led to an enormous breakthrough in a wide range of computer vision tasks, primarily by using large-scale annotated datasets. However, obtaining such datasets in the medical domain remains a challenge. In this paper, we present methods for generating synthetic medical images using recently presented deep learning Generative Adversarial Networks (GANs). Furthermore, we show that generated medical images can be used for synthetic data augmentation, and improve the performance of CNN for medical image classification. Our novel method is demonstrated on a limited dataset of computed tomography (CT) images of 182 liver lesions (53 cysts, 64 metastases and 65 hemangiomas). We first exploit GAN architectures for synthesizing high quality liver lesion ROIs. Then we present a novel scheme for liver lesion classification using CNN. Finally, we train the CNN using classic data augmentation and our synthetic data augmentation and compare performance. In addition, we explore the quality of our synthesized examples using visualization and expert assessment. The classification performance using only classic data augmentation yielded 78.6% sensitivity and 88.4% specificity. By adding the synthetic data augmentation the results increased to 85.7% sensitivity and 92.4% specificity. We believe that this approach to synthetic data augmentation can generalize to other medical classification applications and thus support radiologists' efforts to improve diagnosis. (C) 2018 Elsevier B.V. All rights reserved.
引用
收藏
页码:321 / 331
页数:11
相关论文
共 39 条
[1]  
Abadi M., 2016, TENSORFLOW LARGESCAL
[2]   Classification of hepatic lesions using the matching metric [J].
Adcock, Aaron ;
Rubin, Daniel ;
Carlsson, Gunnar .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2014, 121 :36-42
[3]  
[Anonymous], P SPIE MED IMAG IMAG
[4]  
Ben-Cohen A., 2017, INT WORKSHOP SIMULAT
[5]   Automatic detection and classification of hypodense hepatic lesions on contrast-enhanced venous-phase CT [J].
Bilello, M ;
Gokturk, SB ;
Desser, T ;
Napel, S ;
Jeffrey, RB ;
Beaulieu, CF .
MEDICAL PHYSICS, 2004, 31 (09) :2584-2593
[6]   Computer-aided diagnosis of liver tumors on computed tomography images [J].
Chang, Chin-Chen ;
Chen, Hong-Hao ;
Chang, Yeun-Chung ;
Yang, Ming-Yang ;
Lo, Chung-Ming ;
Ko, Wei-Chun ;
Lee, Yee-Fan ;
Liu, Kao-Lang ;
Chang, Ruey-Feng .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2017, 145 :45-51
[7]  
Costa Pedro, 2017, Towards adversarial retinal image synthesis. arXiv
[8]  
Dai W., 2017, arXiv:1703.08770
[9]  
Denton E. L., 2015, ADV NEURAL INFORM PR, P1486, DOI DOI 10.5555/2969239.2969405
[10]   Task-Driven Dictionary Learning Based on Mutual Information for Medical Image Classification [J].
Diamant, Idit ;
Klang, Eyal ;
Amitai, Michal ;
Konen, Eli ;
Goldberger, Jacob ;
Greenspan, Hayit .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2017, 64 (06) :1380-1392