mlsanityData to 14 Oct 2026

Attention Function as an Intrinsic Inductive Bias: How Models' Behavior Diverges in Novel Contexts

arXiv 2609.39188 ↗First posted 30 Sep 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Developmental psychology holds that certain priors are given to infants prior to experience rather than induced from data, and that the influence of such priors is suppressed under strong, well-constrained conditions but reasserts itself under weak ones. We ask whether an analogous principle holds for the Transformer: can the activation function given to attention heads serve as an intrinsic inductive bias? We propose Mixture of Function Attention (MoFA), a parameter-free modification to multi-head attention that fixes a ratio of softmax and sigmoid heads before training. Across five ratios, a 124M-parameter GPT-2 model, and five seeds, we find that this given ratio has little effect in-distribution -- differences between ratios are statistically negligible for moderate mixtures and remain small even at the extremes -- but its influence re-emerges sharply under zero-shot distribution shift across 15 out-of-distribution domains. Perplexity gaps between ratios widen by more than an order of magnitude on several domains, and the best-performing ratio tracks a single axis of domain structure, separating short, informal text (softmax-favoring) from technical, long-form text (sigmoid-favoring), that explains 78.3% of the variance in domain response. This reorganization is visible at the head level: sigmoid heads show an accelerating drop in attention entropy as their ratio increases, while softmax heads respond more modestly, yielding a consistent division of labor between the two head types. Our results suggest that activation choice functions as a given prior whose influence is masked in-distribution and re-emerges out-of-distribution.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 33 references that are in the map. First posted on arXiv 30 Sep 2026.

Builds on

25 of 33 references.

References · 25

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  3. 2016Pointer Sentinel Mixture Modelscited by 3,096
  4. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  5. 2017Get To The Point: Summarization with Pointer-Generator Networkscited by 1,292
  6. 2015Aligning Books and Movies: Towards Story-like Visual Explanations by Watching Movies and Reading Bookscited by 1,043
  7. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  8. 2019CodeSearchNet Challenge: Evaluating the State of Semantic Code Searchcited by 654
  9. 2017DailyDialog: A Manually Labelled Multi-turn Dialogue Datasetcited by 652
  10. 2018A Discourse-Aware Attention Model for Abstractive Summarization of Long Documentscited by 478
  11. 2020TweetEval: Unified Benchmark and Comparative Evaluation for Tweet Classificationcited by 417
  12. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
13 more
  1. 2019Transformer Dissection: A Unified Understanding of Transformer's Attention via the Lens of Kernelcited by 213
  2. 2019BillSum: A Corpus for Automatic Summarization of US Legislationcited by 117
  3. 2018Abstractive Summarization of Reddit Posts with Multi-level Memory Networkscited by 102
  4. 2024Theory, Analysis, and Best Practices for Sigmoid Self-Attentioncited by 64
  5. 2023Transformers Implement Functional Gradient Descent to Learn Non-Linear Functions In Contextcited by 63
  6. 2023Replacing softmax with ReLU in Vision Transformerscited by 44
  7. 2024MoH: Multi-Head Attention as Mixture-of-Head Attentioncited by 44
  8. 2021Transformers are Deep Infinite-Dimensional Non-Mercer Binary Kernel Machinescited by 22
  9. 2024Attention layers provably solve single-location regressioncited by 17
  10. 2023Spotlight Attention: Robust Object-Centric Learning With a Spatial Locality Priorcited by 8
  11. 2024Rethinking Attention: Polynomial Alternatives to Softmax in Transformerscited by 8
  12. 2023Infusing Lattice Symmetry Priors in Attention Mechanisms for Sample-Efficient Abstract Geometric Reasoningcited by 4
  13. 2026HydraHead: From Head-Level Functional Heterogeneity to Specialized Attention Hybridizationcited by 4

Built on by

Not cited yet.