stat.MLMar 23, 2026

Overfitting and Generalizing with (PAC) Bayesian Prediction in Noisy Binary Classification

Authors: Xiaohan Zhu, Mesrob I. Ohannessian, Nathan Srebro

Organizations: The University of Chicago · The University of Illinois at Chicago · Toyota Technological Institute at Chicago

Abstract

We consider a PAC-Bayes type learning rule for binary classification, balancing the training error of a randomized ''posterior'' predictor with its KL divergence to a pre-specified ''prior''. This can be seen as an extension of a modified two-part-code Minimum Description Length (MDL) learning rule, to continuous priors and randomized predictions. With a balancing parameter of λ=1λ=1 this learning rule recovers an (empirical) Bayes posterior and a modified variant recovers the profile posterior, linking with standard Bayesian prediction (up to the treatment of the single-parameter noise level). However, from a risk-minimization prediction perspective, this Bayesian predictor overfits and can lead to non-vanishing excess loss in the agnostic case. Instead a choice of λ≫1λ\gg 1, which can be seen as using a sample-size-dependent-prior, ensures uniformly vanishing excess loss even in the agnostic case. We precisely characterize the effect of under-regularizing (and over-regularizing) as a function of the balance parameter λλ, understanding the regimes in which this under-regularization is tempered or catastrophic. This work extends previous work by Zhu and Srebro [2025] that considered only discrete priors to PAC Bayes type learning rules and, through their rigorous Bayesian interpretation, to Bayesian prediction more generally.

Figures & tables

Explore similar work

CardsList
  1. PAC-Bayes Beyond Parameter Space: Behavioral Equivalence, Z-Information, and Exact Complexity Decomposition

    Aug 11, 2026Vasant G. Honavar, Satish Kumar Keshri, Neil Ashtekar +1PAC-Bayesian Generalization BoundsPAC-Bayesian Theory

  2. PAC-Bayes Bounds for Gibbs Posteriors via Singular Learning Theory

    Apr 19, 2026Chenyang Wang, Yun YangNeural Network GeneralizationPAC-Bayesian Generalization Bounds

  3. PAC--Bayes Bounds on Quotient Parameter Spaces: Geometry-induced Implicit-Bias Priors

    Jul 20, 2026Nicola Aladrah, Fabio AnselmiPAC-Bayesian Generalization BoundsPAC-Bayesian Theory