Improving Accuracy of Lung Nodule Classification Using Deep Learning with Focal Loss

J Healthc Eng. 2019 Feb 4:2019:5156416. doi: 10.1155/2019/5156416. eCollection 2019.

Abstract

Early detection and classification of pulmonary nodules using computer-aided diagnosis (CAD) systems is useful in reducing mortality rates of lung cancer. In this paper, we propose a new deep learning method to improve classification accuracy of pulmonary nodules in computed tomography (CT) scans. Our method uses a novel 15-layer 2D deep convolutional neural network architecture for automatic feature extraction and classification of pulmonary candidates as nodule or nonnodule. Focal loss function is then applied to the training process to boost classification accuracy of the model. We evaluated our method on the LIDC/IDRI dataset extracted by the LUNA16 challenge. The experiments showed that our deep learning method with focal loss is a high-quality classifier with an accuracy of 97.2%, sensitivity of 96.0%, and specificity of 97.3%.

Publication types

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

MeSH terms

  • Databases, Factual
  • Deep Learning* / statistics & numerical data
  • Early Detection of Cancer / methods
  • Early Detection of Cancer / statistics & numerical data
  • Humans
  • Lung Neoplasms / classification*
  • Lung Neoplasms / diagnostic imaging*
  • Neural Networks, Computer
  • Radiographic Image Interpretation, Computer-Assisted / methods*
  • Radiographic Image Interpretation, Computer-Assisted / statistics & numerical data
  • Solitary Pulmonary Nodule / classification*
  • Solitary Pulmonary Nodule / diagnostic imaging*
  • Tomography, X-Ray Computed / statistics & numerical data