Convolutional neural network-based PSO for lung nodule false positive reduction on CT images

BACKGROUND AND OBJECTIVE Detection of lung nodules is critical in CAD systems; this is because of their similar contrast with other structures and low density, which result in the generation of numerous false positives (FPs). Therefore, this study proposes a methodology to reduce the FP number using a deep learning technique in conjunction with an evolutionary technique. METHOD The particle swarm optimization (PSO) algorithm was used to optimize the network hyperparameters in the convolutional neural network (CNN) in order to enhance the network performance and eliminate the requirement of manual search. RESULTS The methodology was tested on computed tomography (CT) scans from the Lung Image Database Consortium and Image Database Resource Initiative (LIDC-IDRI) with the highest accuracy of 97.62%, sensitivity of 92.20%, specificity of 98.64%, and area under the receiver operating characteristic (ROC) curve of 0.955. CONCLUSION The results demonstrate the high performance-potential of the PSO algorithm in the identification of optimal CNN hyperparameters for lung nodule candidate classification into nodules and non-nodules, increasing the sensitivity rates in the FP reduction step of CAD systems.

[1]  Xiaohong W. Gao,et al.  Classification of CT brain images based on deep learning networks , 2017, Comput. Methods Programs Biomed..

[2]  Steve B. Jiang,et al.  Markerless gating for lung cancer radiotherapy based on machine learning techniques , 2009, Physics in medicine and biology.

[3]  Haider A. Alwzwazy,et al.  Robust Convolutional Neural Networks for Image Recognition , 2015 .

[4]  Anselmo Cardoso de Paiva,et al.  Automatic detection of solitary lung nodules using quality threshold clustering, genetic algorithm and diversity index , 2014, Artif. Intell. Medicine.

[5]  Wei Li,et al.  Pulmonary Nodule Classification with Deep Convolutional Neural Networks on Computed Tomography Images , 2016, Comput. Math. Methods Medicine.

[6]  B. van Ginneken,et al.  Computer-aided diagnosis: how to move from the laboratory to the clinic. , 2011, Radiology.

[7]  Noboru Niki,et al.  A computer-aided diagnosis (CAD) system in lung cancer screening with computed tomography. , 2005, Anticancer research.

[8]  Richard C. Pais,et al.  The Lung Image Database Consortium (LIDC) and Image Database Resource Initiative (IDRI): a completed reference database of lung nodules on CT scans. , 2011, Medical physics.

[9]  K. Bae,et al.  Pulmonary nodules: automated detection on CT images with morphologic matching algorithm--preliminary results. , 2005, Radiology.

[10]  Anselmo Cardoso de Paiva,et al.  Lung nodules diagnosis based on evolutionary convolutional neural network , 2017, Multimedia Tools and Applications.

[11]  Donato Cascio,et al.  Automatic detection of lung nodules in CT datasets based on stable 3D mass-spring models , 2012, Comput. Biol. Medicine.

[12]  Aybars Ugur,et al.  Shape and Texture Based Novel Features for Automated Juxtapleural Nodule Detection in Lung CTs , 2015, Journal of Medical Systems.

[13]  Ilka Lopes Santoro,et al.  Estudo comparativo dos fatores prognósticos entre os pacientes com maior e menor sobrevida em portadores de carcinoma broncogênico , 2002 .

[14]  Xiangrong Zhou,et al.  Classification of teeth in cone-beam CT using deep convolutional neural network , 2017, Comput. Biol. Medicine.

[15]  David G. Stork,et al.  Pattern Classification , 1973 .

[16]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[17]  Wen-Huang Cheng,et al.  Computer-aided classification of lung nodules on computed tomography images via deep learning technique , 2015, OncoTargets and therapy.

[18]  D. Berry,et al.  Benefits and harms of CT screening for lung cancer: a systematic review. , 2012, JAMA.

[19]  Geoffrey E. Hinton,et al.  Deep Learning , 2015, Nature.

[20]  Russell C. Eberhart,et al.  A new optimizer using particle swarm theory , 1995, MHS'95. Proceedings of the Sixth International Symposium on Micro Machine and Human Science.

[21]  D. Shen,et al.  Computer-Aided Diagnosis with Deep Learning Architecture: Applications to Breast Lesions in US Images and Pulmonary Nodules in CT Scans , 2016, Scientific Reports.

[22]  Zhihua Cui,et al.  Individual Parameter Selection Strategy for Particle Swarm Optimization , 2009 .

[23]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[24]  M. Zweig,et al.  Receiver-operating characteristic (ROC) plots: a fundamental evaluation tool in clinical medicine. , 1993, Clinical chemistry.

[25]  Trevor Darrell,et al.  Fully Convolutional Networks for Semantic Segmentation , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[26]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[27]  R. Eberhart,et al.  Comparing inertia weights and constriction factors in particle swarm optimization , 2000, Proceedings of the 2000 Congress on Evolutionary Computation. CEC00 (Cat. No.00TH8512).

[28]  Qi Dou.  Multi-level Contextual 3D CNNs for False Positive Reduction in Pulmonary Nodule Detection , 2017 .

[29]  Kenji Suzuki,et al.  A new method for false-positive reduction in detection of lung nodules in CT images , 2014, 2014 19th International Conference on Digital Signal Processing.

[30]  A. Jemal,et al.  Cancer statistics, 2012 , 2012, CA: a cancer journal for clinicians.

[31]  Chris Rowen,et al.  Using Convolutional Neural Networks for Image Recognition By , 2015 .

[32]  Abbas Z. Kouzani,et al.  Random forest based lung nodule classification aided by clustering , 2010, Comput. Medical Imaging Graph..

[33]  D. Midthun,et al.  An official American Thoracic Society/American College of Chest Physicians policy statement: implementation of low-dose computed tomography lung cancer screening programs in clinical practice. , 2015, American journal of respiratory and critical care medicine.

[34]  J. Klein,et al.  The solitary pulmonary nodule. , 2002, Radiologic clinics of North America.

[35]  Yuhui Shi,et al.  Particle swarm optimization: developments, applications and resources , 2001, Proceedings of the 2001 Congress on Evolutionary Computation (IEEE Cat. No.01TH8546).

[36]  Bram van Ginneken,et al.  Pulmonary Nodule Detection in CT Images: False Positive Reduction Using Multi-View Convolutional Networks , 2016, IEEE Transactions on Medical Imaging.

[37]  Robert M. Haralick,et al.  Textural Features for Image Classification , 1973, IEEE Trans. Syst. Man Cybern..

[38]  Jörg Denzinger,et al.  Lung nodule detection in CT images using deep convolutional neural networks , 2016, 2016 International Joint Conference on Neural Networks (IJCNN).

[39]  Hao Chen,et al.  Multilevel Contextual 3-D CNNs for False Positive Reduction in Pulmonary Nodule Detection , 2017, IEEE Transactions on Biomedical Engineering.

[40]  Ning Wu,et al.  Lung Nodule Detection on Chest CT: Evaluation of a Computer-Aided Detection (CAD) System , 2005, Korean journal of radiology.

[41]  Robin N. Strickland.  Tumor detection in nonstationary backgrounds , 1994, IEEE Trans. Medical Imaging.

[42]  Yann LeCun,et al.  Convolutional networks and applications in vision , 2010, Proceedings of 2010 IEEE International Symposium on Circuits and Systems.

[43]  Hiroki Nakano,et al.  Cascaded Neural Networks with Selective Classifiers and its evaluation using Lung X-ray CT Images , 2016, ArXiv.

[44]  Miguel Ángel Guevara-López,et al.  Representation learning for mammography mass lesion classification with convolutional neural networks , 2016, Comput. Methods Programs Biomed..

[45]  Hiroshi Fujita,et al.  Automated detection of pulmonary nodules in PET/CT images: Ensemble false-positive reduction using a convolutional neural network technique. , 2016, Medical physics.

[46]  Jamshid Dehmeshki,et al.  Shape-Based Computer-Aided Detection of Lung Nodules in Thoracic CT Images , 2009, IEEE Transactions on Biomedical Engineering.

[47]  Sabina Sonia Tangaro,et al.  Automatic Lung Segmentation in CT Images with Accurate Handling of the Hilar Region , 2011, Journal of Digital Imaging.

[48]  Tae-Sun Choi,et al.  Automated pulmonary nodule detection based on three-dimensional shape-based feature descriptor , 2014, Comput. Methods Programs Biomed..

[49]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[50]  Kenji Suzuki,et al.  Maximal partial AUC feature selection in computer-aided detection of hepatocellular carcinoma in contrast-enhanced hepatic CT , 2012, Medical Imaging.

[51]  Wei Li,et al.  A multi-kernel based framework for heterogeneous feature selection and over-sampling for computer-aided detection of pulmonary nodules , 2017, Pattern Recognit..

[52]  Anselmo Cardoso de Paiva,et al.  Methodology for automatic detection of lung nodules in computerized tomography images , 2010, Comput. Methods Programs Biomed..

[53]  Yoshua Bengio,et al.  Convolutional networks for images, speech, and time series , 1998 .

[54]  G. L. F. D. Silva,et al.  Taxonomic indexes for differentiating malignancy of lung nodules on CT images , 2016 .

[55]  K. Bae,et al.  Automated detection of pulmonary nodules on CT images: effect of section thickness and reconstruction interval--initial results. , 2005, Radiology.

[56]  Anselmo Cardoso de Paiva,et al.  Automatic detection of small lung nodules in 3D CT data using Gaussian mixture models, Tsallis entropy and SVM , 2014, Eng. Appl. Artif. Intell..

[57]  Anselmo Cardoso de Paiva,et al.  Lung-Nodule Classification Based on Computed Tomography Using Taxonomic Diversity Indexes and an SVM , 2017, J. Signal Process. Syst..

[58]  Fei-Fei Li,et al.  Large-Scale Video Classification with Convolutional Neural Networks , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[59]  Noboru Yanai,et al.  Prognosis of non-surgically treated, clinical stage I lung cancer patients in Japan. , 2002, Lung cancer.

[60]  Guozhen Zhang,et al.  Feature Selection and Performance Evaluation of Support Vector Machine (SVM)-Based Classifier for Differentiating Benign and Malignant Pulmonary Nodules by Computed Tomography , 2010, Journal of Digital Imaging.

[61]  Anselmo Cardoso de Paiva,et al.  3D shape analysis to reduce false positives for lung nodule detection systems , 2017, Medical & Biological Engineering & Computing.

[62]  Dirk P. Kroese,et al.  The Cross-Entropy Method: A Unified Approach to Combinatorial Optimization, Monte-Carlo Simulation and Machine Learning , 2004 .

[63]  S. Armato,et al.  Massive training artificial neural network (MTANN) for reduction of false positives in computerized detection of lung nodules in low-dose computed tomography. , 2003, Medical physics.

[64]  Zhengrong Liang,et al.  Fast and Adaptive Detection of Pulmonary Nodules in Thoracic CT Images Using a Hierarchical Vector Quantization Scheme , 2015, IEEE Journal of Biomedical and Health Informatics.