Explainable deep learning-based survival prediction for non-small cell lung cancer patients undergoing radical radiotherapy.
Journal:
Radiotherapy and oncology : journal of the European Society for Therapeutic Radiology and Oncology
Published Date:
Jan 18, 2024
Abstract
BACKGROUND AND PURPOSE: Survival is frequently assessed using Cox proportional hazards (CPH) regression; however, CPH may be too simplistic as it assumes a linear relationship between covariables and the outcome. Alternative, non-linear machine learning (ML)-based approaches, such as random survival forests (RSFs) and, more recently, deep learning (DL) have been proposed; however, these techniques are largely black-box in nature, limiting explainability. We compared CPH, RSF and DL to predict overall survival (OS) of non-small cell lung cancer (NSCLC) patients receiving radiotherapy using pre-treatment covariables. We employed explainable techniques to provide insights into the contribution of each covariable on OS prediction.