JMIR Publications

ABSTRACT

Background:

Traditional statistical models often fail to capture the complex dynamics influencing survival outcomes in bladder cancer patients after radical cystectomy, a procedure where approximately 50% of patients develop metastases within two years. The integration of artificial intelligence (AI) offers a promising avenue for enhancing prognostic accuracy and personalizing treatment strategies.

Objective:

This study aimed to develop and evaluate a machine learning algorithm for predicting disease-free survival (DFS), overall survival (OS), and the cause of death in patients with bladder cancer undergoing cystectomy, utilizing a comprehensive dataset of clinical and pathological variables.

Methods:

Retrospective data from 370 bladder cancer patients undergoing radical cystectomy at Fondazione Policlinico Gemelli, Rome, Italy, were collected by two urologists. The dataset comprised 20 input variables, encompassing demographics, tumor characteristics, treatment, and inflammatory markers. The CatBoost algorithm was employed for regression tasks (DFS in 346 patients, OS in 347 patients) and a binary classification task (tumor-related death in 312 patients). Model performance was assessed by AN using Mean Absolute Error (MAE) for regression and F1-score for classification, prioritizing a minimum recall of 75% for tumor-related deaths. Five-fold cross-validation and SHAP (Shapley Additive exPlanations) values were employed to ensure robustness and interpretability.

Results:

For DFS prediction, the CatBoost model achieved an MAE of 18.68 months, with clinical tumor stage (CTS) and pathological tumor classification (TC) identified as the most influential predictors. Overall survival prediction yielded an MAE of 17.2 months, which improved to 14.6 months after feature filtering, where TC and the Systemic Immune-Inflammation Index (SII) were most impactful. For tumor-related death classification, the model achieved a recall of 78.6% and an F1-score of 0.44 for the positive class (tumor deaths), correctly identifying 11 out of 14 cases. Bladder tumor position was the most influential feature for the cause of death.

Conclusions:

The developed machine learning algorithm demonstrates promising accuracy in predicting survival and the cause of death in bladder cancer patients after cystectomy. Key predictors include clinical and pathological tumor staging, systemic inflammation (SII), and bladder tumor position. These findings highlight the potential of AI in providing clinicians with an objective, data-driven tool to improve personalized prognostic assessment and guide clinical decision-making.