Interpretable machine learning model based on clinical factors for predicting muscle radiodensity loss after treatment in ovarian cancer

Muscle radiodensity loss after surgery and adjuvant chemotherapy is associated with poor outcomes in ovarian cancer. Assessing muscle radiodensity is a real-world clinical challenge owing to the requirement for computed tomography (CT) with consistent protocols and labor-intensive processes. This study aimed to use interpretable machine learning (ML) to predict muscle radiodensity loss.

This study included 723 patients with ovarian cancer who underwent primary debulking surgery and platinum-based chemotherapy between 2010 and 2019 at two tertiary centers (579 in cohort 1 and 144 in cohort 2). Muscle radiodensity was assessed from pre- and post-treatment CT acquired with consistent protocols, and a decrease in radiodensity ≥ 5% was defined as loss. Six ML models were trained, and their performances were evaluated using the area under the curve (AUC) and F1-score. The SHapley Additive exPlanations (SHAP) method was applied to interpret the ML models.

The CatBoost model achieved the highest AUC of 0.871 (95% confidence interval, 0.870–0.874) and F1-score of 0.688 (95% confidence interval, 0.685–0.691) among the models in the training set and outperformed in the external validation set, with an AUC of 0.839 and F1-score of 0.673. Albumin change, ascites, and residual disease were the most important features associated with a higher likelihood of muscle radiodensity loss. The SHAP force plot provided an individualized interpretation of model predictions.

An interpretable ML model can assist clinicians in identifying ovarian cancer patients at risk of muscle radiodensity loss after treatment and understanding the contributors of muscle radiodensity loss.