Optimization Risk Bounds for Kolmogorov-Arnold Networks Trained by DP-SGD with Correlated Noise
Not on the map. Why?
Abstract (arXiv)
The theoretical understanding of differentially private stochastic gradient descent (DP-SGD) with temporally correlated noise remains limited, particularly for non-convex neural network training. As a first step, we study two-layer Kolmogorov-Arnold Networks (KANs), a recently introduced architecture with learnable spline-based edge functions. We establish the first optimization risk bounds for clipped mini-batch DP-SGD with correlated noise in this setting, with explicit dependence on temporal correlation, clipping, mini-batch sampling, and network width. Existing arguments fail for three reasons: temporal dependence breaks the conditional-centering step; projection obstructs the cross-iteration cancellation of correlated perturbations; and active clipping breaks the empirical-gradient structure. We address these issues through shifted and auxiliary dynamics, a weighted empirical loss, and a high-probability localization argument. Our bound shows that temporal correlation reduces the leading noise terms, the clipping threshold enters essentially through an effective step size, and private training admits an explicit network width range. Experiments on synthetic data and MNIST support the predicted optimization effect of temporal correlation. As an application, we derive population risk guarantees via algorithmic stability. Our framework recovers non-private mini-batch SGD, independent-noise DP-SGD, and their full-batch counterparts as special cases.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 9 of its 60 references that are in the map. First posted on arXiv 12 May 2026.
Builds on
9 of 60 references.
References · 9
- 2019Gradient Descent can Learn Less Over-parameterized Two-layer Neural Networks on Classification Problemscited by 24
- 2024Near Exact Privacy Amplification for Matrix Mechanismscited by 18
- 2025The Hitchhiker's Guide to Efficient, End-to-End, and Tight DP Auditingcited by 5
- 2024Generalization analysis with deep ReLU networks for metric and similarity learningcited by 4
- 2025On the Rate of Convergence of Kolmogorov-Arnold Network Regression Estimatorscited by 4
- 2025Understanding Private Learning From Feature Perspectivecited by 3
- 2026DP-λCGD: Efficient Noise Correlation for Differentially Private Model Trainingcited by 3
- 2026Understanding the Impact of Differentially Private Training on Memorization of Long-Tailed Datacited by 1
- 2026Differential Privacy in Two-Layer Networks: How DP-SGD Harms Fairness and Robustnesscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.