Machine learning for prediction of concurrent endometrial carcinoma in patients diagnosed with endometrial intraepithelial neoplasia

Abstract

ObjectiveTo identify predictive clinico-pathologic factors for concurrent endometrial carcinoma (EC) among patients with endometrial intraepithelial neoplasia (EIN) using machine learning. Methodsa retrospective analysis of 160 patients with a biopsy proven EIN. We analyzed the performance of multiple machine learning models (n = 48) with different parameters to predict the diagnosis of postoperative EC. The prediction variables included: parity, gestations, sampling method, endometrial thickness, age, body mass index, diabetes, hypertension, serum CA-125, preoperative histology and preoperative hormonal therapy. Python ‘sklearn’ library was used to train and test the models. The model performance was evaluated by sensitivity, specificity, PPV, NPV and AUC. Five iterations of internal cross-validation were performed, and the mean values were used to compare between the models. ResultsOf the 160 women with a preoperative diagnosis of EIN, 37.5% (60) had a post-op diagnosis of EC. In univariable analysis, there were no significant predictors of EIN. For the five best machine learning models, all the models had a high specificity (71%–88%) and a low sensitivity (23%–51%). Logistic regression model had the highest specificity 88%, XG Boost had the highest sensitivity 51%, and the highest positive predictive value 62% and negative predictive value 73%. The highest area under the curve was achieved by the random forest model 0.646. ConclusionsEven using the most elaborate AI algorithms, it is not possible currently to predict concurrent EC in women with a preoperative diagnosis of EIN. As women with EIN have a high risk of concurrent EC, there may be a value of surgical staging including sentinel lymph node evaluation, to more precisely direct adjuvant treatment in the event EC is identified on final pathology.