Cargando…

Measuring the bias of incorrect application of feature selection when using cross-validation in radiomics

BACKGROUND: Many studies in radiomics are using feature selection methods to identify the most predictive features. At the same time, they employ cross-validation to estimate the performance of the developed models. However, if the feature selection is performed before the cross-validation, data lea...

Descripción completa

Detalles Bibliográficos
Autor principal: Demircioğlu, Aydin
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Springer International Publishing 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8613324/
https://www.ncbi.nlm.nih.gov/pubmed/34817740
http://dx.doi.org/10.1186/s13244-021-01115-1
Descripción
Sumario:BACKGROUND: Many studies in radiomics are using feature selection methods to identify the most predictive features. At the same time, they employ cross-validation to estimate the performance of the developed models. However, if the feature selection is performed before the cross-validation, data leakage can occur, and the results can be biased. To measure the extent of this bias, we collected ten publicly available radiomics datasets and conducted two experiments. First, the models were developed by incorrectly applying the feature selection prior to cross-validation. Then, the same experiment was conducted by applying feature selection correctly within cross-validation to each fold. The resulting models were then evaluated against each other in terms of AUC-ROC, AUC-F1, and Accuracy. RESULTS: Applying the feature selection incorrectly prior to the cross-validation showed a bias of up to 0.15 in AUC-ROC, 0.29 in AUC-F1, and 0.17 in Accuracy. CONCLUSIONS: Incorrect application of feature selection and cross-validation can lead to highly biased results for radiomic datasets. SUPPLEMENTARY INFORMATION: The online version contains supplementary material available at 10.1186/s13244-021-01115-1.