Towards a learning theory of representation alignment
Francesco Insulla, Shuo Huang, Lorenzo Rosasco
Abstract
It has recently been argued that AI models' representations are becoming aligned as their scale and performance increase. Empirical analyses have been designed to support this idea and conjecture the possible alignment of different representations toward a shared statistical model of reality. In this paper, we propose a learning-theoretic perspective to representation alignment. First, we review and connect different notions of alignment based on metric, probabilistic, and spectral ideas. Then, we focus on stitching, a particular approach to understanding the interplay between different representations in the context of a task. Our main contribution here is relating properties of stitching to the kernel alignment of the underlying representation. Our results can be seen as a first step toward casting representation alignment as a learning-theoretic problem.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers2
- Text-to-3D by Stitching a Multi-view Reconstruction Network to a Video GeneratorHyojun Go, Dominik Narnhofer, Goutam Bhat, Prune Truong et al.ICLR 2026 · 9 citations
- Training the Untrainable: Introducing Inductive Bias via Representational AlignmentVighnesh Subramaniam, David Mayo, Colin Conwell, Tomaso A. Poggio et al.NeurIPS 2025 · 5 citations
Builds on8
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- FLAVA: A Foundational Language And Vision Alignment ModelAmanpreet Singh, Ronghang Hu, Vedanuj Goswami, Guillaume Couairon et al.CVPR 2022 · 483 citations
- Deep learning versus kernel learning: an empirical study of loss landscape geometry and the time evolution of the Neural Tangent KernelStanislav Fort, Gintare Karolina Dziugaite, Mansheej Paul, Sepideh Kharaghani et al.NeurIPS 2020 · 255 citations
- Revisiting Model Stitching to Compare Neural RepresentationsYamini Bansal, Preetum Nakkiran, Boaz BarakNeurIPS 2021 · 253 citations
- Neural Networks as Kernel Learners: The Silent Alignment EffectAlexander B. Atanasov, Blake Bordelon, Cengiz PehlevanICLR 2022 · 110 citations
Related papers
- Functional Alignment Can Mislead: Examining Model StitchingDamian Smith, Harvey Mannering, Antonia MarcuICML 2025
- On the Functional Similarity of Robust and Non-Robust Neural RepresentationsAndrás Balogh, Márk JelasityICML 2023 · 4 citations
- Latent Functional Maps: a spectral framework for representation alignmentMarco Fumero, Marco Pegoraro, Valentino Maiorca, Francesco Locatello et al.NeurIPS 2024 · 2 citations
- How Not to Stitch Representations to Measure Similarity: Task Loss Matching Versus Direct MatchingAndrás Balogh, Márk JelasityAAAI 2025 · 3 citations
- The Indra Representation Hypothesis for Multimodal AlignmentJianglin Lu, Hailing Wang, Kuo Yang, Yitian Zhang et al.NeurIPS 2025 · 8 citations
