Does Compression Imply Generalization? A Minimum Description Length Perspective
Authors: Lincen Yang, Jiayang Shi, Nan Pu, Daniël M. Pelt, Matthijs van Leeuwen, Zhong Li, Shujian Yu
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 · Department of Computer Science, Vrije Universiteit Amsterdam, The Netherlands · Department of Physics and Technology, UiT The Arctic University of Norway
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
D1∑l=1DIˇ(X;Zls)
0.8481
0.7410
0.2116
0.1831
0.6425
0.5436
minl∈[D]Iˇ(X;Zls)
0.7145
0.5602
0.7203
0.5719
0.4461
0.3404
D1∑l=1DIˇ(X;Zls∣Y)
0.8481
0.7406
0.2140
0.1853
0.6427
0.5435
minl∈[D]Iˇ(X;Zls∣Y)
0.7004
0.5434
0.7062
0.5560
0.4386
0.3305
Iˇ(S;θD+1S)
0.4688
0.3112
0.2512
0.0775
0.2121
0.1208
Table 1: Correlations between generalization-bound quantities and the empirical generalization gap on CIFAR-10, measured both in loss and in classification error. Notation follows the previous work on IB generalization bounds [ 25 ] . We write S for the training sample, D for the network depth, Iˇ for a finite-sample mutual-information estimator, Zls for the activations at layer l on data subsample s , θlS for the parameters at layer l trained on S , and θD+1S for the full network parameter trained on S . The baseline Iˇ(S;θD+1S) is the parameter-information quantity I(S;W) of the information-theoretic line [ 33 , 42 ] and is the only pure model-complexity baseline besides ours. Best results are bolded.
Figure 1: Correlation between the prequential regret and the generalization gaps on CIFAR-100.