Computer-aided diagnosis of lung nodule classification between benign nodule, primary lung cancer, and metastatic lung cancer at different image size using deep convolutional neural network with transfer learning
Open Access
- 27 July 2018
- journal article
- research article
- Published by Public Library of Science (PLoS) in PLOS ONE
- Vol. 13 (7), e0200721
- https://doi.org/10.1371/journal.pone.0200721
Abstract
We developed a computer-aided diagnosis (CADx) method for classification between benign nodule, primary lung cancer, and metastatic lung cancer and evaluated the following: (i) the usefulness of the deep convolutional neural network (DCNN) for CADx of the ternary classification, compared with a conventional method (hand-crafted imaging feature plus machine learning), (ii) the effectiveness of transfer learning, and (iii) the effect of image size as the DCNN input. Among 1240 patients of previously-built database, computed tomography images and clinical information of 1236 patients were included. For the conventional method, CADx was performed by using rotation-invariant uniform-pattern local binary pattern on three orthogonal planes with a support vector machine. For the DCNN method, CADx was evaluated using the VGG-16 convolutional neural network with and without transfer learning, and hyperparameter optimization of the DCNN method was performed by random search. The best averaged validation accuracies of CADx were 55.9%, 68.0%, and 62.4% for the conventional method, the DCNN method with transfer learning, and the DCNN method without transfer learning, respectively. For image size of 56, 112, and 224, the best averaged validation accuracy for the DCNN with transfer learning were 60.7%, 64.7%, and 68.0%, respectively. DCNN was better than the conventional method for CADx, and the accuracy of DCNN improved when using transfer learning. Also, we found that larger image sizes as inputs to DCNN improved the accuracy of lung nodule classification.This publication has 30 references indexed in Scilit:
- Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer LearningIEEE Transactions on Medical Imaging, 2016
- Cancer statistics, 2016CA: A Cancer Journal for Clinicians, 2016
- Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet ClassificationPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2015
- Improving Computer-Aided Detection UsingConvolutional Neural Networks and Random View AggregationIEEE Transactions on Medical Imaging, 2015
- Computer-aided classification of lung nodules on computed tomography images via deep learning techniqueOncoTargets and Therapy, 2015
- ImageNet Large Scale Visual Recognition ChallengeInternational Journal of Computer Vision, 2015
- Computer-Aided Diagnosis Systems for Lung Cancer: Challenges and MethodologiesInternational Journal of Biomedical Imaging, 2013
- Reduced Lung-Cancer Mortality with Low-Dose Computed Tomographic ScreeningNew England Journal of Medicine, 2011
- The Lung Image Database Consortium (LIDC) and Image Database Resource Initiative (IDRI): A Completed Reference Database of Lung Nodules on CT ScansMedical Physics, 2011
- Multiresolution gray-scale and rotation invariant texture classification with local binary patternsIeee Transactions On Pattern Analysis and Machine Intelligence, 2002