Abstract:Background: The prediction of patients' outcomes is a key component in personalized medicine. Oftentimes, a prediction model is developed using a large number of candidate predictors, called high-dimensional data, including genomic data, lab tests, electronic health records, etc. Variable selection, also called dimension reduction, is a critical step in developing a prediction model using high-dimensional data. Methods: In this paper, we compare the variable selection and prediction performance of popular machine learning (ML) methods with our proposed method. LASSO is a popular ML method that selects variables by imposing an L1-norm penalty to the likelihood. By this approach, LASSO selects features based on the size of regression estimates, rather than their statistical significance. As a result, LASSO can miss significant features while it is known to over-select features. Elastic net (EN), another popular ML method, tends to select even more features than LASSO since it uses a combination of L1- and L2-norm penalties that is less strict than an L1-norm penalty. Insignificant features included in a fitted prediction model act like white noises, so that the fitted model will lose prediction accuracy. Furthermore, for the future use of a fitted prediction model, we have to collect the data of all the features included in the model, which will cost a lot and possibly lower the accuracy of the data if the number of features is too many. Therefore, we propose an ML method, called repeated sieving, extending the standard regression methods with stepwise variable selection. By selecting features based on their statistical significance, it resolves the over-selection issue with high-dimensional data. Results: Through extensive numerical studies and real data examples, our results show that the repeated sieving method selects far fewer features than LASSO and EN, but has higher prediction accuracy than the existing ML methods. Conclusions: We conclude that our repeated sieving method performs well in both variable selection and prediction, and it saves the cost of future investigation on the selected factors.

An Efficient Variable Selection Method for Predictive Discriminant Analysis

Optimal Feature Selection in High-Dimensional Discriminant Analysis

Examining variable selection methods for the predictive performance of regression models and the proportion of selected variables and selected random variables

Variable selection in discriminant analysis based on Gram-Schmidt process

Variational Discriminant Analysis with Variable Selection

Variable Selection in Discriminant Partial Least-Squares Analysis

Effect Size Estimation and Misclassification Rate Based Variable Selection in Linear Discriminant Analysis

Variable Selection via Adaptive False Negative Control in Linear Regression

"Pre-conditioning" for feature selection and regression in high-dimensional problems

Repeated Sieving for Prediction Model Building with High-Dimensional Data

A Novel Variable Selection Approach for Redundant Information Elimination Purpose of Process Control

A Two-Stage Variable Selection Approach for Correlated High Dimensional Predictors

“Preconditioning” for feature selection and regression in high-dimensional problems

DPPred: An Effective Prediction Framework with Concise Discriminative Patterns

Post Selection Shrinkage Estimation for High Dimensional Data Analysis

A comparison of random forest variable selection methods for classification prediction modeling

Efficient Integration of Sufficient Dimension Reduction and Prediction in Discriminant Analysis

A Transparent and Nonlinear Method for Variable Selection

Conditional sufficient variable selection with prior information

Automatic Variable Selection for Partially Linear Functional Additive Model and Its Application to the Tecator Data Set

Prediction and Variable Selection in High-Dimensional Misspecified Binary Classification