Training and testing of machine learning techniques to aid in prediction of patients requiring adaptive treatments for head and neck radiotherapy
摘要
Adaptive radiotherapy (ART) offers a tailored approach to radiotherapy treatment and has been shown to be beneficial to patients undergoing treatment for head and neck carcinoma. The challenge lies in prospectively identifying patients who will benefit from ART intervention at the planning stage. This study presents the assessment of AI-based predictive models aimed to address this challenge. Retrospective data from 100 head and neck patients were analysed, encompassing various patient features, including weight, neck dimensions, body volume, and target volumes. The training phase began with a decision tree algorithm, which was compared to a selection of other suitable classifiers, being: random forest, bagging, adaBoost and gradient boosting. Model performance was assessed using accuracy, F1 score, and cross-validation accuracy. Initial features in the classifier were selected based on expert (RO) opinion; feature selection was done to refine the final model. The final model was tested on new patient data (