A VGG16 CNN-based Method for Multiclass Lung Cancer Classification using CT Imaging
DOI:
https://doi.org/10.31961/eltikom.v9i2.1483Keywords:
Classification, CT imaging, lung cancer, modified VGG16-CNNAbstract
Lung cancer is the leading cause of death worldwide among all types of cancer. Early detection and accurate classification are essential to prevent disease progression and improve patient survival rates. One effective approach is the use of computer-aided diagnosis (CAD) systems based on medical imaging, particularly CT scans, which provide high-resolution and non-invasive visualization of lung structures, including blood vessels, soft tissues, and lesions or nodules. This study proposes a VGG16 CNN-based multiclass classification method for lung cancer. Unlike previous studies that primarily focus on binary classification, this research addresses four distinct classes of lung nodule CT images to better reflect complex clinical needs. The modified VGG16 architecture incorporates additional layers, including Flatten, Dense, and Dropout, along with the Softmax activation function, effectively improving classification performance and reducing overfitting risk. An ablation experiment was also conducted by replacing ReLU with LeakyReLU to address the potential “dying ReLU” issue. However, the results indicated that LeakyReLU did not provide significant improvement over the standard ReLU. The proposed model achieved an accuracy of 90.72%, precision of 91.5%, sensitivity of 89.25%, specificity of 96.76%, F1-score of 90%, and a low loss value of 0.37. Furthermore, the modified VGG16-CNN outperformed other CNN architectures, including ResNet50, EfficientNetB1, MobileNetV2, and AlexNet, in multiclass lung cancer image classification. The results demonstrate that the proposed method is effective for diagnosing lung nodules from CT scans and has the potential to support medical professionals in making accurate and timely diagnoses.
Downloads
References
[1] H. Jiang, H. Ma, W. Qian, M. Gao, and Y. Li, “An Automatic Detection System of Lung Nodule Based on Multigroup Patch-Based Deep Learning Network,” IEEE J. Biomed. Heal. Informatics, vol. 22, no. 4, pp. 1227–1237, 2018, doi: 10.1109/JBHI.2017.2725903.
[2] R. L. Siegel, K. D. Miller, H. E. Fuchs, and A. Jemal, “Cancer statistics, 2022,” CA. Cancer J. Clin., vol. 72, no. 1, pp. 7–33, 2022, doi: 10.3322/caac.21708.
[3] S. Kido, Y. Hirano, and N. Hashimoto, “Detection and classification of lung abnormalities by use of convolutional neural network (CNN) and regions with CNN features (R-CNN),” 2018 Int. Work. Adv. Image Technol. IWAIT 2018, pp. 1–4, 2018, doi: 10.1109/IWAIT.2018.8369798.
[4] S. Sari, T. Sutikno, I. Soesanti, and N. A. Setiawan, “A review of convolutional neural network-based computer-aided lung nodule detection system,” IAES Int. J. Artif. Intell., vol. 12, no. 3, pp. 1044–1061, 2023, doi: 10.11591/ijai.v12.i3.pp1044-1061.
[5] Y. Chen, X. Duan, S. Chatani, W. Du, J. Wang, and Q. Hu, “An Improved Region-based Fully Convolutional Network for Automatic Pulmonary Nodules Detection,” pp. 256–261, 2019.
[6] A. Agarwal, K. Patni, and D. Rajeswari, “Lung Cancer Detection and Classification Based on Alexnet CNN,” Proc. 6th Int. Conf. Commun. Electron. Syst. ICCES 2021, pp. 1390–1397, 2021, doi: 10.1109/ICCES51350.2021.9489033.
[7] F. William, A. Serener, and S. Serte, “Deep learning classification of Covid-19, pneumonia, and lung cancer on chest radiographs,” ISMSIT 2021 - 5th Int. Symp. Multidiscip. Stud. Innov. Technol. Proc., pp. 125–129, 2021, doi: 10.1109/ISMSIT52890.2021.9604629.
[8] N. Jahan, M. S. Anower, and R. Hassan, “Automated Diagnosis of Pneumonia from Classification of Chest X-Ray im ages using EfficientNet,” 2021 Int. Conf. Inf. Commun. Technol. Sustain. Dev. ICICT4SD 2021 - Proc., pp. 235–239, 2021, doi: 10.1109/ICICT4SD50815.2021.9397055.
[9] M. Yaqoob, H. Qayoom, and F. Hassan, “Covid-19 Detection Based on the Fine-Tuned MobileNetv2 Through Lung X-rays,” 2021 4th Int. Symp. Adv. Electr. Commun. Technol., pp. 01–06, 2022, doi: 10.1109/isaect53699.2021.9668425.
[10] N. Y. Wu et al., “Magnetic resonance imaging for lung cancer detection: Experience in a population of more than 10,000 healthy individuals,” BMC Cancer, vol. 11, 2011, doi: 10.1186/1471-2407-11-242.
[11] R. Zhang, C. Cheng, X. Zhao, and X. Li, “Multiscale Mask R-CNN–Based Lung Tumor Detection Using PET Imaging,” Mol. Imaging, vol. 18, pp. 1–8, 2019, doi: 10.1177/1536012119863531.
[12] T. Messay, R. C. Hardie, and T. R. Tuinstra, “Segmentation of pulmonary nodules in computed tomography using a regression neural network approach and its application to the Lung Image Database Consortium and Image Database Resource Initiative dataset,” Med. Image Anal., vol. 22, no. 1, pp. 48–62, 2015, doi: 10.1016/j.media.2015.02.002.
[13] H. Shi, Z. Peng, and H. Wan, “Pulmonary Nodules Detection Based on CNN Multi-scale Feature Fusion,” 2019 IEEE 11th Int. Conf. Adv. Infocomm Technol. ICAIT 2019, pp. 86–90, 2019, doi: 10.1109/ICAIT.2019.8935936.
[14] S. Pang et al., “VGG16-T: A novel deep convolutional neural network with boosting to identify pathological type of lung cancer in early stage by ct images,” Int. J. Comput. Intell. Syst., vol. 13, no. 1, pp. 771–780, 2020, doi: 10.2991/ijcis.d.200608.001.
[15] G. K. Abraham, P. Bhaskaran, and V. S. Jayanthi, “Lung nodule classification in CT images using convolutional neural network,” Proc. 2019 9th Int. Conf. Adv. Comput. Commun. ICACC 2019, pp. 199–203, 2019, doi: 10.1109/ICACC48162.2019.8986213.
[16] N. Akter, M. T. Reza, and M. A. Alam, “A Comparison Based Analysis on the Performance of Deep Neural Network Models in Terms of Classifying Pneumonia from Chest X-ray Images,” 2020 IEEE Asia-Pacific Conf. Comput. Sci. Data Eng. CSDE 2020, 2020, doi: 10.1109/CSDE50874.2020.9411560.
[17] A. Y. Saleh, C. K. Chin, V. Penshie, and H. R. H. Al-Absi, “Lung cancer medical images classification using hybrid cnn-svm,” Int. J. Adv. Intell. Informatics, vol. 7, no. 2, pp. 151–162, 2021, doi: 10.26555/ijain.v7i2.317.
[18] N. Srivastava, G. Hinton, A. Krizhevsky, I. Sutskever, and R. Salakhutdinov, “Dropout : A Simple Way to Prevent Neural Networks from Overfitting,” J. Mach. Learn. Res., vol. 15, pp. 1929–1958, 2014.
[19] S. Hossain, N. Basak, A. Mollah, Nahiduzzaman, M. Ahsan, and J. Haider, “Ensemble-based multiclass lung cancer classification using hybrid CNN-SVD feature extraction and selection method,” pp. 1–30, 2025, doi: 10.1371/journal.pone.0318219.
[20] M. Hany, “Chest CT-Scan images Dataset.” Accessed: Mar. 24, 2024. [Online]. Available: https://www.kaggle.com/mohamedhanyyy/chest-ctscan-images
[21] N. D. Miranda, L. Novamizanti, and S. Rizal, “Convolutional Neural Network Pada Klasifikasi Sidik Jari Menggunakan Resnet-50,” J. Tek. Inform., vol. 1, no. 2, pp. 61–68, 2020.
[22] N. Yudistira, A. W. Widodo, and B. Rahayudi, “Deteksi Covid-19 Pada Citra Sinar-X Dada Menggunakan Deep Covid-19 Detection on X-Ray Images Using Efficient Deep Learning,” J. Teknol. Inf. dan Ilmu Komput., vol. 7, no. 6, pp. 1289–1296, 2020, doi: 10.25126/jtiik.202073651.
[23] S. Dodia, A. B., and P. A. Mahesh, “Recent advancements in deep learning based lung cancer detection: A systematic review,” Eng. Appl. Artif. Intell., vol. 116, 2022, doi: https://doi.org/10.1016/j.engappai.2022.105490.
[24] M. M. Ansari et al., “Evaluating CNN Architectures and Hyperparameter Tuning for Enhanced Lung Cancer Detection Using Transfer Learning,” J. Electr. Comput. Eng., vol. 2024, 2024, doi: 10.1155/2024/3790617.
[25] Y. N. Dauphin, H. De Vries, J. Chung, and Y. Bengio, “RMSProp and equilibrated adaptive learning rates for non-convex optimization,” 2014.
[26] S. Sasikala, M. Bharathi, and B. R. Sowmiya, “Lung Cancer Detection and Classification Using Deep CNN,” Int. J. Innov. Technol. Explor. Eng., vol. 8, no. 2, pp. 259–262, 2018.
[27] P. N. Andono, T. Sutojo, and Muljono, “Pengolahan Citra Digital.” [Online]. Available: https://books.google.co.id/books?id=zUJRDwAAQBAJ&printsec=frontcover&hl=id&source=gbs_ge_summary_r&cad=0#v=onepage&q&f=false
[28] C. C. Chiet, K. W. How, P. Y. Han, and Y. H. Yen, “A Lung Cancer Detection With Pre - Trained CNN Models,” J. Informatics Web Eng., vol. 3, no. 1, 2024.
Downloads
Published
Issue
Section
License
Copyright (c) 2025 Sekar Sari, Muna Afdi Muniroh, Kevin Ilham Apriandy
This work is licensed under a Creative Commons Attribution 4.0 International License.
All accepted papers will be published under a Creative Commons Attribution 4.0 International (CC BY 4.0) License. Authors retain copyright and grant the journal right of first publication. CC-BY Licenced means lets others to Share (copy and redistribute the material in any medium or format) and Adapt (remix, transform, and build upon the material for any purpose, even commercially).
