Treffer: Interpretable ensemble learning for tumor-type prediction with a SHAP-based evaluation of CatBoost and voting classifiers.

Title:
Interpretable ensemble learning for tumor-type prediction with a SHAP-based evaluation of CatBoost and voting classifiers.
Authors:
Wolak W; Department of Computer Science, Faculty of Computer Science and Mathematics, Cracow University of Technology, Cracow, Poland., Plichta A; Department of Computer Science, Faculty of Computer Science and Mathematics, Cracow University of Technology, Cracow, Poland., Orlicki H; Department of Computer Science, Faculty of Computer Science and Mathematics, Cracow University of Technology, Cracow, Poland. hubert.orlicki@petra-po.pl.
Source:
Scientific reports [Sci Rep] 2025 Dec 04; Vol. 16 (1), pp. 1401. Date of Electronic Publication: 2025 Dec 04.
Publication Type:
Journal Article
Language:
English
Journal Info:
Publisher: Nature Publishing Group Country of Publication: England NLM ID: 101563288 Publication Model: Electronic Cited Medium: Internet ISSN: 2045-2322 (Electronic) Linking ISSN: 20452322 NLM ISO Abbreviation: Sci Rep Subsets: MEDLINE
Imprint Name(s):
Original Publication: London : Nature Publishing Group, copyright 2011-
References:
Sci Rep. 2025 Jul 26;15(1):27207. (PMID: 40715468)
BMC Med. 2019 Oct 29;17(1):195. (PMID: 31665002)
AJNR Am J Neuroradiol. 2005 Nov-Dec;26(10):2466-74. (PMID: 16286386)
Korean J Radiol. 2020 Oct;21(10):1126-1137. (PMID: 32729271)
PLoS Med. 2018 Nov 6;15(11):e1002683. (PMID: 30399157)
Elife. 2025 Aug 21;14:. (PMID: 40838493)
Comput Methods Programs Biomed. 2016 Apr;127:248-57. (PMID: 26826901)
Wiley Interdiscip Rev Data Min Knowl Discov. 2019 Jul-Aug;9(4):e1312. (PMID: 32089788)
Annu Int Conf IEEE Eng Med Biol Soc. 2015 Aug;2015:797-800. (PMID: 26736382)
Sci Rep. 2025 Apr 22;15(1):13912. (PMID: 40263348)
Nat Mach Intell. 2019 May;1(5):206-215. (PMID: 35603010)
Cochrane Database Syst Rev. 2013 Jun 04;(6):CD001877. (PMID: 23737396)
Genes (Basel). 2025 May 28;16(6):. (PMID: 40565540)
Sci Rep. 2025 Aug 14;15(1):29908. (PMID: 40813393)
Radiol Artif Intell. 2021 Oct 27;3(6):e210097. (PMID: 34870222)
Sci Rep. 2025 Feb 27;15(1):7010. (PMID: 40016334)
Front Comput Neurosci. 2022 Sep 02;16:1005617. (PMID: 36118133)
Sci Rep. 2025 Aug 6;15(1):28669. (PMID: 40764518)
BMC Gastroenterol. 2025 Mar 11;25(1):157. (PMID: 40069597)
Health Technol Assess. 2000;4(5):1-120. (PMID: 10859208)
NPJ Digit Med. 2021 Oct 28;4(1):154. (PMID: 34711955)
Sci Rep. 2025 Jan 10;15(1):1649. (PMID: 39794374)
Turk J Biol. 2025 Sep 11;49(5):600-634. (PMID: 41246228)
Contributed Indexing:
Keywords: Ensemble learning; Explainable artificial intelligence (XAI); SHAP; Tumor classification
Entry Date(s):
Date Created: 20251204 Date Completed: 20260112 Latest Revision: 20260115
Update Code:
20260115
PubMed Central ID:
PMC12796192
DOI:
10.1038/s41598-025-31079-x
PMID:
41345454
Database:
MEDLINE

Weitere Informationen

Accurate early-stage diagnosis of tumours is crucial for improving patient prognosis. Modern machine learning techniques provide advanced and effective tools to support this process. In this study, both base classifiers and ensemble models were compared in the task of predicting tumour type from morphometric data. Particular attention was given to the CatBoost model as well as the Voting and Stacking classifiers. Model performance was evaluated comprehensively using standard diagnostic metrics and confusion matrices. To enhance interpretability, the SHAP framework was employed to assess the contribution of individual features to model predictions. The CatBoost model stood out due to its ability to provide explainable results through feature importance analysis and SHAP values, which highlighted the key role of parameters related to tumour size and border irregularity. The Voting Classifier improved stability and reduced variance, significantly lowering the number of false negative errors - a factor of particular clinical importance in oncological diagnostics. The Stacking Classifier achieved the highest overall predictive performance, minimising both false positive and false negative classifications by integrating heterogeneous base learners with a meta-classifier. The findings of this study confirm that interpretable ensemble methods represent a valuable approach to supporting the diagnostic process in neuro-oncology. The methodology applied here not only increases the reliability and transparency of artificial intelligence systems but also demonstrates potential for application in treatment monitoring and in predicting the risk of tumour recurrence.
(© 2025. The Author(s).)

Declarations. Competing interests: The authors declare no competing interests.