Contextures: Representations from Contexts
Runtian Zhai, Kai Yang, Burak Varici, Che-Ping Tsai, J. Zico Kolter, Pradeep Kumar Ravikumar
Abstract
Despite the empirical success of foundation models, we do not have a systematic characterization of the representations that these models learn. In this paper, we establish the contexture theory. It shows that a large class of representation learning methods can be characterized as learning from the association between the input and a context variable. Specifically, we show that many popular methods aim to approximate the top-d singular functions of the expectation operator induced by the context, in which case we say that the representation learns the contexture. We demonstrate the generality of the contexture theory by proving that representation learning within various learning paradigms-supervised, self-supervised, and manifold learning-can all be studied from such a perspective. We also prove that the representations that learn the contexture are optimal on those tasks that are compatible with the context. One important implication of the contexture theory is that once the model is large enough to approximate the top singular functions, further scaling up the model size yields diminishing returns. Therefore, scaling is not all we need, and further improvement requires better contexts. To this end, we study how to evaluate the usefulness of a context without knowing the downstream tasks. We propose a metric and show by experiments that it correlates well with the actual performance of the encoder on many real datasets.
Ask Lune
Ask anything. The answer quotes the paper.
Lune reads all of it and the papers around it, then shows you the passages behind every answer.
Builds on18
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- data2vec: A General Framework for Self-supervised Learning in Speech, Vision and LanguageAlexei Baevski, Wei-Ning Hsu, Qiantong Xu, Arun Babu et al.ICML 2022 · 1,123 citations
- Understanding Dimensional Collapse in Contrastive Self-supervised LearningLi Jing, Pascal Vincent, Yann LeCun, Yuandong TianICLR 2022 · 467 citations
- Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive LossJeff Z. HaoChen, Colin Wei, Adrien Gaidon, Tengyu MaNeurIPS 2021 · 425 citations
Related papers
- Towards Robust Metrics for Concept Representation EvaluationMateo Espinosa Zarlenga, Pietro Barbiero, Zohreh Shams, Dmitry Kazhdan et al.AAAI 2023 · 32 citations
- On the Power of Foundation ModelsYang YuanICML 2023 · 55 citations
- The Trade-off between Universality and Label Efficiency of Representations from Contrastive LearningZhenmei Shi, Jiefeng Chen, Kunyang Li, Jayaram Raghuram et al.ICLR 2023 · 1 citation
- Efficient Distribution Matching of Representations via Noise-Injected Deep InfoMaxIvan Butakov, Alexander Semenenko, Alexander Tolmachev, Andrey Gladkov et al.ICLR 2025
- A theoretical study of inductive biases in contrastive learningJeff Z. HaoChen, Tengyu MaICLR 2023 · 2 citations
