Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks
Not on the map. Why?
Abstract (arXiv)
Kolmogorov--Arnold Networks (KANs) have recently emerged as a structured alternative to standard MLPs, yet a principled theory for their training dynamics, generalization, and privacy properties remains limited. In this paper, we analyze gradient descent (GD) for training two-layer KANs and derive general bounds that characterize their training dynamics, generalization, and utility under differential privacy (DP). As a concrete instantiation, we specialize our analysis to logistic loss under an NTK-separable assumption, where we show that polylogarithmic network width suffices for GD to achieve an optimization rate of order 1/T and a generalization rate of order 1/n, with T denoting the number of GD iterations and n the sample size. In the private setting, we characterize the noise required for (ε,δ)-DP and obtain a utility bound of order √d/(nε) (with d the input dimension), matching the classical lower bound for general convex Lipschitz problems. Our results imply that polylogarithmic width is not only sufficient but also necessary under differential privacy, revealing a qualitative gap between non-private (sufficiency only) and private (necessity also emerges) training regimes. Experiments further illustrate how these theoretical insights can guide practical choices, including network width selection and early stopping.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 47 of its 68 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
47 of 68 references.
References · 47
- 2016Deep Learning with Differential Privacycited by 2,682
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2024KAN: Kolmogorov-Arnold Networkscited by 634
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
- 2019Generalization Bounds of Stochastic Gradient Descent for Wide and Deep Neural Networkscited by 221
- 2019Private Stochastic Convex Optimization with Optimal Ratescited by 183
- 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 120
- 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 118
- 2024A comprehensive and FAIR comparison between MLP and KAN representations for differential equations and operator networkscited by 103
35 more
- 2024Kolmogorov-Arnold Networks (KANs) for Time Series Analysiscited by 97
- 2018Overparameterized Nonlinear Learning: Gradient Descent Takes the Shortest Path?cited by 87
- 2019How Much Over-parameterization Is Sufficient to Learn Deep ReLU Networks?cited by 84
- 2024Convolutional Kolmogorov-Arnold Networkscited by 76
- 2024A Survey on Kolmogorov-Arnold Networkcited by 67
- 2024Kolmogorov Arnold Informed neural network: A physics-informed deep learning framework for solving forward and inverse problems based on Kolmogorov Arnold Networkscited by 64
- 2017Efficient Private ERM for Smooth Objectivescited by 61
- 2024A Temporal Kolmogorov-Arnold Transformer for Time Series Forecastingcited by 50
- 2021Differentially Private Stochastic Optimization: New Results in Convex and Non-Convex Settingscited by 45
- 2021On the Convergence and Calibration of Deep Learning with Differential Privacycited by 36
- 2024On the expressiveness and spectral bias of KANscited by 36
- 2020Optimal Rates for Averaged Stochastic Gradient Descent under Neural Tangent Kernel Regimecited by 35
- 2022Random Feature Amplification: Feature Learning and Generalization in Neural Networkscited by 33
- 2022Stability vs Implicit Bias of Gradient Methods on Separable Data and Beyondcited by 29
- 2021Stability & Generalisation of Gradient Descent for Shallow Neural Networks without the Neural Tangent Kernelcited by 25
- 2024Physics Informed Kolmogorov-Arnold Neural Networks for Dynamical Analysis via Efficent-KAN and WAV-KANcited by 25
- 2019Gradient Descent can Learn Less Over-parameterized Two-layer Neural Networks on Classification Problemscited by 23
- 2021Differentially Private SGD with Non-Smooth Lossescited by 22
- 2022Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networkscited by 21
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2024KANICE: Kolmogorov-Arnold Networks with Interactive Convolutional Elementscited by 12
- 2024Generalization Bounds and Model Complexity for Kolmogorov-Arnold Networkscited by 11
- 2023How many Neurons do we need? A refined Analysis for Shallow Networks trained with Gradient Descentcited by 10
- 2024How to Make the Gradients Small Privately: Improved Rates for Differentially Private Non-Convex Optimizationcited by 10
- 2023Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networkscited by 9
- 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methodscited by 7
- 2025Finding Local Diffusion Schr"odinger Bridge using Kolmogorov-Arnold Networkcited by 7
- 2024KAN we improve on HEP classification tasks? Kolmogorov-Arnold Networks applied to an LHC physics examplecited by 4
- 2024GINN-KAN: Interpretability pipelining with applications in Physics Informed Neural Networkscited by 4
- 2025Towards Understanding Generalization in DP-GD: A Case Study in Training Two-Layer CNNscited by 4
- 2024Generalization analysis with deep ReLU networks for metric and similarity learningcited by 3
- 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classificationcited by 3
- 2023Fine-grained Analysis of Non-parametric Estimation for Pairwise Learningcited by 2
- 2024Convex Approximation of Two-Layer ReLU Networks for Hidden State Differential Privacycited by 2
- 2025On the Rate of Convergence of Kolmogorov-Arnold Network Regression Estimatorscited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.