Development of MobileNetV2 for CT-Scan Lung Classification Using Transfer Learning
DOI:
https://doi.org/10.30871/jaic.v9i5.10282Keywords:
Lung Cancer, CT Scan, MobileNetV2, Transfer Learning, Deep LearningAbstract
Lung cancer is one of the leading causes of cancer-related deaths worldwide, making early detection crucial for improving patient survival rates. This study proposes an automated classification approach based on deep learning using the MobileNetV2 architecture to identify three categories of lung CT scan images: normal, benign, and malignant. The dataset used is the augmented IQ-OTH/NCCD Lung Cancer Dataset, consisting of 3,609 images with a resolution of 224×224 pixels. All images underwent preprocessing steps including RGB conversion, pixel rescaling, and normalization. The MobileNetV2 model was modified by adding a GlobalAveragePooling2D layer, a dense layer, and dropout to reduce overfitting risk. Training was conducted for 28 epochs using the optimizer Adam, followed by evaluation using accuracy, precision, recall, and F1-score metrics. The model was tested on unseen data and validated using Stratified 5-Fold Cross Validation. The testing results showed an overall accuracy of 97%, with a perfect recall score (1.00) for the malignant class. The cross-validation yielded an average accuracy of 97.26% with a standard deviation of ±0.66%, indicating consistent model performance. Given its lightweight architecture and high accuracy, MobileNetV2 has the potential to be implemented as a decision support system in medical image analysis.
Downloads
References
[1] H. Sung et al., “Global Cancer Statistics 2020: GLOBOCAN Estimates of Incidence and Mortality Worldwide for 36 Cancers in 185 Countries,” CA Cancer J Clin, vol. 71, no. 3, pp. 209–249, May 2021, doi: 10.3322/caac.21660.
[2] J. Ferlay et al., “Cancer statistics for the year 2020: An overview,” Int J Cancer, vol. 149, no. 4, pp. 778–789, 2021, doi: 10.1002/ijc.33588.
[3] E. Edelman Saul et al., “The challenges of implementing low-dose computed tomography for lung cancer screening in low- and middle-income countries,” Nat Cancer, vol. 1, no. 12, pp. 1140–1152, Nov. 2020, doi: 10.1038/s43018-020-00142-z.
[4] M. A. Thanoon, M. A. Zulkifley, M. A. A. Mohd Zainuri, and S. R. Abdani, “A Review of Deep Learning Techniques for Lung Cancer Screening and Diagnosis Based on CT Images,” Diagnostics, vol. 13, no. 16, p. 2617, Aug. 2023, doi: 10.3390/diagnostics13162617.
[5] M. Q. Shatnawi, Q. Abuein, and R. Al-Quraan, “Deep learning-based approach to diagnose lung cancer using CT-scan images,” Intell Based Med, vol. 11, p. 100188, Dec. 2025, doi: 10.1016/j.ibmed.2024.100188.
[6] W. Wulaningsih, C. Villamaria, A. Akram, J. Benemile, F. Croce, and J. Watkins, “Deep Learning Models for Predicting Malignancy Risk in CT-Detected Pulmonary Nodules: A Systematic Review and Meta-analysis,” Lung, vol. 202, no. 5, pp. 625–636, 2024, doi: 10.1007/s00408-024-00706-1.
[7] H. Jiang, S. Tang, W. Liu, and Y. Zhang, “Deep learning for COVID-19 chest CT (computed tomography) image analysis: A lesson from lung cancer,” Comput Struct Biotechnol J, vol. 19, pp. 1391–1399, 2021, doi: 10.1016/j.csbj.2021.02.016.
[8] O. Ukwandu, H. Hindy, and E. Ukwandu, “An evaluation of lightweight deep learning techniques in medical imaging for high precision COVID-19 diagnostics,” Healthcare Analytics, vol. 2, p. 100096, Nov. 2022, doi: 10.1016/j.health.2022.100096.
[9] J. Zheng et al., “Pulmonary nodule risk classification in adenocarcinoma from CT images using deep CNN with scale transfer module,” IET Image Process, vol. 14, no. 8, pp. 1481–1489, Jun. 2020, doi: 10.1049/iet-ipr.2019.0248.
[10] R. Klangbunrueang, P. Pookduang, W. Chansanam, and T. Lunrasri, “AI-Powered Lung Cancer Detection: Assessing VGG16 and CNN Architectures for CT Scan Image Classification,” Informatics, vol. 12, no. 1, p. 18, Feb. 2025, doi: 10.3390/informatics12010018.
[11] M. Kalkan, M. S. Guzel, F. Ekinci, E. Akcapinar Sezer, and T. Asuroglu, “Comparative Analysis of Deep Learning Methods on CT Images for Lung Cancer Specification,” Cancers (Basel), vol. 16, no. 19, p. 3321, Sep. 2024, doi: 10.3390/cancers16193321.
[12] B. D. S. Isnayni Sugma Rachmatika, Lailil Muflikhah, “Deteksi Mutasi Pada Kanker Paru Melalui Citra CT-Scan Penerapan Model Algoritma Convolutional Neural Networks (CNN)dan Optimizer Adam,” Jurnal Pengembangan Teknologi Informasi Dan Ilmu Komputer, vol. Vol 8 No 9, 2024, [Online]. Available: https://j-ptiik.ub.ac.id/index.php/j-ptiik/article/view/14134/6317
[13] Z. Gao, Y. Tian, S.-C. Lin, and J. Lin, “A CT Image Classification Network Framework for Lung Tumors Based on Pre-trained MobileNetV2 Model and Transfer learning, And Its Application and Market Analysis in the Medical field,” Jan. 2025, [Online]. Available: http://arxiv.org/abs/2501.04996
[14] A. Saha, S. M. Ganie, P. K. D. Pramanik, R. K. Yadav, S. Mallik, and Z. Zhao, “VER-Net: a hybrid transfer learning model for lung cancer detection using CT scan images,” BMC Med Imaging, vol. 24, no. 1, p. 120, May 2024, doi: 10.1186/s12880-024-01238-z.
[15] S. Velu, “An efficient, lightweight MobileNetV2-based fine-tuned model for COVID-19 detection using chest X-ray images,” Mathematical Biosciences and Engineering, vol. 20, no. 5, pp. 8400–8427, 2023, doi: 10.3934/mbe.2023368.
[16] A. Gudur, H. Sivaraman, and V. Vimal, “International Journal of Intelligent Systems And Applications In Engineering Deep Learning-Based Detection of Lung Nodules in CT Scans for Cancer Screening,” Original Research Paper International Journal of Intelligent Systems and Applications in Engineering IJISAE, vol. 2023, no. 7s, pp. 20–28, 2023, [Online]. Available: www.ijisae.org
[17] M. Aftab et al., “AI in Oncology: Transforming Cancer Detection through Machine Learning and Deep Learning Applications,” Jan. 2025, [Online]. Available: http://arxiv.org/abs/2501.15489
[18] M. Aparna and B. S. Rao, “A novel automated deep learning approach for Alzheimer’s disease classification,” IAES International Journal of Artificial Intelligence (IJ-AI), vol. 12, no. 1, p. 451, Mar. 2023, doi: 10.11591/ijai.v12.i1.pp451-458.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2025 Daffa Fadhil Rajendra, Ajie Kusuma Wardhana
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License (Attribution-ShareAlike 4.0 International (CC BY-SA 4.0) ) that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
