Lune

ICML2026Top-tier venue

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

Puyu Wang, Junyu Zhou, Philipp Liznerski, Marius Kloft

2026Year
2Citations

Abstract

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/T1/T and a generalization rate of order 1/n1/n, with TT denoting the number of GD iterations and nn the sample size. In the private setting, we characterize the noise required for (ϵ,δ)(\epsilon,\delta)-DP and obtain a utility bound of order d/(nϵ)\sqrt{d}/(n\epsilon) (with dd 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.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext 64e4af98-cc52-435d-bc26-3fc26b61b37e

Builds on17

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines