cs.LGAug 25, 2026

The Sharp Tail of Uniform Stability

Authors: Pahan Dewasurendra

Organizations: Johns Hopkins University

Abstract

Uniform stability controls how much one training example can change the loss at any test point. A new logarithmic-free upper bound shows that a γγ-uniformly stable algorithm with loss in [0,L][0,L] has generalization gap at most O(γlog⁡(1/δ)+Llog⁡(1/δ)n)O \left(γ\log(1/δ) +L\sqrt{\frac{\log(1/δ)}{n}}\right) with probability 1−δ1-δ. Whether an actual bounded-loss learning algorithm can realize the linear dependence on log⁡(1/δ)\log(1/δ) has remained open. The known construction realizes it only for auxiliary weakly dependent random variables whose pointwise range grows with nn. The known learning lower bound holds only at constant probability. We close this gap. For every nn, stability level γγ, and loss bound LL, we construct one deterministic γγ-uniformly stable learning problem whose tail satisfies, simultaneously for 1≤p≤cn1\le p\le c n, P(R(AS)−RS(AS)≥c′min⁡{L,γp+Lp/n})≥e−p.\mathbb P \left( R(A_S)-R_S(A_S) \ge c'\min \left\{L,γp+L\sqrt{p/n}\right\} \right)\ge e^{-p}. The construction is ordinary bounded absolute-loss regression with constant labels. Its key is a multiscale collection of rare Rademacher features. A coordinatewise ramp is stable in sup norm, while an odd symmetrized maximum converts a unique extreme feature into a gap of order γpγp without violating the loss bound. Geometrically spaced ramps put all confidence levels into the same problem. Together with the logarithmic-free upper bound, this determines the optimal high-probability and moment dependence of uniform stability up to universal constants.

Explore similar work

Aug 10, 2026stat.ML

Logarithmic-Free Moment and Generalization Bounds for Uniformly Stable Algorithms

Uniform stability is a classical tool for controlling the generalization error of a learning algorithm. Bousquet, Klochkov, and Zhivotovskiy (2020) showed that the problem can be reduced to a moment inequality for a sum of weakly interacting functions of independent random variables. Their bound contains an additional factor log⁡n\log n, and they asked whether this factor can be removed. We answer this upper-bound question affirmatively. More specifically, let Z=(Z1,…,Zn)Z=(Z_1,\ldots,Z_n) have independent coordinates and let gi(Z)g_i(Z) satisfy E[gi(Z)∣Z−i]=0, ∣E[gi(Z)∣Zi]∣≤M, for every i=1,…,n,\mathbb E[g_i(Z)\mid Z_{-i}]=0, \ \left| \mathbb E[g_i(Z)\mid Z_i]\right|\le M, \ \text{for every } i = 1, \dots, n, where Z−iZ_{-i} denotes all coordinates except ZiZ_i. Assume additionally that changing any coordinate ZjZ_j, j≠ij\neq i, changes gig_i by at most ββ, we prove that, for every p≥2p\ge2, for every p≥2p\ge2, ∥∑i=1ngi(Z)∥p≤16pnβ+M2pn.\left\| \sum_{i=1}^n g_i(Z)\right\|_p \le 16pnβ+M\sqrt{2pn}. This removes the log⁡n\log n factor from the previous bound and matches the lower bound of Bousquet, Klochkov, and Zhivotovskiy up to universal constants in the range covered by their construction. Our proof first establishes the required estimate on the Rademacher cube, then transfers it to arbitrary product distributions by a two-copy randomization argument.
Jun 5, 2026stat.ML

Stability beyond Bounded Differences: Sharp Generalization Bounds under Finite LpL_p Moments

While algorithmic stability is a central tool for understanding generalization of learning algorithms, existing high-probability guarantees typically rely on uniform boundedness or sub-Gaussian/sub-Weibull tail assumptions, which can be overly restrictive for modern settings with heavy-tailed or unbounded losses. We develop a stability-based framework that requires only a finite LpL_p moment condition. Our first contribution is sharp concentration inequalities for functions of independent random variables under LpL_p constraints, extending McDiarmid's bounded-differences techniques beyond the classical regime. Leveraging these results, we derive sharp high-probability generalization bounds across a range of learning paradigms, including empirical risk minimization, transductive regression, and meta-learning. These guarantees show that LpL_p stability suffices for robust generalization even when boundedness fails, substantially weakening the standard assumptions in the stability literature.
Jun 5, 2026cs.LG

Uniform Stability and Generalization Error of GD and SGD on Fixed-Point Parameters

We analyze generalization error, uniform stability, and uniform argument stability of gradient descent (GD) and stochastic gradient descent (SGD) over discrete parameter spaces, where each update involves deterministic or stochastic rounding. We show that deterministic rounding degrades the generalization error of GD on convex, Lipschitz, and smooth loss functions, increasing the rate from O(T/n)O(T/n) to O(T/n)O(T/\sqrt{n}), and establish matching lower bounds. We further prove that uniform stability of GD becomes Ω(T)Ω(T), showing that stability-based generalization bounds are vacuous in this setting. In contrast, for the same losses, stochastic gradient descent with deterministic rounding admits nontrivial uniform stability guarantees, which differ qualitatively from the real-valued case and exhibit distinct dependencies on the number of iterations and the dimension: we prove tight bounds O(T/n)O(T/n) for one dimension and O(T2/n)O(T^2/n) for higher dimensions. We also show that stochastic rounding can introduce generalization error that increases with the dimension; such a phenomenon is absent in standard real-valued optimization and in the deterministic rounding case. Finally, we provide upper bounds on uniform argument stability for stochastic rounding schemes and show that these bounds are tight when the loss can be represented as a sum of coordinate-wise functions.