Preventing over-fitting during model selection via Bayesian regularisation of the hyper-parameters

Gavin C. Cawley, Nicola L. C. Talbot

Research output: Contribution to journalArticlepeer-review

228 Citations (Scopus)

Abstract

While the model parameters of a kernel machine are typically given by the solution of a convex optimisation problem, with a single global optimum, the selection of good values for the regularisation and kernel parameters is much less straightforward. Fortunately the leave-one-out cross-validation procedure can be performed or a least approximated very efficiently in closed form for a wide variety of kernel learning methods, providing a convenient means for model selection. Leave-one-out cross-validation based estimates of performance, however, generally exhibit a relatively high variance and are therefore prone to over-fitting. In this paper, we investigate the novel use of Bayesian regularisation at the second level of inference, adding a regularisation term to the model selection criterion corresponding to a prior over the hyper-parameter values, where the additional regularisation parameters are integrated out analytically. Results obtained on a suite of thirteen real-world and synthetic benchmark data sets clearly demonstrate the benefit of this approach.
Original languageEnglish
Pages (from-to)841-861
Number of pages21
JournalJournal of Machine Learning Research
Volume8
Publication statusPublished - Apr 2007

Cite this