mlsanityData to 7 Oct 2026

Optimization, Generalization and Differential Privacy Bounds for Gradient Descent on Kolmogorov-Arnold Networks

arXiv 2601.22409 ↗First posted 29 Jan 2026cs.LG cs.AI stat.MLcited by 1

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

  1. 2016Deep Learning with Differential Privacycited by 2,682
  2. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  3. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
  4. 2017Spectrally-normalized margin bounds for neural networkscited by 681
  5. 2024KAN: Kolmogorov-Arnold Networkscited by 634
  6. 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
  7. 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
  8. 2019Generalization Bounds of Stochastic Gradient Descent for Wide and Deep Neural Networkscited by 221
  9. 2019Private Stochastic Convex Optimization with Optimal Ratescited by 183
  10. 2019Polylogarithmic width suffices for gradient descent to achieve arbitrarily small test error with shallow ReLU networkscited by 120
  11. 2020Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient Descentcited by 118
  12. 2024A comprehensive and FAIR comparison between MLP and KAN representations for differential equations and operator networkscited by 103
35 more
  1. 2024Kolmogorov-Arnold Networks (KANs) for Time Series Analysiscited by 97
  2. 2018Overparameterized Nonlinear Learning: Gradient Descent Takes the Shortest Path?cited by 87
  3. 2019How Much Over-parameterization Is Sufficient to Learn Deep ReLU Networks?cited by 84
  4. 2024Convolutional Kolmogorov-Arnold Networkscited by 76
  5. 2024A Survey on Kolmogorov-Arnold Networkcited by 67
  6. 2024Kolmogorov Arnold Informed neural network: A physics-informed deep learning framework for solving forward and inverse problems based on Kolmogorov Arnold Networkscited by 64
  7. 2017Efficient Private ERM for Smooth Objectivescited by 61
  8. 2024A Temporal Kolmogorov-Arnold Transformer for Time Series Forecastingcited by 50
  9. 2021Differentially Private Stochastic Optimization: New Results in Convex and Non-Convex Settingscited by 45
  10. 2021On the Convergence and Calibration of Deep Learning with Differential Privacycited by 36
  11. 2024On the expressiveness and spectral bias of KANscited by 36
  12. 2020Optimal Rates for Averaged Stochastic Gradient Descent under Neural Tangent Kernel Regimecited by 35
  13. 2022Random Feature Amplification: Feature Learning and Generalization in Neural Networkscited by 33
  14. 2022Stability vs Implicit Bias of Gradient Methods on Separable Data and Beyondcited by 29
  15. 2021Stability & Generalisation of Gradient Descent for Shallow Neural Networks without the Neural Tangent Kernelcited by 25
  16. 2024Physics Informed Kolmogorov-Arnold Neural Networks for Dynamical Analysis via Efficent-KAN and WAV-KANcited by 25
  17. 2019Gradient Descent can Learn Less Over-parameterized Two-layer Neural Networks on Classification Problemscited by 23
  18. 2021Differentially Private SGD with Non-Smooth Lossescited by 22
  19. 2022Stability and Generalization Analysis of Gradient Methods for Shallow Neural Networkscited by 21
  20. 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
  21. 2024KANICE: Kolmogorov-Arnold Networks with Interactive Convolutional Elementscited by 12
  22. 2024Generalization Bounds and Model Complexity for Kolmogorov-Arnold Networkscited by 11
  23. 2023How many Neurons do we need? A refined Analysis for Shallow Networks trained with Gradient Descentcited by 10
  24. 2024How to Make the Gradients Small Privately: Improved Rates for Differentially Private Non-Convex Optimizationcited by 10
  25. 2023Generalization Guarantees of Gradient Descent for Multi-Layer Neural Networkscited by 9
  26. 2024Sharper Guarantees for Learning Neural Network Classifiers with Gradient Methodscited by 7
  27. 2025Finding Local Diffusion Schr"odinger Bridge using Kolmogorov-Arnold Networkcited by 7
  28. 2024KAN we improve on HEP classification tasks? Kolmogorov-Arnold Networks applied to an LHC physics examplecited by 4
  29. 2024GINN-KAN: Interpretability pipelining with applications in Physics Informed Neural Networkscited by 4
  30. 2025Towards Understanding Generalization in DP-GD: A Case Study in Training Two-Layer CNNscited by 4
  31. 2024Generalization analysis with deep ReLU networks for metric and similarity learningcited by 3
  32. 2025Optimal Rates for Generalization of Gradient Descent for Deep ReLU Classificationcited by 3
  33. 2023Fine-grained Analysis of Non-parametric Estimation for Pairwise Learningcited by 2
  34. 2024Convex Approximation of Two-Layer ReLU Networks for Hidden State Differential Privacycited by 2
  35. 2025On the Rate of Convergence of Kolmogorov-Arnold Network Regression Estimatorscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.