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

Autor: Pillai M; Carolina Health Informatics Program, University of North Carolina, Chapel Hill, North Carolina., Adapa K; Carolina Health Informatics Program, University of North Carolina, Chapel Hill, North Carolina.; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina., Shumway JW; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina., Dooley J; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina., Das SK; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina., Chera BS; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina., Mazur L; Department of Radiation Oncology, University of North Carolina, Chapel Hill, North Carolina.
Jazyk: angličtina
Zdroj: Studies in health technology and informatics [Stud Health Technol Inform] 2022 Jun 06; Vol. 290, pp. 460-464.
DOI: 10.3233/SHTI220118
Abstrakt: 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.
Databáze: MEDLINE