Search papers, labs, and topics across Lattice.
This paper frames LLM pre-training as lossy compression, positing that models learn by retaining only information relevant to their objectives. The authors demonstrate that pre-trained LLMs approach the Information Bottleneck bound for next-sequence prediction, indicating optimal compression. They further show that the optimality of a model's compression correlates with its downstream performance across various benchmarks, providing a direct link between representational structure and model capabilities.
LLM performance isn't just about size, but about how efficiently they compress information during training, offering a new lens for understanding and predicting model capabilities.
Despite the increasing prevalence of large language models (LLMs), we still have a limited understanding of how their representational spaces are structured. This limits our ability to interpret how and what they learn or relate them to learning in humans. We argue LLMs are best seen as an instance of lossy compression, where over training they learn by retaining only information in their training data relevant to their objective(s). We show pre-training results in models that are optimally compressed for next-sequence prediction, approaching the Information Bottleneck bound on compression. Across an array of open weights models, each compresses differently, likely due to differences in the data and training recipes used. However even across different families of LLMs the optimality of a model's compression, and the information present in it, can predict downstream performance on across a wide array of benchmarks, letting us directly link representational structure to actionable insights about model performance. In the general case the work presented here offers a unified Information-Theoretic framing for how these models learn that is deployable at scale.