3D deep learning for detecting pulmonary nodules in CT scans

J Am Med Inform Assoc. 2018 Oct 1;25(10):1301-1310. doi: 10.1093/jamia/ocy098.

Abstract

Objective: To demonstrate and test the validity of a novel deep-learning-based system for the automated detection of pulmonary nodules.

Materials and methods: The proposed system uses 2 3D deep learning models, 1 for each of the essential tasks of computer-aided nodule detection: candidate generation and false positive reduction. A total of 888 scans from the LIDC-IDRI dataset were used for training and evaluation.

Results: Results for candidate generation on the test data indicated a detection rate of 94.77% with 30.39 false positives per scan, while the test results for false positive reduction exhibited a sensitivity of 94.21% with 1.789 false positives per scan. The overall system detection rate on the test data was 89.29% with 1.789 false positives per scan.

Discussion: An extensive and rigorous validation was conducted to assess the performance of the proposed system. The system demonstrated a novel combination of 3D deep neural network architectures and demonstrates the use of deep learning for both candidate generation and false positive reduction to be evaluated with a substantial test dataset. The results strongly support the ability of deep learning pulmonary nodule detection systems to generalize to unseen data. The source code and trained model weights have been made available.

Conclusion: A novel deep-neural-network-based pulmonary nodule detection system is demonstrated and validated. The results provide comparison of the proposed deep-learning-based system over other similar systems based on performance.

Publication types

  • Research Support, Non-U.S. Gov't
  • Validation Study

MeSH terms

  • Deep Learning*
  • Diagnosis, Computer-Assisted*
  • Humans
  • Lung / diagnostic imaging*
  • Multiple Pulmonary Nodules / diagnostic imaging*
  • ROC Curve
  • Radiographic Image Interpretation, Computer-Assisted
  • Sensitivity and Specificity
  • Solitary Pulmonary Nodule / diagnostic imaging*
  • Tomography, X-Ray Computed*