Performance Analysis of State-of-the-Art CNN Architectures for LUNA16

Sensors (Basel). 2022 Jun 11;22(12):4426. doi: 10.3390/s22124426.

Abstract

The convolutional neural network (CNN) has become a powerful tool in machine learning (ML) that is used to solve complex problems such as image recognition, natural language processing, and video analysis. Notably, the idea of exploring convolutional neural network architecture has gained substantial attention as well as popularity. This study focuses on the intrinsic various CNN architectures: LeNet, AlexNet, VGG16, ResNet-50, and Inception-V1, which have been scrutinized and compared with each other for the detection of lung cancer using publicly available LUNA16 datasets. Furthermore, multiple performance optimizers: root mean square propagation (RMSProp), adaptive moment estimation (Adam), and stochastic gradient descent (SGD), were applied for this comparative study. The performances of the three CNN architectures were measured for accuracy, specificity, sensitivity, positive predictive value, false omission rate, negative predictive value, and F1 score. The experimental results showed that the CNN AlexNet architecture with the SGD optimizer achieved the highest validation accuracy for CT lung cancer with an accuracy of 97.42%, misclassification rate of 2.58%, 97.58% sensitivity, 97.25% specificity, 97.58% positive predictive value, 97.25% negative predictive value, false omission rate of 2.75%, and F1 score of 97.58%. AlexNet with the SGD optimizer was the best and outperformed compared to the other state-of-the-art CNN architectures.

Keywords: AlexNet; LUNA16; LeNet; artificial intelligence; big data; cancer research; deep learning; lung cancer; machine learning; medical image analysis.

MeSH terms

  • Humans
  • Lung Neoplasms* / diagnosis
  • Machine Learning
  • Neural Networks, Computer*
  • Tomography, X-Ray Computed

Grants and funding

This research received no external funding.