Language: Python
Code at Github

Pediatric bone marrow transplantation is a crucial treatment for various hematologic disorders, encompassing both malignant and non-malignant conditions. This research aims to understand the variables influencing the success of unmanipulated allogeneic unrelated donor hematopoietic stem cell transplantation in pediatric patients. By evaluating a wide range of patient demographics, medical history, transplantation details, and recovery metrics, this study seeks to provide valuable insights into optimizing transplant procedures and enhancing patient care.

Introduction
Advancements in healthcare, particularly in pediatric medicine, have integrated machine learning (ML) into critical areas like prognostics and treatment planning. Pediatric bone marrow transplantation, a vital intervention for hematologic disorders, relies on numerous variables that make outcome prediction challenging. This study harnesses ML algorithms to identify and evaluate critical factors affecting the success of bone marrow transplants in children. The goal is to use these insights to improve prediction accuracy and optimize transplant procedures.

Data
The dataset for this study was retrieved from the UCI Machine Learning Repository. It comprises 187 instances and 36 features, providing an extensive understanding of pediatric healthcare in the context of bone marrow transplantation. The data includes patient and donor demographics, medical records, transplant procedure details, and outcomes, with features spanning Integer, Categorical, and Binary data types.

Methodology
Data Pre-processing and Analysis
Data pre-processing involved addressing missing values, outliers, and irrelevant information to enhance data quality. Key steps included:

Null Value Handling: Used Random Forest Regressor for numerical features and mode or Random Forest Classifier for categorical features with missing values.

Outlier Treatment: Applied transformations to mitigate the impact of extreme values, such as log, cube root, and square root transformations.

Feature Engineering: One-hot encoding for nominal features and ordinal encoding for ordinal features.

Feature Selection: Utilized correlation analysis, Variance Inflation Factor (VIF), and redundant feature removal to refine the dataset.

Modeling Techniques
Several classification models were employed to analyze the impact of explanatory variables on survival status:

Logistic Regression: Predicted binary variables using the logistic function, with feature selection done through backward elimination.

Decision Tree: Used to build a tree structure based on data labels, optimizing hyperparameters via Grid Search.

Support Vector Machine (SVM): Constructed a hyperplane to separate classes, using kernel methods for higher-dimensional mapping.

XGBoost: An ensemble method combining decision trees to enhance accuracy through gradient boosting.

Results and Inferences
The performance of the classification models was evaluated based on accuracy, precision, recall, and F1 score. The best models achieved the following metrics:

Model Accuracy Precision Recall F1 Score
Logistic Regression-1 0.79 0.74 0.89 0.81
Decision Tree-1 0.66 0.62 0.84 0.71
SVM-2 0.74 0.68 0.89 0.77
XGBoost-1 0.79 0.72 0.95 0.82


For the Logistic Regression, Decision tree, and XGBoost, all features were selected but for SVM - ’ABOmatch’, ’DonorCMV’, ’RecipientCMV’, ’Allele’, ’Recipientage’, ’extcGvHD’, ’CD34kgx10d6’, ’CD3dkgx10d8’, ’Rbodymass’, ’PLTrecovery’, ’Stemcellsource 1’, ’Relapse 1’, and ’aGvHDIIIIV’ features were selected. The outputs of the confusion matrix and Wald’s test for logistic regression are given in Table 2 and Figure 8. The logistic regression models didn’t show signs of overfitting, yet encountered convergence issues. However, the other models demonstrated slight overfitting, with XGBoost exhibiting a wider margin in comparison. The training accuracy of both XGBoost models was perfect. The XGBoost model achieved good accuracy, the satisfactory precision indicates that the predicted positive cases are true positives, the high recall suggests a lower likelihood of missing actual positives, and the balanced F1 score indicates false positives and negatives are minimal. The feature importance from the best XGBoost model are given in Figure 11 (for those of model-1 refer to Figure 12).

Impact of CD34 Cells
Our investigation also aimed to validate the hypothesis that an increased dosage of CD34+ cells per kilogram extends overall survival time while avoiding concurrent adverse events impacting patients’ quality of life. Rigorous analysis focused on specific features—Relapse, aGvHDIIIIV, and extcGvHD—each representing their absence. The primary objective was to construct a predictive model for Survival time, emphasizing the augmentation of CD34+ cell dosage. Our findings unveiled a noteworthy trend in the relationship between CD34+ cells and disease types. Notably, a significant linear correlation was evident between CD34+ cell dosage and Non-Malignant disease cases (Figure 24). However, in Malignant disease cases, the relationship exhibited less linearity (Figure 25). Specifically, a higher dose of infused CD34+ cells demonstrated an association with improved outcomes in the low-risk disease subgroup. This analysis is in line with the research conducted in this paper.

To evaluate the model’s performance excluding undesirable events, such as the recipient developing acute or chronic graft versus host disease, a subset of test values meeting this criterion was isolated. Subsequently, a linear regression model was employed to establish the slope and intercept of the best-fit line for the data points. This analysis provided insights into the relationship between CD34+ cell dose and survival time, distinctly for malignant and non-malignant diseases.

Conclusion
In essence, our exploration into pediatric bone marrow transplantation uncovered pivotal factors influencing transplant success. The investigation scrutinized many variables and their profound impact on transplantation outcomes. Noteworthy among our findings was the discernible influence of CD34+ cell dosage on treatment efficacy, particularly in different disease subgroups. This highlighted the nuanced relationship between dosage escalation and improved outcomes, underscoring its potential in specific patient cohorts. Despite minor model convergence challenges, limited samples, and mild overfitting, our analysis yields valuable insights for predicting and optimizing bone marrow transplant outcomes in children. These insights promise to refine prognostic approaches, elevate patient care standards, and contribute to enhancing the well-being of pediatric patients undergoing hematologic treatments.