Prediction-based regularization using data augmented regression
The role of regularization is to control fitted model complexity and variance by penalizing (or constraining) models to be in an area of model space that is deemed reasonable, thus facilitating good predictive performance. This is typically achieved by penalizing a parametric or non-parametric representation of the model. In this paper we advocate instead the use of prior knowledge or expectations about the predictions of models for regularization. This has the twofold advantage of allowing a more intuitive interpretation of penalties and priors and explicitly controlling model extrapolation into relevant regions of the feature space. This second point is especially critical in high-dimensional modeling situations, where the curse of dimensionality implies that new prediction points usually require extrapolation. We demonstrate that prediction-based regularization can, in many cases, be stochastically implemented by simply augmenting the dataset with Monte Carlo pseudo-data. We investigate the range of applicability of this implementation. An asymptotic analysis of the performance of Data Augmented Regression (DAR) in parametric and non-parametric linear regression, and in nearest neighbor regression, clarifies the regularizing behavior of DAR. We apply DAR to simulated and real data, and show that it is able to control the variance of extrapolation, while maintaining, and often improving, predictive accuracy.
KeywordsRegression Nearest-neighbor Extrapolation Machine learning Regularization
Unable to display preview. Download preview PDF.
- Bickel, P., Bo, L.: Regularization in statistics. Test, pp. 271–344 Google Scholar
- Bickel, P., Ritov, Y., Tsybakov, A.B.: Simultaneous analysis of Lasso and Dantzig selector. Ann. Stat. 37 (2009) Google Scholar
- Breiman, L.: Bagging predictors. Mach. Learn. 24(2) (1996) Google Scholar
- Breiman, L.: Statistical modeling: the two cultures. Stat. Sci. (2001) Google Scholar
- Christensen, R.: Analysis of Variance, Design and Regression: Applied Statistical Methods. Chapman and Hall, New York (1996) Google Scholar
- Dasarathy, B.: Nearest Neighbor Pattern Classification Techniques. IEEE Comput. Soc., Los Alamitos (1991) Google Scholar
- Hooker, G.: Diagnosing extrapolation: Tree-based density estimation. In: Proceedings of the Tenth ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (2004) Google Scholar
- Munson, MA, Webb, K., Sheldon, D., Fink, D., Hochachka, W.M., Iliff, M., Riedewald, M., Sorokina, D., Sullivan, B., Wood, C., Kelling, S.: The ebird reference dataset. http://www.avianknowledge.net (2009)
- R Development Core Team: R: a language and environment for statistical computing. R foundation for statistical computing, Vienna, Austria, URL http://www.R-project.org (2007). ISBN 3-900051-07-0
- Vapnik, V.N.: The Nature of Statistical Learning Theory. Springer, Berlin (1996) Google Scholar
- Wahba, G.: Spline Models for Observational Data. CBMS-NSF Regional Conference Series in Applied Mathematics (1990) Google Scholar