دورية أكاديمية

Application of machine learning techniques for predicting survival in ovarian cancer

التفاصيل البيبلوغرافية
العنوان: Application of machine learning techniques for predicting survival in ovarian cancer
المؤلفون: Amir Sorayaie Azar, Samin Babaei Rikan, Amin Naemi, Jamshid Bagherzadeh Mohasefi, Habibollah Pirnejad, Matin Bagherzadeh Mohasefi, Uffe Kock Wiil
المصدر: BMC Medical Informatics and Decision Making, Vol 22, Iss 1, Pp 1-24 (2022)
بيانات النشر: BMC, 2022.
سنة النشر: 2022
المجموعة: LCC:Computer applications to medicine. Medical informatics
مصطلحات موضوعية: Ovarian cancer, Clinical features, Survival prediction, Machine learning, Interpretable machine learning, Computer applications to medicine. Medical informatics, R858-859.7
الوصف: Abstract Background Ovarian cancer is the fifth leading cause of mortality among women in the United States. Ovarian cancer is also known as forgotten cancer or silent disease. The survival of ovarian cancer patients depends on several factors, including the treatment process and the prognosis. Methods The ovarian cancer patients’ dataset is compiled from the Surveillance, Epidemiology, and End Results (SEER) database. With the help of a clinician, the dataset is curated, and the most relevant features are selected. Pearson’s second coefficient of skewness test is used to evaluate the skewness of the dataset. Pearson correlation coefficient is also used to investigate the associations between features. Statistical test is utilized to evaluate the significance of the features. Six Machine Learning (ML) models, including K-Nearest Neighbors , Support Vector Machine (SVM), Decision Tree (DT), Random Forest (RF), Adaptive Boosting (AdaBoost), and Extreme Gradient Boosting (XGBoost), are implemented for survival prediction in both classification and regression approaches. An interpretable method, Shapley Additive Explanations (SHAP), is applied to clarify the decision-making process and determine the importance of each feature in prediction. Additionally, DTs of the RF model are displayed to show how the model predicts the survival intervals. Results Our results show that RF (Accuracy = 88.72%, AUC = 82.38%) and XGBoost (Root Mean Squad Error (RMSE)) = 20.61%, R 2 = 0.4667) have the best performance for classification and regression approaches, respectively. Furthermore, using the SHAP method along with extracted DTs of the RF model, the most important features in the dataset are identified. Histologic type ICD-O-3, chemotherapy recode, year of diagnosis, age at diagnosis, tumor stage, and grade are the most important determinant factors in survival prediction. Conclusion To the best of our knowledge, our study is the first study that develops various ML models to predict ovarian cancer patients’ survival on the SEER database in both classification and regression approaches. These ML algorithms also achieve more accurate results and outperform statistical methods. Furthermore, our study is the first study to use the SHAP method to increase confidence and transparency of the proposed models’ prediction for clinicians. Moreover, our developed models, as an automated auxiliary tool, can help clinicians to have a better understanding of the estimated survival as well as important features that affect survival.
نوع الوثيقة: article
وصف الملف: electronic resource
اللغة: English
تدمد: 1472-6947
Relation: https://doaj.org/toc/1472-6947
DOI: 10.1186/s12911-022-02087-y
URL الوصول: https://doaj.org/article/9f54ca0130d544e4b72e42f90e9ad7c3
رقم الأكسشن: edsdoj.9f54ca0130d544e4b72e42f90e9ad7c3
قاعدة البيانات: Directory of Open Access Journals
الوصف
تدمد:14726947
DOI:10.1186/s12911-022-02087-y