A universal compression theory: Lottery ticket hypothesis and superpolynomial scaling laws
This provides a theoretical foundation for drastically reducing model and data sizes in machine learning, potentially impacting efficiency and scalability across AI applications.
The paper proves that generic permutation-invariant functions can be compressed to polylogarithmic size with vanishing error, enabling neural networks to be compressed to polylogarithmic width while preserving learning dynamics and datasets to polylogarithmic size while maintaining loss landscapes.
When training large-scale models, the performance typically scales with the number of parameters and the dataset size according to a slow power law. A fundamental theoretical and practical question is whether comparable performance can be achieved with significantly smaller models and substantially less data. In this work, we provide a positive and constructive answer. We prove that a generic permutation-invariant function of $d$ objects can be asymptotically compressed into a function of $\operatorname{polylog} d$ objects with vanishing error. This theorem yields two key implications: (Ia) a large neural network can be compressed to polylogarithmic width while preserving its learning dynamics; (Ib) a large dataset can be compressed to polylogarithmic size while leaving the loss landscape of the corresponding model unchanged. (Ia) directly establishes a proof of the \textit{dynamical} lottery ticket hypothesis, which states that any ordinary network can be strongly compressed such that the learning dynamics and result remain unchanged. (Ib) shows that a neural scaling law of the form $L\sim d^{-α}$ can be boosted to an arbitrarily fast power law decay, and ultimately to $\exp(-α' \sqrt[m]{d})$.