An improved CNN-based architecture for automatic lung nodule classification

Med Biol Eng Comput. 2022 Jul;60(7):1977-1986. doi: 10.1007/s11517-022-02578-0. Epub 2022 May 7.

Abstract

Lung cancer is one of the most critical diseases due to its significant death rate compared to all other types of cancer. The early diagnosis of lung cancer that improves the patient's chance of surviving is mostly done in two phases: screening through CT scan imaging modality and, more importantly the medical expert's reading of the scan, which is a time-consuming task and is vulnerable to errors. It is difficult to differentiate between malignant and benign nodules and biopsies are highly invasive, and patients with benign nodules may undergo unnecessary procedures. In this study, we propose a CNN-based computer-aided diagnosis system to automatically classify pulmonary nodules into benign or malignant. The proposed network architecture is based on AlexNet architecture that experiments with several types of layer ordering, hyperparameters, and functions for the various sides of the network. To build a well-trained model, several pre-processing steps are applied to the entire dataset, for instance segmentation, normalization, and zero centering. Finally, the proposed system obtained results with 98.7% accuracy, 98.6% sensitivity, and 98.9% specificity. The proposed model achieved superior performance compared to the AlexNet. The modifications in the original AlexNet is done to get a reasonable structure that has high nodule analysis sensitivity.

Keywords: Computer-aided diagnosis; Convolutional neural network; Deep learning; Lung nodule classification.

MeSH terms

  • Diagnosis, Computer-Assisted / methods
  • Humans
  • Lung / diagnostic imaging
  • Lung / pathology
  • Lung Neoplasms* / diagnosis
  • Neural Networks, Computer*
  • Tomography, X-Ray Computed / methods