Does Compression Imply Generalization? A Minimum Description Length Perspective
Organizations: Leiden Institute of Advanced Computer Science, Leiden University, The Netherlands · Centrum Wiskunde & Informatica (CWI), Amsterdam, The Netherlands · Hefei University of Technology, Hefei, China · Great Bay University, Dongguan, China ·
Abstract
A central question in modern learning theory is what notion of compression, if any, explains why over-parameterized neural networks generalize. Existing answers measure complexity at different levels, ranging from weights to representations; however, estimating the associated quantities either requires held-out data in principle or relies on the challenging (ultra) high-dimensional mutual information estimation. To overcome these challenges, we approach the question through the minimum description length (MDL) principle, and specifically, we propose the prequential regret as a complementary signal that is computable from the training trajectory alone. We establish that, in the (nearly) interpolating regime, the expected prequential regret upper-bounds the description length of the trained parameter under the algorithm-induced distribution, and that this description length translates into a high-probability generalization bound in the PAC-MDL setting. Along the way, we show that the encoder-complexity correction recently added to the algorithm-level information bottleneck (IB) is upper-bounded by an MDL parameter description length, \new{bringing together} the two views of compression from MDL and IB. Empirically, the prequential regret tracks the generalization error on CIFAR-10 and CIFAR-100, with Pearson correlations exceeding 0.9.
Figures & tables
| Spearman corr. | Pearson corr. | Kendall corr. | ||||
|---|---|---|---|---|---|---|
| Generalization gap: | Loss | Error | Loss | Error | Loss | Error |
| 0.8481 | 0.7410 | 0.2116 | 0.1831 | 0.6425 | 0.5436 | |
| 0.7145 | 0.5602 | 0.7203 | 0.5719 | 0.4461 | 0.3404 | |
| 0.8481 | 0.7406 | 0.2140 | 0.1853 | 0.6427 | 0.5435 | |
| 0.7004 | 0.5434 | 0.7062 | 0.5560 | 0.4386 | 0.3305 | |
| 0.4688 | 0.3112 | 0.2512 | 0.0775 | 0.2121 | 0.1208 | |
Neural Network Generalization
Neural network generalization studies why trained models perform well on unseen data and how architecture, data, and training shape that ability.
See all 299 papers