Deep Learning for Lung Cancer Prediction: Performance & Explainability Analysis

Lord Coffie, Jongyeop Kim, Lei Chen

Research output: Contribution to book or proceedingConference articlepeer-review

Abstract

Globally, lung cancer continues to be among the common causes of cancer death; therefore, the importance of early patient survival increases in its detection. This study examines how deep learning can predict lung cancer using structured clinical and lifestyle survey data. A comparison was made between a multi-layer perceptron (MLP) and the random forest, XGBoost, and support vector machine (SVM) methods, as well as logistic regression machine learning models. The results demonstrate that ensemble-based models (Random Forest & XGBoost) outperformed the deep learning model, achieving an accuracy of 99.7 % compared to MLP's 94.7 %. We employed SHAP and LIME explainability techniques to improve model interpretability, identifying allergy, swallowing diffculty, and coughing as the most significant predictors of lung cancer. These results show how important it is to make AI-based medical predictions more interpretable in order to increase their use in clinical settings. While deep learning performed well, traditional machine learning models proved to be more effective for structured survey-based datasets. The study shows that machine learning could be useful for screening for lung cancer. It also stresses the need for more research on data augmentation biases, external validation on real-world clinical datasets, and advanced deep learning architectures to make the models more accurate.

Original languageEnglish
Title of host publication2025 6th International Conference on Artificial Intelligence, Robotics, and Control, AIRC 2025
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages354-360
Number of pages7
ISBN (Electronic)9798331543488
ISBN (Print)9798331543488
DOIs
StatePublished - May 7 2025
Event6th International Conference on Artificial Intelligence, Robotics, and Control, AIRC 2025 - Savannah, United States
Duration: May 7 2025May 9 2025

Publication series

Name2025 6th International Conference on Artificial Intelligence, Robotics and Control (AIRC)

Conference

Conference6th International Conference on Artificial Intelligence, Robotics, and Control, AIRC 2025
Country/TerritoryUnited States
CitySavannah
Period05/7/2505/9/25

Scopus Subject Areas

  • Mechanical Engineering
  • Control and Optimization
  • Artificial Intelligence
  • Computer Science Applications

Keywords

  • Deep Learning
  • Explainability
  • Lung Cancer Prediction
  • Machine Learning

Fingerprint

Dive into the research topics of 'Deep Learning for Lung Cancer Prediction: Performance & Explainability Analysis'. Together they form a unique fingerprint.

Cite this