Width-Independent Compressibility of Deep Neural Networks
Organizations: Princeton University Princeton, NJ 08544 · Peking University Beijing, China 100871 · Massachusetts Institute of Technology Cambridge, MA 02139
Abstract
It has long been known that well-trained neural networks can be compressed very strongly without affecting their performance, an important phenomenon that remains poorly understood. We prove a uniform compressibility theorem for deep multilayer perceptrons with analytic activations. For a deep, wide fixed teacher network, there exists a narrow (same depth) network that approximately represents the same function as the original. The reachable compressed width is strikingly independent of the original width, but is , where is the error budget and is the effective input dimension. Our construction involves a novel derivative-matching technique which is aware of the low-dimensional input, and a layer-wise reweighting that preserves the input-output mapping.
Figures & tables
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Seed | Teacher MSE | First task-preserving (reduction) | ||
|---|---|---|---|---|
| 0 | 1.2632 | 7.87 | 66 ( ) | |
| 1 | 1.2829 | 7.00 | 21 ( ) | |
| 2 | 1.0715 | 14.46 | 36 ( ) | |
| 3 | 1.2011 | 8.32 | 28 ( ) | |
| 4 | 1.0520 | 13.81 | 21 ( ) |