Kolmogorov-Arnold Networks
Kolmogorov–Arnold Networks are a type of artificial neural network architecture inspired by the Kolmogorov–Arnold representation theorem, also known as the superposition theorem. Unlike traditional multilayer perceptrons, which rely on fixed activation functions and linear weights, KANs replace each weight with a learnable univariate function, often represented using splines.
History
KANs were proposed by Liu et al. as a generalization of the Kolmogorov–Arnold representation theorem, aiming to outperform MLPs in small-scale AI and scientific tasks. Before KANs, numerous studies explored KART's connections to neural networks or used it as a basis for designing new network architectures.In the 1980s and 1990s, early research applied KART to neural network design. Kůrková et al., Hecht-Nielsen, and Nees established theoretical foundations for multilayer networks based on KART. Igelnik et al. introduced the Kolmogorov Spline Network using cubic splines to model complex functions. Sprecher introduced numerical methods for building network layers, while Nakamura et al. created activation functions with guaranteed approximation accuracy. These works linked KART's theoretical potential with practical neural network implementation.
KART has also been used in other computational and theoretical fields. Coppejans developed nonparametric regression estimators using B-splines, Bryant applied it to high-dimensional image tasks, Liu investigated theoretical applications in optimal transport and image encryption, and more recently, Polar and Poluektov used Urysohn operators for efficient KART construction, while Fakhoury et al. introduced ExSpliNet, integrating KART with probabilistic trees and multivariate B-splines for improved function approximation.
Architecture
KANs are based on the Kolmogorov–Arnold representation theorem, which was linked to the 13th Hilbert problem.Given consisting of n variables, a multivariate continuous function can be represented as:
This formulation contains two nested summations: an outer and an inner sum. The outer sum aggregates terms, each involving a function.
The inner sum computes n terms for each q, where each term is a continuous function of the single variable. The inner continuous functions are universal, independent of, while the outer functions depend on the specific function being represented. The representation holds for all multivariate functions as proved in. If is continuous, then the outer functions are continuous; if is discontinuous, then the corresponding are generally discontinuous, while the inner functions remain the same universal functions.
Liu et al. proposed the name KAN. A general KAN network consisting of L layers takes x to generate the output as:
Here, is the function matrix of the l-th KAN layer or a set of pre-activations.
Let i denote the neuron of the l-th layer and j the neuron of the -th layer. The activation function connects to :
where nl is the number of nodes of the l-th layer.
Thus, the function matrix can be represented as an matrix of activations:
Implementations
To make the KAN layers optimizable, the inner function is formed by the combination of spline and basic functions as the formula:where is the basic function, usually defined as and is the base weight matrix. Also, is the spline weight matrix and is the spline function. The spline function can be a sum of B-splines.
Many studies suggested to use other polynomial and curve functions instead of B-spline to create new KAN variants.
Functions used
The choice of functional basis strongly influences the performance of KANs. Common function families include:- B-splines: Provide locality, smoothness, and interpretability; they are the most widely used in current implementations.
- RBFs : Capture localized features in data and are effective in approximating functions with non-linear or clustered structures.
- Chebyshev polynomials: Offer efficient approximation with minimized error in the maximum norm, making them useful for stable function representation.
- Rational function: Useful for approximating functions with singularities or sharp variations, as they can model asymptotic behavior better than polynomials.
- Fourier series: Capture periodic patterns effectively and are particularly useful in domains such as physics-informed machine learning.
- Wavelet functions : Used for feature extraction as they can capture both high-frequency and low-frequency data components.
- Piecewise linear functions: Provide efficient approximation for multivariate functions in KANs.
Usage
In some modern neural architectures like convolutional neural networks, recurrent neural networks, and Transformers, KANs are typically used as drop-in substitutes for MLP layers. Despite KANs' general-purpose design, researchers have created and used them for a number of tasks:- Scientific machine learning : Function fitting, partial differential equations and physical/mathematical laws.
- Continual learning: KANs better preserve previously learned information during incremental updates, avoiding catastrophic forgetting due to the locality of spline adjustments.
- Graph neural networks: Extensions such as Kolmogorov–Arnold Graph Neural Networks integrate KAN modules into message-passing architectures, showing improvements in molecular property prediction tasks.