Predicting the Recurrence of Ovarian Cancer Based on Machine Learning.

Abstract

Background: Recurrence is the main factor for poor prognosis in ovarian cancer, but few prognostic biomarkers were reported. In this study, we used machine learning methods based on multiple biomarkers to develop a specific prediction model for the recurrence of ovarian cancer.

Methods: A total of 277 ovarian cancer patients were enrolled in this study and randomly classified into training and testing cohorts. The prediction information was obtained through 47 clinical parameters using six supervised clustering machine learning algorithms, including K-Nearest Neighbor (K-NN), Decision Tree (DT), Random Forest (RF), Adaptive Boosting (AdaBoost), Gradient Boosting Machine (GBM), and Extreme Gradient Boosting (XGBoost).

Results: In predicting the recurrence of ovarian cancer, machine learning algorithm was superior to conventional logistic regression analysis. In this study, XGBoost showed the best performance in predicting the recurrence of ovarian cancer, with an accuracy of 0.95. In addition, neoadjuvant chemotherapy, Monocyte ratio (MONO%), Hematocrit (HCT), Prealbumin (PAB), Aspartate aminotransferase (AST), and carbohydrate antigen 125 (CA125) are the most important biomarkers to predict the recurrence of ovarian cancer.

Conclusion: The machine learning techniques can achieve a more accurate assessment of the recurrence of ovarian cancer, which can help clinicians make decisions, and develop personalized treatment strategies.