Antonio Franca
Structure, Symmetry, and Applications · 10

Information Geometry

Statistical manifolds, the Fisher-Rao metric, natural gradient, and the geometry of generative models

Statistics has its own geometry. The collection of all probability distributions on a given sample space is not merely a set — it is a manifold, and the natural way to measure distances between distributions equips it with a Riemannian metric. That metric is the Fisher information metric, and the geometry it induces — information geometry — is the subject of this final chapter. The chapter draws together everything that has come before: smooth manifolds (Chapter 1), Riemannian metrics (Chapter 4), connections (Chapter 5), curvature (Chapter 6), and the duality structures that connect symplectic geometry (Chapter 9) to the dually flat spaces of exponential families. Information geometry is not an appendix to differential geometry but one of its richest applications — and, through its connections to machine learning, optimal transport, and quantum mechanics, one of its most active frontiers.

Statistical Manifolds

Let X\mathcal{X} be a measurable sample space and M={pθ:θΘ}\mathcal{M} = \{p_\theta : \theta \in \Theta\} a parametric family of probability distributions on X\mathcal{X}, where ΘRn\Theta \subset \mathbb{R}^n is an open parameter domain. Under mild regularity conditions (differentiability of the log-likelihood, integrability of derivatives), M\mathcal{M} is a smooth manifold with coordinates θ=(θ1,,θn)\theta = (\theta^1, \ldots, \theta^n).

[!definition] Score Function and Fisher Information Matrix For a parametric model pθp_\theta, the score function is i(θ;x)=ilogpθ(x)\ell_i(\theta; x) = \partial_i \log p_\theta(x), where i=/θi\partial_i = \partial/\partial\theta^i. The Fisher information matrix is gij(θ)=Expθ[ilogpθ(x)jlogpθ(x)]=Expθ[ijlogpθ(x)].g_{ij}(\theta) = \mathbb{E}_{x \sim p_\theta}[\partial_i \log p_\theta(x)\, \partial_j \log p_\theta(x)] = -\mathbb{E}_{x \sim p_\theta}[\partial_i \partial_j \log p_\theta(x)].

The Fisher information matrix is symmetric and, under identifiability, positive definite. It therefore defines a Riemannian metric on the parameter manifold M\mathcal{M}, called the Fisher information metric or Fisher-Rao metric.

The metric gijg_{ij} is the unique (up to scalar) Riemannian metric on M\mathcal{M} that is invariant under sufficient statistics: if T:XYT: \mathcal{X} \to \mathcal{Y} is a sufficient statistic for pθp_\theta, then the Fisher metric on the induced model on Y\mathcal{Y} equals the pullback of the metric on M\mathcal{M}. This is Chentsov’s theorem — the Fisher metric is canonically singled out by statistical invariance.

Exponential Families

The most important class of statistical manifolds is the family of exponential distributions, which arise naturally in maximum entropy problems, Bayesian statistics, and machine learning.

[!definition] Exponential Family A family {pθ}\{p_\theta\} is an exponential family if it can be written as pθ(x)=exp ⁣(i=1nθiTi(x)ψ(θ)+k(x)),p_\theta(x) = \exp\!\left(\sum_{i=1}^n \theta^i T_i(x) - \psi(\theta) + k(x)\right), where T=(T1,,Tn)T = (T_1, \ldots, T_n) are sufficient statistics, θ=(θ1,,θn)\theta = (\theta^1, \ldots, \theta^n) are natural parameters, ψ(θ)=logexp(iθiTi(x))dμ(x)\psi(\theta) = \log \int \exp(\sum_i \theta^i T_i(x))\, d\mu(x) is the log-partition function (or cumulant-generating function), and k(x)k(x) is a base measure.

Examples: Gaussian N(μ,σ2)\mathcal{N}(\mu, \sigma^2), Bernoulli, Poisson, Gamma, Dirichlet, von Mises — all are exponential families. The natural parameters θi\theta^i are the coordinates; the expectation parameters ηi=Epθ[Ti(x)]=iψ(θ)\eta_i = \mathbb{E}_{p_\theta}[T_i(x)] = \partial_i \psi(\theta) provide dual coordinates.

[!theorem] Geometry of Exponential Families For an exponential family:

  1. The Fisher information matrix is the Hessian of the log-partition function: gij(θ)=ijψ(θ)g_{ij}(\theta) = \partial_i \partial_j \psi(\theta).
  2. The map θη=ψ(θ)\theta \mapsto \eta = \nabla\psi(\theta) (natural to expectation parameters) is a diffeomorphism; its inverse is ηθ=φ(η)\eta \mapsto \theta = \nabla\varphi(\eta), where φ(η)=supθ{θ,ηψ(θ)}\varphi(\eta) = \sup_\theta \{\langle\theta, \eta\rangle - \psi(\theta)\} is the Legendre transform of ψ\psi.
  3. The pair (ψ,φ)(\psi, \varphi) is a Legendre pair: ψ(θ)+φ(η)=θ,η\psi(\theta) + \varphi(\eta) = \langle\theta, \eta\rangle, and the Fisher metric in expectation coordinates is the Hessian of φ\varphi: gij(η)=ijφ(η)g^{ij}(\eta) = \partial^i \partial^j \varphi(\eta).

The log-partition function ψ\psi is strictly convex (as a generating function of cumulants), so its Hessian is indeed a metric. The natural and expectation parameters are related by a convex duality analogous to the Legendre duality in mechanics — a pattern that information geometry makes precise through the theory of dual connections.

Divergences and the KL Geometry

The Fisher metric arises naturally from a divergence — a measure of dissimilarity between distributions that is not necessarily symmetric.

[!definition] KL Divergence The Kullback-Leibler divergence from pp to qq is DKL(pq)=p(x)logp(x)q(x)dμ(x)0,D_{\mathrm{KL}}(p \| q) = \int p(x) \log \frac{p(x)}{q(x)}\, d\mu(x) \geq 0, with equality if and only if p=qp = q almost everywhere.

The KL divergence is not symmetric and does not satisfy the triangle inequality — it is not a distance. However, its second-order Taylor expansion near p=qp = q recovers the Fisher metric: for q=pθ+dθq = p_{\theta + d\theta} and p=pθp = p_\theta,

This shows that the Fisher metric is the “local divergence” of KL. More generally, any Bregman divergence DF(pq)=F(p)F(q)F(q),pqD_F(p \| q) = F(p) - F(q) - \langle \nabla F(q), p - q\rangle for a strictly convex FF induces a Riemannian metric (the Hessian of FF) by the same second-order expansion.

[!theorem] Pythagorean Relations In an exponential family, the KL divergence satisfies:

  1. Generalized Pythagorean theorem: if CC is an ee-flat submanifold (affine in θ\theta-coordinates) and p=argminqCDKL(pq)p^* = \arg\min_{q \in C} D_{\mathrm{KL}}(p \| q) is the mm-projection of pp onto CC, then for all qCq \in C: DKL(pq)=DKL(pp)+DKL(pq).D_{\mathrm{KL}}(p \| q) = D_{\mathrm{KL}}(p \| p^*) + D_{\mathrm{KL}}(p^* \| q).
  2. Dually, if CC is mm-flat (affine in η\eta-coordinates) and p=argminqCDKL(qp)p^* = \arg\min_{q \in C} D_{\mathrm{KL}}(q \| p) is the ee-projection, the same identity holds with reversed arguments.

This is the geometric content of the EM algorithm (Expectation-Maximization): the E-step is an mm-projection (updating sufficient statistics), the M-step is an ee-projection (fitting parameters). Each step decreases the KL divergence, and the interleaving of the two projections converges to the maximum likelihood estimate.

Dual Connections: the α\alpha-Geometry

The Fisher metric alone does not capture the full differential-geometric structure of a statistical manifold. The key enrichment is a pair of dual connections.

  • α=0\alpha = 0: the Levi-Civita connection (0)\nabla^{(0)} of the Fisher metric.
  • α=1\alpha = 1: the ee-connection (exponential connection) (e)\nabla^{(e)}.
  • α=1\alpha = -1: the mm-connection (mixture connection) (m)\nabla^{(m)}.

The α\alpha- and (α)(-\alpha)-connections are dual with respect to the Fisher metric gg: for any vector fields X,Y,ZX, Y, Z, Xg(Y,Z)=g(X(α)Y,Z)+g(Y,X(α)Z).Xg(Y, Z) = g(\nabla^{(\alpha)}_X Y, Z) + g(Y, \nabla^{(-\alpha)}_X Z). This duality is the central structure of information geometry. The pair (g,(α),(α))(g, \nabla^{(\alpha)}, \nabla^{(-\alpha)}) is called a dualistic structure on the statistical manifold.

[!theorem] Flatness of Exponential Families For an exponential family:

  1. The ee-connection (e)\nabla^{(e)} is flat, with globally affine coordinates θ\theta (natural parameters). The geodesics of (e)\nabla^{(e)} are straight lines in θ\theta-coordinates.
  2. The mm-connection (m)\nabla^{(m)} is flat, with globally affine coordinates η\eta (expectation parameters). The geodesics of (m)\nabla^{(m)} are straight lines in η\eta-coordinates.
  3. The natural and expectation coordinates are related by η=ψ(θ)\eta = \nabla\psi(\theta), and the Fisher metric is the Hessian: gij=ijψg_{ij} = \partial_i\partial_j\psi.

A manifold with a pair of dual flat connections is called dually flat. The geometry of dually flat spaces — with its Pythagorean theorems, divergences, and projection theorems — is the heart of information geometry as developed by Amari. Exponential families are the canonical example, but dually flat geometry appears also in optimal transport (see below) and in certain neural network parameter spaces.

Fisher-Rao Geodesics and Statistical Distance

The geodesics of the Levi-Civita connection (0)\nabla^{(0)} — the Fisher-Rao geodesics — define the natural notion of distance between distributions:

dFR(p,q)=infγ01gij(γ(t))γ˙i(t)γ˙j(t)dt.d_{\mathrm{FR}}(p, q) = \inf_\gamma \int_0^1 \sqrt{g_{ij}(\gamma(t))\,\dot\gamma^i(t)\,\dot\gamma^j(t)}\, dt.

For the family of Gaussian distributions N(μ,σ2)\mathcal{N}(\mu, \sigma^2) with coordinates (μ,σ)(\mu, \sigma), the Fisher metric is g=σ2(dμ2+2dσ2)g = \sigma^{-2}(d\mu^2 + 2\, d\sigma^2) — proportional to the Poincaré metric of the upper half-plane H2={(μ,σ):σ>0}\mathbb{H}^2 = \{(\mu, \sigma) : \sigma > 0\}. The statistical manifold of Gaussians is therefore a hyperbolic space of constant negative curvature 12-\frac{1}{2}.

More generally, the sectional curvatures of statistical manifolds encode geometric properties of the model. For mixtures of exponential families, curvature measures the interaction between components. Flat (α=±1\alpha = \pm 1) connections identify the most tractable families; curved geometries indicate harder inference problems.

The Cramér-Rao Bound as a Geometric Inequality

The most famous result in statistical estimation has a clean geometric interpretation.

[!theorem] Cramér-Rao Bound Let θ^(x)\hat\theta(x) be an unbiased estimator of θ\theta. Then the covariance matrix of θ^\hat\theta satisfies Cov(θ^)g(θ)1\mathrm{Cov}(\hat\theta) \geq g(\theta)^{-1} in the sense of positive semi-definite matrices, where g(θ)1g(\theta)^{-1} is the inverse of the Fisher information matrix.

Geometrically, the Cramér-Rao bound is a statement about the projection of the estimator onto the tangent space of the statistical manifold. The inequality becomes an equality — the estimator is efficient — if and only if the estimator is a function of a sufficient statistic, equivalently when the score function lies in the tangent space of the model. The efficiency of maximum likelihood estimators in regular models follows from this geometric picture.

The bound has a natural interpretation in terms of quantum mechanics: for a quantum system with density matrix ρ\rho, the quantum Fisher information F(ρ,A)=4Varρ(A)\mathcal{F}(\rho, A) = 4\, \mathrm{Var}_\rho(A) for a Hermitian observable AA satisfies a quantum Cramér-Rao bound, and the corresponding geometric structure on the space of density matrices is the Bures metric.

Natural Gradient and Optimization

A central application of information geometry to machine learning is the natural gradient, introduced by Amari (1998).

In standard gradient descent, the parameter update θθϵθL(θ)\theta \leftarrow \theta - \epsilon\, \nabla_\theta \mathcal{L}(\theta) treats all parameter directions equally. But the Fisher metric encodes the geometry of the model: moving distance ϵ\epsilon in different directions in θ\theta-space corresponds to very different changes in the distribution pθp_\theta. The natural gradient corrects for this by moving in the steepest-descent direction in the Fisher-Rao metric:

This is the natural gradient — the gradient of L\mathcal{L} with respect to the Fisher information metric rather than the Euclidean metric on parameter space. Natural gradient descent converges in O(1)O(1) steps (rather than O(κ)O(\kappa) steps, where κ\kappa is the condition number) for strongly convex losses in dually flat geometries. For neural networks, the natural gradient corresponds to the K-FAC (Kronecker-Factored Approximate Curvature) approximation to the Fisher matrix.

Information Geometry and Optimal Transport

A deep connection links information geometry to optimal transport (OT) — the subject of Chapter 9’s closing remarks on geometric quantization and, more broadly, of the companion volume on generative flows.

The Wasserstein distance W2W_2 between probability measures is the geodesic distance in the L2L^2-Wasserstein space (P2(X),W2)(\mathcal{P}_2(\mathcal{X}), W_2). The space (P2(Rn),W2)(\mathcal{P}_2(\mathbb{R}^n), W_2) is an infinite-dimensional Riemannian manifold (in the sense of Otto calculus): the tangent space at ρ\rho is the space of square-integrable functions modulo constants, the metric is s,tρ=stdρ\langle s, t\rangle_\rho = \int s\, t\, d\rho, and the geodesics are displacement interpolants (McCann interpolants).

The Fisher-Rao metric and the Wasserstein metric both live on spaces of probability distributions but measure different things:

  • Fisher-Rao: infinitesimal distinguishability via statistical experiments — sensitivity to perturbations at every point of X\mathcal{X}.
  • Wasserstein W2W_2: cost of transporting mass — sensitive to the geometry of X\mathcal{X} but not to pointwise likelihood.

These two metrics on the same space generate very different geometries. The spherical Hellinger distance dH(p,q)=pqL2d_H(p, q) = \|\sqrt{p} - \sqrt{q}\|_{L^2} is related to the Fisher-Rao geodesic distance by dFR=2arccos(1dH2/2)d_{\mathrm{FR}} = 2\arccos(1 - d_H^2/2), giving the sphere SS^\infty of square-root densities its natural geometry.

The de Bruijn identity links Fisher information to entropy along Gaussian convolution: ddtH(ρN(0,t))=12I(ρN(0,t))\frac{d}{dt} H(\rho * \mathcal{N}(0,t)) = \frac{1}{2} I(\rho * \mathcal{N}(0,t)), where I(ρ)=ρlogρ2dxI(\rho) = \int \rho |\nabla \log \rho|^2\, dx is the Fisher information functional. This is the bridge between score-based diffusion models and information geometry: the score function logρt\nabla \log \rho_t at each diffusion time tt is precisely the integrand of the Fisher information, and learning the score is equivalent to fitting the Fisher geometry of the diffused distribution.

Quantum Information Geometry

The geometry of quantum states parallels classical information geometry, with density matrices replacing probability distributions and the trace replacing integration.

The space of quantum states (density matrices) S={ρ:ρ0,trρ=1}\mathcal{S} = \{\rho : \rho \geq 0,\, \mathrm{tr}\,\rho = 1\} is a manifold. The quantum Fisher information metric (Bures metric) is: gijBures(ρ)=12tr(Liρ),g^{\mathrm{Bures}}_{ij}(\rho) = \tfrac{1}{2}\,\mathrm{tr}(L_i\, \rho), where LiL_i is the symmetric logarithmic derivative defined implicitly by iρ=12(Liρ+ρLi)\partial_i \rho = \frac{1}{2}(L_i \rho + \rho L_i).

The Bures metric generates the quantum Fisher information, which governs the precision of quantum parameter estimation via the quantum Cramér-Rao bound Cov(θ^)F(ρ)1\mathrm{Cov}(\hat\theta) \geq \mathcal{F}(\rho)^{-1}. This bound is saturated by projective measurements in the eigenbasis of the symmetric logarithmic derivative — the optimal measurement in quantum metrology.

Quantum information geometry also admits α\alpha-connections (Petz’s monotone metrics), dual connections, and a quantum analogue of the exponential family (the ee-family of Gibbs states ρθeiθiAi\rho_\theta \propto e^{\sum_i \theta^i A_i}). The quantum-classical correspondence is deepest for commuting observables, where quantum information geometry reduces to its classical counterpart.

Geometry of Neural Networks and Generative Models

Information geometry provides a natural language for modern machine learning.

A deep neural network pθ(yx)p_\theta(y|x) defines a statistical manifold parameterized by weights θRd\theta \in \mathbb{R}^d. The Fisher information matrix of this model is g(θ)=Ex,ypθ[θlogpθ(yx)θlogpθ(yx)]g(\theta) = \mathbb{E}_{x, y \sim p_\theta}[\nabla_\theta \log p_\theta(y|x) \otimes \nabla_\theta \log p_\theta(y|x)]. Since dd may be tens of billions, computing gg exactly is infeasible; K-FAC and related methods approximate it block-diagonally by Kronecker products.

The information-geometric view of generative models is particularly illuminating:

  • Normalizing flows parameterize a push-forward of a base measure through a diffeomorphism fθf_\theta. The Fisher metric on the flow manifold pulls back the Wasserstein metric on the distribution manifold via the push-forward map.
  • Score-based diffusion learns the score function logpt\nabla \log p_t, which is precisely the natural gradient of the entropy H(ρt)H(\rho_t) with respect to the Fisher metric — connecting training objectives to geodesics in the space of distributions.
  • Variational inference minimizes DKL(qϕp)D_{\mathrm{KL}}(q_\phi \| p) over a variational family {qϕ}\{q_\phi\}; the optimal qq^* is the mm-projection of pp onto the family. The ELBO gradient is proportional to the natural gradient on the variational manifold.
  • GAN training corresponds (in idealized form) to minimizing a divergence between the model and data distributions, where the choice of divergence determines the geometry. The Wasserstein GAN uses a geometry closer to W1W_1 than KL; f-GANs correspond to ff-divergences with different local geometries.

Closing: The Unity of Differential Geometry

This chapter, and this series, ends at the intersection of differential geometry and modern science. The ten chapters have built a single coherent edifice: smooth manifolds provide the stage; tangent bundles and differential forms give the calculus; Riemannian metrics introduce measurement; connections and parallel transport provide the machinery for comparing across fibers; curvature measures the failure of flatness; Lie groups and symmetric spaces encode symmetry; fiber bundles unify local and global structure; symplectic geometry governs Hamiltonian dynamics; and information geometry applies the entire apparatus to the space of probability distributions.

The unity is not accidental. Probability distributions are smooth objects — they live on manifolds, their spaces have curvature, their symmetries form Lie groups, their dynamics satisfy Hamiltonian equations (in the Wasserstein sense), and their estimation theory is a chapter of Riemannian geometry. The tools developed for pure geometry turn out to be precisely those needed for modern statistics, machine learning, and physics. The direction of transfer also runs the other way: Fisher information, KL divergence, and quantum states have generated new geometric structures — dually flat spaces, α\alpha-connections, Bures geometry — that enrich the discipline from which they emerged.

Differential geometry is not background material for applications. It is the native language in which the applications are most clearly written.


Information geometry was founded by C.R. Rao (1945), who introduced the Fisher-Rao metric, and developed into a systematic theory by Shun-ichi Amari, whose book Methods of Information Geometry (with Nagaoka, 2000) remains the standard reference. The dual connection structure was clarified by Amari and Nagaoka; the dually flat geometry of exponential families by Efron (1975) and Amari (1982). The connection between Wasserstein gradient flows and Fokker-Planck equations is due to Jordan, Kinderlehrer, and Otto (1998). Quantum information geometry was systematically developed by Petz (1996). The application of natural gradient to neural network optimization is in Amari (1998); the K-FAC approximation in Martens and Grosse (2015). The neural tangent kernel is introduced in Jacot, Gabriel, and Hongler (2018).