cs.LGApr 30, 2026

Exponential families from a single KL identity

Authors: Marc Dymetman

Organizations: Scientific Consultant

Abstract

Exponential families encompass the distributions central to modern machine learning -- softmax, Gaussians, and Boltzmann distributions -- and underlie the theory of variational inference, entropy-regularized reinforcement learning, and RLHF. We isolate a simple identity for exponential families that expresses the KL difference KL(qpλ2)KL(qpλ1)\mathrm{KL}(q \| p_{λ_2}) - \mathrm{KL}(q \| p_{λ_1}) in terms of the log-partition function A(λ)A(λ) and the moment μqμ_q. Remarkably, this identity together with the single fact that KL0\mathrm{KL} \geq 0 (with equality iff p=qp = q) suffices, by direct substitution and rearrangement, to derive a cluster of results that are classically obtained by separate, heavier arguments: a generalized three-point identity for arbitrary reference distributions, Pythagorean theorems for I-projections and reverse I-projections, convexity of the log-partition function, identification of its Legendre dual in KL terms, the Gibbs variational principle, and the explicit optimizer in KL-regularized reward maximization, including the exponential tilting formula underlying entropy-regularized control and RLHF. Beyond these purely algebraic consequences, standard analytic arguments recover the gradient formula for the log-partition function, the Bregman representation of within-family KL divergence, and the surjectivity of the moment map. The note is self-contained.

Explore similar work

CardsList