English

Information-Geometric Decomposition of Generalization Error in Unsupervised Learning

Machine Learning 2026-04-15 v1 Statistical Mechanics Information Theory Machine Learning math.IT Statistics Theory Statistics Theory

Abstract

We decompose the Kullback--Leibler generalization error (GE) -- the expected KL divergence from the data distribution to the trained model -- of unsupervised learning into three non-negative components: model error, data bias, and variance. The decomposition is exact for any e-flat model class and follows from two identities of information geometry: the generalized Pythagorean theorem and a dual e-mixture variance identity. As an analytically tractable demonstration, we apply the framework to ϵ\epsilon-PCA, a regularized principal component analysis in which the empirical covariance is truncated at rank NKN_K and discarded directions are pinned at a fixed noise floor ϵ\epsilon. Although rank-constrained ϵ\epsilon-PCA is not itself e-flat, it admits a technical reformulation with the same total GE on isotropic Gaussian data, under which each component of the decomposition takes closed form. The optimal rank emerges as the cutoff λcut=ϵ\lambda_{\mathrm{cut}}^{*} = \epsilon -- the model retains exactly those empirical eigenvalues exceeding the noise floor -- with the cutoff reflecting a marginal-rate balance between model-error gain and data-bias cost. A boundary comparison further yields a three-regime phase diagram -- retain-all, interior, and collapse -- separated by the lower Marchenko--Pastur edge and an analytically computable collapse threshold ϵ(α)\epsilon_{*}(\alpha), where α\alpha is the dimension-to-sample-size ratio. All claims are verified numerically.

Keywords

Cite

@article{arxiv.2604.12340,
  title  = {Information-Geometric Decomposition of Generalization Error in Unsupervised Learning},
  author = {Gilhan Kim},
  journal= {arXiv preprint arXiv:2604.12340},
  year   = {2026}
}

Comments

21 pages, 3 figures

R2 v1 2026-07-01T12:08:04.903Z