Theorem 11: polynomial generalization bounds from hypothesis stability
ProvedStabGen.Hypothesis.hypothesis_stability_generalization_boundLet be a probability distribution on and let the cost function satisfy . Let be a symmetric deterministic learning algorithm whose losses are measurable, and suppose that at sample size it has hypothesis stability (Definition 3) and pointwise hypothesis stability (Definition 4). Draw the training set . Then for every , with probability at least ,
and, with probability at least ,
Here is the generalization error, the empirical error and the leave-one-out error of . The theorem shows that the two classical estimators of the risk deviate from it by whenever the algorithm's stability decays like , without any capacity measure of the hypothesis space.
Formalization Note The two bounds are stated separately: the probability, under , of the failure event is at most , and likewise for . The paper prints the empirical bound with in place of . Its proof bounds the replace-one term of Lemma 9, eq. (8), by , but one of the two summands is a hypothesis-stability term bounded by ; the bound the proof establishes is the one stated here, which coincides with the printed one when . The leave-one-out bound is as printed. The sample size is (a training set has at least one point; at the stability hypotheses are vacuous and the statement fails). The paper's proof goes through Lemmas 9 and 25, which need two distinct indices; at the bounds still hold, by comparing with directly, so no restriction is imposed.
import Mathlib import Definitions.Def_FoundationsML_Stability_Loss import Definitions.Def_FoundationsML_Stability_EmpiricalError import Definitions.Def_FoundationsML_Stability_GeneralizationError import Definitions.Def_StabGen_Hypothesis_Setting import Definitions.Def_StabGen_Hypothesis_Stability open MeasureTheory
namespace StabGen.Hypothesis
open FoundationsML.Stability
theorem hypothesis_stability_generalization_bound {X Y Y' : Type*} [MeasurableSpace X] [MeasurableSpace Y]
(D : Measure (X × Y)) [IsProbabilityMeasure D] (L : Y' → Y → ℝ) (M : ℝ)
(hL : ∀ (y' : Y') (y : Y), 0 ≤ L y' y ∧ L y' y ≤ M)
(A : LearningAlgorithm X Y Y')
(hA : ∀ n : ℕ, Measurable (fun p : (Fin n → X × Y) × (X × Y) =>
Loss L (A (trainingSet p.1)) p.2))
(m : ℕ) (hm : 1 ≤ m) (β₁ β₂ : ℝ) (h₁ : HypothesisStable D L A m β₁)
(h₂ : PointwiseHypothesisStable D L A m β₂) (δ : ℝ) (hδ : 0 < δ) :
(Measure.pi fun _ : Fin m => D)
{S | EmpiricalError L S (A (trainingSet S))
+ Real.sqrt ((M ^ 2 + 6 * M * (m : ℝ) * (β₁ + β₂)) / (2 * (m : ℝ) * δ))
< GeneralizationError D L (A (trainingSet S))} ≤ ENNReal.ofReal δ ∧
(Measure.pi fun _ : Fin m => D)
{S | looError L A S
+ Real.sqrt ((M ^ 2 + 6 * M * (m : ℝ) * β₁) / (2 * (m : ℝ) * δ))
< GeneralizationError D L (A (trainingSet S))} ≤ ENNReal.ofReal δ := by sorry
end StabGen.Hypothesis
Confirmed by the mission captain (proposal self-audit).
Confirmed by the moderator at approval.