cs.LGMay 20, 2026

Three Costs of Amortizing Gaussian Process Inference with Neural Processes

Authors: Robin Young

Organizations: University of Cambridge, Cambridge, UK

Abstract

Neural processes amortize Gaussian process inference, replacing the exact O(n3)O(n^3) posterior with a learned O(n)O(n) map from context sets to predictive distributions. For a class of latent neural processes, we bound the Kullback--Leibler (KL) divergence between the GP and LNP predictives, decomposing it into three interpretable sources, namely label contamination as the neural process uses label values to estimate a quantity that is label-independent in the exact GP, an information bottleneck because the finite-dimensional representation cannot resolve the full context geometry, and amortization error from a single encoder network shared across all contexts. The bottleneck truncation term decays in the representation dimension dd as O(ecd2/dx)O(e^{-cd^{2/d_x}}) for squared-exponential kernels on Rdx\mathbb{R}^{d_x} where c>0c > 0 is a kernel-dependent constant and as O(d2ν/dx)O(d^{-2ν/d_x}) for Matérn-νν kernels, directly linking architecture sizing to kernel smoothness and input dimension. The label contamination term is O(1)O(1) in general, with only the observation-noise component decaying as O(1/n)O(1/n), identifying a persistent cost of routing uncertainty estimation through a label-dependent representation. These results characterize the costs of amortization within the analyzed class and yield architectural recommendations to predict variance from context locations alone in the GP-amortization regime, and replace mean aggregation with second-order pooling to close the dominant amortization gap.

Explore similar work

CardsList
  1. Conditioning Gaussian Processes on Almost Anything

    May 20, 2026Henry Moss, Lachlan Astfalck, Thomas Cowperthwaite +5Gaussian Process3D Gaussian Splatting