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

Journal: Journal of healthcare engineering
Published Date:

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%.

Authors

  • Giang Son Tran
    ICTLab, University of Science and Technology of Hanoi, Vietnam Academy of Science and Technology, 18 Hoang Quoc Viet, Cau Giay, Hanoi, Vietnam.
  • Thi Phuong Nghiem
    ICTLab, University of Science and Technology of Hanoi, Vietnam Academy of Science and Technology, 18 Hoang Quoc Viet, Cau Giay, Hanoi, Vietnam.
  • Van Thi Nguyen
    Department of Radiology, Vietnam National Cancer Hospital, Hanoi, Vietnam.
  • Chi Mai Luong
    Institute of Information Technology, Vietnam Academy of Science and Technology, 18 Hoang Quoc Viet, Cau Giay, Hanoi, Vietnam.
  • Jean-Christophe Burie
    L3i Laboratory, University of La Rochelle, La Rochelle, France.