Universal multi-factor feature selection method for radiomics-based brain tumor classification

Abstract

Brain tumor mortality is high, and accurate classification before treatment can improve patient prognosis. Radiomics, which extracts numerous features from medical images, has been widely applied in brain tumor classification studies. Feature selection (FS) is a critical step in radiomics because it reduces redundant information and enhances classification performance. However, the lack of universal FS methods limits the development of radiomics-based brain tumor classification studies. To address this issue, we summarize the characteristics of the FS methods used in related studies and propose a universal method based on three selection factors called triple-factor cascaded selection (TFCS). Particularly, these factors correspond to the correlation between features and task labels, interdependence among features, and role of features in the model. The TFCS method divides FS into two steps. First, it utilizes mutual information to select features that are strongly correlated with the task and contain less redundant information. Recursive feature elimination is then employed to obtain the subset with the best classification performance. To validate the universality of the TFCS, we conducted experiments on seven datasets containing 13 brain tumor classification tasks and evaluated the overall performance using five types of indicators. Results: TFCS exhibited excellent overall performance for all tasks. Compared to the 13 related methods, it takes less time, has moderate parsimony, the best classification performance, adaptability, and stability, and shows better universality. Our study demonstrates that the reasonable utilization of multiple factors can enhance FS performance and provide new insights for future method design.

Keywords: Brain tumor classification; Feature selection; High-dimensionality features; Medical image; Radiomics analysis.