論文

査読有り 国際誌
2018年

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

PloS one
  • Mizuho Nishio
  • ,
  • Mitsuo Nishizawa
  • ,
  • Osamu Sugiyama
  • ,
  • Ryosuke Kojima
  • ,
  • Masahiro Yakami
  • ,
  • Tomohiro Kuroda
  • ,
  • Kaori Togashi

13
4
開始ページ
e0195875
終了ページ
記述言語
英語
掲載種別
研究論文(学術雑誌)
DOI
10.1371/journal.pone.0195875

We aimed to evaluate a computer-aided diagnosis (CADx) system for lung nodule classification focussing on (i) usefulness of the conventional CADx system (hand-crafted imaging feature + machine learning algorithm), (ii) comparison between support vector machine (SVM) and gradient tree boosting (XGBoost) as machine learning algorithms, and (iii) effectiveness of parameter optimization using Bayesian optimization and random search. Data on 99 lung nodules (62 lung cancers and 37 benign lung nodules) were included from public databases of CT images. A variant of the local binary pattern was used for calculating a feature vector. SVM or XGBoost was trained using the feature vector and its corresponding label. Tree Parzen Estimator (TPE) was used as Bayesian optimization for parameters of SVM and XGBoost. Random search was done for comparison with TPE. 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 was 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. Bayesian optimization of SVM and XGBoost parameters was more efficient than random search. Based on observer study, AUC values of two board-certified radiologists were 0.898 and 0.822. The results show that diagnostic accuracy of our CADx system was comparable to that of radiologists with respect to classifying lung nodules.

リンク情報
DOI
https://doi.org/10.1371/journal.pone.0195875
PubMed
https://www.ncbi.nlm.nih.gov/pubmed/29672639
PubMed Central
https://www.ncbi.nlm.nih.gov/pmc/articles/PMC5908232
ID情報
  • DOI : 10.1371/journal.pone.0195875
  • PubMed ID : 29672639
  • PubMed Central 記事ID : PMC5908232

エクスポート
BibTeX RIS