Comparison of time-to-event machine learning models in predicting oral cavity cancer prognosis

Abstract

Background: Applying machine learning to predicting oral cavity cancer prognosis is important in selecting candidates for aggressive treatment following diagnosis. However, models proposed so far have only considered cancer survival as discrete rather than dynamic outcomes. Objectives: To compare the model performance of different machine learning-based algorithms that incorporate time-to-event data. These algorithms included DeepSurv, DeepHit, neural net-extended time-dependent cox model (Cox-Time), and random survival forest (RSF). Materials and Methods: Retrospective cohort of 313 oral cavity cancer patients were obtained from electronic health records. Models were trained on patient data following preprocessing. Predictors were based on demographic, clinicopathologic, and treatment information of the cases. Outcomes were the disease-specific and overall survival. Multivariable analyses were conducted to select significant prognostic features associated with tumor prognosis. Two models were generated per algorithm based on all-prognostic features and significant-prognostic features following statistical analysis. Concordance index (c-index) and integrated Brier scores were used as performance evaluators and model stability was assessed using intraclass correlation coefficients (ICC) calculated from these measures obtained from the cross-validation folds. Results: While all models were satisfactory, better discriminatory performance and calibration was observed for disease-specific than overall survival (mean c-index: 0.85 vs 0.74; mean integrated Brier score: 0.12 vs 0.17). DeepSurv performed best in terms of discrimination for both outcomes (c-indices: 0.76 -0.89) while RSF produced better calibrated survival estimates (integrated Brier score: 0.06 -0.09). Model stability of the algorithms varied with the outcomes as Cox-Time had the best intraclass correlation coefficient (mean ICC: 1.00) for disease-specific survival while DeepSurv was most stable for overall survival prediction (mean ICC: 0.99). Conclusions: Machine learning algorithms based on time-to-event outcomes are successful in predicting oral cavity cancer prognosis with DeepSurv and RSF producing the best discriminative performance and calibration.