Prediction of Cranial Radiotherapy Treatment in Pediatric Acute Lymphoblastic Leukemia Patients Using Machine Learning: A Case Study at MAHAK Hospital

BACKGROUND: Acute Lymphoblastic Leukemia (ALL) is the most common blood disease in children and is responsible for the most deaths amongst children. Due to major improvements in the treatment protocols in the 50-years period, the survivability of this disease has witnessed dramatic rise until this date which is about 90 percent. There are many investigations tending to indicate the efficiency of cranial radiotherapy found out that without that, outcome of the patients did not change and even it improved at some cases. METHODS: the main aim of this study is predicting cranial radiotherapy treatment in pediatric acute lymphoblastic leukemia patients using machine learning. Scope of this paper is intertwined with predicting the necessity of one of the treatment modalities that has been used for many years for this group of patients named Cranial Radiotherapy (CRT). For this purpose, a case study is considered at Mahak charity hospital. In this paper, our focus is on ALL patients aged 0 to 17 treated at Mahak hospital, one of the best centers for treatment of childhood malignancies in Iran. Dataset analyzed in this study is gathered by the research team from patient’s paper-based files. Our dataset consists of 241 observations on patients with 31 attributes after the data cleaning process. Our designed machine learning model for predicting cranial radiotherapy treatment in pediatric acute lymphoblastic leukemia patients is a stacked ensemble classifier of independently strong models with a meta-learner to tune the weights and parameters of the base classifiers. RESULTS: The stacked ensemble classifier show highly reasonable performance with AUC of 87.52%. Moreover, the attributes are ranked based on their predictive power and the most important variable for CRT necessity prediction is the disease relapse. CONCLUSION: In order to conclude, derived from previous studies regarding CRT it is not only cost-effective but also more healthy to eradicate the use of CRT for the treatment of childhood ALL. Furthermore, it is valuable to increase the clinical databases by creating more synthetic health databases not only for research purposes but also for physicians to keep track of their patient’s status.