Generalization Bounds: Perspectives from Information Theory and PAC-Bayes

TL;DR

This paper unifies information-theoretic and PAC-Bayesian perspectives on generalization in machine learning, highlighting their connections, differences, and applications, especially in deep learning.

Contribution

It provides a comprehensive, unified framework linking PAC-Bayesian and information-theoretic bounds, with insights into their common structures and implications for deep neural networks.

Findings

Unified treatment of PAC-Bayesian and information-theoretic bounds

Analysis of the conditional mutual information framework

Application of methods to deep learning models

Abstract

A fundamental question in theoretical machine learning is generalization. Over the past decades, the PAC-Bayesian approach has been established as a flexible framework to address the generalization capabilities of machine learning algorithms, and design new ones. Recently, it has garnered increased interest due to its potential applicability for a variety of learning algorithms, including deep neural networks. In parallel, an information-theoretic view of generalization has developed, wherein the relation between generalization and various information measures has been established. This framework is intimately connected to the PAC-Bayesian approach, and a number of results have been independently discovered in both strands. In this monograph, we highlight this strong connection and present a unified treatment of PAC-Bayesian and information-theoretic generalization bounds. We present techniques and results that the two perspectives have in common, and discuss the approaches and interpretations that differ. In particular, we demonstrate how many proofs in the area share a modular structure, through which the underlying ideas can be intuited. We pay special attention to the conditional mutual information (CMI) framework; analytical studies of the information complexity of learning algorithms; and the application of the proposed methods to deep learning. This monograph is intended to provide a comprehensive introduction to information-theoretic generalization bounds and their connection to PAC-Bayes, serving as a foundation from which the most recent developments are accessible. It is aimed broadly towards researchers with an interest in generalization and theoretical machine learning.