Machine learning and explainable artificial intelligence to predict pathologic stage in men with localized prostate cancer.

Though several nomograms exist, machine learning (ML) approaches might improve prediction of pathologic stage in patients with prostate cancer. To develop ML models to predict pathologic stage that outperform existing nomograms that use readily available clinicopathologic variables.

Patients with prostate adenocarcinoma who underwent surgery were identified in the National Cancer Database. Seven ML models were trained to predict organ-confined (OC) disease, extracapsular extension, seminal vesicle invasion (SVI), and lymph node involvement (LNI). Model performance was measured using area under the curve (AUC) on a holdout testing data set. Clinical utility was evaluated using decision curve analysis (DCA). Performance metrics were confirmed on an external validation data set.

The ML-based extreme gradient boosted trees model achieved the best performance with an AUC of 0.744, 0.749, 0.816, 0.811 for the OC, ECE, SVI, and LNI models, respectively. The MSK nomograms achieved an AUC of 0.708, 0.742, 0.806, 0.802 for the OC, ECE, SVI, and LNI models, respectively. These models also performed the best on DCA. Findings were consistent on both a holdout internal validation data set as well as an external validation data set.

Our ML models better predicted pathologic stage relative to existing nomograms at predicting pathologic stage. Accurate prediction of pathologic stage can help oncologists and patients determine optimal definitive treatment options for patients with prostate cancer.

The Prostate. 2024 Oct 13 [Epub ahead of print]

Hemal Semwal, Colton Ladbury, Ali Sabbagh, Osama Mohamad, Derya Tilki, Arya Amini, Jeffrey Wong, Yun Rose Li, Scott Glaser, Bertram Yuh, Savita Dandapani

Department of Bioengineering, University of California Los Angeles, Los Angeles, California, USA., Department of Radiation Oncology, City of Hope National Medical Center, Duarte, California, USA., Department of Radiation Oncology, University of California San Francisco, San Francisco, California, USA., Department of Radiation Oncology, MD Anderson Cancer Center, Houston, Texas, USA., Division of Urology and Urologic Oncology, City of Hope National Medical Center, Duarte, California, USA.