Amortized inference

ID: amortized-inference

Amortized inference learns one shared rule that maps observations to approximate posterior distributions, spreading the work of fitting that rule across many inputs. Instead of separately optimizing variational parameters for every observation , a recognition network returns the parameters of . The shared parameters are learned across data; evaluating the rule for a new input is then a different operation from fitting it.[ref]
A scalar Gaussian model gives an exact example. Let and . Completing the square gives
The shared rule , with , and fixed variance , represents the exact posterior for every real . Evaluating replaces a separate optimization for each observation in this stipulated model. It does not imply that every nonlinear latent-variable model has an exact rule this simple.
Sharing can introduce an additional restriction even when the chosen distribution family is adequate. For observations and , the exact posterior means are and . The Gaussian family can represent each exactly when is fitted separately. If the shared rule is constrained to output one constant mean for both inputs, its best equal-weight choice is . Each case then has
This original example isolates an amortization gap: the loss from restricting how observation-specific variational parameters are produced, rather than an inability of the per-observation Gaussian family to express the target. The shared mapping's capacity and its training therefore matter alongside the family of probability distributions.

New to topics? Read the docs here!