Back to Search Start Over

A tutorial on variable selection for clinical prediction models: feature selection methods in data mining could improve the results.

Authors :
Bagherzadeh-Khiabani F
Ramezankhani A
Azizi F
Hadaegh F
Steyerberg EW
Khalili D
Source :
Journal of clinical epidemiology [J Clin Epidemiol] 2016 Mar; Vol. 71, pp. 76-85. Date of Electronic Publication: 2015 Oct 22.
Publication Year :
2016

Abstract

Objectives: Identifying an appropriate set of predictors for the outcome of interest is a major challenge in clinical prediction research. The aim of this study was to show the application of some variable selection methods, usually used in data mining, for an epidemiological study. We introduce here a systematic approach.<br />Study Design and Setting: The P-value-based method, usually used in epidemiological studies, and several filter and wrapper methods were implemented to select the predictors of diabetes among 55 variables in 803 prediabetic females, aged ≥ 20 years, followed for 10-12 years. To develop a logistic model, variables were selected from a train data set and evaluated on the test data set. The measures of Akaike information criterion (AIC) and area under the curve (AUC) were used as performance criteria. We also implemented a full model with all 55 variables.<br />Results: We found that the worst and the best models were the full model and models based on the wrappers, respectively. Among filter methods, symmetrical uncertainty gave both the best AUC and AIC.<br />Conclusion: Our experiment showed that the variable selection methods used in data mining could improve the performance of clinical prediction models. An R program was developed to make these methods more feasible and visualize the results.<br /> (Copyright © 2016 Elsevier Inc. All rights reserved.)

Details

Language :
English
ISSN :
1878-5921
Volume :
71
Database :
MEDLINE
Journal :
Journal of clinical epidemiology
Publication Type :
Academic Journal
Accession number :
26475568
Full Text :
https://doi.org/10.1016/j.jclinepi.2015.10.002