Background: Machine learning (ML) and deep learning(DL) technology has been used widely in the quality assurance. Due to the complexity of intensity modulated radiotherapy(IMRT)technology, the implementation of patient-specific quality assurance (PSQA) before the treatment has become an essential part in the IMRT. Therefore, this paper is aim to establish the different machine learning classification predict models of gamma pass rates for specific dosimetric verification of pelvic IMRT plan which based on the radiomic features and to explore the best prediction model.
Methods: Retrospective analysis of the 3D dosimetric verification results based on measurements with gamma pass rate criteria of 3%/2 mm and 10% dose threshold of 196 pelvic intensity-modulated radiotherapy plans was carried. Prediction models were established by extracting radiomic features data. Four machine learning algorithms, random forest, support vector machine, adaptive boosting and gradient boosting decision trees, were used to calculate the AUC value, sensitivity and specificity respectively. The classification performance of the four prediction models were evaluated.
Results: The sensitivity and specificity of the random forest, support vector machine, adaptive boosting, and gradient boosting decision trees models were 0.93,0.85,0.93,0.96, and 0.38,0.69,0.46, and 0.46, respectively. The AUC values for the random forest model and the adaptive boosting model were 0.81 and 0.82, respectively, and the AUC values for the support vector machine and gradient boosting decision tree models were 0.87.
Conclusions: Machine learning methods based on radiomics can be used to establish a prediction model of gamma pass rate for specific dosimetric verification of pelvic intensity modulated radiotherapy. The classification performance of support vector machine model and gradient boosting decision trees model is better than that of random forest model and adaptive boosting model.The prediction model for a specific site is helpful to improve the performance of the model.