Enhancing Lung Cancer Type Prediction with a Novel Hybrid Approach: Transfer Learning, SVM, and Model Stacking

Adrian Lopez, Remigio Hurtado

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

Abstract

In this paper, we propose a novel hybrid model for the prediction of multiclass lung cancer, an ailment of paramount concern in the medical field. We leverage transfer learning applied to Convolutional Neural Networks (CNN), alongside Support Vector Machines (SVM), and Principal Component Analysis (PCA), combined via stacking to form a powerful meta-model. The model utilizes pre-trained CNN architectures to extract intricate features from medical imaging data, which then undergo dimensionality reduction via PCA, enhancing the computational efficiency while preserving vital information. Following feature extraction and reduction, we introduce an SVM, known for its exceptional classification prowess, to construct a preliminary predictive model. To maximize the performance, we employ a stacking methodology, treating the preliminary models as base classifiers and training a meta-learner on their predictions to predict the final class labels. Experimental results on real-world lung cancer datasets demonstrate that our model significantly outperforms traditional machine learning and deep learning models in terms of prediction accuracy, sensitivity, and specificity. This work contributes to enhancing the accuracy of multiclass lung cancer predictions, potentially saving lives by facilitating early and accurate diagnoses. Further research will focus on applying this model to other forms of cancer and diseases.

Original languageEnglish
Title of host publicationInformation Technology and Systems - ICITS 2024
EditorsAlvaro Rocha, Jorge Hochstetter Diez, Carlos Ferras, Mauricio Dieguez Rebolledo
PublisherSpringer Science and Business Media Deutschland GmbH
Pages288-297
Number of pages10
ISBN (Print)9783031542343
DOIs
StatePublished - 2024
EventInternational Conference on Information Technology and Systems, ICITS 2024 - Temuco, Chile
Duration: 24 Jan 202426 Jan 2024

Publication series

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

Conference

ConferenceInternational Conference on Information Technology and Systems, ICITS 2024
Country/TerritoryChile
CityTemuco
Period24/01/2426/01/24

Bibliographical note

Publisher Copyright:
© The Author(s), under exclusive license to Springer Nature Switzerland AG 2024.

Keywords

  • Cancer Detection
  • Classification
  • Convolutional Neural Network (CNN)
  • Lung PET Images
  • Machine Learning
  • Morphological Operations
  • Multi-class Classifier
  • Preprocessing
  • Support Vector Machine (SVM)

Fingerprint

Dive into the research topics of 'Enhancing Lung Cancer Type Prediction with a Novel Hybrid Approach: Transfer Learning, SVM, and Model Stacking'. Together they form a unique fingerprint.

Cite this