The science of systems that learn from data and experience. Its scope runs from the statistical and mathematical foundations of learning, including generalization, expressivity, and computational limits, through the design of learning algorithms, deep learning, reinforcement learning, and probabilistic methods, to the empirical study of large models and the trustworthiness, interpretability, and societal impact of learned systems.
Every time a streaming service guesses what you would rate a film you have never seen, it is solving a matrix completion problem: fill in the missing entries of a vast user-by-item table from the few that are observed. The question became famous during the Netflix Prize (2006-2009), and it looks hopeless - infinitely many matrices fit the observed entries - until one assumes the structure that makes recommendation possible: the table is essentially low rank, because tastes are governed by a few latent factors. In their landmark 2009 paper 'Exact Matrix Completion via Convex Optimization' (Foundations of Computational Mathematics), Emmanuel Candes and Benjamin Recht proved that an n-by-n matrix of rank r can be recovered exactly, with high probability, from only about n^1.2 * r * log n randomly observed entries - not by the NP-hard route of minimizing rank, but by minimizing the nuclear norm, a convex surrogate (the sum of the singular values) solvable efficiently. The proof, in the lineage of Candes-Romberg-Tao compressed sensing, turns on two ideas: an incoherence condition ensuring the singular vectors are spread out rather than spiky, and a dual certificate witnessing optimality, whose existence rests on delicate random-matrix concentration. It transformed a practical engineering puzzle into rigorous theory and seeded a decade of work across machine learning, signal processing, computer vision, and sensor localization. This mission formalizes the Candes-Recht exact-recovery theorem in Lean, decomposed into its dual-certificate construction and the probabilistic concentration reductions at its core.
Bandit Algorithms III: Asymptotic and Minimax Optimality of UCBTextbook
The basic UCB regret bound of Mission II is logarithmic but not tight: its leading constant 16/Δi is eight times the information-theoretic limit, and its worst-case rate carries a spurious logn. Chapters 8–9 of Lattimore–Szepesvári close both gaps. A refined confidence schedule f(t)=1+tlog2t yields the asymptotically optimal limsupn→∞Rn/logn≤∑i:Δi>02/Δi — exactly matching the instance-dependent lower bound of Mission VII for Gaussian noise. The MOSS index μ^i+Ti4log+(kTin) achieves minimax regret Rn≤39kn+∑iΔi, matching the Ω(kn) lower bound up to a constant. These two theorems are the gold standard for finite-armed stochastic bandits.
Bandit Algorithms II: Stochastic Bandits and the UCB AlgorithmTextbook
A learner repeatedly chooses one of k slot machines, observes only the reward of the chosen arm, and wants to earn almost as much as the best arm in hindsight. This is the stochastic multi-armed bandit, the canonical model of the exploration–exploitation dilemma. This mission formalizes the model (environments, policies, regret, and the regret decomposition Rn=∑iΔiE[Ti(n)]) and the two classical algorithms of Chapters 6–7 of Lattimore–Szepesvári: Explore-Then-Commit and the Upper Confidence Bound algorithm built on the optimism principle. The goal theorem is the instance-dependent UCB regret bound Rn≤3∑iΔi+∑i:Δi>016log(n)/Δi — logarithmic regret with explicit constants, the single most cited result of bandit theory — together with its distribution-free companion Rn≤8nklogn+3∑iΔi.
Bandit Algorithms XIV: Bayesian Bandits, the Gittins Index and Thompson SamplingTextbook
The oldest bandit algorithm (Thompson, 1933) is also the most modern: sample a parameter from the posterior and act greedily. Chapters 34–36 of Lattimore–Szepesvári develop the Bayesian view in two crowning results. The Gittins index theorem: for infinite-horizon discounted Markov bandits, the seemingly intractable dynamic program is solved exactly by an index policy — each arm gets a retirement-value index computable arm-by-arm, and playing the largest index is Bayesian optimal. And the frequentist analysis of Thompson sampling — the goal theorem: with Gaussian posteriors, Thompson sampling on 1-subgaussian bandits achieves limn→∞Rn/logn=∑i:Δi>02/Δi, exactly asymptotically optimal, alongside the minimax-grade Rn≤Cknlogn. Together they explain why posterior sampling is both principled and practically dominant.
Bandit Algorithms XI: Lower Bounds for Stochastic Linear BanditsTextbook
Is the dn regret of LinUCB (Mission X) an artifact of the algorithm or a law of nature? Chapters 24–25 of Lattimore–Szepesvári prove it is essentially unimprovable. On the unit ball there is a parameter θ with ∥θ∥22=d2/(48n) forcing Rn≥163dn — the goal theorem — and the hypercube gives the same Ω(dn) rate. The asymptotic chapter is more striking still: for fixed finite action sets, the instance-optimal constant c(A,θ) is characterized by an allocation program, and optimism itself is provably suboptimal — LinUCB and Thompson sampling cannot achieve it, because exploration must sometimes deliberately play actions optimism would never touch. These lower bounds define the targets for the entire linear-bandit literature.
Bandit Algorithms V: Adversarial Bandits and Exp3Textbook
What if the rewards are not random at all, but chosen by an adversary who knows your algorithm? Remarkably, a randomized learner can still compete with the best fixed arm in hindsight. Chapters 11–12 of Lattimore–Szepesvári develop the adversarial k-armed bandit: rewards xti∈[0,1] are an arbitrary fixed matrix, the learner samples At∼Pt, and regret is measured against maxi∑txti. The exponential-weights algorithm Exp3, fed by importance-weighted loss estimates X^ti=1−1{At=i}(1−Xt)/Pti, achieves Rn≤2nklogk — the goal theorem. The companion Exp3-IX, which deliberately biases its estimator, upgrades this to a bound holding with high probability rather than only in expectation. These results are the foundation of all adversarial online learning with partial feedback.
Improved Algorithms for Linear Stochastic Bandits I: High-Probability Regret Bound for the OFUL AlgorithmResearch Paper
Motivation
In a linear stochastic bandit, a learner repeatedly chooses an action from a set of vectors and receives a noisy reward whose mean is linear in the action. The model underlies contextual recommendation, adaptive routing and dynamic pricing, where each option is described by features and the payoff of a feature vector must be learned while it is exploited. The quality of a strategy is measured by its regret: the reward lost, relative to always playing the best action, over the first n rounds.
The optimism-in-the-face-of-uncertainty principle (play as if the most favourable parameter consistent with the data were true) was introduced for linear bandits by Auer (2002), and developed by Dani, Hayes and Kakade (2008) (ConfidenceBall, regret O(dnlog3/2n) with confidence sets from a union bound over time) and Rusmevichientong and Tsitsiklis (2010). Abbasi-Yadkori, Pál and Szepesvári (NIPS 2011) replaced the union bound by a self-normalized martingale inequality that holds uniformly in time. It gives smaller confidence ellipsoids and, through them, a high-probability regret bound for the resulting algorithm, OFUL, that improves the earlier ones by logarithmic factors. The inequality became the standard tool for linear and kernelized bandits and for linear reinforcement learning.
Setting
Fix a dimension d≥1 and an unknown parameter θ∗∈Rd. In round t=1,2,… the learner is given a nonempty decision setDt⊆Rd, chooses Xt∈Dt, and observes the reward
Yt=⟨Xt,θ∗⟩+ηt.
There is a filtration {Ft}t≥0 such that Xt is Ft−1-measurable and ηt is Ft-measurable and conditionally R-sub-Gaussian: E[eληt∣Ft−1]≤exp(λ2R2/2) for all λ∈R, with R≥0 fixed.
For a regularization parameter λ>0 let Vt=λI+∑s=1tXsXs⊤ and let θt=Vt−1∑s=1tYsXs be the regularized least-squares estimate. With ∥v∥A=v⊤Av and a known bound ∥θ∗∥2≤S, the confidence ellipsoid is
The OFUL algorithm chooses, in round t, a pair (Xt,θt) maximizing ⟨x,θ⟩ over Dt×Ct−1. The pseudo-regret is Rn=∑t=1n⟨xt∗−Xt,θ∗⟩, where ⟨xt∗,θ∗⟩=maxx∈Dt⟨x,θ∗⟩.
Formalization targets
Goal: Theorem 3, the regret of OFUL
If ∥Xt∥2≤L, ⟨x,θ∗⟩∈[−1,1] for all x∈Dt, and λ≥max(1,L2), then for every δ>0, with probability at least 1−δ,
For any positive definite V, Vt=V+∑s≤tXsXs⊤ and St=∑s≤tηsXs: with probability at least 1−δ, for all t≥0,
∥St∥Vt−12≤2R2log(det(Vt)1/2det(V)−1/2/δ).
Milestones: Theorem 2, the confidence ellipsoids
With probability at least 1−δ, θ∗∈Ct for all t≥0 (first claim). If ∥Xt∥2≤L, then with probability at least 1−δ, for all t, ∥θt−θ∗∥Vt≤Rdlog((1+tL2/λ)/δ)+λ1/2S (second claim, stated here for d≥2).
Significance
Theorem 3 bounds the regret of OFUL by O(dnlogn) with high probability, uniformly over the horizon, so it holds for an unknown horizon without restarting. The bound applies to arbitrary, even adversarially changing, decision sets. Theorem 1 is the ingredient that makes this possible: a deviation bound for a vector-valued martingale, normalized by its own random covariance, that holds for all times simultaneously and whose logarithmic term is a determinant rather than a union-bound count. The same inequality underlies regret analyses of generalized linear bandits, kernelized bandits, linear Markov decision processes and many confidence-sequence constructions.
All three results are proved in the paper's appendices (not included in the source file used here). None of them is formalized in the stated generality. Prove2Me holds the special cases V=λI, R=1, δ<1 of Theorems 1 and 2 (from the Bandit Algorithms textbook series), a pathwise LinUCB regret lemma that assumes the confidence event, and the elliptical potential lemma. A formal proof of Theorem 3 would be the first machine-checked high-probability regret bound for OFUL with the paper's confidence sets.
Difficulty
The actions are chosen adaptively, by an argmax over a data-dependent set, so the sequence Xt has no independence structure and the least-squares estimate is not a sum of independent terms. A fixed-design concentration bound followed by a union bound over time and over a covering of the sphere loses logarithmic factors and does not produce the determinant in the radius; that is the route of the earlier work that Theorem 1 improves. Theorem 1 must hold for all times at once for a quantity normalized by the random matrix Vt, which is itself built from the adaptively chosen actions; a bound for each fixed t does not give it.
Formalization scope
Vectors are Fin d → ℝ, matrices Matrix (Fin d) (Fin d) ℝ, and ∥x∥A is Real.sqrt (x ⬝ᵥ A *ᵥ x). Rounds are indexed t+1 for t:N, so sums over s≤t are sums over Finset.range t at index s + 1, and the time-0 objects are empty sums. The probability space is standard Borel, as Mathlib's conditional sub-Gaussianity (HasCondSubgaussianMGF, variance proxy R2) requires; this is an added hypothesis. Every "with probability at least 1−δ, for all t" is stated as an outer-measure bound ≤δ on the failure event, with the time quantifier inside the event. det(⋅)1/2 is the real square root of the determinant; the matrices inverted are positive definite, so Lean's junk inverse never occurs.
OFUL is a predicate on the whole process: in every round the chosen pair maximizes ⟨x,θ⟩ over Dt×Ct−1, with any tie-breaking. Runs exist whenever the decision sets are nonempty and compact. The measurability of the actions is assumed, as in Theorem 1. The optimal reward ⟨xt∗,θ∗⟩ is the supremum over Dt, finite because of the reward bound.
Two corrections to the printed Theorem 3 are made and disclosed. The printed nL/d is replaced by nL2/d, which is what the determinant–trace bound detVn≤(λ+nL2/d)d gives; for L≤1 the corrected bound implies the printed one. The hypothesis λ≥max(1,L2) is added: for λ<1 the printed logarithm can be negative, and the printed bound would then assert Rn≤0. In the second claim of Theorem 2, d≥2 is added, because at d=1 the claim does not follow from the first claim and the paper's proof is not available.
The goal is a probability bound over the noise, not the pathwise statement "if θ∗∈Ct−1 for all t then Rn≤…". The pathwise statement assumes the confidence event instead of proving it, and is already on the platform. The confidence sets inside the OFUL predicate use the same δ as the conclusion.
Needed infrastructure: maximal inequalities for nonnegative supermartingales, Gaussian integrals of quadratic forms on Rd, log-determinant bounds for sums of rank-one updates, and the determinant–trace inequality. The platform rows BanditAlgorithm.self_normalized_martingale_bound, BanditAlgorithm.least_squares_confidence_ellipsoid and BanditAlgorithm.elliptical_potential_lemma are referenced as tools. Proofs of the milestones, generalizations of the existing special cases to general V and R, and reusable determinant lemmas are all welcome.
Selected references
Y. Abbasi-Yadkori, D. Pál, Cs. Szepesvári, Improved Algorithms for Linear Stochastic Bandits, Advances in Neural Information Processing Systems 24 (NIPS), 2011. https://proceedings.neurips.cc/paper/2011
P. Rusmevichientong, J. N. Tsitsiklis, Linearly Parameterized Bandits, Mathematics of Operations Research 35(2), 2010. https://doi.org/10.1287/moor.1100.0446
Bandit Algorithms XVI: Markov Decision Processes and UCRL2Textbook
The final step from bandits to reinforcement learning: actions now change the state of the world. Chapter 38 of Lattimore–Szepesvári studies online learning in an unknown Markov decision process with S states, A actions and rewards in [0,1]. The optimism principle of Mission II scales up: UCRL2 maintains confidence sets over transition kernels, solves an extended MDP by extended value iteration, and recomputes only when a state-action count doubles. The goal theorem: with probability 1−δ, R^n<CD(M)SAnlog(nSA/δ), where D(M) is the diameter of the MDP — sublinear regret with no prior knowledge of the dynamics. The matching lower bound E[R^n]≥C′DSAn brackets the true complexity of tabular reinforcement learning up to DS.
Bandit Algorithms VII: Lower Bounds for Finite-Armed BanditsTextbook
How well can any algorithm possibly do? Chapters 13–17 of Lattimore–Szepesvári answer with three matching impossibility results. The divergence decomposition identifies the information a policy collects: D(Pνπ,Pν′π)=∑iE[Ti(n)]D(Pi,Pi′). Feeding it into the Bretagnolle–Huber inequality of Mission VI yields the goal theorem — the minimax lower bound Rn≥271(k−1)n over Gaussian bandits, showing MOSS (Mission III) is optimal up to a constant. The same machinery gives the instance-dependent bound of Lai–Robbins type: every consistent policy suffers liminfnRn/logn≥∑i:Δi>0Δi/dinf(Pi,μ∗,Mi), certifying the asymptotic optimality of the UCB of Mission III and KL-UCB of Mission IV, and a high-probability lower bound showing the Exp3-IX guarantees of Mission V cannot be improved.
Optimal Best Arm Identification with Fixed Confidence IV: Asymptotic Optimality of the Track-and-Stop StrategyResearch Paper
Motivation
In best arm identification with fixed confidence, a learner samples K unknown distributions (arms) sequentially and must, as early as possible, name the arm with the largest mean, while being wrong with probability at most a prescribed risk δ. The problem models adaptive A/B/n testing, clinical and simulation-based selection among alternatives, and the "ranking and selection" problem of operations research and simulation optimization. The quantity of interest is the sample complexityEμ[τδ], the expected number of samples a strategy takes before stopping.
Timeline of the question this mission formalizes:
Chernoff (1959) introduced sequential tests based on generalized likelihood ratios for adaptive design of experiments, with a finite set of hypotheses (doi:10.1214/aoms/1177706205).
Kaufmann, Cappé and Garivier (2016, JMLR) proved a change-of-measure lower bound on the sample complexity of every δ-PAC strategy (arXiv:1407.4443).
Garivier and Kaufmann (COLT 2016) identified the exact constant T∗(μ) in that lower bound and gave the first strategy, Track-and-Stop, whose sample complexity matches it asymptotically as δ→0 (arXiv:1602.04589). This mission covers the upper-bound half of that paper.
Setting
A canonical one-parameter exponential family is a family of laws νθ, θ∈Θ, on R with density exp(θx−b(θ)) with respect to a reference measure ξ; b is twice differentiable and strictly convex, and νθ has mean b˙(θ). Bernoulli, Poisson and Gaussian laws with known variance are examples. The divergence d(μ,μ′) is the Kullback–Leibler divergence between the members with means μ and μ′.
A bandit model μ=(μ1,…,μK) assigns a member of the family to each arm. The class S consists of models with a unique optimal arm a∗(μ). At each round t=1,2,… the learner picks an arm At as a function of past observations, observes a reward drawn from that arm's law, and at a stopping time τδ recommends an arm. Na(t) is the number of draws of arm a in the first t rounds and μ^a(t) its empirical mean.
With Alt(μ)={λ∈S:a∗(λ)=a∗(μ)} and ΣK the probability simplex, the characteristic time is
T∗(μ)−1=w∈ΣKsupλ∈Alt(μ)infa=1∑Kwad(μa,λa),
and the maximizer w∗(μ) are the optimal proportions of arm draws.
Track-and-Stop combines two ingredients:
a sampling rule that tracks the plug-in proportions w∗(μ^(t)) while forcing each arm to be drawn about t times: C-Tracking tracks the cumulated sum of projections of w∗(μ^(s)) onto ΣKϵs={w∈ΣK:wa≥ϵs}, ϵs=(K2+s)−1/2/2; D-Tracking draws an under-sampled arm when some Na(t)<t−K/2, and otherwise the arm maximizing twa∗(μ^(t))−Na(t);
Chernoff's stopping rule, which stops at the first t at which some arm a beats every other arm b in a generalized likelihood ratio test, Za,b(t)>β(t,δ), here with β(t,δ)=log(r(t)/δ).
Formalization targets
Goal: Theorem 14 (p. 13)
For α∈[1,e/2] and r(t)=O(tα), Chernoff's stopping rule with β(t,δ)=log(r(t)/δ) combined with C-Tracking or D-Tracking satisfies
Lemma 7 (p. 7): C-Tracking ensures Na(t)≥t+K2−2K and maxa∣Na(t)−∑s<twa∗(μ^(s))∣≤K(1+t).
Lemma 8 (p. 7): D-Tracking ensures Na(t)≥(t−K/2)+−1, and proportions within 3(K−1)ϵ of w∗(μ) after a time tϵ that does not depend on the trajectory, once the plug-in targets are within ϵ.
Proposition 9 (p. 8): under either rule, Na(t)/t→wa∗(μ) almost surely.
Lemma 18 (p. 27): an explicit x with c1x≥log(c2xα) for α∈[1,e/2].
Proposition 13 (p. 11): with any sampling rule whose proportions converge almost surely to w∗, τδ<∞ almost surely and limsupδ→0τδ/log(1/δ)≤αT∗(μ) almost surely.
Significance
Theorem 1 of the same paper shows Eμ[τδ]≥T∗(μ)kl(δ,1−δ) for every δ-PAC strategy, and kl(δ,1−δ)∼log(1/δ). Theorem 14 with α=1 therefore shows that the lower bound is attained: T∗(μ) is the exact asymptotic sample complexity of best arm identification in exponential family models, and Track-and-Stop is asymptotically optimal.
The result is proved in the paper. What is not available is a machine-checked proof for exponential families. The platform already holds a Lean development of the Gaussian case following Lattimore and Szepesvári, Bandit Algorithms, Ch. 33, stated for one existentially chosen policy with a different threshold. This mission asks for the universal statement: every run of either tracking rule, for every exponential family, with the paper's thresholds. The tracking lemmas (Lemmas 15, 7, 8) are deterministic combinatorics and reusable by any tracking-based algorithm.
Difficulty
The obvious argument plugs the almost-sure behaviour of Proposition 13 into an expectation. That step fails: almost-sure convergence of τδ/log(1/δ) does not control E[τδ], because on the rare events where the empirical means are far from μ the stopping time may be very large. Theorem 14 needs a quantitative concentration of μ^(t) on events whose complements have summable probability, which in turn relies on the forced exploration guaranteed by the t lower bounds on Na(t) (the concentration step of App. D, Lemmas 19–20).
A second obstacle is the regularity of w∗: the tracking lemmas only transfer convergence of μ^(t) to convergence of Na(t)/t through the continuity of μ↦w∗(μ) on S, proved from the characterization of w∗ in §2.2 (Proposition 6). The GLR statistic also needs its closed form (7) near μ, which requires the empirical means to lie in the interior of the mean space.
Formalization scope
Model. The exponential family is a structure (ξ,Θ,b) with Θ a nonempty open interval, each νθ normalized, b twice continuously differentiable and b¨>0 on Θ. Openness and b¨>0 are added to the paper's "convex, twice differentiable"; strict convexity is what makes νμ unique. Bandit models are parameter vectors θ∈ΘK with K≥2; arms are indexed 0,…,K−1. S is the set of parameter vectors with a unique arm of largest mean b˙(θa).
Protocol. Policies, the trajectory law Pμ, pull counts, empirical means and T∗(μ) are the platform's published definitions (BanditPolicy, BanditTrajectory, TrackAndStop). T∗ uses Kullback–Leibler divergences of the arm laws over the class S and takes values in [0,∞]. Trajectory coordinate t is round t+1. An arm never drawn has empirical mean 0.
Target map.w∗(μ^(t)) is undefined in the paper when μ^(t)∈/S (an unsampled arm, ties, a mean outside b˙(Θ)). Every tracking statement quantifies over every target map with values in ΣK that returns optimal proportions on S, over every choice of L∞ projections, and over every tie-breaking, including randomized ones.
Stopping rule. The two maxima in Za,b(t) are suprema over Θ in the extended reals. Za,b(t)>β is written without subtracting infinities. The stopping time is the first t≥1 at which the test succeeds, +∞ if none. "r(t)=O(tα)" is r(t)≤Dtα for t≥1; r>0 is added so that log(r(t)/δ) is defined.
Values in [0,∞]. Expectations of τδ, the ratios and T∗ live in [0,∞]. No statement converts them to reals, so an infinite expected stopping time is never read as 0.
Corrections, disclosed. Proposition 9's printed Pw is Pμ. Lemma 18 adds c2/c1α>1 and x>0, without which its expressions are undefined.
Ruled out. Specializing to Gaussian arms, or asserting that some sampling policy achieves the bound, would restate existing platform results and is not this theorem: the goal is about every C-Tracking or D-Tracking run in every exponential family.
Welcome contributions. Exponential-family facts (b˙ is the mean, the KL formula, concentration of empirical means); the continuity of w∗ (Proposition 6, App. A.3); Lemma 17 (App. B.2), from which Lemma 8 follows; the closed form (7) of the GLR statistic.
Selected references
A. Garivier, E. Kaufmann, Optimal Best Arm Identification with Fixed Confidence, COLT 2016, JMLR W&CP 49. arXiv:1602.04589v2
E. Kaufmann, O. Cappé, A. Garivier, On the Complexity of Best-Arm Identification in Multi-Armed Bandit Models, JMLR 17, 2016. arXiv:1407.4443
Optimal Best Arm Identification with Fixed Confidence I: Non-Asymptotic Lower Bound on the Sample ComplexityResearch Paper
Motivation
Best arm identification with fixed confidence is the pure-exploration counterpart of the multi-armed bandit problem. A learner faces K unknown reward distributions ("arms"), samples them sequentially, and must stop and name the arm with the largest mean, being wrong with probability at most a prescribed δ. The question is how many samples this requires. It arises in adaptive A/B testing, in the selection of the best of several simulated systems (ranking and selection in simulation optimization), and in clinical trials that must declare the best treatment with a guaranteed error rate.
Lower bounds for this problem were first stated in terms of the gaps between means (Mannor and Tsitsiklis, 2004). Kaufmann, Cappé and Garivier (2016) replaced ad hoc changes of measure by a single "transportation" lemma relating expected sample counts, Kullback–Leibler divergences and the error probability. Garivier and Kaufmann (COLT 2016, arXiv:1602.04589v2) combine this lemma over all alternative models at once, in the spirit of Graves and Lai (1997), and obtain a lower bound whose constant T∗(μ) is exactly matched, as δ→0, by their Track-and-Stop strategy. This mission formalizes that lower bound (Theorem 1 of the paper, p. 3).
Setting
A canonical one-parameter exponential family is given by a reference measure ξ on R, an open interval Θ⊂R and a function b, twice continuously differentiable on Θ with b¨>0, such that the laws νθ with density exp(θx−b(θ)) with respect to ξ are probability measures for θ∈Θ. The mean of νθ is b˙(θ). Bernoulli laws and Gaussian laws of known variance are examples.
A bandit model is a vector θ=(θ1,…,θK)∈ΘK; arm a returns i.i.d. rewards with law νθa and mean μa=b˙(θa). Arm a∗(μ) is the unique optimal arm if μa∗>μa for every a=a∗. Let S be any set of bandit models of the family each having a unique optimal arm, and put Alt(μ)={λ∈S:a∗(λ)=a∗(μ)}.
A strategy consists of a sampling ruleπ (the arm At drawn at round t depends, possibly with extra randomization, on the first t−1 observations), a stopping timeτ of the natural filtration Ft=σ(A1,X1,…,At,Xt), and an Fτ-measurable decisiona^τ. It is δ-PAC on S if for every μ∈S, Pμ(τ<∞)=1 and Pμ(a^τ=a∗(μ))≤δ. Na(t) is the number of draws of arm a among the first t rounds.
Write d(μa,λa)=KL(νθa,νλa) for the divergence between two arm laws, kl(x,y)=xlogyx+(1−x)log1−y1−x, and ΣK for the probability simplex on the K arms. The characteristic time is defined by eq. (1):
T∗(μ)−1=w∈ΣKsupλ∈Alt(μ)infa=1∑Kwad(μa,λa).
Formalization targets
Goal: Theorem 1 (p. 3)
For δ∈(0,1/2], every δ-PAC strategy on S and every μ∈S,
Eμ[τ]≥T∗(μ)kl(δ,1−δ).
The statement fixes no constant beyond those of the paper, and it holds for every δ, not only in the limit.
Milestone: eq. (2) (p. 4)
For every λ∈S with a∗(λ)=a∗(μ),
a=1∑Kd(μa,λa)Eμ[Na(τ)]≥kl(δ,1−δ).
This is Lemma 1 of Kaufmann et al. (2016), which the paper quotes without proof; Theorem 1 follows from it for every alternative simultaneously.
Significance
Theorem 1 identifies T∗(μ) as the exact problem-dependent complexity of fixed-confidence best arm identification: since kl(δ,1−δ)∼log(1/δ), it gives liminfδ→0Eμ[τδ]/log(1/δ)≥T∗(μ), and the paper's Track-and-Stop strategy attains this rate (the subject of mission IV of this series). The bound also explains which proportions of draws an optimal strategy must use: the maximizer w∗(μ) of eq. (1) (mission II).
Both results are proved in the literature. Neither is formalized in the paper's generality. The platform holds the textbook form of Lattimore and Szepesvári (Theorem 33.5), which is stated for an arbitrary class with the weaker constant log(1/(4δ)); for δ≤1/2, kl(δ,1−δ)≥log(1/(2.4δ))>log(1/(4δ)), so Theorem 1 is strictly stronger. A formal proof here yields the transportation lemma for exponential families on the platform's infinite-horizon bandit model, which later missions (II–IV, and any lower bound by change of measure) can reuse.
Difficulty
The obvious proof applies the finite-horizon divergence decomposition KL(Pμn,Pλn)=∑aEμ[Na(n)]d(μa,λa) at a deterministic horizon n. That fails here: τ is random and unbounded, the decision is Fτ-measurable, and the relevant divergence is between the laws of the stopped observations. The step from a fixed horizon to a stopping time, together with the data-processing inequality that turns the error guarantees under two models into kl(δ,1−δ), is the central difficulty. A second, smaller difficulty is to identify the paper's divergence d and its means b˙(θ) with the measure-theoretic KL divergence and mean of the arm laws of the exponential family.
Formalization scope
Lean namespace OptimalBAI.LowerBound. The bandit protocol is the platform's (BanditAlgorithm.BanditPolicy, banditTrajMeasure, IsBanditStoppingTime, IsSoundBAI, baiComplexity); kl is the platform's bernoulliRelativeEntropy and Na(t) is trajPullCount. Conventions:
arms are Fin K, 0-based (the paper's arm a is index a−1); trajectory coordinate t is round t+1;
Θ is a nonempty open interval and b¨>0 on Θ (added: the paper says b is convex and twice differentiable; strict convexity is what makes "the unique distribution with mean μ" meaningful); the paper's d is written as the KL divergence of the arm laws (its first equality on p. 3), and the unique optimal arm is defined through the parameter means b˙(θa);
S is an arbitrary set of models with a unique optimal arm, not the specific set the paper fixes from p. 4 on;
δ-PAC keeps both halves of the paper's definition (almost-sure stopping and error at most δ);
T∗(μ), divergences and expectations of τ take values in [0,∞], never truncated to reals; T∗=0 when Alt(μ)=∅ and T∗=∞ when the supremum in eq. (1) is 0;
δ≤1/2 is added. The paper states δ∈(0,1), but the theorem and eq. (2) are false for δ∈(1/2,1): with two unit-variance Gaussian arms, drawing arm 1 once and naming arm 1 exactly when the fractional part of the reward is below 1/2 is 0.9-PAC, while T∗(μ)→∞ as the two means merge. At δ=1/2 the bound is 0.
A statement with log(1/(4δ)) in place of kl(δ,1−δ), or restricted to Gaussian arms, is the platform's existing textbook theorem and does not count as this mission's goal; nor does any version that drops the almost-sure stopping clause or truncates E[τ] or T∗ to real numbers.
Needed infrastructure: the transportation lemma at a stopping time (data processing for KL through an Fτ-measurable event, Wald-type identity for the stopped log-likelihood ratio), the identities "mean of νθ=b˙(θ)" and "KL of two family members =b(θ′)−b(θ)−b˙(θ)(θ′−θ)", and E[τ]=∑aE[Na(τ)]. All are reusable beyond this mission. Contributions of any of these lemmas, of eq. (2) alone, or of the Gaussian and Bernoulli special cases as stepping stones are welcome.
Selected references
A. Garivier, E. Kaufmann, Optimal Best Arm Identification with Fixed Confidence, COLT 2016 (JMLR W&CP 49), arXiv:1602.04589v2. https://arxiv.org/abs/1602.04589
E. Kaufmann, O. Cappé, A. Garivier, On the Complexity of Best-Arm Identification in Multi-Armed Bandit Models, Journal of Machine Learning Research 17(1), 2016. https://arxiv.org/abs/1407.4443
T. L. Graves, T. L. Lai, Asymptotically Efficient Adaptive Choice of Control Laws in Controlled Markov Chains, SIAM Journal on Control and Optimization 35(3), 1997. https://doi.org/10.1137/S0363012994275440
S. Mannor, J. N. Tsitsiklis, The Sample Complexity of Exploration in the Multi-Armed Bandit Problem, Journal of Machine Learning Research 5, 2004. https://www.jmlr.org/papers/v5/mannor04b.html
Improved Algorithms for Linear Stochastic Bandits II: Constant High-Probability Regret of UCB(δ)Research Paper
Motivation
The stochastic multi-armed bandit is the basic model of sequential decision making under uncertainty: a learner repeatedly chooses one of d actions, observes a noisy reward for the chosen action only, and must balance exploring poorly known actions against exploiting the one that currently looks best. It underlies adaptive clinical trials, online advertising, recommendation, dynamic pricing and many simulation-optimization procedures in operations research.
The standard algorithm is UCB (Auer, Cesa-Bianchi and Fischer, 2002, doi:10.1023/A:1013689704352), which plays the arm with the largest upper confidence bound on its mean. Its confidence widths grow with the current time t (or with a horizon n fixed in advance), and its guarantee is on the expected regret, which grows like logn. Lai and Robbins (1985, doi:10.1016/0196-8858(85)90002-8) showed that logarithmic growth of the expected regret cannot be avoided by a consistent algorithm.
Abbasi-Yadkori, Pál and Szepesvári (NIPS 2011) proved a self-normalized concentration inequality for vector-valued martingales that holds uniformly over time. Section 6 of their paper applies it to the d-armed bandit. The resulting confidence intervals depend on neither the horizon nor the current time. The UCB variant built on them, UCB(δ), has pseudo-regret bounded by a constant, independent of the horizon, on a single event of probability at least 1−δ. This mission formalizes that section.
Setting
There are d≥1 arms with unknown meansμ1,…,μd∈R. Write μ∗=max1≤i≤dμi for the best mean and Δi=μ∗−μi≥0 for the gap of arm i.
Randomness lives on a probability space (Ω,F,P) with a filtration (Ft)t≥0. In round t=1,2,… the learner plays an arm It that is Ft−1-measurable and receives the reward μIt+ηt. The noise ηt is Ft-measurable and conditionally 1-sub-Gaussian:
E[eληt∣Ft−1]≤eλ2/2for all λ∈R.
The noise need not be independent or identically distributed across rounds, and the rewards need not be bounded.
After t rounds, Ni,t is the number of plays of arm i and Xi,t is the average reward received from it. For a confidence level δ>0 the confidence width is
ci,t=Ni,t21+Ni,t(1+2logδd(1+Ni,t)1/2)(3)
with ci,t=+∞ when Ni,t=0. UCB(δ) plays, in round t, an arm that maximizes Xi,t−1+ci,t−1; in particular every arm is played once before any comparison is made. The pseudo-regret after n rounds is
Rn=t=1∑n(μ∗−μIt).
This is the linear bandit of the paper with the standard basis of Rd as decision set and θ∗=μ.
Formalization targets
Goal: Theorem 7 (constant regret of UCB(δ))
For every δ>0, with probability at least 1−δ, for all n≥0 simultaneously,
Rn≤i:Δi>0∑(3Δi+Δi16logΔiδ2d).
The right-hand side depends only on the gaps, d and δ.
Milestone: Lemma 6 (confidence intervals)
For any adapted choice of arms (not only UCB(δ)) and every δ>0, with probability at least 1−δ,
∣Xi,t−μi∣≤ci,tfor all arms i and all t≥0.
Significance
The result. Theorem 7 shows that, once a confidence level is fixed, a UCB-type algorithm can stop paying for exploration after finitely many rounds: with probability 1−δ the total regret over an infinite horizon is bounded. The paper notes that this does not contradict the Lai–Robbins lower bound, which concerns expected regret; on the failure event of probability δ the regret may grow linearly. Lemma 6 is the practical content behind this: anytime-valid confidence intervals for adaptively sampled means under martingale noise, usable for stopping rules, best-arm identification and any sequential procedure that inspects its estimates at data-dependent times.
Formalizing it. Both results are proved in the paper's appendices (F and G). No machine-checked proof of either is known to exist. The platform already has the self-normalized martingale bound for V=λI and unit sub-Gaussian noise (BanditAlgorithm.self_normalized_martingale_bound, listed here as a reference item), and UCB bounds with horizon-dependent widths and in-expectation or pathwise conclusions. Neither is Theorem 7. The mission produces a formal account of time-uniform confidence intervals for adaptively sampled arm means, and of a regret bound that is uniform in the horizon.
Difficulty
Two points separate this from textbook UCB analyses. First, the number of samples Ni,t of an arm is itself random and depends on past noise, so a fixed-sample concentration inequality with a union bound over times does not give widths that are free of t: a union bound over all t costs a factor that diverges. The time-uniform event must come from a maximal (self-normalized) inequality applied to the martingale ∑sηs1{Is=i}. Second, the regret statement is uniform in n with constants depending on the gaps; converting a condition of the form "c(N)≥Δi/2" into an explicit bound on N requires solving an inequality in which N appears both polynomially and inside a logarithm, and the explicit constants 3 and 16 must come out of that step.
Formalization scope
The Lean development lives in the namespace ImprovedLinBandits.UCBDelta.
Arms are Fin d with 0 < d in the goal; means are μ : Fin d → ℝ, μ∗ is ⨆ i, μ i (the maximum over a nonempty finite type).
Rounds are indexed t + 1 for t : ℕ: the arm I (t + 1) is ℱ t-measurable, the noise η (t + 1) is ℱ (t + 1)-measurable and satisfies Mathlib's HasCondSubgaussianMGF (ℱ t) … (η (t + 1)) 1 P. Values at index 0 are unused.
The sample space is assumed to be a standard Borel space, which Mathlib's conditional sub-Gaussianity requires; this hypothesis is not in the paper.
"With probability at least 1−δ, for all …" is stated as: the outer probability of the failure event, with the quantifiers over arms and times inside it, is at most δ. For δ≥1 the statements are trivially true, as in the paper.
The rule (4) is read with the statistics of rounds 1,…,t−1, because the printed Xi,t, ci,t already count round t. An unplayed arm, whose width is +∞ in the paper, is played before the indices are compared. Every tie-breaking rule is allowed, and measurability of the chosen arm is assumed rather than derived.
The printed Theorem 7 has no quantifier on n; it is formalized in the uniform-in-n form, matching the section's claim of constant regret and the time-uniform event of Lemma 6.
Lean's division by zero makes Xi,t and ci,t equal to 0 when Ni,t=0. Lemma 6 therefore excludes Ni,t=0 explicitly (where the paper's inequality is vacuous), and the run predicate handles unplayed arms separately. The regret bound sums over arms with Δi>0, so its divisions are well defined.
A trivializing formalization is ruled out: the confidence event is not assumed as a hypothesis of Theorem 7, the i.i.d. bandit model is not substituted for the martingale noise model, and no bound on the means or rewards is imposed.
Useful infrastructure for solvers: the self-normalized bound of Theorem 1 in the scalar case (d=1, λ=1, As=1{Is=i}, Vt=1+Ni,t), a union bound over arms, and elementary inequalities inverting N↦N21+N(1+2log(d1+N/δ)). Lemmas about pull counts and empirical means under adaptive sampling are reusable beyond this mission and are welcome as contributions.
P. Auer, N. Cesa-Bianchi, P. Fischer, Finite-time Analysis of the Multiarmed Bandit Problem, Machine Learning 47, 2002. https://doi.org/10.1023/A:1013689704352
Logarithmic Regret Algorithms for Online Convex Optimization 1: Logarithmic Regret of Online Gradient DescentResearch Paper
Motivation
Online convex optimization is a repeated game between a learner and an adversary. In each round t=1,…,T the learner commits to a point xt of a convex set P⊆Rn; only then is a convex cost function ft revealed, and the learner pays ft(xt). The learner is judged by its regret: its total cost minus the total cost of the best fixed point chosen in hindsight. The model covers online portfolio selection, online regression, prediction with expert advice and the analysis of stochastic gradient methods, and it is the standard language of online learning theory.
Zinkevich (ICML 2003) showed that projected gradient descent with step sizes of order 1/t has regret O(GDT) for any convex costs with gradients bounded by G on a set of diameter D, and this rate cannot be improved for linear costs. Hazan, Agarwal and Kale (Machine Learning 69, 2007) asked what curvature buys. Their first result, the subject of this mission, is that the same algorithm with the faster step sizes 1/(Ht) has regret only logarithmic in T once every cost function is H-strongly convex. The paper's other three results (the Online Newton Step, Follow the Approximate Leader and Exponentially Weighted Online Optimization, for exp-concave costs) are the subject of the other missions of this series.
Setting
Fix n∈N and a nonempty, closed, bounded, convex set P⊆Rn, with the Euclidean norm ∥⋅∥2 (§2.1, p. 171).
Cost functions. A sequence f1,f2,⋯:Rn→R. Write ∇ft(x) for the gradient and ∇2ft(x) for the Hessian.
Gradient bound. A number G with ∥∇ft(x)∥2≤G for all x∈P and all rounds t (p. 172).
H-strong convexity (p. 172). For H>0, f is H-strongly convex on P when it is twice differentiable and ∇2f(x)⪰HIn for every x∈P, i.e. v⊤∇2f(x)v≥H∥v∥22 for all v.
Euclidean projection.ΠP(y) is the point of P nearest to y, ΠP(y)=argminx∈P∥x−y∥2 (IsProj P y z).
Online Gradient Descent (Fig. 1, p. 174), with step sizes η1,η2,…: x1∈P is arbitrary, and in iteration t>1
xt=ΠP(xt−1−ηt∇ft−1(xt−1))
(IsOGDRun P η f x).
Regret (p. 171): RegretT=∑t=1Tft(xt)−minx∈P∑t=1Tft(x), and RegretT(OGD) is its supremum over all cost sequences.
Formalization targets
Goal: Theorem 1 (p. 175)
For H>0, cost functions that are H-strongly convex on P with gradients bounded by G on P, and any run of Online Gradient Descent whose step after round t is ηt+1=1/(Ht), for every T≥1 and every u∈P:
t=1∑T(ft(xt)−ft(u))≤2HG2(1+logT).
This is LogRegretOCO.OGD.ogd_regret_bound. The constant is the paper's, and at T=1 the bound reads G2/(2H).
Milestones
Eq. (1), the strong-convexity inequality: for x,y∈P, 2(f(x)−f(y))≤2∇f(x)⊤(x−y)−H∥y−x∥22.
Lemma 8 with A=In: for a convex P, z=ΠP(y) and a∈P, ∥y−a∥22≥∥z−a∥22. This is already on the platform, proved, as UnderstandingML.projection_lemma, and is reused as a reference item.
Eq. (2), the one-step inequality: for z=ΠP(x−ηg), η>0, ∥g∥2≤G and u∈P, 2g⊤(x−u)≤(∥x−u∥22−∥z−u∥22)/η+ηG2.
The last step of the argument is the harmonic-sum bound ∑t=1T1/t≤1+logT, which Mathlib already provides (harmonic_le_one_add_log).
Significance
The result. Theorem 1 separates two regimes of online convex optimization: Θ(T) regret for general convex costs and O(logT) for strongly convex ones, with an algorithm that costs one gradient and one projection per round. Through the online-to-batch conversion, the same step-size schedule gives the O(logT/T) rate of stochastic gradient descent on strongly convex objectives. The theorem is also the reference point for the paper's weaker exp-concavity assumption, under which the other three algorithms obtain O(nlogT) regret.
Formalizing it. The result is proved and well known; what this mission adds is a machine-checked proof with the paper's exact constant. The Prove2Me platform holds a statement of the textbook version of this theorem (Hazan, Introduction to Online Convex Optimization, Theorem 3.3), OnlineConvexOpt.FirstOrder.online_gradient_descent_strongly_convex_regret, but it is Disproved: its regret is written with a real infimum over the decision set, which Lean evaluates to a junk value. No proved version of the logarithmic bound is on the platform. The strong-convexity inequality, the one-step projected-gradient inequality and the telescoping argument are reusable by every later formalization of gradient methods on the platform.
Difficulty
The argument is short, and its difficulty lies in the bookkeeping. The telescoping sum of squared distances cancels exactly only with the right step-size indexing: the step taken after round t must be 1/(Ht). Reading Fig. 1 literally with ηt=1/(Ht) makes the step after round t equal to 1/(H(t+1)), and the sum then leaves an uncancelled term 2H∥x1−x∗∥22 that the printed bound does not contain. Strong convexity is a statement about the Hessian, so the curvature inequality (1) needs a second-order Taylor expansion along a segment of P; convexity of P keeps the segment inside the set where the Hessian bound holds. The projection step needs the obtuse-angle property of Euclidean projection onto a convex set.
Formalization scope
Points are EuclideanSpace ℝ (Fin n), so ∥⋅∥ is the Euclidean norm (the sup norm of Fin n → ℝ would change the gradient bound). Cost functions are functions on all of Rn, as the paper's use of gradients and Hessians presupposes; the gradient is Mathlib's gradient, and the Hessian quadratic form is the second Fréchet derivative applied to (v,v). Rounds are 1-based: x 0, f 0 and η 1 are never read, and sums run over Finset.Icc 1 T. The run of the algorithm is a predicate on the whole trajectory, required at every round; the projection is a predicate (nearest point of P), which is unique for nonempty closed convex P.
Choices and corrections relative to the printed text:
Step-size index. Theorem 1 prints "step sizes ηt=Ht1", while its proof sets ηt+1=1/(Ht) for the step after round t. The goal uses the proof's indexing, as a hypothesis η (t + 1) = 1 / (H * t) for t≥1 on the step sizes of Fig. 1.
Eq. (2). The paper prints "5∇t⊤(xt−x∗)"; the 5 is a typo for 2, and the milestone states 2. The verbatim milestone text keeps the printed 5.
Regret. Regret is stated against every comparator u∈P; the minimum over the compact set P is attained, so this is the paper's statement. The expectation in the paper's regret is vacuous for this deterministic algorithm, and the supremum over cost sequences is the universal quantifier over f.
Hypotheses. Strong convexity and the gradient bound are required for the rounds 1,…,T only. Convexity of each ft, a standing assumption of §2.2, follows from H-strong convexity on the convex set P and is not added. No diameter bound enters Theorem 1.
A regret bound written with a real ⨅/sInf over P, or a bound for an arbitrary sequence satisfying Eq. (2) instead of a run of the paper's algorithm, would not be Theorem 1; the goal quantifies over every comparator in P and carries the run predicate, the Hessian hypothesis and the gradient bound. The hypotheses are jointly satisfiable: on the closed unit ball, ft(x)=2H∥x∥22 with G=H meets all of them.
Contributions welcome: proofs of the two inequalities and of the goal; a general projection lemma for positive semidefinite A (Lemma 8 in full, needed by the Online Newton Step mission) is reusable beyond this mission.
Selected references
E. Hazan, A. Agarwal, S. Kale, Logarithmic regret algorithms for online convex optimization, Machine Learning 69 (2007), 169–192. https://doi.org/10.1007/s10994-007-5016-8
S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press 2014 (Lemma 14.9, the projection lemma). https://doi.org/10.1017/CBO9781107298019
Chapter 2's tail bounds mostly rest on moment-generating-function control, obtained either
directly (sub-Gaussianity) or through explicit combinatorial arguments (Hoeffding, bounded
differences). The entropic method offers a different, more structural route: bound a
specific information-theoretic quantity — the φ-entropy of eλX — and convert
that bound mechanically into a tail bound via a short ODE argument (the Herbst argument). This
method's real payoff appears once it is combined with the tensorization property of entropy
across independent coordinates, which is what lets it handle Lipschitz functions of many
independent variables — including cases, such as separately convex functions, that elude the
purely martingale-based techniques of Chapter 2. This mission formalizes the entropic method's
two foundational entropy-to-tail conversions and its central Lipschitz-concentration
application, following Wainwright, High-Dimensional Statistics: A Non-Asymptotic Viewpoint
(Cambridge University Press, 2019), Chapter 3.
Setting
For φ(u):=ulogu (u>0), φ(0):=0, the φ-entropy of a nonnegative
random variable Z is H(Z):=E[ZlogZ]−E[Z]logE[Z] (Eqs. (3.1)-(3.2)).
Writing φX(λ):=E[eλX] for the moment generating function of X,
the entropy of eλX has the explicit form H(eλX)=λφX′(λ)−φX(λ)logφX(λ) (Eq. (3.3)).
A function f:Rn→R is separately convex if, for each coordinate k, the
univariate function obtained by fixing every coordinate but the k-th is convex — strictly
weaker than joint convexity of f itself. f is L-Lipschitz with respect to the Euclidean
norm if ∣f(x)−f(x′)∣≤L∥x−x′∥2 for all x,x′.
Let {Xi}i=1n be independent, each supported on [a,b], and f separately convex and
L-Lipschitz. Then for all δ>0,
P[f(X)≥E[f(X)]+δ]≤exp(−4L2(b−a)2δ2).
Milestone — Proposition 3.2 (the Herbst argument)
If H(eλX)≤21σ2λ2φX(λ) for all λ∈I
(I=[0,∞) or R), then logE[eλ(X−E[X])]≤21λ2σ2 for all λ∈I — the basic entropy-to-sub-Gaussian-tail
conversion.
Milestone — Proposition 3.3 (the Bernstein entropy bound)
The sub-exponential analogue: if H(eλX)≤λ2{bφX′(λ)+φX(λ)(σ2−bE[X])} for λ∈[0,1/b), then logE[eλ(X−E[X])]≤σ2λ2(1−bλ)−1 on the same range.
Significance
Propositions 3.2 and 3.3 are the two basic entropy-to-tail conversions the entire chapter's
entropic method rests on — every subsequent Lipschitz-concentration result in the chapter
(including Theorem 3.4 and the more advanced Theorem 3.24) is obtained by first establishing an
entropy bound of one of these two forms and then invoking the corresponding proposition.
Theorem 3.4 is itself the direct analogue, for independent bounded variables, of Chapter 2's
Gaussian Lipschitz concentration (Theorem 2.26) — but crucially requires the extra hypothesis of
separate convexity, which the Gaussian case does not need and which cannot be dropped in
general.
Formalizing it. No faithful prior art exists on the platform. The one candidate flagged in
BRIEF.md, Talagrand.lipschitz_concentration, was read in full: it is a weighted-Hamming-
distance concentration bound for functions on a finite-alphabet product spaceFin n → α,
proved via Talagrand's convex-distance method — a different underlying space (finite alphabet
vs. real-valued bounded coordinates) and a different Lipschitz norm (weighted Hamming vs.
Euclidean) from Theorem 3.4, and not reused here. A search for "log-Sobolev" and "Herbst" turned
up bousquet_herbst_cgf_le_phi_via_herbst/bousquet_herbst_cgf_le_phi_double_integration: these
are abstract calculus lemmas about a generic function G satisfying an ODE-type growth
condition (G′′≤vex), concluding G(L)≤v(eL−1−L) — a genuinely different statement
shape from Proposition 3.2/3.3's entropy-to-CGF conversions (which conclude a quadratic, not
exponential, bound on logE[eλ(X−EX)]), and not a faithful match. All
three theorems here are drafted as open goals (:= by sorry).
Difficulty
The naive approach to Theorem 3.4 — try to adapt the bounded-differences (martingale) method of
Chapter 2 directly — fails, because the bounded-differences method needs f to have small
coordinatewise oscillation in an absolute sense, while separate convexity alone gives no such
uniform bound (a separately convex function can vary arbitrarily fast within the interior of its
domain, only its slope is controlled by the Lipschitz condition). The entropic method
sidesteps this by working with the φ-entropy of eλf(X) directly: entropy has
a tensorization property across independent coordinates (not itself part of this mission, but
what the entropic method's proof of Theorem 3.4 uses) that reduces a multivariate entropy bound
to a sum of "one coordinate at a time" contributions, each of which convexity and the Lipschitz
condition jointly control — a route with no analogue in the bounded-differences approach.
Formalization scope
Separate convexity and Euclidean-Lipschitzness are both restated locally in this chapter's own
sub-namespace (HighDimStat.Concentration), per this book series' rule against importing
another chapter's draft definitions, even though Chapter 2 already defines an
IsLLipschitz for the same Euclidean condition. φ_X'(\lambda)$ (Proposition 3.3) is realized via Mathlib's deriv, a legitimate way to state a hypothesis on a derivative without separately proving differentiability, appropriate at the draft-statement stage. Explicit Integrable`
hypotheses guard the Bochner integral's junk value on non-integrable functions throughout (trap
2), not literal in the book's own propositions but implied by what "the entropy H(eλX) exists" (an explicit qualifier the book itself makes when introducing Eq. (3.2)) means.
Goal substitution, disclosed.BRIEF.md recommends Theorem 3.24 (the two-sided, jointly
convex analogue) as the primary goal, but explicitly names Theorem 3.4 as a fallback "if 3.24's
dependence on the unnumbered transportation-cost inequality (Eq. 3.73, attributed to Samson)
proves too heavy to state faithfully in the time available." Theorem 3.24's proof route depends
on Theorem 3.19 (a general "transportation cost implies concentration" result for an abstract
metric measure space, itself needing a from-scratch formalization of the transportation-cost
inequality (3.58) and the concentration function αP,(X,ρ)) plus the
unproven-in-chapter Eq. (3.73). Building this full stack faithfully was judged to exceed this
chunk's time budget; Theorem 3.4 is drafted instead, using this mission's own budget on
Propositions 3.2 and 3.3 (the two most load-bearing entropy-to-tail conversions of the chapter)
rather than the heavier transportation-cost machinery. Theorem 3.19, Theorem 3.24, and Eq.
(3.73) are all out of scope for this mission and named here as natural follow-on work.
Selected references
M. J. Wainwright, High-Dimensional Statistics: A Non-Asymptotic Viewpoint, Cambridge
University Press, 2019. DOI: 10.1017/9781108627771.
Chapter 3.
M. Ledoux, The Concentration of Measure Phenomenon, American Mathematical Society, 2001.
I. Herbst, unpublished (the argument bearing his name is attributed in Ledoux (2001) and
standard references on log-Sobolev inequalities).
Markov Entanglement: Value Decomposition Error in Multi-agent MDPsResearch Paper
Value decomposition — approximating the value of a joint state by a sum of per-agent local values — is a staple of multi-agent dynamic programming and reinforcement learning, from index policies for restless bandits to modern MARL architectures, yet it is normally used without justification. Chen and Peng (arXiv:2506.02385) supply one. They show a multi-agent MDP admits an exact value decomposition precisely when its transition matrix is not entangled — a notion built in direct analogy with quantum entanglement — and then turn that qualitative characterisation into a quantitative one: a measure of Markov entanglement bounds the decomposition error in general. This mission formalizes that core theory. The goal is Theorem 6, the general N-agent bound in the occupancy-weighted norm; the milestones are the equivalence between separability and exact decomposition, the perturbation machinery that carries a one-step transition error into a value-function error, and the extensions to shared global state and shared rewards. The paper's restless-bandit application, which needs mean-field machinery of its own, is left to a second mission in the series.
Bandit Algorithms XIII: Pure Exploration and Best-Arm IdentificationTextbook
Sometimes reward during learning is irrelevant — a pharmaceutical company running phase-II trials only cares about identifying the best treatment, as quickly and as reliably as possible. Chapter 33 of Lattimore–Szepesvári formalizes fixed-confidence best-arm identification: a policy together with a stopping time τ and a recommendation must be sound (wrong with probability at most δ) while minimizing E[τ]. The information-theoretic complexity is c∗(ν)−1=supα∈Pk−1infν′∈Ealt(ν)∑iαiD(νi,νi′): every sound strategy needs E[τ]≥c∗(ν)log4δ1, and the Track-and-Stop algorithm — the goal theorem — achieves limδ→0E[τ]/log(1/δ)=c∗(ν) exactly. The mission also covers the fixed-budget counterpart, sequential halving.
Bandit Algorithms IV: Bernoulli Bandits and KL-UCBTextbook
When rewards are binary — a click or no click, a cure or no cure — the subgaussian machinery of Missions I–III is not tight: the variance of a Bernoulli arm degrades near the boundary of [0,1], and the correct exponential rate is governed by the binary relative entropy d(p,q)=plogqp+(1−p)log1−q1−p rather than a squared distance. Chapter 10 of Lattimore–Szepesvári develops Chernoff's tail bound in its information-theoretic form and the KL-UCB algorithm, whose upper confidence bounds are level sets of d. The goal theorem shows KL-UCB attains
n→∞limsupRn/logn=i:Δi>0∑Δi/d(μi,μ∗)
— asymptotic optimality with exactly the constant demanded by the lower bound of Mission VII, strictly improving subgaussian UCB on every Bernoulli instance.
Every lower bound in bandit theory rests on one question: how hard is it to tell two probability measures apart from a sample? The answer is quantified by the relative entropy D(P,Q), and the sharpest elementary tool is the Bretagnolle–Huber inequality: for any event A, P(A)+Q(Ac)≥21exp(−D(P,Q)) — no test can distinguish P from Q with total error probability below 21e−D(P,Q). This mission formalizes Chapter 14 of Lattimore–Szepesvári: the Bretagnolle–Huber inequality (the goal theorem, proved via Le Cam's inequality ∫p∧q≥21(∫pq)2), Pinsker's inequality δ(P,Q)≤D(P,Q)/2, and the closed-form divergences between Gaussians and Bernoullis. These half-page inequalities power every impossibility result in Missions VII, XI and beyond.
Variance-based Regularization with Convex Objectives III: Localized-Rademacher Risk Bounds for the Robust MinimizerResearch Paper
Why variance-regularized risk bounds
In statistical learning, one picks a function f from a class F to make the population risk E[f] small, with access only to an i.i.d. sample x1,…,xn from an unknown distribution P. Empirical risk minimization replaces E[f] by the empirical mean EPn[f], and its classical guarantees decay like 1/n regardless of how concentrated f is. Bernstein-type inequalities show that the deviation of EPn[f] from E[f] scales with the standard deviation of f, so a procedure that minimizes "empirical risk plus a standard-deviation penalty" can, in principle, achieve faster rates when the variance at the optimum is small (Maurer and Pontil, 2009). The penalized objective is non-convex even when every f is convex in its parameters, which makes it hard to optimize.
J. C. Duchi and H. Namkoong (arXiv:1610.02581v3, 2017) replace the penalty by a distributionally robust objective: the worst-case risk over all reweightings of the sample within a χ2-divergence ball. This objective is convex whenever the losses are, and (Theorem 1 of the paper) it equals the empirical mean plus a standard-deviation penalty up to an error of order 1/n. This mission formalizes the paper's guarantee for the minimizer of that robust objective in terms of localized Rademacher complexities (Section 3.2, Theorem 4), the sharpest of the paper's three generalization analyses. It is the third of four missions on the paper.
Setting
Let P be a probability measure on a measurable space X and x1,…,xn, n≥1, an i.i.d. sample from P with empirical distribution Pn. Let M≥1 and let F be a collection of measurable functions f:X→[0,M] (losses).
The χ2 ball of radius ρ≥0 is the set Pn of weight vectors p∈Rn with pi≥0, ∑ipi=1 and 21∑i(npi−1)2≤ρ; equivalently, the distributions P on the sample with Dϕ(P∥Pn)≤ρ/n for ϕ(t)=21(t−1)2.
The robust risk of f is supP:Dϕ(P∥Pn)≤ρ/nEP[f]=supp∈Pn∑ipif(xi), and a robust minimizerf minimizes it over F.
The empirical Rademacher complexity is Rn(F)=Eε[supf∈Fn1∑iεif(xi)] with i.i.d. uniform signs εi∈{−1,1}, and E[Rn(F)] averages it over the sample.
A function ψ:R+→R+ is sub-root if it is nonnegative, nondecreasing, and r↦ψ(r)/r is nonincreasing on r>0.
The localization inequality (20) asks that, for all r≥0,
ψn(r)≥E[Rn({cf:f∈F,c∈[0,1],E[c2f2]≤r})],
with ψn sub-root, and rn⋆>0 is a point with rn⋆≥ψn(rn⋆).
Formalization targets
Goal: Theorem 4, inequality (23), as its proof establishes it
Let 0<t<n and let ρ satisfy (21): nρ≥8(n45M(t+log⌈logtn⌉)+18rn⋆). With probability at least 1−4e−t, every robust minimizer f satisfies
In attack order: Bousquet's form of Talagrand's inequality (Lemma B.2); the elementary root bound (Lemma D.4); the contraction principle (Lemma D.5, a published theorem); the uniform Bernstein inequality with Rademacher complexity (Lemma D.1); its localized version in terms of rn⋆ (Lemma D.2); localized second-moment bounds (Lemma D.3); the deterministic expansion (10) of Theorem 1,
The bound (23) says that the robust minimizer competes with the best trade-off between risk and standard deviation in the class, and that the complexity of the class enters only through the fixed point rn⋆ of a localized complexity bound. For bounded VC classes rn⋆ is of order ndlog(n/d) (Bartlett, Bousquet and Mendelson, 2005, Corollary 3.7), so when the optimal function has small variance the excess risk is of order ρ/n, faster than the 1/n of uniform covering arguments; and localized complexities apply to classes, such as balls of reproducing kernel Hilbert spaces, whose covering numbers are too large for the covering-number analysis of the paper's Theorem 3 (mission II of this series).
The paper's result is proved, not open. No part of it, and none of the localized-complexity machinery of Bartlett, Bousquet and Mendelson, is formalized in Lean or Mathlib to our knowledge. The mission produces a checked version of the theorem with every constant explicit and, along the way, the localization lemmas D.1–D.3, which are reusable for any localized-complexity analysis. Reading the proof also exposed three arithmetic slips in the printed statements; the mission states what the proof establishes (see Formalization scope).
Difficulty
The obvious route applies a uniform concentration inequality to F and then a Bernstein bound to each f. Talagrand's inequality applied to the whole class gives a deviation governed by the largest variance in the class and by the global complexity E[Rn(F)], which yields only 1/n rates. Obtaining a deviation that scales with each function's own second moment requires peeling the class into shells of comparable second moment and a fixed-point argument on the sub-root bound, with a union bound whose cost appears as log⌈logtn⌉. The two directions of the localized inequalities (population to sample, and sample to population for second moments) must then be combined with the deterministic expansion (10) while keeping the constants explicit. A further subtlety is the self-normalized rescaling f↦r/(E[f2]∨r)f, which differs from the variance normalization of Bartlett et al. and is what makes the bound compatible with the robust objective.
Formalization scope
Lean conventions. The sample is the coordinate map of the product measure Pn on Fin n → X. Distributions on the sample are weight vectors in the χ2 ball; the robust risk is the real supremum over that ball (attained, since the ball is nonempty and compact for n≥1, ρ≥0). Population means and variances are ∫ x, f x ∂P and ProbabilityTheory.variance f P for measurable bounded f; empirical means and variances are normalized by 1/n. The empirical Rademacher complexity is the published UnderstandingML_Rademacher definition evaluated on {(f(x1),…,f(xn))}. Its expectation is a Bochner integral, and every hypothesis that bounds it also asserts that the integrand is integrable: otherwise the integral is 0, (20) would hold for free, and the theorem would be false. Probability bounds are stated for the failure event under Pn (an outer measure when the event is not measurable). The goal speaks about every minimizer of the robust risk, so it is not vacuous when the set of minimizers is empty. The condition rn⋆>0 is part of the page's "root" (and the proof divides by rn⋆); with rn⋆=0 allowed, ψ(r)=r would remove the complexity term from (21). The condition t<n makes log⌈logtn⌉ defined.
Corrections of printed statements, each recorded in the item's docstring and Formalization Note (the milestone texts stay verbatim):
(22) is stated with probability 1−2e−t; the paper prints 1−e−t, and its proof (p. 41) concludes 1−2e−t.
(23) is stated with probability 1−4e−t (printed 1−3e−t; the proof adds two fixed-f events to the two of (22)) and with 45n182ρ (printed 45n91ρ; the proof's step ρ+t≤91ρ/45 multiplies 2Var(f)/n).
Lemma D.3 is stated with the additive term 72M2(1+η)rn⋆+(4(1+η)+314)nM2t and, in the reversed direction, the coefficient 1+1+η1, as its proof yields (printed: nMt(4+37M) and 1+1+ηη), under Theorem 4's standing hypothesis M≥1.
Lemma D.5 is linked to the published contraction lemma UnderstandingML.contraction_lemma, which states it at a fixed sample for nonempty bounded classes and allows a different Lipschitz map per coordinate.
Contributions welcome: proofs of the milestones in any order; Lemma B.2 (Bousquet's inequality) is the deepest single ingredient and is reusable well beyond this mission, as are the peeling Lemma D.1 and the sub-root fixed-point Lemma D.2.
Selected references
J. C. Duchi and H. Namkoong, Variance-based regularization with convex objectives, arXiv:1610.02581v3, 2017. https://arxiv.org/abs/1610.02581
O. Bousquet, A Bennett concentration inequality and its application to suprema of empirical processes, Comptes Rendus Mathématique 334(6), 2002. https://doi.org/10.1016/S1631-073X(02)02292-6
A. Maurer and M. Pontil, Empirical Bernstein bounds and sample variance penalization, COLT 2009. https://arxiv.org/abs/0907.3740
Optimal Best Arm Identification with Fixed Confidence III: δ-PAC Guarantee of Chernoff's Stopping Rule for Bernoulli BanditsResearch Paper
Motivation
In best arm identification with fixed confidence, a learner samples K unknown distributions ("arms") one at a time, and must eventually stop and name the arm with the largest mean, with an error probability at most a prescribed risk δ, while using as few samples as possible. The problem goes back to the sequential design of experiments (Chernoff, 1959; Even-Dar, Mannor and Mansour, 2006) and underlies adaptive A/B testing, clinical trial design and hyperparameter selection.
Any fixed-confidence strategy consists of three parts: a sampling rule, a stopping rule and a decision rule. Garivier and Kaufmann (arXiv:1602.04589, COLT 2016) proposed the Track-and-Stop strategy, the first shown to match the asymptotic lower bound on the expected sample complexity. Its stopping rule is a generalized likelihood ratio (GLR) test, Chernoff's stopping rule. Its correctness, the guarantee that the recommended arm is wrong with probability at most δ, must hold whatever the sampling rule, which is what allows the sampling rule to be tuned freely for efficiency. This mission formalizes that guarantee for Bernoulli arms, Theorem 10 of the paper, with its explicit threshold β(t,δ)=log(2t(K−1)/δ).
Setting
The arms are A={1,…,K}. A Bernoulli bandit model is a mean vector μ=(μ1,…,μK)∈[0,1]K: pulling arm a returns reward 1 with probability μa and 0 otherwise, independently of the past. The class S contains the models with a unique optimal arma∗(μ), i.e. μa∗>μi for all i=a∗.
At round t the learner chooses an arm At as a (possibly randomized) function of the past observations and observes a reward Xt. Write Na(t) for the number of pulls of arm a among the first t rounds, sa(t) for the number of those pulls that returned 1, and μ^a(t)=Na(t)−1∑s≤tXs1{As=a} for the empirical mean. The likelihood of arm a's observations under mean u is pu(XNa(t)a)=usa(t)(1−u)Na(t)−sa(t).
The GLR statistic for "arm a is at least as good as arm b" is
Chernoff's stopping rule with exploration rateβ(t,δ) is
τδ=inf{t≥1:∃a∈A,∀b=a,Za,b(t)>β(t,δ)},
and the decision rule recommends a^τδ∈argmaxaμ^a(τδ).
The Krichevsky–Trofimov (KT) distribution on binary sequences x∈{0,1}n is kt(x)=∫01(πu(1−u))−1pu(x)du, the Bernoulli likelihood mixed over the Beta(1/2,1/2) prior.
Formalization targets
Goal: Theorem 10
For every δ∈(0,1), every sampling strategy, and the threshold β(t,δ)=log(2t(K−1)/δ),
∀μ∈S,Pμ(τδ<∞,a^τδ=a∗)≤δ.
Milestone: Lemma 11 (Willems, Shtarkov and Tjalkens, 1995)
kt is a probability law on {0,1}n, and for n≥1,
x∈{0,1}nsupu∈[0,1]supkt(x)pu(x)≤2n.
Milestone: the pairwise crossing bound of Appendix C.1
With Ta,b=inf{t:Za,b(t)>β(t,δ)}, for all arms with μa<μb,
Pμ(Ta,b<∞)≤K−1δ.
Significance
Theorem 10 decouples correctness from efficiency. Because the guarantee holds for every sampling strategy, any sampling rule, including the C-Tracking and D-Tracking rules of Track-and-Stop, the uniform rule, or a heuristic, inherits δ-correctness as soon as it is paired with Chernoff's stopping rule at this threshold. The asymptotic optimality result of the paper (Theorem 14) then only has to control the sample complexity. The threshold is explicit, with no unspecified constant, in contrast to the deviational threshold of Proposition 12.
The result is proved in the paper, in Appendix C.1, and rests on Lemma 11, which the paper quotes from the universal coding literature without proof. As far as the platform's catalog shows, none of these results is formalized. The platform holds a machine-checkable statement of the analogous result for Gaussian arms with the Lattimore–Szepesvári threshold (BanditAlgorithm.chernoff_stopping_rule_sound, Lemma 33.7 of Bandit Algorithms), which is a different model and a different threshold. Formalizing Theorem 10 adds a proof of Lemma 11 (the KT regret bound, reusable in information theory and universal prediction), the Bernoulli GLR statistic, and a change of measure from the true bandit law to a Bayesian mixture law on the trajectory space.
Difficulty
The obvious approach bounds, for each fixed t, the probability that Za,b(t) exceeds β(t,δ) by a concentration inequality and sums over t. This fails: the sampling strategy is arbitrary and adaptive, so Na(t) and Nb(t) are random and depend on the past rewards, and a fixed-sample-size deviation bound does not apply; a union bound over the possible values of the counts loses more than the threshold allows. The maximum likelihood in the numerator of Za,b is also not a probability density, so the likelihood ratio cannot directly be read as a change of measure. The argument must control the whole trajectory law under an arbitrary randomized policy, and must handle empty samples (an arm never pulled contributes likelihood 1) and the boundary means 0 and 1.
Formalization scope
The formalization is in Lean 4 with Mathlib and reuses the platform's canonical bandit model: StochasticBandit, BanditPolicy (a Markov kernel per round from the observed history to the next arm, so randomized strategies are included), banditTrajMeasure (the law of the infinite trajectory, where coordinate s is round s+1) and IsSoundBAI from BanditTrajectory; bernoulliBandit from bernoulliRelativeEntropy; and only the pull counts trajPullCount and empirical means trajEmpiricalMean from TrackAndStop. The Gaussian GLR and threshold of TrackAndStop are not used.
Conventions committed to:
Arms are Fin K. Bernoulli means range over [0,1], degenerate laws included; the paper's exponential-family mean space is (0,1), so the [0,1] statement implies the paper's.
Za,b(t) is defined as the ratio of the two maxima over [0,1]2, not by the closed form (7), which holds only when μ^a(t)≥μ^b(t). Both maxima are attained and positive.
The stopping rule ranges over t≥1; at t=0 there is no observation and the paper's β(0,δ)=log0 is undefined. τδ=∞ when the rule never fires.
The decision rule is quantified: the goal holds for every recommendation that maximizes the empirical mean at τδ, whatever the tie-breaking.
Probabilities of events are outer measures under the trajectory law; no measurability is assumed.
K≥1 only. For K=1 the statement is trivially true (no suboptimal arm).
Disclosed deviations from the page: Lemma 11's ratio bound is stated for n≥1, since at n=0 the printed bound reads 1≤0; the use of the lemma in Appendix C.1 is unaffected. Appendix C.1 calls the result "Proposition 10" (a slip for Theorem 10) and prints the KT density as 1/πu(1−u) (a slip for 1/(πu(1−u)) of Lemma 11, which is the normalized one); the formalization follows Lemma 11.
A trivializing formalization is ruled out: stating the result for Gaussian arms or with the Lattimore–Szepesvári threshold is the platform's existing Lemma 33.7 and is not Theorem 10, and a free decision rule without the argmax hypothesis would make the claim false rather than faithful.
Contributions welcome: a proof of Lemma 11; a change-of-measure lemma for banditTrajMeasure under a mixture of environments; and the union-bound reduction from the goal to the pairwise claim.
Selected references
A. Garivier, E. Kaufmann, Optimal Best Arm Identification with Fixed Confidence, COLT 2016 (JMLR W&CP 49), arXiv:1602.04589v2, 2016. https://arxiv.org/abs/1602.04589
F. M. J. Willems, Y. M. Shtarkov, T. J. Tjalkens, The context-tree weighting method: basic properties, IEEE Transactions on Information Theory 41(3), 1995. https://doi.org/10.1109/18.382012
R. Krichevsky, V. Trofimov, The performance of universal encoding, IEEE Transactions on Information Theory 27(2), 1981. https://doi.org/10.1109/TIT.1981.1056331
A Distributional Interpretation of Robust Optimization II: Box-Robust Sample Average Optimization Is ConsistentResearch Paper
Why robustify a sampled stochastic program
Many decision problems under uncertainty take the form of a stochastic program: choose a decision v from a feasible set F to maximise the expected utility Ex∼μ[f(v,x)], where the distribution μ of the uncertain parameter x∈Rm is known only through i.i.d. samples x1,…,xn. The standard remedy, sample average approximation, maximises n1∑if(v,xi) instead. Its consistency (convergence of the optimal expected utility of its solutions to the true optimum) is classical, but it needs regularity assumptions of its own, for example those of King and Wets (Stochastics and Stochastic Reports, 1991), cited on p. 98 of the paper; the paper presents its construction as a route to consistency under weaker conditions.
Robust optimization (RO) takes a different route: it protects each sample by an uncertainty set and optimises against the worst point in it. Xu, Caramanis and Mannor (Math. Oper. Res. 2012) show that RO over several overlapping uncertainty sets is equivalent to a distributionally robust stochastic program (their Theorem 2.1, the subject of mission I of this series). Section 3 of the paper uses that equivalence to show that a specific robustification of the sampled problem, with ℓ∞ boxes of shrinking radius around each sample, is consistent under only boundedness and equicontinuity of f. This mission formalizes that result, Theorem 3.1.
Setting
Equip Rm with the sup norm ∥z∥∞=maxk∣zk∣, its Borel σ-algebra and Lebesgue measure dx. The data are:
a set of decisions V and a nonempty feasible setF⊆V;
a utilityf:V×Rm→R, Borel measurable in x for each v;
a true densityh∗ on Rm (nonnegative, ∫h∗dx=1) and i.i.d. samples x1,x2,… with distribution h∗(x)dx;
radiiϵ(n)>0.
For a sample x1,…,xn the boxes are Zi={xi+δ∣∥δ∥∞≤ϵ(n)}, and the box-robust sample objective is
The RO solution v(n) is a maximiser of Jn over F. The equicontinuity modulus of f is
d(ϵ)=v,x,∥δ∥∞≤ϵsup∣f(v,x)−f(v,x+δ)∣.
The proof works with the distribution setPn of probability measures μ with μ(⋃i∈SZi)≥∣S∣/n for every S⊆{1,…,n}, and with the uniform box kernel density estimator
hn is the density of a probability measure in Pn.
Jn(v)≤∫f(v,x)hn(x)dx for every v.
Oscillation over a box: supZif(v,⋅)−infZif(v,⋅)≤d(2ϵ(n)).
Eq. (7): with Mn=C∫∣hn−h∗∣dx, for every v,
Jn(v)−Mn≤∫f(v,x)h∗(x)dx≤Jn(v)+Mn+d(2ϵ(n)).
Strong L1 consistency of the box kernel density estimator: if ϵ(n)→0 and nϵ(n)m→∞, then ∫∣hn−h∗∣dx→0 almost surely.
Milestones 1–4 are deterministic statements about a fixed sample; milestone 5 is the only probabilistic input.
Significance
Theorem 3.1 gives consistency of a tractable robust reformulation of a sampled stochastic program under conditions the paper notes are weaker than those of King and Wets for sampled stochastic programs: f need only be bounded and equicontinuous in x, uniformly in v, and the true distribution need only have a density. It also gives an explicit schedule for the size of the uncertainty set, ϵ(n)→0 with nϵ(n)m→∞, the bandwidth condition of kernel density estimation. Section 4 of the paper applies the same distributional interpretation to regularised learning methods such as the support vector machine and the Lasso.
The result is proved in the paper, with the L1 consistency of kernel density estimators (Devroye 1983; Devroye and Györfi 1985) cited rather than proved. No part of it is formalized in Lean or on this platform as far as a search of the platform found. A complete development would produce, besides Theorem 3.1, a machine-checked strong L1 consistency theorem for kernel density estimators, which is a basic result of nonparametric statistics in its own right.
Difficulty
The deterministic part (milestones 1–4) is measure-theoretic bookkeeping: the kernel integrates to one only because the box is a sup-norm ball of volume (2ϵ)m, and every infimum and supremum must be handled with care, since f need not attain them.
The obstacle is milestone 5. Almost-sure L1 convergence of hn to an arbitrary density h∗, with no continuity or support assumption, does not follow from the strong law of large numbers applied pointwise: hn(x) is an average of n terms whose law changes with n through ϵ(n), and almost-sure convergence at each fixed x does not give convergence of the integral along a single sample path. The theorem needs both a bias estimate valid for every integrable density and a concentration estimate for the random L1 error. Mathlib has Lebesgue differentiation and the strong law, but no kernel density estimator and no such concentration result.
Formalization scope
Rm is Fin m → ℝ, whose Mathlib norm is the sup norm; boxes are Metric.closedBall. The integrals ∫f(v,x)h∗(x)dx are Bochner integrals against Lebesgue measure of integrable integrands.
The samples are a sequence X : ℕ → Ω → Fin m → ℝ on a probability space, independent (iIndepFun) and each with law volume.withDensity h*; x1,x2,… become X 0, X 1, …, and the n-th problem uses the first n. "With probability one" is ∀ᵐ ω ∂P.
The goal quantifies over every selection v(n) of maximisers, with no measurability assumed; a version with one chosen maximiser would be weaker and is ruled out.
Readings and corrections of the printed text:
the kernel argument printed (x−xi)/ϵ on p. 98 is read as (x−xi)/ϵ(n), as the proof on p. 99 writes it;
"maxv,x∣f(v,x)∣≤C" is read as the uniform bound ∣f∣≤C and the "max" in d(ϵ) as a supremum;
"d(ϵ)↓0" is read as d(ϵ)→0 as ϵ↓0;
implicit hypotheses made explicit: F=∅, ϵ(n)>0, measurability of f(v,⋅), h∗ a Lebesgue density;
the monotonicity in "ϵ(n)↓0, nϵ(n)m↑∞" is kept in the goal; milestone 5 uses the limits only, as the paper states it;
the paper's Mn ("there exists {Mn}→0") is made explicit as Mn=C∫∣hn−h∗∣, so Eq. (7) is stated for every sample.
Remark 3.2 and Appendix B (an integrable envelope in place of boundedness) are not part of this mission.
Every real infimum and supremum ranges over a nonempty set of values bounded by C in absolute value, so no statement holds through a junk value; a formalization in which the supremum over F or the box infimum could be vacuous is excluded.
The definitions (boxes, Pn, the kernel, the estimator, Jn, d) live in one definition file. Pn duplicates, with weights 1/n, the distribution set of mission I; the duplication is deliberate because draft missions cannot import each other.
Welcome contributions: the kernel density estimator and its strong L1 consistency as reusable infrastructure, and any of the deterministic milestones.
Selected references
H. Xu, C. Caramanis, S. Mannor, A Distributional Interpretation of Robust Optimization, Mathematics of Operations Research 37(1):95–110, 2012. https://doi.org/10.1287/moor.1110.0531
L. Devroye, The equivalence of weak, strong and complete convergence in L1 for kernel density estimates, Annals of Statistics 11(3):896–904, 1983.
L. Devroye, L. Györfi, Nonparametric Density Estimation: The L1 View, Wiley, 1985.
A. J. King, R. J.-B. Wets, Epi-consistency of convex stochastic programs, Stochastics and Stochastic Reports 34(1), 1991 (reference [22] of the paper).
Learnability, Stability and Uniform Convergence III: For an ERM, Leave-One-Out Stability, Universal Consistency and Universal Generalization Are EquivalentResearch Paper
Motivation
Algorithmic stability asks how much the output of a learning algorithm changes when its training sample is perturbed. Since Devroye and Wagner (IEEE Trans. Inf. Theory 1979) it has served as a route to generalization bounds that does not go through the complexity of the hypothesis class. Bousquet and Elisseeff (JMLR 2002) popularised uniform stability, and Mukherjee, Niyogi, Poggio and Rifkin (Adv. Comput. Math. 2006) showed that for empirical risk minimisation in supervised learning, a leave-one-out type of stability is necessary and sufficient for consistency.
Shalev-Shwartz, Shamir, Srebro and Sridharan (JMLR 11, 2010) study stability in Vapnik's General Learning Setting, where uniform convergence can fail even though the problem is learnable. In Appendix A.2 they compare replace-one and leave-one-out (LOO) stability. For an empirical risk minimiser they prove that LOO stability is equivalent to consistency and to generalization, provided each property holds with one rate for all distributions (Theorem 31, p. 2667). This mission formalizes that theorem and the lemmas of Section 5.3 on which its proof rests.
Timeline:
1979, Devroye–Wagner: leave-one-out estimates for local rules.
2002, Kutin–Niyogi (UAI 2002): a taxonomy of stability notions.
2006, Mukherjee et al.: LOO stability characterises consistency of ERM in supervised learning.
2010, Shalev-Shwartz et al.: in the General Learning Setting, for ERMs, LOO stability, universal consistency and universal generalization are equivalent (Theorem 31). Universally consistent AERMs need not be LOO stable (Example 6).
Setting
A learning problem consists of an instance space Z with a σ-algebra, a nonempty hypothesis class H, and an objective f:H×Z→R with ∣f(h;z)∣≤B for all h,z. For a probability measure D on Z:
the risk is F(h)=Ez∼D[f(h;z)] and the optimal risk is F∗=infhF(h);
for a sample S=(z1,…,zm)∼Dm of m i.i.d. draws, the empirical risk is FS(h)=m1∑if(h;zi), and FS(h^S)=infhFS(h) denotes the minimal empirical risk;
a learning ruleA maps each sample of size m≥1 to a hypothesis A(S). It is an ERM if FS(A(S))=FS(h^S) for every sample. It is an AERM with rate εerm if E[FS(A(S))−FS(h^S)]≤εerm(m);
A is consistent with rate ε if ES∼Dm[F(A(S))−F∗]≤ε(m). It generalizes with rate ε if E[∣F(A(S))−FS(A(S))∣]≤ε(m), and it on-average generalizes if ∣E[F(A(S))−FS(A(S))]∣≤ε(m);
writing S∖i for S with zi removed, A is LOO stable with rate ε (Definition 29) if
Lemma 14: an AERM that on-average generalizes with rate εoag generalizes with rate εoag+2εerm+2B/m.
Lemma 15: under the same hypotheses the rule is consistent with rate εoag+εerm.
Lemma 16 (Main Converse Lemma): in a learnable problem, E∣FS(h^S)−F∗∣≤2εcons(m′)+2B/m+2Bm′2/m for 2≤m′≤m/2.
Lemma 17: Eq. (12), together with an AERM that is consistent, gives generalization with rate εemp+εerm+εcons.
First display of the proof of Theorem 31: a generalizing ERM is LOO stable with rate εgen(m−1).
Second display: a LOO stable ERM on-average generalizes on samples of size m−1 with rate εstable(m)+2B/m.
Significance
Theorem 31 shows that for exact ERMs, LOO stability is not only sufficient but necessary for consistency. It transfers the supervised-learning characterisation of Mukherjee et al. to the General Learning Setting, where uniform convergence is no longer available as an intermediate. The hypothesis is sharp in one direction: Example 6 of the paper gives a universally consistent AERM that is not LOO stable. The equivalence therefore depends on exact minimisation, and an asymptotic minimiser does not suffice.
The lemmas are useful on their own. Lemmas 14 and 15 are the standard bridges between on-average generalization, generalization and consistency. Lemma 16 says that the minimal empirical risk estimates F∗ consistently in every learnable problem, even when no ERM learns. Lemma 16 also underlies Theorem 7, the paper's main characterisation of learnability, which is the subject of mission I of this series.
The results are proved in the paper. As far as is known they have not been formalized in any proof assistant. The platform has the textbook side of this framework (Shalev-Shwartz and Ben-David, Understanding Machine Learning, Chapter 13). Those statements are in Rd and use a replace-one stability notion; they do not cover leave-one-out stability or the General Learning Setting.
Difficulty
The implications between stability and generalization change the sample size: S∖i has m−1 points, so a statement about the rule at size m has to be compared with the rule at size m−1 under the marginal law of the reduced sample. The step from universal consistency to generalization needs Lemma 16. That lemma estimates F∗ from a sample on which the ERM itself may be inconsistent, and it is the only place where universality of the consistency rate is used. Per-distribution consistency of an ERM does not imply generalization (Example 1 of the paper). An argument that fixes D throughout therefore cannot succeed.
Formalization scope
Samples are tuples S:Finm→Z with law Measure.pi (the i.i.d. product), and S∖i is Fin.removeNth i S. A learning rule is a family Am:Zm→H. Its value at m=0 is never used, and LOO stability is required only for m≥2. FS(h^S) is the infimum infhFS(h) and no minimiser is chosen. An ERM is a rule attaining this infimum at every sample. A rate is non-increasing on m≥1 and tends to 0. Universal properties are stated as "there exists ε with IsRate ε such that for every probability measure D …", with the rate chosen before the distribution. The bound B is any bound on ∣f∣; the paper's B is sup∣f∣, and all its rates increase with B.
Measurability is not discussed in the paper. The formalization assumes that each f(h;⋅) is measurable and that the rule is measurable in the sense that (S,z)↦f(Am(S);z) is jointly measurable. Lemmas 14–17 also assume that S↦infhFS(h) is measurable. For a measurable ERM this holds automatically, so Theorem 31 makes no such assumption. Without these assumptions Lean's integral of a non-measurable function is 0 and every rate bound would hold trivially. For the same reason Utility Lemma 13 assumes X,Y integrable. The ERM hypothesis of the goal must not be weakened to an AERM: the statement would then be false (Example 6).
One statement corrects the printed text. In the second display of the proof of Theorem 31 (p. 2668), the chain adds 2B/m and then drops it. The milestone states the bound the argument proves, εstable(m)+2B/m. The rate-free Theorem 31 is unaffected.
The development needs product measures, independence and variance bounds, all available in Mathlib, and the marginals of Measure.pi under removal of a coordinate. The definitions of risks, rules and stability notions can be reused by other stability results. Proofs of any milestone are welcome, and so are alternative proofs of the goal.
Selected references
S. Shalev-Shwartz, O. Shamir, N. Srebro, K. Sridharan, Learnability, Stability and Uniform Convergence, Journal of Machine Learning Research 11 (2010) 2635–2670. https://jmlr.org/papers/v11/shalev-shwartz10a.html
S. Mukherjee, P. Niyogi, T. Poggio, R. Rifkin, Learning theory: stability is sufficient for generalization and necessary and sufficient for consistency of empirical risk minimization, Advances in Computational Mathematics 25 (2006) 161–193. https://doi.org/10.1007/s10444-004-7634-z
S. Kutin, P. Niyogi, Almost-everywhere algorithmic stability and generalization error, UAI 2002. https://arxiv.org/abs/1301.0579
L. Devroye, T. Wagner, Distribution-free performance bounds for potential function rules, IEEE Transactions on Information Theory 25(5) (1979) 601–604. https://doi.org/10.1109/TIT.1979.1056087
Learnability, Stability and Uniform Convergence II: Tikhonov-Regularized ERM Learns Convex Lipschitz Stochastic Optimization in Hilbert Space with High ProbabilityResearch Paper
Motivation
Statistical learning theory asks when a rule that sees only an i.i.d. sample z1,…,zm from an unknown distribution D can return a hypothesis whose expected loss is close to the best possible. In supervised classification the classical answer is uniform convergence: learnability holds exactly when empirical risks converge to expected risks uniformly over the hypothesis class, and then empirical risk minimization (ERM) learns. Shalev-Shwartz, Shamir, Srebro and Sridharan (JMLR 11, 2010) showed that in Vapnik's broader General Learning Setting this picture breaks down. Their motivating example is stochastic convex optimization in a Hilbert space: minimizing an expected convex, Lipschitz objective over a bounded convex set from samples. This problem underlies regularized linear prediction, kernel methods and online-to-batch conversions, and the paper shows (§4.1) that in infinite dimension uniform convergence can fail and the plain empirical minimizer can fail to converge, while the problem is still learnable.
This mission formalizes the positive half of that example: Tikhonov-regularized ERM learns every such problem, with an explicit bound holding with probability 1−δ (Theorem 3, p. 2644), through the stability of strongly convex empirical minimization (Theorem 2).
Setting
Let Z be a measurable space of instances and E a real Hilbert space. A stochastic convex optimization problem consists of a nonempty, closed, convex, bounded set H⊆E and an objective f:E×Z→R such that for every z the map h↦f(h;z) is convex and L-Lipschitz on H, each f(h;⋅) is measurable, and ∣f(h;z)∣≤C on H×Z. For a distribution D on Z define the risk and optimal risk
F(h)=Ez∼D[f(h;z)],F∗=h∈HinfF(h),
and for a sample S=(z1,…,zm)∼Dm the empirical riskFS(h)=m1∑i=1mf(h;zi). A function g is λ-strongly convex on H if g−2λ∥⋅∥2 is convex there. The regularized empirical minimizer is
h^λ∈h∈Hargmin(FS(h)+2λ∥h∥2).(5)
For the general part, a learning ruleA maps samples to hypotheses; it is an AERM with rate εerm if E[FS(A(S))−infhFS(h)]≤εerm(m), consistent with rate εcons if E[F(A(S))−F∗]≤εcons(m), and uniform-RO stable with rate εstable if replacing any one sample point changes the loss at any test point by at most εstable(m) on average over the replaced index (Definition 4).
Formalization targets
Goal: Theorem 3
If ∥h∥≤B on H, L,B>0, δ∈(0,1), m≥1 and λ=16L2/(δB2m), then with probability at least 1−δ over S∼Dm
F(h^λ)−F∗≤4δmL2B2(1+δm8).
The constants are the paper's.
Milestones, in the order the proof uses them
Quadratic growth at a minimizer of a λ-strongly convex g: g(h′)−g(h)≥2λ∥h′−h∥2 (§4.2, p. 2644).
Eq. (6): if f(⋅;z) is λ-strongly convex and L-Lipschitz, empirical minimizers of S and of S(i) satisfy ∣f(h^S,z)−f(h^S(i),z)∣≤4L2/(λm) for all z (p. 2645).
Theorem 8: a uniform- or average-RO stable AERM is consistent with rate εstable+εerm and generalizes with rate εstable+2εerm+2C/m (p. 2649).
ES∼Dm[F(h^S)−F∗]≤4L2/(λm) for the strongly convex empirical minimizer (p. 2645).
Theorem 2: with probability 1−δ, F(h^S)−F∗≤4L2/(δλm) (p. 2644).
Theorem 2 applied to r(h;z)=2λ∥h∥2+f(h;z): with probability 1−δ, 2λ∥h^λ∥2+F(h^λ)≤infh(2λ∥h∥2+F(h))+4(L+λB)2/(δλm) (p. 2645).
Significance
The result. Theorem 3 shows that every convex, Lipschitz, bounded stochastic optimization problem over a bounded subset of a Hilbert space is learnable at rate O(LB/δm), with no dimension dependence and no uniform convergence. Together with the counterexamples of §4.1 it separates learnability from uniform convergence and from ERM, and it motivates the paper's general characterization: a problem is learnable if and only if it admits a uniform-RO stable asymptotic empirical risk minimizer (Theorem 7). Theorem 8 is the sufficiency half of that characterization and is reused wherever stability arguments give generalization bounds.
Formalizing it. The results are proved in the paper; to our knowledge none has a machine-checked proof. The closest platform material is the textbook treatment in Understanding Machine Learning, chapter 13 (Shalev-Shwartz and Ben-David): Corollary 13.9 (UnderstandingML.convex_lipschitz_bounded_learnable), Corollary 13.6 (rlm_lipschitz_stable) and Lemma 13.5 (strongly_convex_lemma). Those are stated in Rd, bound the risk in expectation, use the regularizer λ∥w∥2 over all of Rd, and have different constants; the present mission works in an arbitrary Hilbert space, over a constraint set H, with high-probability bounds and the paper's constants. Its definitions of learning rules, AERM, consistency and replace-one stability in the General Learning Setting are reusable by the other missions of this series.
Difficulty
The obvious route, bounding suph∈H∣F(h)−FS(h)∣, is unavailable: §4.1 exhibits problems of exactly this type in which that supremum stays bounded away from zero for every sample size. Any successful argument therefore has to rely on a property of the learning rule rather than of the class H, and the plain empirical minimizer does not have it: §4.1 shows it can stay a constant away from F∗ at every sample size. A second difficulty is purely formal: the regularization parameter λ depends on δ and m, so the regularized minimizer changes with them, and all expectations involve a data-dependent hypothesis in a possibly non-separable Hilbert space, where measurability is not automatic.
Formalization scope
Lean conventions, fixed for every item:
E is a real inner product space with CompleteSpace E, never assumed finite-dimensional; H is Hset : Set E, and all infima, suprema, strong convexity and Lipschitz conditions are taken on Hset only. F∗ is ⨅ h : Hset, F h.
Samples are Fin m → Z, Dm is Measure.pi, S(i) is Function.update S i z', and m≥1 throughout.
The paper's standing loss bound ∣f∣≤B (p. 2637) is named C, because Theorem 3 uses B for the norm bound ∥h∥≤B. L>0 and B>0 are implicit in Theorem 3's choice of λ and are stated.
Strong convexity is Mathlib's StrongConvexOn Hset λ, which is the paper's definition.
Minimizers are selections S↦h^S∈H satisfying the minimization property; the theorems hold for every such selection, hence for the minimizer, which is unique by strong convexity.
Measurability, not discussed in the paper, is the series' single standing convention: each f(h;⋅) is measurable and the selection makes (S,z)↦f(h^S;z) jointly measurable; for Theorem 8, the rule is measurable in the same sense and S↦infhFS(h) is measurable.
"With probability at least 1−δ" is the bound Dm{failure}≤δ with 0<δ<1.
No statement of the paper is corrected: all printed constants were checked against the proofs and are reproduced exactly.
A formalization in which the expected excess risk is a Bochner integral of a non-measurable or non-integrable function, or in which F∗ is an infimum over all of E or over an unbounded family, would make the bounds trivially true; the measurability hypotheses, the bound ∣f∣≤C and the infimum over the nonempty set H rule this out.
Contributions welcome: proofs of the milestones in order, and in particular a reusable replace-one identity E[FS(A(S))]=m1∑iE[f(A(S(i));zi′)] under Measure.pi, and Markov's inequality in the form used for high-probability bounds.
Selected references
S. Shalev-Shwartz, O. Shamir, N. Srebro, K. Sridharan, Learnability, Stability and Uniform Convergence, Journal of Machine Learning Research 11 (2010) 2635–2670. https://jmlr.org/papers/v11/shalev-shwartz10a.html
S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press, 2014, chapter 13. https://doi.org/10.1017/CBO9781107298019
V. N. Vapnik, Statistical Learning Theory, Wiley, 1998.
A Unified Framework for High-Dimensional Analysis of M-Estimators with Decomposable Regularizers: Error Bounds under Decomposability and Restricted Strong ConvexityResearch Paper
Motivation
High-dimensional statistics studies estimation when the number of parameters p is comparable to, or larger than, the number of observations n. The standard estimators in this regime are regularized M-estimators: minimise an empirical loss plus a penalty that encodes structure, such as the Lasso (ℓ1 penalty, sparse vectors), the group Lasso (block norms, group sparsity) and nuclear-norm regularization (low-rank matrices). Before 2009 each of these estimators came with its own consistency proof. Negahban, Ravikumar, Wainwright and Yu (arXiv:1010.2731; Statistical Science 27(4), 2012, doi:10.1214/12-STS400) isolated two properties that these proofs share, decomposability of the regularizer and restricted strong convexity of the loss, and proved one deterministic theorem from them. The Lasso rates of Bickel, Ritov and Tsybakov (arXiv:0801.1095), rates under ℓq-sparsity, and group-sparse and low-rank rates then follow as corollaries. The framework is the organising principle of Chapter 9 of Wainwright's textbook High-Dimensional Statistics (Cambridge University Press, 2019).
Setting
Let E be a finite-dimensional real inner product space with inner product ⟨⋅,⋅⟩ and induced error norm∥⋅∥. Given a loss L:E→R, a regularizer R:E→R and a constant λn>0, program (1) is
θ^λn∈argθ∈Emin{L(θ)+λnR(θ)}.
For a subspace S write uS for the orthogonal projection of u onto S, and S⊥ for the orthogonal complement.
Decomposability. For subspaces M⊆M, the norm R is decomposable with respect to (M,M⊥) if R(θ+γ)=R(θ)+R(γ) for all θ∈M and γ∈M⊥. Example: the ℓ1-norm with M=M={θ:θj=0∀j∈/S}.
Dual norm.R∗(v)=supR(u)≤1⟨u,v⟩.
Subspace compatibility constant.Ψ(M)=supu∈M∖{0}R(u)/∥u∥; for the ℓ1-norm on an s-dimensional coordinate subspace, Ψ=s.
The set C. For a point θ∗∈E,
C(M,M⊥;θ∗)={Δ∣R(ΔM⊥)≤3R(ΔM)+4R(θM⊥∗)}.
Restricted strong convexity (RSC). With the Taylor error δL(Δ,θ∗)=L(θ∗+Δ)−L(θ∗)−⟨∇L(θ∗),Δ⟩, the loss satisfies RSC with curvature κL>0 and tolerance τL(θ∗) if δL(Δ,θ∗)≥κL∥Δ∥2−τL2(θ∗) for every Δ∈C(M,M⊥;θ∗).
The conditions of the paper's main theorem are (G1): R is a norm, decomposable with respect to (M,M⊥) with M⊆M; and (G2): L is convex, differentiable and satisfies RSC. The Lean development lives in the namespace UnifiedMEstimator.General, with these objects named IsNormFn, IsDecomposable, dualNorm, compat, setC, taylorErr, RSC and IsOptimal.
Formalization targets
Goal: Theorem 1 (p. 10), tolerance term corrected
Under (G1) and (G2), if λn>0 and λn≥2R∗(∇L(θ∗)), then every optimal solution of program (1) satisfies
The bound holds for every pair (M,M) over which R decomposes, and for every optimum, not only a distinguished one.
Milestones
Lemma 1 (p. 7): under the dual-norm condition on λn, the error Δ^=θ^λn−θ∗ lies in C(M,M⊥;θ∗). This milestone links an existing platform statement of the same lemma (Wainwright, Proposition 9.13).
Section 2.4, p. 10, first display: if θ∗∈M and Δ∈C, then R(Δ)≤4Ψ(M)∥Δ∥.
Further statements
Corollary 1 (p. 11): if θ∗∈M and τL(θ∗)=0, then ∥θ^λn−θ∗∥≤3λnΨ(M)/κL and R(θ^λn−θ∗)≤12λnΨ2(M)/κL.
Section 2.4, p. 10, second display: a lower bound δL≥κ1∥Δ∥2−κ2gR2(Δ) on the unit ball gives curvature κ1−16κ2Ψ2(M)g on C when θ∗∈M.
Example 1 (p. 5) and the value Ψ(M(S))=∣S∣ (p. 9): the ℓ1-norm instance, which shows that the hypotheses of the goal can be met.
Significance
Theorem 1 reduces a consistency proof for a new regularized estimator to two checks: that the regularizer decomposes over a pair of subspaces adapted to the model, and that the loss is curved on the set C, together with a bound on R∗(∇L(θ∗)) that is usually a concentration inequality. The paper derives from it the slogp/n Lasso rate under restricted eigenvalue conditions, rates for weakly sparse (ℓq-ball) vectors, and group-Lasso rates; companion papers use it for low-rank matrix estimation, matrix completion and generalized linear models. Because the bound holds for every pair (M,M), it gives an explicit trade-off between an estimation error and an approximation error R(θM⊥∗).
The theorem is proved in the paper's supplementary appendix. No machine-checked proof of it is known. On Prove2Me, Wainwright's textbook restatement (Theorem 9.19, HighDimStat.Decomposability.thm9_19_general_bound) is a related but different statement: its RSC condition is local, on a ball, with a tolerance proportional to R2(Δ), and it has extra side conditions and a different bound. A formal proof of the present goal certifies the deterministic core that every corollary of the paper relies on.
Difficulty
The obvious argument compares the objective at θ^ and at θ∗ and applies RSC to the error. RSC, however, is available only on the set C, not on all of E: in high dimensions the loss is flat in many directions, so strong convexity fails. The work is to show first that the error lies in C (Lemma 1, which rests on decomposability and the choice of λn), and then to relate the regularizer to the error norm through the projections onto M and M⊥. The distinction between M and M matters throughout: the compatibility constant is taken on the larger space M, while the approximation error projects θ∗ onto the complement of the smaller one. The bound comes from a quadratic inequality in ∥Δ^∥, and the constants depend on how its terms are split.
Formalization scope
Representation. The parameter space is an arbitrary finite-dimensional real inner product space E (equivalently Rp with any inner product, as the paper allows); matrices are covered by the same abstraction. Subspaces are Submodule ℝ E, projections are Submodule.starProjection, and the gradient is Mathlib's gradient, under the hypothesis that L is differentiable. The dual norm and Ψ are real suprema (sSup). They equal the paper's quantities because R is required to be a genuine norm (nonnegative, definite, absolutely homogeneous, subadditive) and E is finite-dimensional; Ψ({0})=0. The tolerance is a real number τ entering as τ2; RSC contains κ>0 and is quantified over exactly C(M,M⊥;θ∗) for the same pair and point as the decomposability. Every statement is for every optimal solution of program (1). The data Z1n are fixed and absorbed into L, and θ∗ is an arbitrary point: the paper's requirement that θ∗ minimise the population risk is never used by the theorem and is dropped.
Corrections of the printed statements.
Display (22) prints the tolerance term as κLλn⋅2τL2(θ∗). As printed the statement is false: for E=R, R=∣⋅∣, M=M=R, L(θ)=(max(0,∣θ∣−1))2, θ∗=0.9, λn=0.01, κL=1/2, τL2=10, the optimum is 0 and ∥Δ^∥2=0.81 exceeds the printed bound 0.4036. The goal states 2τL2(θ∗)/κL; the two forms agree when τL=0, and the constants 9 and 4 are the paper's.
Corollary 1's (25a) prints ∥θ^−θ∗∥≤9λn2Ψ2(M)/κL, which fails for L(θ)=(θ−0.002)2, θ∗=0.001, λn=0.004 on R; the mission states 3λnΨ(M)/κL. Its "C(M,M,θ∗)" is read as C(M,M⊥;θ∗).
Trivializations ruled out. A regularizer predicate weaker than a norm would let the real suprema collapse to the junk value 0 and make the λn condition or the Ψ term free; RSC over all of E would be classical strong convexity, and RSC over the cone without the 4R(θM⊥∗) slack would make the goal false; decomposability without M⊆M or with the bars misplaced changes the theorem. None of these is used. The ℓ1 example and a checked one-dimensional instance show that all hypotheses of the goal can hold simultaneously.
Infrastructure and contributions. A complete development needs: Hölder's inequality for a norm and its dual norm, boundedness of the two suprema in finite dimension, the decomposability inequality R(θ∗+Δ)−R(θ∗)≥R(ΔM⊥)−R(ΔM)−2R(θM⊥∗), the first-order characterization of convexity, and the solution of a scalar quadratic inequality. The dual-norm and compatibility-constant lemmas are reusable for every decomposable-regularizer mission. Proofs of the milestones, of the goal, and of the ℓ1 instance are all welcome.
Selected references
S. N. Negahban, P. Ravikumar, M. J. Wainwright, B. Yu, A Unified Framework for High-Dimensional Analysis of M-Estimators with Decomposable Regularizers, Statistical Science 27(4), 2012, 538–557. arXiv:1010.2731v3, doi:10.1214/12-STS400
P. J. Bickel, Y. Ritov, A. B. Tsybakov, Simultaneous Analysis of Lasso and Dantzig Selector, Annals of Statistics 37(4), 2009, 1705–1732. arXiv:0801.1095
M. J. Wainwright, High-Dimensional Statistics: A Non-Asymptotic Viewpoint, Cambridge University Press, 2019, Chapter 9. doi:10.1017/9781108627771
Calibrated Learning and Correlated Equilibrium I: Calibrated Forecasts with Best Responses Converge to the Set of Correlated EquilibriaResearch Paper
Motivation
A correlated equilibrium (Aumann 1974) is a joint distribution over the players' strategy profiles such that no player gains by deviating from the strategy the distribution recommends to them. It is the equilibrium notion that learning dynamics in repeated games most naturally reach, and a basic question in learning in games is which simple rules, played repeatedly, drive the empirical distribution of play to the set of correlated equilibria.
Foster and Vohra (1997) answer this with a hypothesis on forecasts instead of a particular algorithm. Each player forecasts the other's next move and best-responds to the forecast. They only require the forecasts to be calibrated in the sense of Dawid (1982): among the rounds in which a player forecast a given probability vector, the empirical frequencies of the opponent's moves must approach that vector. Their Theorem 1 says that this already forces the empirical joint distribution of play to approach the set of correlated equilibria. The paper uses this to argue that Bayesian players under a common prior, whose forecasts are calibrated by Dawid's theorem, end up playing a correlated equilibrium. That is an alternative to Aumann's (1987) derivation of correlated equilibrium from common priors and rationality.
Timeline:
Aumann (1974, 1987) introduces correlated equilibrium and derives it from Bayesian rationality.
Dawid (1982) proposes calibration as a minimal requirement on probability forecasts.
Foster and Vohra (1997) prove Theorem 1 (this mission) and show that calibrated forecasts exist once the forecaster may randomize.
Hart and Mas-Colell (2000) give regret matching, an adaptive procedure with the same limit set.
Setting
A finite two-player game G has strategy sets S(1)={0,…,m−1} and S(2)={0,…,n−1} and payoff matrices u1,u2:S(1)×S(2)→R, which the players maximize. A joint distributionD is a nonnegative m×n matrix with entries summing to 1. It is a correlated equilibrium if
x,y∑D(x,y)u1(Φ(x),y)≤x,y∑D(x,y)u1(x,y)for all Φ:S(1)→S(1),
and symmetrically for player 2. The set of correlated equilibria is π(G).
The game is played in rounds s=0,1,2,…. In round s player 1 issues a forecastf1(s), a probability vector over S(2), and player 2 issues a forecast f2(s) over S(1). Each player then plays a best response to its forecast, x(s)=R1(f1(s)) and y(s)=R2(f2(s)). Here R1 and R2 are best-reply functions: R1(p) maximizes ∑ypyu1(⋅,y) for every probability vector p, and R1 is a fixed function of the forecast alone. This is the paper's standing assumption of a stationary, deterministic tie-breaking rule.
For a forecast sequence f and the opponent's plays z, N(p,t) counts the rounds among the first t in which f forecast p. ρ(p,j,t) is the fraction of those rounds in which the opponent played j, and 0 if there are none. The forecast is calibrated with respect to z if for every j
p∑∣ρ(p,j,t)−pj∣tN(p,t)⟶0(t→∞).
The empirical joint distributionDt(x,y) is the fraction of the first t rounds in which player 1 played x and player 2 played y.
Formalization targets
Goal: Theorem 1
If f1 is calibrated with respect to y and f2 is calibrated with respect to x, then
The goal fixes no rate and no particular forecasting method: it asserts only convergence of Dt to the set π(G), for every calibrated forecast.
Milestones: the steps of the proof (pp. 44–45)
Dt lies in the simplex for t≥1.
For each x∈S(1), the set Mb(x) of mixtures to which x is a best response is closed and convex.
The mixtures Mp(x) at which player 1 actually plays x satisfy Mp(x)⊆Mb(x).
The identity writing Dt(x,y) as a forecast-weighted term plus a calibration error.
Calibration makes the error term vanish.
The weighted average of the forecasts at which player 1 plays x lies in Mb(x).
For a convergent subsequence Dti→D, every row of D with positive mass, normalized, lies in Mb(x).
Every subsequential limit of Dt is a correlated equilibrium.
Further result: matching pennies (p. 46)
With the constant forecast (1/2,1/2) and the non-stationary tie-break "heads on even rounds, tails on odd rounds", both forecasts are calibrated and every play is a best reply, yet Dt does not approach π(G). The stationarity assumption cannot be dropped.
Significance
The result. Theorem 1 separates what learning needs from how it is achieved. Any forecasting procedure that is calibrated, combined with myopic best responses, yields correlated equilibrium behaviour in the long run. The paper's Theorem 3 constructs a randomized calibrated forecaster, so the theorem gives an uncoupled learning procedure for correlated equilibrium, one that needs no knowledge of the opponent's payoffs. The converse direction, that every correlated equilibrium arises this way for almost every game, is the paper's Theorem 2 (a separate mission of this series).
Formalizing it. The theorem is proved in the paper and has no machine-checked proof on Prove2Me or, to the knowledge of this mission, elsewhere. The platform's existing correlated-equilibrium results (the Algorithmic Game Theory swap-regret development, AGT.swap_regret_correlated_equilibrium) reach correlated equilibrium through swap regret of mixed strategies, a different hypothesis and a different object. This mission adds a formal notion of calibration and the convergence argument, both reusable for the paper's Theorems 2 and 3 and for later work on calibration and learning.
Difficulty
The obvious reading of calibration is that each player's forecast converges to the opponent's empirical distribution; if that held, best responses to it would give convergence. It does not hold. Calibration constrains the opponent's frequencies only conditionally on the forecast issued, and the forecasts need not converge at all. What must be shown is a statement about the conditional distributions of the joint play given each strategy of player 1, while the set of forecasts issued keeps growing. Rows of the limit with zero mass carry no conditional distribution. The "min → 0" form also asks for more than a property of limit points: it is a uniform statement about all large t.
Formalization scope
Strategies are Fin m and Fin n; payoffs are real matrices; forecasts are real vectors required to be probability vectors in every round.
A correlated equilibrium is the joint-distribution form of p. 44 (the correlated strategy on a finite probability space is represented by its law). It is the ε=0, two-player, payoff (not cost) instance of the published AGT.IsCorrelatedEquilibrium, restated rather than imported.
The stationary deterministic tie-break is modelled by arbitrary best-reply functions Ri of the forecast. They do not depend on the round, and the statements quantify over all of them, which includes the lowest-index rule.
Forecasts are sequences fi:N→Rk. The theorem uses only the realized forecasts, and every sequence is realized by a rule reading the round number from the history.
Rounds are indexed from 0: "the first t rounds" are 0,…,t−1. D0=0 by Lean's division convention; only t≥1 and limits are used.
The calibration sum runs over the forecasts issued in the first t rounds, which is the paper's sum over all p with its zero terms removed. ρ(p,j,t)=0 when N(p,t)=0, as on the page.
"min … → 0" is stated as: for every ε>0, eventually some D∈π(G) is within ε of Dt in every coordinate. The two forms are equivalent because π(G) is compact and nonempty. The formalization avoids an infimum over π(G), which Lean would evaluate to 0 on an empty set.
A statement that drops the best-reply property, the probability-vector condition on forecasts, or the stationarity of Ri is not Theorem 1: the matching pennies example shows the last one is essential. Swapping the calibration hypotheses (player 1's forecast calibrated against player 1's own plays) type-checks when m=n and is not the theorem.
Only the two-player case is claimed. The paper says the results "generalize easily to the n-person case" without proof.
Proofs of any milestone, and reusable lemmas about calibration scores and compactness of the simplex of joint distributions, are welcome.
Selected references
D. P. Foster, R. V. Vohra, Calibrated learning and correlated equilibrium, Games and Economic Behavior 21 (1997) 40–55. https://doi.org/10.1006/game.1997.0595
S. Hart, A. Mas-Colell, A simple adaptive procedure leading to correlated equilibrium, Econometrica 68 (2000) 1127–1150. https://doi.org/10.1111/1468-0262.00153