A comparison of different machine learning classifiers in predicting xerostomia and sticky saliva due to head and neck radiotherapy using a multi-objective, multimodal radiomics model.

Abstract

Background and Purpose. Although radiotherapy techniques are a primary treatment for head and neck cancer (HNC), they are still associated with substantial toxicity and side effects. Machine learning (ML) based radiomics models for predicting toxicity mostly rely on features extracted from pre-treatment imaging data. This study aims to compare different models in predicting radiation-induced xerostomia and sticky saliva in both early and late stages HNC patients using CT and MRI image features along with demographics and dosimetric information.Materials and Methods.A cohort of 85 HNC patients who underwent radiation treatment was evaluated. We built different ML-based classifiers to build a multi-objective, multimodal radiomics model by extracting 346 different features from patient data. The models were trained and tested for prediction, utilizing Relief feature selection method and eight classifiers consisting eXtreme Gradient Boosting (XGBoost), Multilayer Perceptron (MLP), Support Vector Machines (SVM), Random Forest (RF), K-Nearest Neighbor (KNN), Naive Bayes (NB), Logistic Regression (LR), and Decision Tree (DT). The performance of the models was evaluated using sensitivity, specificity, area under the curve (AUC), and accuracy metrics.Results.Using a combination of demographics, dosimetric, and image features, the SVM model obtained the best performance with AUC of 0.77 and 0.81 for predicting early sticky saliva and xerostomia, respectively. Also, SVM and MLP classifiers achieved a noteworthy AUC of 0.85 and 0.64 for predicting late sticky saliva and xerostomia, respectively.Conclusion. This study highlights the potential of baseline CT and MRI image features, combined with dosimetric data and patient demographics, to predict radiation-induced xerostomia and sticky saliva. The use of ML techniques provides valuable insights for personalized treatment planning to mitigate toxicity effects during radiation therapy for HNC patients.