English

Exponential families from a single KL identity

Machine Learning 2026-05-01 v1 Information Theory math.IT

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_{\lambda_2}) - \mathrm{KL}(q \| p_{\lambda_1}) in terms of the log-partition function A(λ)A(\lambda) and the moment μq\mu_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.

Keywords

Cite

@article{arxiv.2604.28036,
  title  = {Exponential families from a single KL identity},
  author = {Marc Dymetman},
  journal= {arXiv preprint arXiv:2604.28036},
  year   = {2026}
}
R2 v1 2026-07-01T12:43:52.959Z