mlsanityData to 14 Oct 2026

Kan Extension Transformers: A Categorical Unification of Attention, Diffusion, and Predict-Detach Self-Conditioning

arXiv 2605.27259 ↗First posted 26 May 2026cs.LGcited by 3

Not on the map. Why?

Abstract (arXiv)

We propose Kan Extension Transformers (KETs) as a categorical design language for a diverse group of Transformer implementations. A layer can be viewed generally as a weighted structured extension operator: attention uses token neighborhoods, geometric mixing uses sparse incidences, and KET uses simplicial sources. This operator is an actual enriched left Kan extension only when the source values are functorial, the weights are representable hom-objects (or the specified profunctor action), and aggregation realizes the corresponding coend; otherwise ``Kan-style'' denotes an interpretation rather than an identity theorem. Predict-detach blocks gradients through a predictive carrier and avoids transporting teacher-forced hidden states, but detach alone does not make a noncausal update strictly autoregressive: every carrier consumed at target t must also be measurable from the prefix available at t. We evaluate 12 implementations on Penn Treebank, WikiText-2, and WikiText-103 across strict-causal and self-conditioned regimes, using widths d=64,256 and depths L=2,8,16 across the reported studies. Quadratic KET is strongest among the compared strict-causal architectures on WikiText-2 and WikiText-103; the largest cross-regime gains arise from additional self-conditioning information, not neighborhood design alone.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 13 of its 21 references that are in the map. First posted on arXiv 26 May 2026.

Builds on

13 of 21 references.

References · 13

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2022Scalable Diffusion Models with Transformerscited by 6,054
  4. 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
  5. 2022Neural Sheaf Diffusion: A Topological Perspective on Heterophily and Oversmoothing in GNNscited by 214
  6. 2022Topological Deep Learning: Going Beyond Graph Datacited by 97
  7. 2022Sheaf Neural Networks with Connection Laplacianscited by 65
  8. 2022Simplicial Attention Networkscited by 41
  9. 2022HEGEL: Hypergraph Transformer for Long Document Summarizationcited by 28
  10. 2023Generalized Simplicial Attention Neural Networkscited by 27
  11. 2025Copresheaf Topological Neural Networks: A Generalized Deep Learning Frameworkcited by 12
  12. 2025Large Causal Models from Large Language Modelscited by 8
1 more
  1. 2026PROMETHEUS: Automating Deep Causal Research Integrating Text, Data and Modelscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.