Papers
Topics
Authors
Recent
Search
2000 character limit reached

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

Published 5 Jun 2026 in stat.ML, cs.LG, and math.ST | (2606.06855v1)

Abstract: 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.

Summary

  • The paper introduces a novel framework for algorithmic stability using finite L_p moments, enabling sharp concentration bounds for heavy-tailed losses.
  • It extends McDiarmid’s inequality to unbounded settings, providing robust high-probability generalization guarantees for ERM, transductive regression, and meta-learning.
  • The results reveal a dual tail regime with sub-Gaussian behavior for moderate deviations and polynomial decay for extreme events, enhancing risk assessment.

Stability under Finite LpL_p Moments: A Technical Analysis of Modern Generalization Bounds

Overview and Motivation

Algorithmic stability is central to theoretical generalization guarantees in statistical learning. Classical approaches—most notably founded on uniform stability and bounded loss functions—enable concentration results via tools such as McDiarmid’s inequality, yielding tight high-probability deviation bounds. However, modern machine learning applications, especially those encountering unbounded or heavy-tailed losses (e.g., deep networks, robust/contaminated regression), frequently violate these restrictive assumptions. While recent work has relaxed tails (e.g., via sub-Gaussian or sub-Weibull surrogates), these still require existence of exponential moments and thus fail to accommodate the power-law phenomena observed empirically in contemporary models.

The paper "Stability beyond Bounded Differences: Sharp Generalization Bounds under Finite LpL_p Moments" (2606.06855) introduces a framework that leverages only a finite LpL_p moment condition for algorithmic stability. The core contribution is a suite of sharp concentration inequalities that extend McDiarmid-type results to the heavy-tailed regime, and high-probability generalization guarantees for canonical learning setups—empirical risk minimization (ERM), transductive regression, and meta-learning—under (Lp,β)(L_p, \beta)-Lipschitz stability. The bounds feature a two-regime tail: sub-Gaussian for moderate deviations and polynomial for extreme events, reflecting precisely the distributional properties permitted by finite-LpL_p increments.

Theoretical Contributions: Concentration without Exponential Moments

(Lp,β)(L_p, \beta)-Lipschitz Stability

The technical linchpin is a generalized notion of stability: For independent samples x1,…,xnx_1,\ldots,x_n and a measurable statistic f(x1,…,xn)f(x_1,\ldots,x_n),

  • The sensitivity of ff to coordinate ii is controlled not by a deterministic bound, but by a random variable LpL_p0, where LpL_p1 is an independent copy.
  • The only assumption is LpL_p2 for some LpL_p3.

This setup is strictly weaker than both bounded differences and Orlicz/sub-Weibull diameter conditions. The approach naturally models scenarios where one-sample perturbations can be unbounded but their distributions possess sufficiently light polynomial tails.

Main Concentration Inequality

A fundamental result (Theorem 1) is as follows: If LpL_p4 for each LpL_p5 and the LpL_p6 satisfy LpL_p7, then, for any LpL_p8,

LpL_p9

with LpL_p0 dependent on LpL_p1. Notably, this captures both moderate and large deviations. For LpL_p2, the tail is essentially polynomial, matching the heavy-tailed regime. Figure 1

Figure 1

Figure 1: Plot of LpL_p3 versus LpL_p4 for a heavy-tailed linear regression; stabilization near LpL_p5 confirms the polynomial tail in high deviations.

Sharpness in the tail-order sense is guaranteed by reference to the classical Fuk–Nagaev inequalities: the coexistence of sub-Gaussian and polynomial regimes is unavoidable under mere LpL_p6 assumptions.

Applications: Robust Generalization in Modern Learning

Empirical Risk Minimization (ERM)

For ERM in the i.i.d. setting, the paper proves high-probability oracle inequalities for the gap between the population risk and empirical risk (or leave-one-out estimate), under LpL_p7-Lipschitz stability: LpL_p8 where LpL_p9 controls the fluctuation of the loss across test samples. Unlike uniform stability, vanishing generalization now requires not only (Lp,β)(L_p, \beta)0 but also that stability coefficients (Lp,β)(L_p, \beta)1 decay faster than (Lp,β)(L_p, \beta)2.

Transductive Regression

For transductive setups (where train and test points are splits of a fixed finite population), the paper develops concentration results for sampling without replacement, extending bounded differences to the finite-moment scenario. The generalization bound for the difference between the test and training error again features both Gaussian and heavy-tailed terms, now scaled by subsample sizes (Lp,β)(L_p, \beta)3 and (Lp,β)(L_p, \beta)4 and their moments. Figure 2

Figure 3: Plot of (Lp,β)(L_p, \beta)5 versus (Lp,β)(L_p, \beta)6 showing robustness of polynomial tail ratio across increasing train sizes in transductive regression.

Meta-Learning

The meta-learning framework considers a learn-to-learn setting: a meta-learner is trained across tasks (task distributions drawn i.i.d. from a meta-distribution), and performance is measured on new unseen tasks. The analysis demands meta-stability both across training tasks and within-task samples. The result is a high-probability bound for the meta-generalization error, which, while retaining a polynomial tail, only presupposes finite (Lp,β)(L_p, \beta)7-moment meta-stability.

Empirical Results: Verifying the Two-Regime Theory

Large-scale simulations in regression (linear and two-layer neural networks) establish the theoretical sharpness of the proposed bounds. The key diagnostic, (Lp,β)(L_p, \beta)8, stabilizes at (Lp,β)(L_p, \beta)9 for large LpL_p0, as predicted for polynomial tails, and displays sub-Gaussian scaling in the moderate-deviation regime. Figure 3

Figure 4: Plot of LpL_p1 versus LpL_p2 on a neural network regression task confirming both sub-Gaussian and polynomial regimes for heavy-tailed data under ADAM optimization.

These results highlight in particular that classical bounds, by omitting such polynomial terms, can dramatically underestimate tail risks in presence of heavy-tailed stability increments.

Implications and Future Directions

The move to LpL_p3-based stability fundamentally broadens the applicability of concentration-based generalization theory. In practical terms, this guarantees that algorithms can be certified for robust out-of-sample performance—even when data or losses violate exponential tail conditions—provided that finite moments and demonstrable asymptotic stability are present.

Theoretical implications include:

  • Non-exponential tails are fundamental: No uniform LpL_p4 (Orlicz) exponential-type bound can be valid in Pareto-like or power-law heavy-tailed settings.
  • Finite-moment stability is both necessary and sufficient: In such regimes, the polynomial term is not a technicality but a lower-bound-matching feature.
  • Sharper understanding of "rare" overfitting: While moderate errors remain tightly controlled, rare extreme deviations are quantifiably possible—a risk that traditional uniform stability obfuscates.

Practically, this framework invites further empirical investigation and supports algorithm design for robust generalization:

  • Design and tuning of regularization/hyperparameters to enforce rapid decay of stability coefficients.
  • Algorithmic policies (e.g., clipped gradients, robust means) that can ensure finite moments even in highly non-ideal data regimes. Figure 5

Figure 5

Figure 2: Plot of LpL_p5 versus LpL_p6 for transductive ridge regression, highlighting the two-regime tail and alignment with theoretical predictions.

Open directions include studying stability in the presence of dependent data, fully unsupervised settings, and matching lower bounds for various learning paradigms under only LpL_p7-moment assumptions.

Conclusion

This work decisively advances the theory of algorithmic stability for generalization beyond bounded or sub-exponential settings, providing precise and sharp probabilistic bounds under minimal finite-moment conditions. The framework and concentration inequalities formalize, for the first time, the exact way in which heavy-tailed data and losses impact generalization in modern learning algorithms. This creates a robust bridge from abstract stability concepts to the realities of deep and heavy-tailed data regimes in modern AI.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Open Problems

We haven't generated a list of open problems mentioned in this paper yet.