Author
Listed:
- Shahid Mohammad Ganie
- Pijush Kanti Dutta Pramanik
- Zhongming Zhao
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.
Suggested Citation
Shahid Mohammad Ganie & Pijush Kanti Dutta Pramanik & Zhongming Zhao, 2026.
"Lung cancer risk prediction using interpretable ensemble models on lifestyle and clinical data,"
PLOS ONE, Public Library of Science, vol. 21(9), pages 1-45, September.
Handle:
RePEc:plo:pone00:0357291
DOI: 10.1371/journal.pone.0357291
Download full text from publisher
Corrections
All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:plo:pone00:0357291. See general information about how to correct material in RePEc.
If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.
We have no bibliographic references for this item. You can help adding them by using this form .
If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.
For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: plosone (email available below). General contact details of provider: https://journals.plos.org/plosone/ .
Please note that corrections may take a couple of weeks to filter through
the various RePEc services.