Kolmogorov--Arnold Networks for Small Language Models
Not on the map. Why?
Abstract (arXiv)
Kolmogorov--Arnold Networks (KANs) replace fixed node activations with learned one-dimensional edge functions, offering an explicit interface for interpretation and a possible alternative to transformer feed-forward networks. We test these claims separately. In a six-layer, 10M-parameter B-spline KAN, we reconstruct all 884,736 feed-forward edges: 87.8% exceed (NLS>0.1) and 0.4% are inactive. Pruning the lowest-activity 20--25% causes negligible loss increase, although structured MLP neuron pruning tolerates comparable sparsity. The audit replicates on BabyLM, but grid-size sweeps show that near-total fPCA compression and high closed-form-fit coverage are properties of the low-capacity grid-2 basis, not universal KAN behavior. For replacement, we evaluate MLP, SwiGLU, grouped Chebyshev, and rational GR-KAN networks on BabyLM. The KAN-family and gated variants improve validation loss over the GELU MLP, but this ordering does not transfer to standardized benchmarks: across ten seeds and 59,875 BLiMP pairs, accuracies span 62.4--63.1%, EWoK remains at chance, and a (+0.7)-point GR-KAN effect on BLiMP reverses on the supplement. Larger tests are also cautionary: parameter-matched MLPEdge underperforms the MLP on Wikitext-103, and 286M-parameter GR-KAN remains below a SwiGLU ClimbMix baseline after stabilization. Thus, small-basis KANs provide a practical, corpus-transferable interface for auditing learned scalar transformations, but the tested replacements show no consistent benchmark, quality, or latency advantage over strong MLP baselines.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 31 references that are in the map. First posted on arXiv 17 Jul 2026.
Builds on
22 of 31 references.
References · 22
- 2017Attention Is All You Needcited by 59,932
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2020GLU Variants Improve Transformercited by 1,724
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2024TinyLlama: An Open-Source Small Language Modelcited by 604
- 2024OLMo: Accelerating the Science of Language Modelscited by 599
- 2019BLiMP: The Benchmark of Linguistic Minimal Pairs for Englishcited by 553
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 425
- 2024Sparse Feature Circuits: Discovering and Editing Interpretable Causal Graphs in Language Modelscited by 412
- 2024DataComp-LM: In search of the next generation of training sets for language modelscited by 381
10 more
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
- 2020Neural Additive Models: Interpretable Machine Learning with Neural Netscited by 318
- 2024Kolmogorov-Arnold Networks are Radial Basis Function Networkscited by 125
- 2024Wav-KAN: Wavelet Kolmogorov-Arnold Networkscited by 118
- 2024Chebyshev Polynomial-Based Kolmogorov-Arnold Networks: An Efficient Architecture for Nonlinear Function Approximationcited by 115
- 2024A Survey on Kolmogorov-Arnold Networkcited by 67
- 2024Kolmogorov-Arnold Transformercited by 56
- 2024Elements of World Knowledge (EWoK): A Cognition-Inspired Framework for Evaluating Basic World Knowledge in Language Modelscited by 55
- 2025A Practitioner's Guide to Kolmogorov-Arnold Networkscited by 16
- 2026Making Gaussian Kolmogorov-Arnold Networks Reliable and Accuratecited by 3
Built on by
Not cited yet.