KAN比MLP更擅长学习高频函数,理论与实证均支持其高效性。
On the expressiveness and spectral bias of KANs
- 用分段线性函数替代传统神经元,提升对复杂函数的表达能力
- 在相同规模下KAN逼近能力不弱于MLP,大网格时效率更高
- 对高频信号偏差更小,适合科学计算中高频率问题
Kolmogorov-Arnold网络(KAN)作为多层感知机(MLP)的潜在替代架构,近期在人工智能用于科学任务中展现出优异性能,涵盖函数回归、偏微分方程求解等。本文从理论角度重新比较了KAN与MLP的表达与逼近能力:我们证明了任意MLP可被尺寸相当的KAN表示,表明KAN至少具备与MLP相当的逼近能力;反之,尽管KAN可由MLP表示,但参数量会增加至原网格尺寸的倍数。这说明大网格的KAN在逼近特定函数时可能更具效率。此外,从学习优化视角出发,我们发现相比MLP,KAN对低频信号的偏好更弱,其特有的分段样条网格扩展机制有助于提升高频成分的学习效果。通过对比不同深度、宽度和网格大小,为实际超参数选择提供了参考。
原文摘要 · Abstract (English)
Kolmogorov-Arnold Networks (KAN) \cite{liu2024kan} were very recently proposed as a potential alternative to the prevalent architectural backbone of many deep learning models, the multi-layer perceptron (MLP). KANs have seen success in various tasks of AI for science, with their empirical efficiency and accuracy demostrated in function regression, PDE solving, and many more scientific problems. In this article, we revisit the comparison of KANs and MLPs, with emphasis on a theoretical perspective. On the one hand, we compare the representation and approximation capabilities of KANs and MLPs. We establish that MLPs can be represented using KANs of a comparable size. This shows that the approximation and representation capabilities of KANs are at least as good as MLPs. Conversely, we show that KANs can be represented using MLPs, but that in this representation the number of parameters increases by a factor of the KAN grid size. This suggests that KANs with a large grid size may be more efficient than MLPs at approximating certain functions. On the other hand, from the perspective of learning and optimization, we study the spectral bias of KANs compared with MLPs. We demonstrate that KANs are less biased toward low frequencies than MLPs. We highlight that the multi-level learning feature specific to KANs, i.e. grid extension of splines, improves the learning process for high-frequency components. Detailed comparisons with different choices of depth, width, and grid sizes of KANs are made, shedding some light on how to choose the hyperparameters in practice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。