A1 Refereed original research article in a scientific journal

Interpretable machine learning model for prediction of overall survival in laryngeal cancer




AuthorsAlabi Rasheed Omobolaji, Almangush Alhadi, Elmusrati Mohammed, Leivo Ilmo, Mäkitie Antti A.

PublisherTaylor and Francis Ltd.

Publication year2024

JournalActa Oto-Laryngologica

Journal name in sourceActa Oto-Laryngologica

eISSN1651-2251

DOIhttps://doi.org/10.1080/00016489.2023.2301648

Web address https://doi.org/10.1080/00016489.2023.2301648


Abstract

Background: The mortality rates of laryngeal squamous cell carcinoma cancer (LSCC) have not significantly decreased in the last decades.

Objectives: We primarily aimed to compare the predictive performance of DeepTables with the state-of-the-art machine learning (ML) algorithms (Voting ensemble, Stack ensemble, and XGBoost) to stratify patients with LSCC into chance of overall survival (OS). In addition, we complemented the developed model by providing interpretability using both global and local model-agnostic techniques.

Methods: A total of 2792 patients in the Surveillance, Epidemiology, and End Results (SEER) database diagnosed with LSCC were reviewed. The global model-agnostic interpretability was examined using SHapley Additive exPlanations (SHAP) technique. Likewise, individual interpretation of the prediction was made using Local Interpretable Model Agnostic Explanations (LIME).

Results: The state-of-the-art ML ensemble algorithms outperformed DeepTables. Specifically, the examined ensemble algorithms showed comparable weighted area under receiving curve of 76.9, 76.8, and 76.1 with an accuracy of 71.2%, 70.2%, and 71.8%, respectively. The global methods of interpretability (SHAP) demonstrated that the age of the patient at diagnosis, N-stage, T-stage, tumor grade, and marital status are among the prominent parameters.

Conclusions: A ML model for OS prediction may serve as an ancillary tool for treatment planning of LSCC patients. © 2024 Acta Oto-Laryngologica AB (Ltd).



Last updated on 2025-15-08 at 14:08