Lung cancer risk prediction using interpretable ensemble models on lifestyle and clinical data
- 1Department of Health Information Management and Technology, College of Applied Medical Sciences, King Faisal University, Al-Ahsa, Saudi Arabia
- 2School of Computer Applications and Technology, Galgotias University, Greater Noida, Uttar Pradesh, India
- 3Center for Precision Health, McWilliams School of Biomedical Informatics, The University of Texas Health Science Center at Houston, Houston, Texas, United States of America
Abstract
Lung cancer remains one of the leading causes of cancer-related mortality worldwide, where early detection and reliable risk assessment are critical for improving outcomes. This study develops and evaluates a range of ensemble learning models for lung cancer prediction using lifestyle and clinical indicators, with an emphasis on both predictive performance and interpretability. Five base learners—logistic regression, k-nearest neighbors, naïve Bayes, support vector machine, and linear discriminant analysis—were used to construct multiple boosting and bagging models. Building on these, voting and stacking ensembles were designed by selectively combining high-performing models. All approaches were evaluated on the original dataset as well as on balanced and upsampled variants derived through synthetic augmentation. Model performance was assessed using accuracy, precision, recall, F1-score, Matthews correlation coefficient (MCC), and AUC-ROC. The results show that ensemble approaches consistently outperform individual models, with voting and stacking demonstrating superior performance over boosting and bagging methods. The stacking model achieved the strongest overall performance across all evaluated models. On the original dataset, which provides a more realistic representation of practical deployment conditions, it attained an accuracy of 93.53%. Performance further improved on the balanced and upsampled datasets on the upsampled dataset. To enhance transparency, SHAP and LIME were employed to provide global and local interpretability, respectively, enabling identification of key clinical factors and patient-specific risk drivers. The analysis highlights both alignment with known clinical patterns and dataset-driven variations, supporting informed interpretation of model outputs. The results suggest that stacking-based ensembles can improve risk prediction from lifestyle and clinical indicators while maintaining model transparency through SHAP and LIME explanations. These findings highlight the potential of interpretable ensemble learning as a decision-support tool for lung cancer risk assessment.
Related
- Uncertainty Quantification-Incorporated Ensemble Model for Personalized Prediction of Severe Radiotherapy-Induced Immunosuppression Among Esophageal Cancer Patients
- Multimodal lifestyle intervention during neoadjuvant chemotherapy in patients with early breast cancer as a strategy to improve treatment response: Protocol of the DIANA randomized clinical trial
- An ensemble framework for robust drug response prediction and large-scale sensitivity profiling across cancers
- An interpretable open platform for sequence-based antibody developability prediction
- SIMORGH: Ensemble-Aware Geometric Deep Learning for Apo-State and Cryptic Ligand Binding Site Prediction
BibTeX
@article{ganie2026lung,
title = {{Lung cancer risk prediction using interpretable ensemble models on lifestyle and clinical data}},
author = {Shahid Mohammad Ganie and Pijush Kanti Dutta Pramanik and Zhongming Zhao},
journal = {PLOS One},
year = {2026},
month = sep,
doi = {10.1371/journal.pone.0357291},
url = {https://doi.org/10.1371/journal.pone.0357291},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Lung cancer risk prediction using interpretable ensemble models on lifestyle and clinical data
AU - Shahid Mohammad Ganie
AU - Pijush Kanti Dutta Pramanik
AU - Zhongming Zhao
PY - 2026
DA - 2026/09/03
JO - PLOS One
DO - 10.1371/journal.pone.0357291
UR - https://doi.org/10.1371/journal.pone.0357291
AB - Lung cancer remains one of the leading causes of cancer-related mortality worldwide, where early detection and reliable risk assessment are critical for improving outcomes. This study develops and evaluates a range of ensemble learning models for lung cancer prediction using lifestyle and clinical indicators, with an emphasis on both predictive performance and interpretability. Five base learners—logistic regression, k-nearest neighbors, naïve Bayes, support vector machine, and linear discriminant analysis—were used to construct multiple boosting and bagging models. Building on these, voting and stacking ensembles were designed by selectively combining high-performing models. All approaches were evaluated on the original dataset as well as on balanced and upsampled variants derived through synthetic augmentation. Model performance was assessed using accuracy, precision, recall, F1-score, Matthews correlation coefficient (MCC), and AUC-ROC. The results show that ensemble approaches consistently outperform individual models, with voting and stacking demonstrating superior performance over boosting and bagging methods. The stacking model achieved the strongest overall performance across all evaluated models. On the original dataset, which provides a more realistic representation of practical deployment conditions, it attained an accuracy of 93.53%. Performance further improved on the balanced and upsampled datasets on the upsampled dataset. To enhance transparency, SHAP and LIME were employed to provide global and local interpretability, respectively, enabling identification of key clinical factors and patient-specific risk drivers. The analysis highlights both alignment with known clinical patterns and dataset-driven variations, supporting informed interpretation of model outputs. The results suggest that stacking-based ensembles can improve risk prediction from lifestyle and clinical indicators while maintaining model transparency through SHAP and LIME explanations. These findings highlight the potential of interpretable ensemble learning as a decision-support tool for lung cancer risk assessment.
SN - 1932-6203
ER -