# Bayes linear regression and basis-functions in Gaussian process regression

a.k.a Fixed Rank Kriging, weight space GPs

February 22, 2022 — July 27, 2022

Another way of cunningly chopping up the work of fitting a Gaussian process is to represent the process as a random function comprising basis functions \(\phi=\left(\phi_{1}, \ldots, \phi_{\ell}\right)\) with the Gaussian random weight vector \(w\) so that \[
f^{(w)}(\cdot)=\sum_{i=1}^{\ell} w_{i} \phi_{i}(\cdot) \quad \boldsymbol{w} \sim \mathcal{N}\left(\mathbf{0}, \boldsymbol{\Sigma}_{\boldsymbol{w}}\right).
\] \(f^{(w)}\) is a random function satisfying \(\boldsymbol{f}^{(\boldsymbol{w})} \sim \mathcal{N}\left(\mathbf{0}, \boldsymbol{\Phi}_{n} \boldsymbol{\Sigma}_{\boldsymbol{w}} \boldsymbol{\Phi}^{\top}\right)\), where \(\boldsymbol{\Phi}_{n}=\boldsymbol{\phi}(\mathbf{X})\) is a \(|\mathbf{X}| \times \ell\) matrix of features. This is referred to as a *weight space* approach in ML.

TODO: I just assumed centred weights here, but that is crazy. Update to relax that assumption.

We might imagine this representation would be exact if we had countably many basis functions, and under sane conditions it is. We would like to know, further, that we can find a basis such that we need *not too many* basis functions to represent the process. Looking at the Karhunen-Loève theorem theorem we might imagine that this can sometimes work out fine, and indeed it does, sometimes.

This is a classic; see Chapter 3 of Bishop (2006) is classic and nicely clear. Cressie and Wikle (2011) targets for the spatiotemporal context.

Hijinks ensue when selecting the basis functions. If we were to treat the natural Hilbert space here seriously we could consider identifying the bases as eigenfunctions of the kernel. This is not generally easy. We tend to use either global bases such as Fourier bases or more generally Karhunen-Loéve bases, or construct local bases of limited overlap (usually piecewise polynomials AFAICT).

The kernel trick writes a kernel \(k\) as an inner product in a corresponding reproducing kernel Hilbert space (RKHS) \(\mathcal{H}_{k}\) with a feature map \(\varphi: \mathcal{X} \rightarrow \mathcal{H}_{k} .\) In sufficiently nice cases the kernel is well approximated \[ k\left(\boldsymbol{x}, \boldsymbol{x}^{\prime}\right)=\left\langle\varphi(\boldsymbol{x}), \varphi\left(\boldsymbol{x}^{\prime}\right)\right\rangle_{\mathcal{H}_{k}} \approx \boldsymbol{\phi}(\boldsymbol{x})^{\top} \overline{\boldsymbol{\phi}\left(\boldsymbol{x}^{\prime}\right)} \] where \(\boldsymbol{\phi}: \mathcal{X} \rightarrow \mathbb{C}^{\ell}\) is a finite-dimensional feature map. TODO: What is the actual guarantee here?

## 1 Fourier features

When the Fourier basis is natural for the problem we are in a pretty good situation. We can use the Wiener Khintchine relations to analyse and simulate the process. Connection perhaps to Fourier features in neural nets?

## 2 Random Fourier features

The random Fourier features method (Rahimi and Recht 2007, 2008) constructs a Monte Carlo estimate to a stationary kernel by representing the inner product in terms of \(\ell\) complex exponential basis functions \(\phi_{j}(\boldsymbol{x})=\ell^{-1 / 2} \exp \left(i \boldsymbol{\omega}_{j}^{\top} \boldsymbol{x}\right)\) with frequency parameters \(\boldsymbol{\omega}_{j}\) sampled proportionally to the spectral density \(\rho\left(\boldsymbol{\omega}_{j}\right).\)

This sometimes has a favourable error rate (Sutherland and Schneider 2015).

## 3 K-L basis

We recall from the Karhunen-Loéve notebook that the mean-square-optimal \(f^{(w)}\) for approximating a Gaussian process \(f\) is found by truncating the Karhunen-Loéve expansion \[ f(\cdot)=\sum_{i=1}^{\infty} w_{i} \phi_{i}(\cdot) \quad w_{i} \sim \mathcal{N}\left(0, \lambda_{i}\right) \] where \(\phi_{i}\) and \(\lambda_{i}\) are, respectively, the \(i\)-th (orthogonal) eigenfunction and eigenvalue of the covariance operator \(\psi \mapsto \int_{\mathcal{X}} \psi(\boldsymbol{x}) k(\boldsymbol{x}, \cdot) \mathrm{d} \boldsymbol{x}\), written in decreasing order of \(\lambda_{i}\). What is the orthogonal basis \(\{\phi_{i}\}_i\) though? That depends on the problem and can be a lot of work to calculate.

In the case that our field is stationary on a “nice” domain, though, this can easy — we simply have the Fourier features as the natural basis.

## 4 Compactly-supported basis functions

As seen in GPs as SDEs and FEMs (Lindgren, Rue, and Lindström 2011; Lord, Powell, and Shardlow 2014).

## 5 “Decoupled” bases

Cheng and Boots (2017);Salimbeni et al. (2018);Shi, Titsias, and Mnih (2020);Wilson et al. (2020).

## 6 References

*arXiv:1403.6015 [Astro-Ph, Stat]*.

*Pattern Recognition and Machine Learning*. Information Science and Statistics.

*Advances in Neural Information Processing Systems*.

*Journal of the American Statistical Association*.

*Journal of the Royal Statistical Society: Series B (Statistical Methodology)*.

*Journal of Computational and Graphical Statistics*.

*Statistics for Spatio-Temporal Data*. Wiley Series in Probability and Statistics 2.0.

*Wiley StatsRef: Statistics Reference Online*.

*arXiv:1903.03986 [Cs, Stat]*.

*Handbook of Mathematical Geosciences: Fifty Years of IAMG*.

*Journal of Statistical Software*.

*Journal of Applied Mechanics*.

*IEEE Transactions on Pattern Analysis and Machine Intelligence*.

*arXiv:2012.11857 [Cs, Math, Stat]*.

*Proceedings of the International Conference on Machine Learning*.

*Journal of the Royal Statistical Society: Series B (Statistical Methodology)*.

*arXiv:1411.4681 [Math, Stat]*.

*An Introduction to Computational Stochastic PDEs*. Cambridge Texts in Applied Mathematics.

*Journal of Agricultural, Biological and Environmental Statistics*.

*Journal of the American Statistical Association*.

*Mathematical Geosciences*.

*SIAM Journal on Scientific Computing*.

*Progress in Aerospace Sciences*.

*Advances in Neural Information Processing Systems*.

*2008 46th Annual Allerton Conference on Communication, Control, and Computing*.

*Advances in Neural Information Processing Systems*.

*arXiv:2004.11408 [Stat]*.

*Proceedings of the 32nd International Conference on Neural Information Processing Systems*. NIPS’18.

*IEEE Signal Processing Magazine*.

*Proceedings of the Twenty Third International Conference on Artificial Intelligence and Statistics*.

*Proceedings of the Twenty-Second International Conference on Artificial Intelligence and Statistics*.

*Statistics and Computing*.

*Journal of the Korean Statistical Society*.

*Proceedings of the Thirty-First Conference on Uncertainty in Artificial Intelligence*. UAI’15.

*Proceedings of the 37th International Conference on Machine Learning*.

*Journal of Statistical Software*.