Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

KAN: Kolmogorov-Arnold Networks

About

Inspired by the Kolmogorov-Arnold representation theorem, we propose Kolmogorov-Arnold Networks (KANs) as promising alternatives to Multi-Layer Perceptrons (MLPs). While MLPs have fixed activation functions on nodes ("neurons"), KANs have learnable activation functions on edges ("weights"). KANs have no linear weights at all -- every weight parameter is replaced by a univariate function parametrized as a spline. We show that this seemingly simple change makes KANs outperform MLPs in terms of accuracy and interpretability. For accuracy, much smaller KANs can achieve comparable or better accuracy than much larger MLPs in data fitting and PDE solving. Theoretically and empirically, KANs possess faster neural scaling laws than MLPs. For interpretability, KANs can be intuitively visualized and can easily interact with human users. Through two examples in mathematics and physics, KANs are shown to be useful collaborators helping scientists (re)discover mathematical and physical laws. In summary, KANs are promising alternatives for MLPs, opening opportunities for further improving today's deep learning models which rely heavily on MLPs.

Ziming Liu, Yixuan Wang, Sachin Vaidya, Fabian Ruehle, James Halverson, Marin Solja\v{c}i\'c, Thomas Y. Hou, Max Tegmark• 2024

Related benchmarks

TaskDatasetResultRank
Graph ClassificationNCI1
Accuracy79.3
658
Image ClassificationCIFAR-10
Accuracy53.5
564
Image ClassificationCIFAR-100
Accuracy27.9
302
Image ClassificationCIFAR100
Accuracy14.04
301
Image ClassificationCIFAR10
Accuracy (%)43
282
Language ModelingWikiText
Word Perplexity3.98e+4
234
Point Cloud ClassificationModelNet40 (test)
Accuracy23.2
229
Image ClassificationFashionMNIST
Accuracy85.44
185
Graph ClassificationREDDIT BINARY
Accuracy91.7
124
Language ModelingOpenWebText
Perplexity2.78e+4
122
Showing 10 of 230 rows
...

Other info

Follow for update