Theorem 5 — quantile-value and truncated-mean generalization bounds for pseudo-robust algorithms
ProvedXuMannorRobust.Quantile.theorem5_quantile_truncated_mean_boundsLet be a measurable space, a set of hypotheses and a loss with each measurable. Let the training set , , consist of i.i.d. draws from a probability measure on , with empirical distribution . Let be pseudo robust (Definition 5), let , , and
Then with probability at least the following holds: if and , then
- ;
- .
Here and are the -quantile value and -truncated mean of the loss , .
The theorem says that the quantile value and truncated mean of the testing error of a pseudo-robust algorithm are bracketed, up to , by the same functionals of the training errors at slightly shifted levels, so they can be estimated from the training sample.
Formalization Note The level condition involves and so depends on the sample; it is placed inside the high-probability event as the premise of an implication, which is the reading the paper's proof gives (everything holds pointwise on the event ). "With probability at least " is encoded as: the outer measure under of the set of training sets where the implication fails is at most , so no measurability of is required. Measurability of each and of the cells of the partition is added (the paper ignores measurability). uses Definition 3 with the misprinted second branch corrected. Levels are real numbers, with and cast to .
import Mathlib import Definitions.Def_XuMannorRobust_Quantile_LossQuantile import Definitions.Def_XuMannorRobust_Quantile_IsPseudoRobust open MeasureTheory
namespace XuMannorRobust.Quantile
/-- **Theorem 5** (Xu & Mannor 2012, p. 402; proof in Appendix C, pp. 415–418). Let the loss take
values in `[0, M]` with each `l h` measurable, let `s` consist of `n ≥ 1` i.i.d. draws from the
probability measure `μ`, with empirical distribution `μ_emp`, let `β ∈ (0, 1)`, `δ > 0`,
`λ₀ = √((2K ln 2 + 2 ln(1/δ))/n)` and `r(s) = (n − n̂(s))/n`. If `A` is `(K, ε(·), n̂(·))` pseudo
robust, then with probability at least `1 − δ` the following holds: if
`0 ≤ β − λ₀ − r(s)` and `β + λ₀ + r(s) ≤ 1`, then
(I) `Q(A_s, β − λ₀ − r(s), μ_emp) − ε(s) ≤ Q(A_s, β, μ) ≤ Q(A_s, β + λ₀ + r(s), μ_emp) + ε(s)`;
(II) `T(A_s, β − λ₀ − r(s), μ_emp) − ε(s) ≤ T(A_s, β, μ) ≤ T(A_s, β + λ₀ + r(s), μ_emp) + ε(s)`.
"With probability at least `1 − δ`" is: the outer measure under `μⁿ` of the set of training sets
where the implication fails is at most `δ`. `T` uses the corrected Definition 3. -/
theorem theorem5_quantile_truncated_mean_bounds {Z H : Type*} [MeasurableSpace Z]
(μ : Measure Z) [IsProbabilityMeasure μ] (l : H → Z → ℝ) (M : ℝ)
(hl_bound : ∀ h z, 0 ≤ l h z ∧ l h z ≤ M) (hl_meas : ∀ h, Measurable (l h))
{n : ℕ} (hn : 0 < n) (A : (Fin n → Z) → H) (K : ℕ) (ε : (Fin n → Z) → ℝ)
(nhat : (Fin n → Z) → ℕ) (hA : IsPseudoRobust l A K ε nhat)
(δ : ℝ) (hδ : 0 < δ) (β : ℝ) (hβ0 : 0 < β) (hβ1 : β < 1) :
(Measure.pi fun _ : Fin n => μ)
{s : Fin n → Z | ¬
((0 ≤ β - Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
- ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ) ∧
β + Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
+ ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ) ≤ 1) →
-- (I) quantile value
((lossQuantile l (A s)
(β - Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
- ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ)) (empiricalMeasure s) - ε s
≤ lossQuantile l (A s) β μ ∧
lossQuantile l (A s) β μ
≤ lossQuantile l (A s)
(β + Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
+ ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ)) (empiricalMeasure s) + ε s) ∧
-- (II) truncated mean
(lossTruncatedMean l (A s)
(β - Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
- ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ)) (empiricalMeasure s) - ε s
≤ lossTruncatedMean l (A s) β μ ∧
lossTruncatedMean l (A s) β μ
≤ lossTruncatedMean l (A s)
(β + Real.sqrt ((2 * (K : ℝ) * Real.log 2 + 2 * Real.log (1 / δ)) / (n : ℝ))
+ ((n : ℝ) - (nhat s : ℝ)) / (n : ℝ)) (empiricalMeasure s) + ε s)))}
≤ ENNReal.ofReal δ := by sorry
end XuMannorRobust.Quantile
Confirmed by the mission captain (proposal self-audit).
Confirmed by the moderator at approval.