Computer-aided diagnosis of lung nodule using gradient tree boosting and Bayesian optimization

Abstract

We aimed to evaluate computer-aided diagnosis (CADx) system for lung nodule classification focusing on (i) usefulness of gradient tree boosting (XGBoost) and (ii) effectiveness of parameter optimization using Bayesian optimization (Tree Parzen Estimator, TPE) and random search. 99 lung nodules (62 lung cancers and 37 benign lung nodules) were included from public databases of CT images. A variant of local binary pattern was used for calculating feature vectors. Support vector machine (SVM) or XGBoost was trained using the feature vectors and their labels. TPE or random search was used for parameter optimization of SVM and XGBoost. Leave-one-out cross-validation was used for optimizing and evaluating the performance of our CADx system. Performance was evaluated using area under the curve (AUC) of receiver operating characteristic analysis. AUC was calculated 10 times, and its average was obtained. The best averaged AUC of SVM and XGBoost were 0.850 and 0.896, respectively; both were obtained using TPE. XGBoost was generally superior to SVM. Optimal parameters for achieving high AUC were obtained with fewer numbers of trials when using TPE, compared with random search. In conclusion, XGBoost was better than SVM for classifying lung nodules. TPE was more efficient than random search for parameter optimization.

29 pages, 4 figures

References in corpus (5)

• XGBoost: A Scalable Tree Boosting System
• Towards automatic pulmonary nodule management in lung cancer screening with deep learning
• Stopping criteria for boosting automatic experimental design using real-time fMRI with Bayesian optimization
• Towards tailoring non-invasive brain stimulation using real-time fMRI and Bayesian optimization
• Best Practices for Applying Deep Learning to Novel Applications

Cited by in corpus (6)

• Automatic segmentation of the Foveal Avascular Zone in ophthalmological OCT-A images
• Evolving the pulmonary nodules diagnosis from classical approaches to deep learning aided decision support: three decades development course and future prospect
• Automatic detection of acute ischemic stroke using non-contrast computed tomography and two-stage deep learning model
• Lung segmentation on chest x-ray images in patients with severe abnormal findings using deep learning
• Gradient Boosting Application in Forecasting of Performance Indicators Values for Measuring the Efficiency of Promotions in FMCG Retail
• Pocket Diagnosis: Secure Federated Learning against Poisoning Attack in the Cloud