Reasoning About Generalization via Conditional Mutual Information
Abstract
We provide an information-theoretic framework for studying the generalization properties of machine learning algorithms. Our framework ties together existing approaches, including uniform convergence bounds and recent methods for adaptive data analysis. Specifically, we use Conditional Mutual Information (CMI) to quantify how well the input (i.e., the training data) can be recognized given the output (i.e., the trained model) of the learning algorithm. We show that bounds on CMI can be obtained from VC dimension, compression schemes, differential privacy, and other methods. We then show that bounded CMI implies various forms of generalization.
- Publication:
-
arXiv e-prints
- Pub Date:
- January 2020
- DOI:
- 10.48550/arXiv.2001.09122
- arXiv:
- arXiv:2001.09122
- Bibcode:
- 2020arXiv200109122S
- Keywords:
-
- Computer Science - Machine Learning;
- Computer Science - Cryptography and Security;
- Computer Science - Data Structures and Algorithms;
- Computer Science - Information Theory;
- Statistics - Machine Learning
- E-Print:
- 58 pages. Changes from previous version: Added discussion on related work and updated references. Simplified part of the proof of Theorem 4.10