arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.08026cs.LGcs.NEstat.ML

Kolmogorov-Arnold Networks的泛化界与模型复杂度

Generalization Bounds and Model Complexity for Kolmogorov-Arnold Networks

Texas A&M University(得克萨斯农工大学);School of Statistics and Data Science, Shanghai University of Finance and Economics(上海财经大学统计与数据科学学院)

Xianyang Zhang, Huijuan Zhou

AI总结:

本文针对KAN建立两类泛化界,分别覆盖基函数线性组合与低秩RKHS激活函数,揭示其对系数范数、秩、Lipschitz常数及网络规模的依赖,并通过模拟与真实数据实验验证。

摘要:

Kolmogorov-Arnold Network(KAN)是Liu等人(2024)最近提出的一种网络结构;与多层感知机相比,它在许多面向科学的任务中提供了更好的可解释性和更简约的设计。本工作通过为配备激活函数的KAN建立泛化界,对KAN进行严格的理论分析;这些激活函数要么由基函数的线性组合表示,要么位于低秩再生核Hilbert空间(RKHS)中。在第一种情形下,该泛化界容纳了在KAN每一层形成激活函数时对基函数的多种选择,并适配每一层不同的算子范数。对于算子范数的一种特定选择,该界随系数矩阵的$l_1$范数和激活函数的Lipschitz常数缩放,并且除对数因子外,不依赖于组合参数(例如节点数)。此外,我们的结果不要求损失函数具有有界性假设,因此适用于一般类别的回归型损失函数。在低秩情形下,该泛化界随底层秩以及每一层激活函数的Lipschitz常数呈多项式缩放。对于在模拟和真实数据集上使用随机梯度下降训练的KAN,本文对这些界进行了实证研究。数值结果证明了这些界的实际相关性。

英文摘要:

Kolmogorov-Arnold Network (KAN) is a network structure recently proposed by Liu et al. (2024) that offers improved interpretability and a more parsimonious design in many science-oriented tasks compared to multi-layer perceptrons. This work provides a rigorous theoretical analysis of KAN by establishing generalization bounds for KAN equipped with activation functions that are either represented by linear combinations of basis functions or lying in a low-rank Reproducing Kernel Hilbert Space (RKHS). In the first case, the generalization bound accommodates various choices of basis functions in forming the activation functions in each layer of KAN and is adapted to different operator norms at each layer. For a particular choice of operator norms, the bound scales with the $l_1$ norm of the coefficient matrices and the Lipschitz constants for the activation functions, and it has no dependence on combinatorial parameters (e.g., number of nodes) outside of logarithmic factors. Moreover, our result does not require the boundedness assumption on the loss function and, hence, is applicable to a general class of regression-type loss functions. In the low-rank case, the generalization bound scales polynomially with the underlying ranks as well as the Lipschitz constants of the activation functions in each layer. These bounds are empirically investigated for KANs trained with stochastic gradient descent on simulated and real data sets. The numerical results demonstrate the practical relevance of these bounds.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑