Deep Kernel Learning
About
We introduce scalable deep kernels, which combine the structural properties of deep learning architectures with the non-parametric flexibility of kernel methods. Specifically, we transform the inputs of a spectral mixture base kernel with a deep architecture, using local kernel interpolation, inducing points, and structure exploiting (Kronecker and Toeplitz) algebra for a scalable kernel representation. These closed-form kernels can be used as drop-in replacements for standard kernels, with benefits in expressive power and scalability. We jointly learn the properties of these kernels through the marginal likelihood of a Gaussian process. Inference and learning cost $O(n)$ for $n$ training points, and predictions cost $O(1)$ per test point. On a large and diverse collection of applications, including a dataset with 2 million examples, we show improved performance over scalable Gaussian processes with flexible kernel learning models, and stand-alone deep architectures.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Classification | COIL-20 | Accuracy0.99 | 96 | |
| Regression | UCI ENERGY (test) | Negative Log Likelihood0.45 | 71 | |
| Regression | Boston UCI (test) | -- | 45 | |
| GP regression | Kernel Cookbook 1.0 (test) | MSE2.52e-4 | 35 | |
| Image Classification | MNIST (test) | NLL0.09 | 34 | |
| Regression | Naval | RMSE0.01 | 25 | |
| Regression | Sarcos | RMSE3.46 | 24 | |
| Regression | Wine | NLL0.98 | 22 | |
| Regression | Boston | RMSE3.45 | 22 | |
| Regression | Power | RMSE4.36 | 21 |