How do the ELBO, the Barber–Agakov, marginal and VNMC bounds on EIG, the contrastive PCE / ACE bounds, neural ratio estimation and NPE’s forward-KL objective relate?
Summary
Every one of these objectives is the same identity, intractable quantity = computable surrogate ± an expected KL, obtained by replacing an intractable density with a learned . Two things decide everything else. (1) Which density the expectation is taken under: under (reverse KL — ELBO, VAE, the VNMC bound) you need an evaluable target and you get zero-forcing, mode-seeking under-dispersion; under the true joint (forward KL — Barber–Agakov, NPE, the marginal bound, ACE) you need only simulations and you get mass-covering over-dispersion. (2) Whether replaces a numerator or a denominator: replacing the posterior in the numerator gives a lower bound on EIG, replacing the marginal in the denominator gives an upper bound. NPE’s loss is the Barber–Agakov bound up to the prior entropy; PCE is NMC with the generating sample added to the denominator, which flips NMC’s upward bias into a lower bound that cannot exceed .
Answer
1. The shared move
The ELBO and KL Divergence Minimization gives the template: for any ,
so the ELBO is a lower bound on the evidence, tight iff is the posterior (^thm-evidence-decomposition). That note already flags the Barber–Agakov estimator as “the same ‘replace an intractable posterior by and get a bound’ move, applied to mutual information instead of evidence.”
The EIG has three equivalent forms (^thm-eig-forms):
and it is doubly intractable: the posterior and the marginal are both unavailable, and both change with every (Nested Estimation and Nested Monte Carlo). Each estimator below substitutes a for one of them:
- Posterior in the numerator → lower bound. (^thm-ba-bound).
- Marginal in the denominator → upper bound. is “an expected KL from the true marginal to its approximation”, i.e. (^thm-marg-bound).
Both gaps are forward KLs, because the outer expectation is always under the model’s own joint — which is why both are trained from simulations alone, and why the BA note can say no reparameterization is needed: the sampling distribution does not depend on .
2. The one table
| Objective | Surrogate inserted | Gap = which KL | Bound | Amortized over | Characteristic failure |
|---|---|---|---|---|---|
| ELBO (CAVI, ADVI, flows) | for | reverse | lower, on | nothing — one optimisation per dataset | zero-forcing: mode-seeking, light tails, mean-field “underestimates marginal variances”; ELBO value is not a fit measure |
| VAE / amortized ELBO | encoder | reverse, averaged over data | lower | data points | same, plus an amortization gap (network output is not the per-datum optimum) |
| Barber–Agakov | for the posterior | forward | lower, on EIG | all outcomes (and ) | family gap (term III) is permanent; a mass-covering is too diffuse, so EIG is under-stated; needs the density of the running posterior in sequential use |
| NPE | forward — identical loss | (its negative loss + prior entropy is ) | all from the prior predictive | over-dispersion in the limit; in practice “a badly trained network returns a smooth, confident, wrong answer”; prior baked in; SNPE correction instabilities | |
| Marginal | for | forward, in -space | upper, on EIG | nothing (one density per design) | needs explicit likelihood; poor when is large; an upper bound cannot be maximised jointly with the design |
| NMC | , prior draws | none (Jensen bias) | upward-biased, ; consistent | nothing | cost , rate with |
| VNMC | importance-weighted | — reverse-type, under | upper; tight if exact or | outcomes | an under-dispersed proposal gives heavy-tailed importance weights (synthesis, by the logic); second stage falls back to |
| ACE | same weights, but added to the denominator | — forward-type | lower; tight if exact or ; is BA | outcomes ; trained jointly with | cost grows with ; gradient variance (score-function vs reparameterised) |
| PCE | ACE with | forward-type, fixed proposal | lower; tight only as | nothing — no at all | saturation; “degraded as dimension increased” |
| Likelihood-free ACE | unnormalised critic | — | lower (Thm 2) | , and the critic over | critic capacity; inherits the contrastive ceiling |
| Implicit | and | two forward-KL fits | not a bound — only is bounded (Lemma 2) | — | sign of the error unknown, so unsafe to maximise over designs |
| NRE | classifier logit for | cross-entropy, not a KL to the posterior | plug-in is not a bound; the multi-class form is an InfoNCE lower bound (synthesis) | all pairs; MCMC is not amortised | classifier saturation; ratio “can take on an arbitrary value” off-support; trustworthy only where the training prior had mass |
3. Four relationships the table hides
(a) NPE = Barber–Agakov. NPE minimises on prior-predictive pairs (Neural Posterior Estimation (NPE)); BA maximises on the same pairs. The vault states it directly: “the same amortized , trained by the same forward-KL objective”. Synthesis: therefore — any NPE network trained for ABM calibration hands you, for free, a lower bound on how informative that simulated dataset design is about . Conversely, Normalizing Flows as Conditional Density Estimators gives the cleanest reason the two communities use opposite KLs: “VI has the density and lacks samples, so it uses reverse KL; SBI has samples … and lacks the density, so it uses forward KL.”
(b) NMC → VNMC → ACE → PCE is one estimator with two switches. Switch one is the proposal (prior vs learned ); switch two is whether the generating sample sits in the denominator.
| excluded | included | |
|---|---|---|
| prior proposal | NMC (biased up) | PCE (lower bound) |
| learned proposal | VNMC (upper bound) | ACE (lower bound) |
Excluding makes the inner average an unbiased estimate of , and Jensen’s inequality on the pushes the EIG estimate up (Nested Estimation and Nested Monte Carlo). Including it “prevents the catastrophic under-estimation of ” and yields a valid lower bound (^thm1-ace); the NMC note calls this “turning the NMC bias into a controlled bound”. Synthesis: at the VNMC inner term is exactly a single-sample ELBO for , so — EIG plus an expected reverse KL, the mirror image of BA’s EIG minus an expected forward KL. Larger is the importance-weighted (IWAE-style) tightening noted in ^lemma1-vnmc.
(c) Why contrastive bounds saturate. Synthesis — this is not stated in the vault notes. In the denominator contains the numerator’s own term, so the integrand is at most and whatever the true EIG. The same argument for ACE gives : a learned proposal lifts the ceiling by the BA value. That is the quantitative content of the vault’s qualitative remarks that PCE has only “case-2 tightness”, “degraded as dimension increased”, while ACE “generally does at least as well as the better of BA and PCE”.
(d) NRE is the critic view of the same quantity. The optimal NRE logit is , and Neural Ratio Estimation notes it “is the pointwise mutual information whose expectation is the expected information gain”. PCE is “the experimental-design instance of InfoNCE with a known critic” (the likelihood) (^def-pce); likelihood-free ACE learns an unnormalised critic and keeps the bound (^thm2-lface); multi-class NRE (Durkan et al.) learns by picking the right out of and is “closely related to the atomic SNPE-C/APT approach”. So the binary-NRE loss is not a KL to the posterior and gives no bound direction; its multi-class cousin is a PCE-type lower bound with a learned critic and the same ceiling.
4. What the KL direction costs you
Zero-forcing under reverse KL ( ^thm-zero-forcing)
Reverse KL “penalizes placing mass in on areas where has little mass, but penalizes less the reverse”. Consequences: variance underestimation for factorised , mode-seeking against multimodal targets, lighter tails than the posterior.
- Reverse KL is local. On a Gaussian sits on one mode with KL nats while missing 20% of the mass, and computed from ‘s samples is blind to it (Diagnosing Variational Inference (PSIS k-hat and VSBC)). Richer families (Normalizing Flows for Variational Inference) shrink the family gap but keep the direction.
- It is not always under-dispersed. On centred eight schools ADVI over-estimates every ‘s sd; “VI uncertainty is unreliable in an unknown direction” (Variational Inference - Overview). Variational Inference and Pathfinder: “which divergence works best depends on the inferential task at hand.”
- Forward KL “is mass-covering and moment-matching but requires expectations under the unknown posterior” — which simulation supplies. The flows note adds (flagged there as a standard observation, not from its source) that it “tends to over-cover … for a posterior approximation, over-dispersion is usually the less harmful error.”
- For EIG the direction changes the decision, not just the interval. Synthesis: a lower bound with a forward-KL gap is loosest at designs whose posteriors are hardest for the family (multimodal, funnel-shaped), so maximising BA over can prefer designs that are merely easy to amortise. ACE’s extra samples and the ACE-lower / VNMC-upper sandwich exist to remove that confound: “if design ‘s lower bound exceeds design ‘s upper bound, is provably superior” (High-Dimensional Design Applications).
- Error budget. Total error MC variance () + optimisation () + family gap (constant); only VNMC — and ACE/PCE via — can drive the third term to zero without a bigger family (Convergence Rates and Estimator Selection).
Practical Implications
Decision rule
- Have a differentiable log-density, want a posterior fast (MMM prototyping): reverse-KL VI or Pathfinder, then PSIS ( good, unreliable). Treat ADVI ROAS intervals as wrong in an unknown direction; never rank models by ELBO.
- Simulator only (ABM calibration): forward KL — NPE with a flow, or NRE when a classifier is easier than a density and MCMC is acceptable. Validate with SBC / coverage, since the asymptotic over-dispersion story does not protect a badly trained network.
- Scoring a fixed set of geo-test designs: if (lift, ROAS) is low-dimensional and (geo × week panel) is high-dimensional use the posterior family (BA / ACE); in the opposite case use the marginal. With geo random effects the likelihood is implicit → or likelihood-free ACE; use only to score, never to optimise, because it has no bound direction.
- Optimising a continuous design by gradient: needs a lower bound — ACE by default, PCE if the prior is an adequate proposal and dimension is low.
- Sizing : you need . An experiment expected to deliver 3 nats needs ; 6 nats needs . If PCE values cluster near , the estimate is the ceiling, not the design.
- Before acting on a comparison: sandwich the finalists with ACE (lower) and VNMC (upper); spend roughly 50–90% of the budget on training (Foster 2019, Fig. 1d).
Checklist for any new “variational” objective: under which distribution is the expectation taken (→ density or samples needed; → which tail error)? Does replace a numerator or a denominator (→ bound direction; → can it be maximised over designs)? What is amortised, and is there an amortisation gap? Is there a second route to tightness (more inner samples), and does it have a ceiling?
Source Notes
Related Concepts
- Lindley’s Information Measure — expected prior-to-posterior KL, the utility all the EIG bounds target
- Unified SGD BOED - Overview and Variational BOED - Overview — the two Foster papers these bounds come from
- Amortized vs Sequential Inference — what is lost when proposals are adapted to one
- Neural Likelihood Estimation and Sequential Neural Likelihood — the likelihood-targeting alternative, analogue of in
- Approximation Methods — expectation propagation as the classical forward-KL-inspired method
- Simulation-Based Calibration - Overview — the diagnostic that applies to every amortised here
- Q - Encoding a Geo-Holdout as a Bayesian Experimental Design and Computing Its EIG — where these estimators get used
- Q - Choosing a Simulation-Based Inference Method for ABM Calibration — NPE vs NRE vs NLE in practice
Gaps
- The ceiling of PCE/InfoNCE is not stated anywhere in the vault; §3(c) derives it, and the ACE variant, as synthesis. The Poole et al. (2019) / McAllester–Stratos analysis of variational MI bounds is not ingested.
- No note on IWAE / multi-sample ELBOs; the VNMC note mentions the parallel in one line. The identity in §3(b) is synthesis.
- Bound status of NRE (binary loss vs the multi-class contrastive loss) is asserted here by analogy; the vault’s NRE note summarises Durkan et al. second-hand and does not give the loss.
- MINE / NWJ / Donsker–Varadhan critics appear only as a pointer to Kleinegesse & Gutmann; there is no comparison of their bias–variance behaviour.
- Empirical direction of NPE miscalibration under finite simulation budgets (over- vs under-confidence) is not covered beyond the general warning in the benchmarking note; the mass-covering claim is itself flagged in the flows note as not from its source.
- Expectation propagation and -/Rényi divergences — the interpolants between the two KL directions — have only passing coverage.
Follow-Up Questions
- Can an NPE network trained for ABM calibration be reused as the in a BA/ACE bound to choose which summary statistics or data sources to collect?
- For a geo-holdout with hierarchical geo effects, how large is the BA family gap with a Gaussian vs a flow , and does it change the ranking of designs?
- How does randomised MLMC (unbiased EIG) compare with the ACE/VNMC sandwich at equal budget?
- Does PSIS applied to predict when VNMC’s second stage will be unstable?