Validation of Machine Learning Prediction Models - Archive ouverte HAL
Article Dans Une Revue The New England Journal of Statistics in Data Science Année : 2023

Validation of Machine Learning Prediction Models

Résumé

We address the estimation of the Integrated Squared Error (ISE) of a predictor η(x) of an unknown function f learned using data acquired on a given design Xn. We consider ISE estimators that are weighted averages of the residuals of the predictor η(x) on a set of selected points Zm. We show that, under a stochastic model for f, minimisation of the mean squared error of these ISE estimators is equivalent to minimisation of a Maximum Mean Discrepancy (MMD) for a nonstationary kernel that is adapted to the geometry of Xn. Sequential Bayesian quadrature then yields sequences of nested validation designs that minimise, at each step of the construction, the relevant MMD. The optimal ISE estimate can be written in terms of the integral of a linear reconstruction, for the assumed model, of the square of the interpolator residuals over the domain of f. We present an extensive set of numerical experiments which demonstrate the good performance and robustness of the proposed solution. Moreover, we show that the validation designs obtained are space-filling continuations of Xn , and that correct weighting of the observed interpolator residuals is more important than the precise configuration Zm of the points at which they are observed.
Fichier principal
Vignette du fichier
NEJSDC_LPJR_submitted.pdf (2.64 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03818234 , version 1 (17-10-2022)

Licence

Identifiants

Citer

Luc Pronzato, Maria Joao Rendas. Validation of Machine Learning Prediction Models. The New England Journal of Statistics in Data Science, 2023, 1 (3), pp.394--414. ⟨doi.org/10.51387/23-NEJSDS50⟩. ⟨hal-03818234⟩
83 Consultations
60 Téléchargements

Altmetric

Partager

More