Understanding the Parameter Space Geometry of Transformers Encoding Boolean Functions
Blanka Kövér, Alexandra Butoi, Anej Svete, Michael Hahn, Ryan Cotterell
Abstract
Transformers consistently fail to learn certain simple functions that are provably expressible with specific parameter settings. This gap between learnability and expressivity is particularly prominent for sensitive functions---functions whose output is likely to change if a single bit of the input is flipped---for example, Parity. While prior work has established that transformers exhibit a bias toward functions with low average sensitivity, the precise mechanism underlying this bias remains poorly understood. To shed light on this phenomenon, we study the geometry of transformers' parameter space. We show that sensitive functions---even when representable---occupy a vanishingly small region that random initialization is very likely to miss. Specifically, we shift the focus from average sensitivity to the full sensitivity profile---the distribution of sensitivity values across all inputs---and prove that randomly initialized transformers almost surely compute functions which have low-sensitivity strings. Consequently, any function that lacks such strings is provably unlearnable.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0ae140ac-1393-4c5b-b1c4-252e72fea6a9Builds on6
- Generalization on the Unseen, Logic Reasoning and Degree CurriculumEmmanuel Abbe, Samy Bengio, Aryo Lotfi, Kevin RizkICML 2023 · 68 citations
- How Uniform Random Weights Induce Non-uniform Bias: Typical Interpolating Neural Networks Generalize with Narrow TeachersGon Buzaglo, Itamar Harel, Mor Shpigel Nacson, Alon Brutzkus et al.ICML 2024 · 11 citations
- Simplicity Bias in Transformers and their Ability to Learn Sparse Boolean FunctionsSatwik Bhattamishra, Arkil Patel, Varun Kanade, Phil BlunsomACL 2023 · 7 citations
- Why are Sensitive Functions Hard for Transformers?Michael Hahn, Mark RofinACL 2024 · 3 citations
- Overcoming a Theoretical Limitation of Self-AttentionDavid Chiang, Peter CholakACL 2022
Related papers
- Trapped by simplicity: When Transformers fail to learn from noisy featuresEvan Peters, Matheus Hrabowec Zambianco, Ando Deng, Devin Blankespoor et al.ICLR 2026
- Probability Distributions Computed by Autoregressive TransformersAndy Yang, Anej Svete, Jiaoda Li, Anthony W. Lin et al.ICLR 2026 · 2 citations
- Towards Understanding Inductive Bias in Transformers: A View From InfinityItay Lavie, Guy Gur-Ari, Zohar RingelICML 2024 · 11 citations
- On Expressive Power of Floating-Point TransformersSejun Park, Yeachan Park, Geonho HwangICML 2026 · 2 citations
- Transformers Learn Low Sensitivity Functions: Investigations and ImplicationsBhavya Vasudeva, Deqing Fu, Tianyi Zhou, Elliott Kau et al.ICLR 2025
