Survival prognostic factors in patients with acute myeloid leukemia using machine learning techniques.

• Main Authors: Keyvan Karami, Mahboubeh Akbari, Mohammad-Taher Moradi, Bijan Soleymani, Hossein Fallahi
• Format: Article
• Language: English
• Published: Public Library of Science (PLoS) 2021-01-01
• Series: PLoS ONE
• Online Access: https://doi.org/10.1371/journal.pone.0254976

This paper identifies prognosis factors for survival in patients with acute myeloid leukemia (AML) using machine learning techniques. We have integrated machine learning with feature selection methods and have compared their performances to identify the most suitable factors in assessing the survival of AML patients. Here, six data mining algorithms including Decision Tree, Random Forrest, Logistic Regression, Naive Bayes, W-Bayes Net, and Gradient Boosted Tree (GBT) are employed for the detection model and implemented using the common data mining tool RapidMiner and open-source R package. To improve the predictive ability of our model, a set of features were selected by employing multiple feature selection methods. The accuracy of classification was obtained using 10-fold cross-validation for the various combinations of the feature selection methods and machine learning algorithms. The performance of the models was assessed by various measurement indexes including accuracy, kappa, sensitivity, specificity, positive predictive value, negative predictive value, and area under the ROC curve (AUC). Our results showed that GBT with an accuracy of 85.17%, AUC of 0.930, and the feature selection via the Relief algorithm has the best performance in predicting the survival rate of AML patients.