Edith Zhang and David Blei (2021), Bayesian Deep Learning workshop, NeurIPS 2021 — https://bayesiandeeplearning.org/2021/papers/3.pdf

Key takeaways

  • The claim. Local optima of the ELBO for latent Dirichlet allocation, fitted with stochastic variational inference, are not isolated: two optima from different random initialisations are connected by a path along which the ELBO stays essentially at its optimal value. The optima “lie on a connected subset in parameter space” (abstract, §1.1).
  • The method. The simplified string method of E, Ren and Vanden-Eijnden, built to find transition paths in chemistry: 15 beads on the line between two optima, each taking a natural-gradient (batch VI) step, then re-spaced evenly along the string, repeated to convergence (§2, App. C).
  • The explanation. Over-parameterisation, with more topics K than the corpus has, lets a reassignment of one word be absorbed by small shifts in the topics, as with neurons in an over-fitted network (§3). Beyond that, “continuous statistical degeneracy”, a connected set of optimal parameters, would produce near-optimal paths even in an under-parameterised model (§1).

Standing in the record

Filed with the owner’s batch on mode connectivity and model merging, under loss-landscapes (ADR-027), with probabilistic-modeling for its subject: variational inference on a topic model, outside deep learning. The anthology flag follows ADR-027’s batch rule; of the nine works in this filing, it is the one least likely to belong there.

Proposed, not Active. The paper says its experiments show near-optimal paths, flattening with data size and K, and paths in an under-parameterised model. None of these results appears in it. The only figure is a schematic of two topic sets, §1 refers to an “experimental section” that the paper does not have, and App. C’s loop says “repeat steps 3 and 4” where steps 2 and 3 are meant. Its value now is the conjecture and the method, transplanted from chemistry by way of Draxler et al. (LIT-653), whose method it says it resembles. It sets its theoretical discussion closest to Kuditipudi et al. (LIT-671), on over-parameterisation and resilience. It builds on neither and measures neither, so no relation is declared.

“Statistical degeneracy” is the record’s singular-model question in another vocabulary: Watanabe’s singular learning theory (LIT-354, unread) is the frame in which non-identifiable models have connected sets of optima, and the batch’s paper on generative and contrastive models (LIT-660) cites it for the same intuition. Neither paper connects the two.