期刊
IEEE ACCESS
卷 7, 期 -, 页码 18450-18463出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/ACCESS.2019.2896409
关键词
Computer-aided diagnosis (CAD); lung nodule; computed tomography (CT); synthetic minority over-sampling; deep learning; data imbalance; adversarial neural networks
资金
- National Key Research and Development Program of China [2017YFA0700900]
- National Natural Science Foundation of China [61772482, 61501305, 61672438]
- Youth Innovation Promotion Association CAS [2017497]
- Anhui Provincial Natural Science Foundation [BJ2150110002]
- Sichuan Provincial Open Foundation of Civil-Military Integration Research Institute [2017SCII0219, 2017SCII0220]
- Key Project of Sichuan Provincial Science and Technology Innovation [19MZGC0123]
Data imbalance issue generally exists in most medical image analysis problems and maybe getting important with the popularization of data-hungry deep learning paradigms. We explore the cutting-edge Wasserstein generative adversarial networks (WGANs) to address the data imbalance problem with oversampling on the minority classes. The WGAN can estimate the underlying distribution of a minority class to synthesize more plausible and helpful samples for the classification model. In this paper, the WGAN-based over-sampling technique is applied to augment the data to balance for the fine-grained classification of seven semantic attributes of lung nodules in computed tomography images. The fine-grained classification is carried out with a normal convolutional neural network (CNN). To further illustrate the efficacy of the WGAN-based over-sampling technique, the conventional data augmentation method commonly used in many deep learning works, the generative adversarial networks (GANs), and the deep convolutional generative adversarial networks (DCGANs) are implemented for comparison. The whole schemes of the minority oversampling and fine-grained classification are tested with the public lung imaging database consortium dataset. The experimental results suggest that the WGAN-based oversampling technique can synthesize helpful samples for the minority classes to assist the training of the CNN model and to boost the fine-grained classification performance better than the conventional data augmentation method and the two schemes of the GAN and DCGAN techniques do. It may thus suggest that the WGAN technique offers an alternative methodological option for the further deep learning on imbalanced classification studies.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据