Feature Engineering for Interpretable Machine Learning for Quality Assurance in Radiation Oncology

• Published in: Studies in health technology and informatics Vol. 290; p. 460
• Main Authors: Pillai, Malvika, Adapa, Karthik, Shumway, John W, Dooley, John, Das, Shiva K, Chera, Bhishamjit S, Mazur, Lukasz
• Format: Journal Article
• Language: English
• Published: Netherlands 06-06-2022
• Subjects: Engineering; Machine Learning; Radiation Oncology; quality assurance; Machine learning; radiation oncology
• ISSN: 1879-8365

Chart checking is a time intensive process with high cognitive workload for physicists. Previous studies have partially automated and standardized chart checking, but limited studies implement data-driven approaches to reduce cognitive workload for quality assurance processes. This study aims to evaluate feature selection methods to improve the interpretability and transparency of machine learning models in predicting the degree of difficulty for a pretreatment physics chart check. We compare chi-square, mutual information, feature importance thresholding, and greedy feature selection for four different classifiers. Random forest has the highest performance with SMOTE oversampling using mutual information for feature selection (accuracy 84.0%, AUC 87.0%, precision 80.0%, recall 80.0%). This study demonstrates that feature selection methods can improve model interpretability and transparency.