On the Brittleness of Maximum Likelihood Estimation for Gaussian Process Hyperparameter Optimization
Organizations: Department of Mechanical and Aerospace Engineering, University of California, Irvine · Department of Civil and Environmental Engineering, University of California, Irvine
Abstract
Machine learning (ML) has become an indispensable part of modern engineering design workflows. A crucial step in training an ML model is the selection of the loss function which can be systematically formulated via various techniques such as maximum likelihood estimation (MLE) and cross-validation . While MLE is one of the most popular, effective, and intuitive mechanisms for training ML models, it is brittle: if the assumptions underpinning it are not met, the trained ML model may generalize poorly. This brittleness affects even Gaussian processes (GPs) which are widely used in engineering design and are often (incorrectly) presumed to be very robust to overfitting. In this paper, we fundamentally evaluate the brittleness of MLE in the context of training GPs for probabilistic regression or classification tasks. We compare theoretically grounded metrics against MLE and propose practical solutions. Our extensive studies demonstrate the effectiveness of our solutions in downstream design tasks such as Bayesian optimization and provide a blueprint for practitioners to build accurate and robust GPs that can even outperform tabular foundation models in terms of prediction accuracy, uncertainty quantification, and inference cost. Our contributions are publicly available via GitHub at https://github.com/Bostanabad-Research-Group/GP-vs-TabPFN-vs-GPyTorch.
Figures & tables
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| RRMSE | NIS | ||||||||||||||
| Problem | N | D x | Noise | GP+ | GP+ (PE) | GP+ (LOO) | PFN 2.5 | PFN 2.0 | GPyTorch | GP+ | GP+ (PE) | GP+ (LOO) | PFN 2.5 | PFN 2.0 | GPyTorch |
| Buckling | 5D x | 4 | 0.002 | 0.344 0.14 | 0.344 0.14 | 0.519 0.18 | 0.385 0.26 | 0.286 0.14 | 0.730 0.15 | 6.940 3.20 | 6.940 3.20 | 10.35 5.09 | 1.377 0.24 | 1.524 0.27 | 8.769 1.92 |
| 0.08 | 0.537 0.12 | 0.537 0.12 | 0.851 0.57 | 0.459 0.11 | 0.423 0.14 | 0.759 0.17 | 9.781 3.18 | 9.781 3.18 | 18.10 10.4 | 1.806 0.27 | 1.902 0.32 | 9.340 2.00 | |||
| 20D x | 4 | 0.002 | 0.058 0.02 | 0.058 0.02 | 0.049 0.02 | 0.346 0.27 | 0.205 0.25 | 0.261 0.07 | 0.198 0.06 | 0.198 0.06 | 0.235 0.16 | 0.354 0.10 | 0.476 0.15 | 1.438 0.58 | |
| 0.08 | 0.226 0.06 | 0.226 0.06 | 0.249 0.23 | 0.585 0.35 | 0.382 0.16 | 0.335 0.08 | 1.139 0.49 | 1.139 0.49 | 1.359 2.92 | 0.981 0.17 | 1.285 0.22 | 1.847 0.72 | |||
| Borehole | 5D x | 8 | 0.002 | 0.024 0.01 | 0.027 0.01 | 0.044 0.01 | 0.049 0.01 | 0.065 0.01 | 0.027 0.00 | 0.131 0.04 | 0.139 0.05 | 0.645 0.30 | 0.271 0.03 | 0.345 0.05 | 0.138 0.04 |
| GP+ | GP+ (PE) | GP+ (LOO) | PFN 2.5 | PFN 2.0 | GPyTorch | ||||||||||
| Problem | N | D x | Noise | Train (s) | Pred (s) | Train (s) | Pred (s) | Train (s) | Pred (s) | Fit (s) | Inf (s) | Fit (s) | Inf (s) | Train (s) | Pred (s) |
| Buckling | 5D x | 4 | 0.002 | 5.23 1.16 | 0.011 0.002 | 4.55 1.01 | 0.011 0.037 | 7.14 0.97 | 0.010 0.003 | 0.17 0.07 | 1.05 0.06 | 0.17 0.02 | 1.23 0.04 | 1.54 0.30 | 0.006 0.004 |
| 0.08 | 5.54 1.59 | 0.012 0.002 | 4.79 1.35 | 0.011 0.002 | 6.88 0.93 | 0.010 0.002 | 0.16 0.15 | 1.05 0.04 | 0.15 0.03 | 1.21 0.06 | 1.63 0.49 | 0.008 0.002 | |||
| Buckling | 20D x | 4 | 0.002 | 6.81 1.37 | 0.016 0.001 | 5.87 1.25 | 0.017 0.003 | 6.15 1.36 | 0.015 0.002 | 0.18 0.04 | 1.04 0.06 | 0.18 0.04 | 1.24 0.04 | 2.62 0.62 | 0.010 0.004 |
| 0.08 | 2.92 1.98 | 0.017 0.003 | 2.73 1.73 | 0.014 0.002 | 6.03 2.15 | 0.015 0.003 | 0.17 0.02 | 1.06 0.04 | 0.15 0.03 | 1.20 0.04 | 2.08 0.41 | 0.008 0.001 | |||
| Borehole | 5D x | 8 | 0.002 | 0.50 0.10 | 0.006 0.002 | 0.68 0.08 | 0.013 0.002 | 0.92 0.18 | 0.009 0.002 | 0.17 0.02 | 1.21 0.06 | 0.19 0.04 | 1.43 0.05 | 2.47 0.21 | 0.006 0.001 |
Gaussian Processes
Gaussian Processes are probabilistic models of functions that use kernels to make predictions with explicit uncertainty.
See all 113 papers