Leveraging Transfer Learning as Feature Extractors for Lung Cancer Classification: Insights from VGG19 and ResNet152 Pipelines

Darren Soong Kai Xuan, Anwar P.P. Abdul Majeed*, Rabiu Muazu Musa, Yang Luo, Saad Aslam, Samuel Soma M. Ajibade, Mehran Behjati, Muhammed Basheer Jasser, Muhammad Amirul Abdullah

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

Lung cancer remains a significant global health challenge, with early detection being crucial for improving patient outcomes. This study explores the use of transfer learning in enhancing the diagnostic accuracy of lung cancer classification from CT images. Specifically, it compares the effectiveness of two transfer learning pipelines utilizing pre-trained models, VGG19 and ResNet152, as feature extractors combined with Logistic Regression (LR) for classification. The results demonstrate that the VGG16 + LR pipeline outperforms the ResNet152 pipeline, achieving superior classification accuracy for both validation and testing datasets at 97%, respectively. This finding underscores the potential of transfer learning in medical imaging, offering a promising approach to improving early detection and treatment strategies for lung cancer.

Original languageEnglish
Title of host publicationSelected Proceedings from the 2nd International Conference on Intelligent Manufacturing and Robotics, ICIMR 2024 - Advances in Intelligent Manufacturing and Robotics
EditorsWei Chen, Andrew Huey Ping Tan, Yang Luo, Long Huang, Yuyi Zhu, Anwar PP Abdul Majeed, Fan Zhang, Yuyao Yan, Chenguang Liu
PublisherSpringer Science and Business Media Deutschland GmbH
Pages750-756
Number of pages7
ISBN (Print)9789819639489
DOIs
Publication statusPublished - 2025
Event2nd International Conference on Intelligent Manufacturing and Robotics, ICIMR 2024 - Suzhou, China
Duration: 22 Aug 202423 Aug 2024

Publication series

NameLecture Notes in Networks and Systems
Volume1316 LNNS
ISSN (Print)2367-3370
ISSN (Electronic)2367-3389

Conference

Conference2nd International Conference on Intelligent Manufacturing and Robotics, ICIMR 2024
Country/TerritoryChina
CitySuzhou
Period22/08/2423/08/24

Keywords

  • Deep Learning
  • Feature Extractors
  • Lung Cancer
  • Machine Learning
  • Transfer Learning

Fingerprint

Dive into the research topics of 'Leveraging Transfer Learning as Feature Extractors for Lung Cancer Classification: Insights from VGG19 and ResNet152 Pipelines'. Together they form a unique fingerprint.

Cite this