One-Layer Transformer Provably Learns Multiclass One-Nearest Neighbor in Context

arXiv:2609.01311 · cs.LG, stat.ML · Submitted 2026-09-01 · Read on arXiv

cs.LG, stat.ML

Submitted: 2026-09-01

Updated: 2026-09-01

Code: https://github.com/skandaka/onenn-multiclass

Terminology

• Accuracy — 22× in this paper · explained in 1 episode
• Transformer — 22× in this paper · explained in 2 episodes
• Distribution Shift — 6× in this paper · explained in 6 episodes
• In-Context Learning — 4× in this paper · explained in 4 episodes
• Generalization — 3× in this paper · explained in 13 episodes
• International Conference on Machine Learning — 3× in this paper
• Loss Function — 3× in this paper · explained in 2 episodes
• Adam — 2× in this paper · explained in 2 episodes
• Alignment — 2× in this paper · explained in 1 episode
• Ablation — 1× in this paper · explained in 1 episode
• Ambient Dimension — 1× in this paper · explained in 1 episode
• Consistency — 1× in this paper · explained in 3 episodes
• Core — 1× in this paper · explained in 2 episodes
• In-Context Learning (ICL) — 1× in this paper · explained in 5 episodes
• Representation Learning — 1× in this paper · explained in 2 episodes
• Robustness — 1× in this paper · explained in 5 episodes
• Trace — 1× in this paper · explained in 2 episodes

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