Scalable Linearized Laplace Approximation via Surrogate Neural Kernel

Luis A. Ortega, Simón Rodríguez-Santana, Daniel Hernández-Lobato

We introduce a scalable method to approximate the kernel of the Linearized Laplace Approximation (LLA). For this, we use a surrogate deep neural network (DNN) that learns a compact…