arXiv · 2411.05296
On Training of Kolmogorov-Arnold Networks
Abstract
Kolmogorov-Arnold Networks have recently been introduced as a flexible alternative to multi-layer Perceptron architectures. In this paper, we examine the training dynamics of different KAN architectures and compare them with corresponding MLP formulations. We train with a variety of different initialization schemes, optimizers, and learning rates, as well as utilize back propagation free approaches like the HSIC Bottleneck. We find that (when judged by test accuracy) KANs are an effective alternative to MLP architectures on high-dimensional datasets and have somewhat better parameter efficiency, but suffer from more unstable training dynamics. Finally, we provide recommendations for improving training stability of larger KAN models.
Explore related subjects
Keep this discovery
Shairoz Sohail. 2024-11-08. On Training of Kolmogorov-Arnold Networks. https://arxiv.org/abs/2411.05296
Cite the original work for its findings. Save a collection to share your selection of sources.