Lune

NeurIPS2021Top-tier venue

Stability and Deviation Optimal Risk Bounds with Convergence Rate O(1/n)O(1/n)

Yegor Klochkov, Nikita Zhivotovskiy

2021Year
72Citations
17Top-tier citations

Abstract

The sharpest known high probability generalization bounds for uniformly stable algorithms (Feldman, Vondrák, 2018, 2019), (Bousquet, Klochkov, Zhivotovskiy, 2020) contain a generally inevitable sampling error term of order Θ(1/n)\Theta(1/\sqrt{n}). When applied to excess risk bounds, this leads to suboptimal results in several standard stochastic convex optimization problems. We show that if the so-called Bernstein condition is satisfied, the term Θ(1/n)\Theta(1/\sqrt{n}) can be avoided, and high probability excess risk bounds of order up to O(1/n)O(1/n) are possible via uniform stability. Using this result, we show a high probability excess risk bound with the rate O(log⁡n/n)O(\log n/n) for strongly convex and Lipschitz losses valid for any empirical risk minimization method. This resolves a question of Shalev-Shwartz, Shamir, Srebro, and Sridharan (2009). We discuss how O(log⁡n/n)O(\log n/n) high probability excess risk bounds are possible for projected gradient descent in the case of strongly convex and Lipschitz losses without the usual smoothness assumption.

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.

Cited by top-tier papers17

Ask how each one uses it

Builds on1

Related papers

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