Leaner Transformers Can Easily Learn to Cluster
cs.LG, stat.ML
Submitted: 2026-10-07
Updated: 2026-10-07
Code: https://github.com/rithram/kmeans-trf
Terminology
Sources
- Formal Algorithms for Transformers
- Neural Turing Machines
- In-Context Convergence of Transformers
- Trained Transformers Learn Linear Models In-Context
- Train faster, generalize better: Stability of stochastic gradient descent
- The emergence of clusters in self-attention dynamics
- A mathematical perspective on Transformers
- Projection onto the probability simplex: An efficient algorithm with a simple proof, and an application
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: Local-Consistency Optimisation for Bayesian Neural Networks
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks