JCM, Vol. 14, Pages 7079: Predicting Biochemical Recurrence After Robot-Assisted Prostatectomy with Interpretable Machine Learning Model

Journal of Clinical Medicine doi: 10.3390/jcm14197079

Authors:
Tianwei Zhang
Hisamitsu Ide
Jun Lu
Yan Lu
Toshiyuki China
Masayoshi Nagata
Tsuyoshi Hachiya
Shigeo Horie

Background: This study aimed to develop and evaluate machine learning (ML) models to predict biochemical recurrence (BCR) after robot-assisted radical prostatectomy (RARP). Methods: We retrospectively analyzed clinical data from 1125 patients who underwent RARP between July 2013 and December 2023. The dataset was divided into a training set (70%) and a testing set (30%) using a stratified sampling strategy. Five ML models were developed using the training set. Model performance was evaluated on the testing set using the area under the receiver operating characteristic curve (AUC), accuracy, sensitivity, specificity, and F1 scores. Additionally, model interpretability was assessed using SHapley Additive exPlanations (SHAP) values to determine the contribution of individual features. Results: Among the five ML models, the LightGBM model achieved the best prediction ability with an AUC of 0.881 (95%CI: 0.840–0.922) in the testing set. For model interpretability, SHAP values explained the contribution of individual features to the model, revealing that pathological T stage (pT), positive surgical margin (PSM), prostate-specific antigen (PSA) nadir, initial PSA, systematic prostate biopsy positive rate, seminal vesicle invasion (SVI), pathological International Society of Urological Pathology Grade Group (pGG), and perineural invasion (PI) were the key contributors to the predictive performance. Conclusions: We developed and validated ML models to predict BCR following RARP and identified that the LightGBM model with 8 variables achieved promising performance and demonstrated a high level of clinical applicability.

Source link

Tianwei Zhang www.mdpi.com