Cross-validation: what does it estimate and how well does it do it?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Bates, Stephen, Hastie, Trevor, Tibshirani, Robert
Format: Preprint
Veröffentlicht: 2021
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929268837056512
author Bates, Stephen
Hastie, Trevor
Tibshirani, Robert
author_facet Bates, Stephen
Hastie, Trevor
Tibshirani, Robert
contents Cross-validation is a widely-used technique to estimate prediction error, but its behavior is complex and not fully understood. Ideally, one would like to think that cross-validation estimates the prediction error for the model at hand, fit to the training data. We prove that this is not the case for the linear model fit by ordinary least squares; rather it estimates the average prediction error of models fit on other unseen training sets drawn from the same population. We further show that this phenomenon occurs for most popular estimates of prediction error, including data splitting, bootstrapping, and Mallow's Cp. Next, the standard confidence intervals for prediction error derived from cross-validation may have coverage far below the desired level. Because each data point is used for both training and testing, there are correlations among the measured accuracies for each fold, and so the usual estimate of variance is too small. We introduce a nested cross-validation scheme to estimate this variance more accurately, and we show empirically that this modification leads to intervals with approximately correct coverage in many examples where traditional cross-validation intervals fail.
format Preprint
id arxiv_https___arxiv_org_abs_2104_00673
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Cross-validation: what does it estimate and how well does it do it?
Bates, Stephen
Hastie, Trevor
Tibshirani, Robert
Methodology
Statistics Theory
Computation
Machine Learning
Cross-validation is a widely-used technique to estimate prediction error, but its behavior is complex and not fully understood. Ideally, one would like to think that cross-validation estimates the prediction error for the model at hand, fit to the training data. We prove that this is not the case for the linear model fit by ordinary least squares; rather it estimates the average prediction error of models fit on other unseen training sets drawn from the same population. We further show that this phenomenon occurs for most popular estimates of prediction error, including data splitting, bootstrapping, and Mallow's Cp. Next, the standard confidence intervals for prediction error derived from cross-validation may have coverage far below the desired level. Because each data point is used for both training and testing, there are correlations among the measured accuracies for each fold, and so the usual estimate of variance is too small. We introduce a nested cross-validation scheme to estimate this variance more accurately, and we show empirically that this modification leads to intervals with approximately correct coverage in many examples where traditional cross-validation intervals fail.
title Cross-validation: what does it estimate and how well does it do it?
topic Methodology
Statistics Theory
Computation
Machine Learning
url https://arxiv.org/abs/2104.00673