Prove2Me
Navigate
DiscoverCollectionsFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

Machine Learning

653 missions · 438 completed

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.

Missions

Open215Completed438All653
Operations ResearchOptimization·Captain: mikedeng1

A Modern Introduction to Online Learning IX: Cover's Universal Portfolio with Uniform or Dirichlet(1/2) Prior Has Logarithmic Regret Against Constantly Rebalanced PortfoliosTextbook

Why sequential portfolios matter

A portfolio manager chooses fractions of current wealth to place in several stocks before seeing that day’s market gains. The gains need not follow a statistical model: a sequence may be selected adversarially, with some stocks losing all value on a given day. A useful benchmark is the best constantly rebalanced portfolio, which holds the same fractions each day but trades at the end of every round to restore them. Its fixed allocation is an active strategy, and it can grow even when buying and holding each individual stock does not. Chapter 10 of Orabona’s online learning text studies guarantees against this benchmark without imposing a positive lower bound or a finite upper bound on nonzero market gains.

The chapter presents a universal portfolio: an allocation determined by a prior distribution over all fixed portfolios and reweighted according to their wealth so far. The mission focuses on the uniform-prior part of Theorem 10.1. It asks for a worst-case logarithmic regret guarantee against every fixed comparator and every admissible market sequence. This is a deterministic guarantee; no distributional assumption about the market enters its statement.

Market and strategy

There are d≥2d\ge2d≥2 stocks and T≥1T\ge1T≥1 rounds. In round ttt, the market supplies a vector wt∈R≥0d∖{0}w_t\in\mathbb R_{≥0}^d\setminus\{0\}wt​∈R≥0d​∖{0} of nonnegative gains. A portfolio is u∈Δd−1u\in\Delta^{d-1}u∈Δd−1, the simplex of vectors with nonnegative coordinates summing to one. If one unit is invested initially and the allocation stays fixed, its wealth after TTT rounds is

WT(u)=∏t=1T⟨wt,u⟩.W_T(u)=\prod_{t=1}^{T}\langle w_t,u\rangle.WT​(u)=t=1∏T​⟨wt​,u⟩.

A sequential strategy chooses xt∈Δd−1x_t\in\Delta^{d-1}xt​∈Δd−1 before seeing wtw_twt​, and has wealth WT=∏t=1T⟨wt,xt⟩W_T=\prod_{t=1}^{T}\langle w_t,x_t\rangleWT​=∏t=1T​⟨wt​,xt​⟩. The quantity of interest is the difference between the logarithms of benchmark wealth and sequential wealth. A fixed portfolio can have zero wealth when all the stocks it holds have a zero gain on some day; its log wealth is then −∞-\infty−∞ in the mathematical convention of the book.

Algorithm 10.1 begins with a prior FFF on the simplex. Before round ttt, it gives each fixed portfolio uuu weight proportional to its wealth Wt−1(u)W_{t-1}(u)Wt−1​(u), and chooses the resulting mean allocation:

xt(i)=∫Δd−1uiWt−1(u) F(du)∫Δd−1Wt−1(u) F(du).x_t(i)=\frac{\int_{\Delta^{d-1}}u_i W_{t-1}(u)\,F(du)}{\int_{\Delta^{d-1}}W_{t-1}(u)\,F(du)}.xt​(i)=∫Δd−1​Wt−1​(u)F(du)∫Δd−1​ui​Wt−1​(u)F(du)​.

For this mission, FFF is the uniform probability measure on the simplex. It is represented by Lebesgue measure in the first d−1d-1d−1 coordinates, scaled by (d−1)!(d-1)!(d−1)!; the final coordinate is one minus their sum. This explicit density fixes what “uniform” means and prevents the prior from being chosen to suit a particular market sequence.

Formalization targets

The goal is the uniform-prior case of Theorem 10.1. With the notation above, for every fixed portfolio of positive wealth,

log⁡WT(u)−log⁡WT≤log⁡(T+d−1d−1)≤(d−1)[log⁡(Td−1+1)+1].\log W_T(u)-\log W_T \le \log\binom{T+d-1}{d-1} \le (d-1)\left[\log\left(\frac{T}{d-1}+1\right)+1\right].logWT​(u)−logWT​≤log(d−1T+d−1​)≤(d−1)[log(d−1T​+1)+1].

The goal also asserts WT>0W_T>0WT​>0, so the algorithm’s logarithmic wealth is defined. The first inequality gives an exact finite-horizon combinatorial bound; the second displays its logarithmic dependence on TTT for fixed ddd. The book also states a Dirichlet(1/2,…,1/2)(1/2,\ldots,1/2)(1/2,…,1/2) prior bound in the same theorem. That companion branch is not among this proposal’s draft items; its gamma-function normalization and integral identities would require a separate audited definition layer. The fixed series title follows the chapter plan, while the goal and milestone list specify the precise scope delivered here.

The attack path records four source statements. Lemma 10.5 bounds a ratio of nonnegative sums by the largest component ratio. Lemma 10.6 reduces a universal portfolio’s wealth ratio to a maximum over sequences of single-stock choices. Theorem 10.4 bounds the size of a type class, the set of sequences with the same empirical fractions of symbols. Equation (10.5) computes the monomial integral under the uniform simplex prior. These statements are from the chapter’s proof of Theorem 10.1; the equation is referred to as an exercise in Problem 10.3.

What the result provides

The theorem ensures that the uniform-prior strategy keeps its log wealth within O(dlog⁡(1+T/d))O(d\log(1+T/d))O(dlog(1+T/d)) of the best constantly rebalanced portfolio, while making no bounded-gain assumption. In particular, a stock may have gain zero on one day or a very large gain on another. The guarantee is pathwise: it applies to the realized market vectors, regardless of how they were chosen. For a fixed number of stocks, dividing regret by TTT therefore gives a vanishing average shortfall in log wealth.

The mathematical result is already proved in the cited text. This mission’s work is to give its exact algorithm, prior measure, combinatorial objects, and finite-horizon constants machine-checked statements and then proofs. The density-based prior, finite type classes, and simplex monomial integrals can also be reused in online aggregation and universal coding developments. The draft statements currently have Lean placeholders for proofs; a successful submission will still require solvers to close them.

Main difficulty

Applying a generic online convex optimization bound to the loss −log⁡⟨wt,x⟩-\log\langle w_t,x\rangle−log⟨wt​,x⟩ runs into unbounded gradients when a chosen portfolio has small gain. A bound based on one global Lipschitz constant cannot cover the market sequences allowed here. The integral strategy creates a different obstacle: its denominator must be positive at every round, including histories with zero gains for many fixed portfolios. Its uniform prior is a measure on a (d−1)(d-1)(d−1)-dimensional simplex embedded in Rd\mathbb R^dRd, so ambient ddd-dimensional volume would give it mass zero. A proof must work with the intrinsic coordinate measure and its normalization. The exact binomial coefficient also depends on counting all sequences of one type without losing the boundary cases where some symbol occurs zero times.

Formalization scope

Lean represents a portfolio as a function Fin d → ℝ in Mathlib’s stdSimplex; a market is a one-based sequence of vectors of the same type. The coordinate chart maps Rd−1\mathbb R^{d-1}Rd−1 into the simplex, and a scaled restricted Lebesgue measure is pushed forward to define the uniform prior. The chart’s measurability is proved in the definition file. Algorithm 10.1 is a componentwise ratio of real integrals, and the main theorem asserts positive realized wealth. All market vectors on rounds 1,…,T1,\ldots,T1,…,T are nonnegative and nonzero. The parameter restrictions d≥2d\ge2d≥2 and T≥1T\ge1T≥1 make the dimension and printed denominators meaningful.

For Lemma 10.5, quotients use extended nonnegative reals, with a/0=+∞a/0=+\inftya/0=+∞ when a>0a>0a>0 and 0/0=00/0=00/0=0 as printed. Lemma 10.6 is stated for a generic prior concentrated on the simplex, with its ratios in extended nonnegative reals under the same conventions, and with the hypothesis that the normalizing integral of Algorithm 10.1 is positive in every round, which is what makes the algorithm's prediction defined. A comparator with zero wealth is omitted only from the real-log statement, corresponding to log wealth −∞-\infty−∞ in the source. The prior cannot be replaced by an arbitrary point mass: the theorem names the chart-based uniform measure, and the prediction is the exact integral ratio. Contributions proving normalization, support, the integral identity, and the final regret bound are all within scope.

Selected references

  • Francesco Orabona, Online Learning: A Modern Introduction Using Convex Optimization, arXiv:1912.13213v10, 2026, Chapter 10 and Appendix A.2.2. arXiv
7 thms2 active usersReviewed
OptimizationProbability·Captain: mikedeng1

Stochastic Model-Based Minimization of Weakly Convex Functions 2: The Proximal Stochastic Subgradient Method Drives E‖∇φ_{1/ρ̄}(x_{t*})‖² to Zero at Rate O(1/√T)Research Paper

Motivation

The proximal stochastic subgradient method is the workhorse of large-scale learning: to minimize a regularized population risk φ(x)=f(x)+r(x)\varphi(x) = f(x) + r(x)φ(x)=f(x)+r(x) with f(x)=Eξ∼P[f(x,ξ)]f(x) = \mathbb E_{\xi \sim P}[f(x,\xi)]f(x)=Eξ∼P​[f(x,ξ)], it samples a data point, takes a step against a sampled subgradient of fff, and applies the proximal map of the regularizer rrr. Its sample complexity is classical for convex problems (O(ε−2)O(\varepsilon^{-2})O(ε−2) samples for function accuracy ε\varepsilonε) and for smooth nonconvex problems (O(ε−4)O(\varepsilon^{-4})O(ε−4) samples for gradient norm ε\varepsilonε, Ghadimi–Lan 2013). For problems that are both nonsmooth and nonconvex, however, no complexity bound was known for any reasonably broad class before Davis–Drusvyatskiy 2019.

Many such problems — robust phase retrieval, nonlinear least squares with an ℓ1\ell_1ℓ1​ loss, conditional value-at-risk minimization with smooth losses, covariance estimation — are weakly convex: they become convex after adding a quadratic ρ2∥x∥2\frac{\rho}{2}\|x\|^22ρ​∥x∥2. This mission formalizes the complexity guarantee of Davis and Drusvyatskiy for the proximal stochastic subgradient method on weakly convex problems.

Timeline. Nurminskii (1973) showed that for weakly convex f(⋅,ξ)f(\cdot,\xi)f(⋅,ξ) and r=0r = 0r=0 the stochastic subgradient method has limit points that are stationary, without a rate. Ghadimi and Lan (2013) proved the O(ε−4)O(\varepsilon^{-4})O(ε−4) rate for smooth fff. Davis and Drusvyatskiy (arXiv 2018, SIAM J. Optim. 2019) proved the first rate for the proximal stochastic subgradient method under weak convexity, measured by the gradient of the Moreau envelope; this is Theorem 3.4 of their paper.

Setting

Work in Rd\mathbb R^dRd with the Euclidean norm. Let r:Rd→R∪{+∞}r : \mathbb R^d \to \mathbb R \cup \{+\infty\}r:Rd→R∪{+∞} be closed (its epigraph is closed) and convex, with nonempty domain D=dom⁡rD = \operatorname{dom} rD=domr, and let f:Rd→Rf : \mathbb R^d \to \mathbb Rf:Rd→R be ρ\rhoρ-weakly convex: x↦f(x)+ρ2∥x∥2x \mapsto f(x) + \frac{\rho}{2}\|x\|^2x↦f(x)+2ρ​∥x∥2 is convex. Set φ=f+r\varphi = f + rφ=f+r.

The subdifferential ∂ψ(x)\partial\psi(x)∂ψ(x) of a function ψ\psiψ at a point xxx of its domain is the set of vectors vvv with ψ(y)≥ψ(x)+⟨v,y−x⟩+o(∥y−x∥)\psi(y) \ge \psi(x) + \langle v, y - x\rangle + o(\|y - x\|)ψ(y)≥ψ(x)+⟨v,y−x⟩+o(∥y−x∥) as y→xy \to xy→x; it is empty off the domain. For λ>0\lambda > 0λ>0 the Moreau envelope and the proximal map are

ψλ(x)=min⁡y{ψ(y)+12λ∥y−x∥2},prox⁡λψ(x)=argmin⁡y{ψ(y)+12λ∥y−x∥2}.\psi_\lambda(x) = \min_y \Big\{\psi(y) + \tfrac{1}{2\lambda}\|y - x\|^2\Big\}, \qquad \operatorname{prox}_{\lambda\psi}(x) = \operatorname*{argmin}_y \Big\{\psi(y) + \tfrac{1}{2\lambda}\|y - x\|^2\Big\}.ψλ​(x)=ymin​{ψ(y)+2λ1​∥y−x∥2},proxλψ​(x)=yargmin​{ψ(y)+2λ1​∥y−x∥2}.

Assumption A (stochastic subgradient oracle): on a probability space (Ω,F,P)(\Omega, \mathcal F, P)(Ω,F,P) one can draw i.i.d. samples ξ0,ξ1,⋯∼P\xi_0, \xi_1, \dots \sim Pξ0​,ξ1​,⋯∼P (A1); there are an open set U⊇DU \supseteq DU⊇D and a measurable map GGG with Eξ[G(x,ξ)]∈∂f(x)\mathbb E_\xi[G(x,\xi)] \in \partial f(x)Eξ​[G(x,ξ)]∈∂f(x) for all x∈Ux \in Ux∈U (A2); and Eξ∥G(x,ξ)∥2≤L2\mathbb E_\xi\|G(x,\xi)\|^2 \le L^2Eξ​∥G(x,ξ)∥2≤L2 for all x∈Dx \in Dx∈D (A3).

Algorithm 3.1 starts at x0∈Dx_0 \in Dx0​∈D and, for t=0,…,Tt = 0, \dots, Tt=0,…,T, samples ξt\xi_tξt​ and sets xt+1=prox⁡αtr(xt−αtG(xt,ξt))x_{t+1} = \operatorname{prox}_{\alpha_t r}(x_t - \alpha_t G(x_t, \xi_t))xt+1​=proxαt​r​(xt​−αt​G(xt​,ξt​)). It returns xt∗x_{t^*}xt∗​ with t∗t^*t∗ drawn from {0,…,T}\{0, \dots, T\}{0,…,T} with probability αt/∑iαi\alpha_t / \sum_i \alpha_iαt​/∑i​αi​.

Formalization targets

Goal: Theorem 3.4, (3.15) and (3.16)

For ρˉ∈(ρ,2ρ]\bar\rho \in (\rho, 2\rho]ρˉ​∈(ρ,2ρ] and stepsizes αt∈(0,1/ρˉ]\alpha_t \in (0, 1/\bar\rho]αt​∈(0,1/ρˉ​],

E[∥∇φ1/ρˉ(xt∗)∥2]≤ρˉρˉ−ρ⋅(φ1/ρˉ(x0)−min⁡φ)+2ρˉL2∑t=0Tαt2∑t=0Tαt,\mathbb E\big[\|\nabla\varphi_{1/\bar\rho}(x_{t^*})\|^2\big] \le \frac{\bar\rho}{\bar\rho - \rho}\cdot\frac{\big(\varphi_{1/\bar\rho}(x_0) - \min\varphi\big) + 2\bar\rho L^2\sum_{t=0}^T\alpha_t^2}{\sum_{t=0}^T\alpha_t},E[∥∇φ1/ρˉ​​(xt∗​)∥2]≤ρˉ​−ρρˉ​​⋅∑t=0T​αt​(φ1/ρˉ​​(x0​)−minφ)+2ρˉ​L2∑t=0T​αt2​​,

and with αt=γ/T+1\alpha_t = \gamma/\sqrt{T+1}αt​=γ/T+1​, γ∈(0,12ρ]\gamma \in (0, \frac{1}{2\rho}]γ∈(0,2ρ1​],

E[∥∇φ1/(2ρ)(xt∗)∥2]≤2⋅(φ1/(2ρ)(x0)−min⁡φ)+4ρL2γ2γT+1.\mathbb E\big[\|\nabla\varphi_{1/(2\rho)}(x_{t^*})\|^2\big] \le 2\cdot\frac{\big(\varphi_{1/(2\rho)}(x_0) - \min\varphi\big) + 4\rho L^2\gamma^2}{\gamma\sqrt{T+1}}.E[∥∇φ1/(2ρ)​(xt∗​)∥2]≤2⋅γT+1​(φ1/(2ρ)​(x0​)−minφ)+4ρL2γ2​.

Milestones

In the order the proof uses them: Lemma 2.1 (subdifferential characterization of weak convexity), the existence of v^t∈∂f(x^t)\hat v_t \in \partial f(\hat x_t)v^t​∈∂f(x^t​) with ρˉ(xt−x^t)∈∂r(x^t)+v^t\bar\rho(x_t - \hat x_t) \in \partial r(\hat x_t) + \hat v_tρˉ​(xt​−x^t​)∈∂r(x^t​)+v^t​ (§3.2), Lemma 3.2 (x^t\hat x_tx^t​ is a proximal point of rrr), Lemma 3.3 (one-step contraction towards x^t\hat x_tx^t​), Lemma 2.2 (the envelope is C1C^1C1 with ∇φλ(x)=λ−1(x−prox⁡λφ(x))\nabla\varphi_\lambda(x) = \lambda^{-1}(x - \operatorname{prox}_{\lambda\varphi}(x))∇φλ​(x)=λ−1(x−proxλφ​(x))), and the one-step descent estimate (3.14).

Significance

The result. A small ∥∇φλ(x)∥\|\nabla\varphi_\lambda(x)\|∥∇φλ​(x)∥ certifies that xxx is within λ∥∇φλ(x)∥\lambda\|\nabla\varphi_\lambda(x)\|λ∥∇φλ​(x)∥ of a point x^\hat xx^ with φ(x^)≤φ(x)\varphi(\hat x) \le \varphi(x)φ(x^)≤φ(x) and dist⁡(0,∂φ(x^))≤∥∇φλ(x)∥\operatorname{dist}(0, \partial\varphi(\hat x)) \le \|\nabla\varphi_\lambda(x)\|dist(0,∂φ(x^))≤∥∇φλ​(x)∥. Theorem 3.4 therefore gives a sample complexity of O(ε−4)O(\varepsilon^{-4})O(ε−4) for reaching an ε\varepsilonε-nearly stationary point, matching the smooth rate, for a method that is run in practice without modification. The underlying observation — the method is an approximate descent method on the Moreau envelope — is reused in the paper's model-based analysis (Theorem 4.3) and in subsequent work on stochastic nonsmooth nonconvex optimization.

Formalizing it. The theorem is proved in the paper; no machine-checked version exists. A complete development produces reusable infrastructure: the Fréchet subdifferential of extended-valued functions, weak convexity and its subgradient characterization, Moreau envelopes of weakly convex functions and their differentiability, nonexpansiveness of convex proximal maps, and the bookkeeping of iterates driven by i.i.d. samples under a product measure.

Difficulty

The obvious argument — show that φ(xt)\varphi(x_t)φ(xt​) decreases in expectation — fails: neither φ(xt)−min⁡φ\varphi(x_t) - \min\varphiφ(xt​)−minφ nor dist⁡(0,∂φ(xt))\operatorname{dist}(0, \partial\varphi(x_t))dist(0,∂φ(xt​)) need tend to zero, and the latter is discontinuous in xxx. Progress must be measured on the Moreau envelope, which requires comparing the iterate xt+1x_{t+1}xt+1​ with the proximal point x^t\hat x_tx^t​ of φ\varphiφ, not of rrr. In Lean, the heaviest single ingredient is Lemma 2.2 (differentiability of the envelope of an extended-valued weakly convex function), and the direction 4⇒14 \Rightarrow 14⇒1 of Lemma 2.1, which in the paper rests on Rockafellar–Wets' characterization of convexity by monotonicity of the subdifferential.

Formalization scope

  • Carrier. Rd\mathbb R^dRd is EuclideanSpace ℝ (Fin d). An extended-valued function is a pair (domain DDD, real values on DDD); closedness is closedness of {(x,t):x∈D, ψ(x)≤t}\{(x,t) : x \in D,\ \psi(x) \le t\}{(x,t):x∈D, ψ(x)≤t}. The subdifferential is the published regular subgradient NonconvexSplitting.Shared.IsRegularSubgrad of the +∞+\infty+∞-extended lift.
  • Proximal map. prox a z is a function constrained, for every a>0a > 0a>0, to return a minimizer over DDD of r(y)+12a∥y−z∥2r(y) + \frac{1}{2a}\|y - z\|^2r(y)+2a1​∥y−z∥2; this determines it uniquely.
  • Randomness. The samples ξ0,…,ξT\xi_0, \dots, \xi_Tξ0​,…,ξT​ form a point of ΩT+1\Omega^{T+1}ΩT+1 under the product measure P⊗(T+1)P^{\otimes(T+1)}P⊗(T+1); E\mathbb EE is the Bochner integral, and every bounded expectation is accompanied by a conclusion that its integrand is integrable. The conditional expectation Et\mathbb E_tEt​ in Lemma 3.3 is the integral over one sample with the iterate held fixed.
  • Output index. Because t∗t^*t∗ is independent of the samples, E∥∇φ1/ρˉ(xt∗)∥2\mathbb E\|\nabla\varphi_{1/\bar\rho}(x_{t^*})\|^2E∥∇φ1/ρˉ​​(xt∗​)∥2 is the α\alphaα-weighted average of E∥∇φ1/ρˉ(xt)∥2\mathbb E\|\nabla\varphi_{1/\bar\rho}(x_t)\|^2E∥∇φ1/ρˉ​​(xt​)∥2, as in the paper's (3.17).
  • min⁡φ\min\varphiminφ is an arbitrary real lower bound φ∗\varphi^*φ∗ of φ\varphiφ on DDD.
  • Corrected constant in (3.14). The paper prints the last term of (3.14) as αt2ρˉL2\alpha_t^2\bar\rho L^2αt2​ρˉ​L2; its proof, and the unfolded sum leading to (3.15), give 2αt2ρˉL22\alpha_t^2\bar\rho L^22αt2​ρˉ​L2, and that is what is stated. (3.15) and (3.16) are stated as printed.
  • Lemma 2.1 is stated for items 1–4; the clause for C2C^2C2-smooth functions is not stated. Lemma 2.2 adds the implicit hypotheses that φ\varphiφ is closed and proper, and includes existence and uniqueness of the proximal point.
  • Ruling out trivializations. The proximal map is tied to rrr, GGG is tied to ∂f\partial f∂f by (A2), min⁡φ\min\varphiminφ is not a real infimum, and integrability is concluded, so neither a junk value nor an unconstrained selection can make the statements hold vacuously; the hypotheses are satisfied by r=f=0r = f = 0r=f=0, G=0G = 0G=0.
  • Contributions welcome: proofs of the milestones, general lemmas on Moreau envelopes of weakly convex functions, and the nonexpansiveness of convex proximal maps on a closed convex domain.

Selected references

  • D. Davis and D. Drusvyatskiy, Stochastic Model-Based Minimization of Weakly Convex Functions, SIAM J. Optim. 29(1), 2019; arXiv:1803.06523v3, 2018. https://arxiv.org/abs/1803.06523
  • S. Ghadimi and G. Lan, Stochastic First- and Zeroth-Order Methods for Nonconvex Stochastic Programming, SIAM J. Optim. 23(4), 2013. https://doi.org/10.1137/120880811
  • R. T. Rockafellar and R. J.-B. Wets, Variational Analysis, Springer, 1998. https://doi.org/10.1007/978-3-642-02431-3
  • H. Robbins and S. Monro, A Stochastic Approximation Method, Ann. Math. Statist. 22(3), 1951. https://doi.org/10.1214/aoms/1177729586
12 thms2 active usersReviewed
Bandit AlgorithmsOperations ResearchStatistics·Captain: mikedeng1

Explore First, Exploit Next: The True Shape of Regret in Bandit Problems IV: Under Pairwise Symmetry a Suboptimal Arm Is Drawn T/K Times or Almost as Often as an Optimal ArmResearch Paper

Motivation

Lower bounds for stochastic multi-armed bandits are usually stated for a large horizon: Lai and Robbins (1985) and Burnetas and Katehakis (1996) show that a consistent strategy pulls a suboptimal arm at least about ln⁡T/Kinf⁡\ln T / \mathcal{K}_{\inf}lnT/Kinf​ times. These bounds say nothing about the small-horizon regime, the first rounds in which a strategy has not yet had time to separate the arms. Garivier, Ménard and Stoltz (arXiv:1602.07182v3, Mathematics of Operations Research, 2019) describe that regime through one information-theoretic inequality, their (6), and derive several lower bounds from it. This mission formalizes their relative lower bound, Theorem 3 (p. 11): under a mild symmetry assumption on the strategy, a suboptimal arm is either pulled at the uniform rate T/KT/KT/K or, on average, almost as often as an optimal arm. It is the statement behind the title's claim that bandit strategies must first explore.

Setting

There are KKK arms. A bandit problem ν‾=(ν1,…,νK)\underline{\nu} = (\nu_1, \dots, \nu_K)ν​=(ν1​,…,νK​) assigns to each arm aaa a probability distribution νa\nu_aνa​ on R\mathbb{R}R with expectation μa\mu_aμa​. The optimal mean is μ⋆=max⁡aμa\mu^\star = \max_a \mu_aμ⋆=maxa​μa​, the gap of arm aaa is Δa=μ⋆−μa\Delta_a = \mu^\star - \mu_aΔa​=μ⋆−μa​; arm aaa is optimal if Δa=0\Delta_a = 0Δa​=0 and suboptimal if Δa>0\Delta_a > 0Δa​>0.

A model D\mathcal{D}D is a set of distributions on R\mathbb{R}R, each with an expectation; a bandit problem is in D\mathcal{D}D when every νa\nu_aνa​ belongs to D\mathcal{D}D.

At each round ttt a strategy ψ\psiψ picks an arm AtA_tAt​, possibly at random, as a function of the past arms and rewards, and receives a reward YtY_tYt​ drawn from νAt\nu_{A_t}νAt​​. The number of pulls of arm aaa in the first TTT rounds is

Nψ,a(T)=∑t=1T1{At=a},N_{\psi,a}(T) = \sum_{t=1}^{T} \mathbb{1}\{A_t = a\},Nψ,a​(T)=t=1∑T​1{At​=a},

and Eν‾\mathbb{E}_{\underline{\nu}}Eν​​ denotes expectation when ψ\psiψ plays against ν‾\underline{\nu}ν​. Write Nψ,a+(T)=max⁡{Nψ,a(T),1}N^+_{\psi,a}(T) = \max\{N_{\psi,a}(T), 1\}Nψ,a+​(T)=max{Nψ,a​(T),1}.

Kullback–Leibler divergence. KL(P,Q)∈[0,+∞]\mathrm{KL}(P, Q) \in [0, +\infty]KL(P,Q)∈[0,+∞] is the Kullback–Leibler divergence between distributions, and kl(p,q)\mathrm{kl}(p, q)kl(p,q) the divergence between Bernoulli laws of parameters ppp and qqq.

Pairwise symmetry (Definition 3, p. 11). A strategy ψ\psiψ is pairwise symmetric for optimal arms on D\mathcal{D}D if for every bandit problem ν‾\underline{\nu}ν​ in D\mathcal{D}D, every pair of optimal arms a⋆,a⋆a^\star, a_\stara⋆,a⋆​ with νa⋆=νa⋆\nu_{a^\star} = \nu_{a_\star}νa⋆​=νa⋆​​ and every T≥1T \ge 1T≥1, the pairs (Nψ,a⋆(T),Nψ,a⋆(T))(N_{\psi,a^\star}(T), N_{\psi,a_\star}(T))(Nψ,a⋆​(T),Nψ,a⋆​​(T)) and (Nψ,a⋆(T),Nψ,a⋆(T))(N_{\psi,a_\star}(T), N_{\psi,a^\star}(T))(Nψ,a⋆​​(T),Nψ,a⋆​(T)) have the same distribution. A strategy that uses only the observed payoffs, and not the labels of the arms, has this property.

Formalization targets

Goal: Theorem 3 (p. 11)

For every model D\mathcal{D}D, every strategy ψ\psiψ pairwise symmetric for optimal arms on D\mathcal{D}D, every bandit problem ν‾\underline{\nu}ν​ in D\mathcal{D}D, every suboptimal arm aaa, every optimal arm a⋆a^\stara⋆ with KL(νa,νa⋆)<∞\mathrm{KL}(\nu_a, \nu_{a^\star}) < \inftyKL(νa​,νa⋆​)<∞ and every T≥1T \ge 1T≥1,

Eν‾[Nψ,a(T)]≥TKorEν‾[Nψ,a+(T)Nψ,a⋆+(T)]≥1−22T KL(νa,νa⋆)K.\mathbb{E}_{\underline{\nu}}\big[N_{\psi,a}(T)\big] \ge \frac{T}{K} \quad\text{or}\quad \mathbb{E}_{\underline{\nu}}\left[\frac{N^+_{\psi,a}(T)}{N^+_{\psi,a^\star}(T)}\right] \ge 1 - 2\sqrt{\frac{2T\,\mathrm{KL}(\nu_a, \nu_{a^\star})}{K}} .Eν​​[Nψ,a​(T)]≥KT​orEν​​[Nψ,a⋆+​(T)Nψ,a+​(T)​]≥1−2K2TKL(νa​,νa⋆​)​​.

Companion: the "in particular" clause

For T≤K/(32 KL(νa,νa⋆))T \le K / (32\,\mathrm{KL}(\nu_a, \nu_{a^\star}))T≤K/(32KL(νa​,νa⋆​)) the second alternative becomes Eν‾[Nψ,a+(T)/Nψ,a⋆+(T)]≥1/2\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}(T)/N^+_{\psi,a^\star}(T)] \ge 1/2Eν​​[Nψ,a+​(T)/Nψ,a⋆+​(T)]≥1/2.

Milestones (p. 12)

  1. The symmetry identity: for the alternative problem ν‾′\underline{\nu}'ν​′ obtained by replacing νa\nu_aνa​ by νa⋆\nu_{a^\star}νa⋆​, Eν‾′[Nψ,a+/(Nψ,a++Nψ,a⋆+)]=1/2\mathbb{E}_{\underline{\nu}'}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})] = 1/2Eν​′​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)]=1/2.
  2. Inequality (12): Eν‾[Nψ,a(T)] KL(νa,νa′)≥kl(Eν‾[Nψ,a+/(Nψ,a++Nψ,a⋆+)],1/2)\mathbb{E}_{\underline{\nu}}[N_{\psi,a}(T)]\,\mathrm{KL}(\nu_a, \nu'_a) \ge \mathrm{kl}\big(\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})], 1/2\big)Eν​​[Nψ,a​(T)]KL(νa​,νa′​)≥kl(Eν​​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)],1/2).
  3. The Jensen display: Eν‾[Nψ,a+/(Nψ,a++Nψ,a⋆+)]≤r/(1+r)\mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})] \le r/(1+r)Eν​​[Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​)]≤r/(1+r) with r=Eν‾[Nψ,a+/Nψ,a⋆+]r = \mathbb{E}_{\underline{\nu}}[N^+_{\psi,a}/N^+_{\psi,a^\star}]r=Eν​​[Nψ,a+​/Nψ,a⋆+​].
  4. The bound r/(1+r)≥1/2−T KL(νa,νa′)/(2K)r/(1+r) \ge 1/2 - \sqrt{T\,\mathrm{KL}(\nu_a,\nu'_a)/(2K)}r/(1+r)≥1/2−TKL(νa​,νa′​)/(2K)​ in the case Eν‾[Nψ,a(T)]≤T/K\mathbb{E}_{\underline{\nu}}[N_{\psi,a}(T)] \le T/KEν​​[Nψ,a​(T)]≤T/K, r≤1r \le 1r≤1.

Significance

The theorem shows that in the initial phase a symmetric strategy cannot discard a suboptimal arm quickly: unless the arm is already sampled at the uniform rate, it is sampled, in ratio, nearly as often as an optimal arm, as long as TTT is small compared with K/KL(νa,νa⋆)K / \mathrm{KL}(\nu_a, \nu_{a^\star})K/KL(νa​,νa⋆​). Together with Theorem 2 (absolute lower bound) and Theorem 4 (collective lower bound) it describes the regret curve before the logarithmic regime: roughly linear growth, then a transition. It requires no assumption on the model beyond the existence of expectations, and it involves the divergence between the two arms themselves rather than the model-dependent quantity Kinf⁡\mathcal{K}_{\inf}Kinf​.

The result is proved in the paper; to our knowledge it has no machine-checked proof. The mission produces a Lean statement of Theorem 3 on the canonical bandit model already used on the platform (StochasticBandit, BanditPolicy), a formal version of the symmetry assumption that sibling missions reuse (mission V, the collective lower bound, uses the same Definition 3), and the four intermediate steps of the proof as separate targets.

Difficulty

The proof compares ν‾\underline{\nu}ν​ with an alternative problem ν‾′\underline{\nu}'ν​′ that differs from it only in arm aaa, and transfers information between them through the fundamental inequality (6) of §2, a data-processing argument on the law of the whole history. The step specific to Theorem 3 is the choice of the test variable: the expectation of Nψ,a+/(Nψ,a++Nψ,a⋆+)N^+_{\psi,a}/(N^+_{\psi,a} + N^+_{\psi,a^\star})Nψ,a+​/(Nψ,a+​+Nψ,a⋆+​) is known exactly under ν‾′\underline{\nu}'ν​′ only because the strategy is pairwise symmetric, and the symmetry is a statement about joint laws of pull counts, not about their expectations. A version of the assumption that only equates expected pull counts does not determine the expectation of this ratio. The ratio expectation then has to be extracted from a bound on E[x/(1+x)]\mathbb{E}[x/(1+x)]E[x/(1+x)], which needs Jensen's inequality in the right direction and a case split on whether r≤1r \le 1r≤1.

Formalization scope

Arms are Fin K (0-based). The horizon TTT is a natural number. A strategy is a published BanditPolicy: a family of Markov kernels from the observed history of (arm, reward) pairs to the next arm. Every strategy of the paper, which may use auxiliary uniform random variables, induces such a kernel with the same law of arms and rewards. The law of the history is the published banditMeasure, and Nψ,a(T)N_{\psi,a}(T)Nψ,a​(T) is the published armPullCount. The model is a set of measures on R\mathbb{R}R, and every statement assumes the paper's standing hypothesis (§1.1, p. 3) that each member is a probability measure with an integrable identity, together with ν‾\underline{\nu}ν​ in D\mathcal{D}D.

Committed conventions:

  • KL\mathrm{KL}KL and kl\mathrm{kl}kl take values in [0,+∞][0, +\infty][0,+∞]; the Bernoulli divergence is the Kullback–Leibler divergence of two Bernoulli laws on R\mathbb{R}R, not a real formula with finite values at q∈{0,1}q \in \{0, 1\}q∈{0,1}.
  • Theorem 3 adds the hypothesis KL(νa,νa⋆)<∞\mathrm{KL}(\nu_a, \nu_{a^\star}) < \inftyKL(νa​,νa⋆​)<∞. When the divergence is infinite the paper's bound reads ≥−∞\ge -\infty≥−∞ and is empty; reading +∞+\infty+∞ as the real number 000 would make the statement false.
  • The "in particular" clause is stated as 32 T KL(νa,νa⋆)≤K32\,T\,\mathrm{KL}(\nu_a,\nu_{a^\star}) \le K32TKL(νa​,νa⋆​)≤K in [0,+∞][0, +\infty][0,+∞], which needs no finiteness assumption.
  • Definition 3 is an equality of joint laws (push-forwards of the history law), for every T≥1T \ge 1T≥1, only for optimal arms of problems in D\mathcal{D}D with equal distributions. Invariance under all permutations of the arms would strengthen the hypothesis and weaken the theorem.
  • Ratios use max⁡{N,1}\max\{N, 1\}max{N,1} as real numbers, so no division by zero occurs; all integrands are bounded by TTT.

A formalization in which the symmetry assumption equates only expected pull counts, in which the divergence is a real number with junk value 000 at +∞+\infty+∞, or in which the disjunction is replaced by its second alternative, is not this theorem.

The mission builds on the published definitions StochasticBandit and BanditPolicy. A complete development needs the fundamental inequality (6) for kernel policies (the divergence decomposition, published as BanditAlgorithm.bandit_divergence_decomposition, plus data processing for the Bernoulli test), Pinsker's inequality for Bernoulli laws (proved on the platform as BanditAlgorithm.bernoulli_relative_entropy_pinsker), Jensen's inequality for concave functions (Mathlib), and a lemma that the law of the history is unchanged in distribution when two identically distributed optimal arms are relabelled. The measurability of pull counts and the relation between the extended-real and real Bernoulli divergences are reusable beyond this mission. Proofs of any milestone, and of the general fundamental inequality, are welcome.

Selected references

  • A. Garivier, P. Ménard, G. Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research 44(2), 2019. arXiv:1602.07182v3, doi:10.1287/moor.2017.0928
  • T. L. Lai, H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6(1), 1985. doi:10.1016/0196-8858(85)90002-8
  • A. N. Burnetas, M. N. Katehakis, Optimal adaptive policies for sequential allocation problems, Advances in Applied Mathematics 17(2), 1996. doi:10.1006/aama.1996.0007
  • T. Lattimore, C. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020. doi:10.1017/9781108571401
8 thms2 active usersReviewed
Dynamical SystemsHarmonic AnalysisProbability·Captain: mikedeng1

Approximation Bounds for Random Neural Networks and Reservoir Systems II: Echo State Networks with Random Weights Approximate Regular Functionals with an Explicit L² Error Bound (Theorem 2)Research Paper

Motivation

Reservoir computing learns an input/output system from data in two stages. A dynamical system, the reservoir, is driven by the input sequence and its state is recorded; then only a linear readout of that state is trained, by least squares. The echo state network (ESN) is the most widely used reservoir: its state evolves by xt=σ(Axt−1+Czt+ζ)x_t = \sigma(Ax_{t-1} + Cz_t + \zeta)xt​=σ(Axt−1​+Czt​+ζ), with the matrices A,CA, CA,C and the vector ζ\zetaζ drawn at random and never trained. ESNs predict chaotic systems well in practice (Jaeger & Haas, Science 2004), and they are known to be universal: for every fading-memory input/output map some ESN approximates it (Grigoryeva & Ortega, Neural Netw. 2018). Universality does not explain the practice, where the reservoir weights are random and fixed in advance. Gonon, Grigoryeva & Ortega (Ann. Appl. Probab. 2023) prove that an ESN whose weights are drawn from an explicit distribution that does not depend on the target approximates every sufficiently regular target, with an explicit L2L^2L2 error bound. This mission formalizes that result, Theorem 2, and the chain of results its proof rests on.

Setting

Time runs over Z−={0,−1,−2,… }\mathbb Z_- = \{0, -1, -2, \dots\}Z−​={0,−1,−2,…}. Inputs take values in a set DdD_dDd​ with 0∈Dd⊂BM⊂Rd0 \in D_d \subset B_M \subset \mathbb R^d0∈Dd​⊂BM​⊂Rd, where BMB_MBM​ is the closed Euclidean ball of radius M>0M > 0M>0. The input Z=(Zt)t∈Z−Z = (Z_t)_{t \in \mathbb Z_-}Z=(Zt​)t∈Z−​​ is a DdD_dDd​-valued process with law μZ\mu_ZμZ​. The target is a functional H∗:(Dd)Z−→RH^* : (D_d)^{\mathbb Z_-} \to \mathbb RH∗:(Dd​)Z−​→R, the present output as a function of the whole input history. All norms are Euclidean.

Regularity. Fix weights w−i>0w_{-i} > 0w−i​>0 with ∑ii w−i<∞\sum_i i\, w_{-i} < \infty∑i​iw−i​<∞. H∗H^*H∗ is www-Lipschitz with constant L>0L > 0L>0 if ∣H∗(u)−H∗(v)∣≤L∑i≥0w−i∥u−i−v−i∥|H^*(u) - H^*(v)| \le L\sum_{i \ge 0} w_{-i}\|u_{-i} - v_{-i}\|∣H∗(u)−H∗(v)∣≤L∑i≥0​w−i​∥u−i​−v−i​∥. Assumption 1 asks, in addition, for every T≥1T \ge 1T≥1: with q=d(T+1)q = d(T+1)q=d(T+1), the truncation HT∗(z0,…,z−T)=H∗(…,0,z−T,…,z0)H^*_T(z_0, \dots, z_{-T}) = H^*(\dots, 0, z_{-T}, \dots, z_0)HT∗​(z0​,…,z−T​)=H∗(…,0,z−T​,…,z0​) has a Fourier representation HT∗(u)=∫Rqei⟨w,u⟩gT(w) dwH^*_T(u) = \int_{\mathbb R^q} e^{i\langle w, u\rangle} g_T(w)\,dwHT∗​(u)=∫Rq​ei⟨w,u⟩gT​(w)dw on (Dd)T+1(D_d)^{T+1}(Dd​)T+1, with gT∈L1g_T \in L^1gT​∈L1 and ∫max⁡(1,∥w∥3)∣gT(w)∣2 dw<∞\int \max(1, \|w\|^3)|g_T(w)|^2\,dw < \infty∫max(1,∥w∥3)∣gT​(w)∣2dw<∞.

The random ESN. Fix T,N≥1T, N \ge 1T,N≥1, R>0R > 0R>0, and a bound MTM_TMT​ with ∥(Z0,…,Z−T)∥≤MT\|(Z_0, \dots, Z_{-T})\| \le M_T∥(Z0​,…,Z−T​)∥≤MT​ almost surely. Draw A1,…,ANA_1, \dots, A_NA1​,…,AN​ i.i.d. uniform on BR⊂RqB_R \subset \mathbb R^qBR​⊂Rq and, independently, ζ1,…,ζN\zeta_1, \dots, \zeta_Nζ1​,…,ζN​ i.i.d. uniform on [−max⁡(MTR,1),max⁡(MTR,1)][-\max(M_TR,1), \max(M_TR,1)][−max(MT​R,1),max(MT​R,1)]. With SSS the block shift on Rq\mathbb R^qRq and ccc the embedding of Rd\mathbb R^dRd into the first block, and aaa the matrix with rows Ai⊤A_i^\topAi⊤​, put

Aˉ=(S0aS0),Cˉ=(cac),ζˉ=(0ζ),A=(Aˉ−Aˉ−AˉAˉ),C=(Cˉ−Cˉ),ζ=(ζˉ−ζˉ).\bar A = \begin{pmatrix} S & 0\\ aS & 0\end{pmatrix},\quad \bar C = \begin{pmatrix} c\\ ac\end{pmatrix},\quad \bar\zeta = \begin{pmatrix} 0\\ \zeta\end{pmatrix},\quad A = \begin{pmatrix}\bar A & -\bar A\\ -\bar A & \bar A\end{pmatrix},\quad C = \begin{pmatrix}\bar C\\ -\bar C\end{pmatrix},\quad \zeta = \begin{pmatrix}\bar\zeta\\ -\bar\zeta\end{pmatrix}.Aˉ=(SaS​00​),Cˉ=(cac​),ζˉ​=(0ζ​),A=(Aˉ−Aˉ​−AˉAˉ​),C=(Cˉ−Cˉ​),ζ=(ζˉ​−ζˉ​​).

The ESN is Xt=σ(AXt−1+CZt+ζ)X_t = \sigma(AX_{t-1} + CZ_t + \zeta)Xt​=σ(AXt−1​+CZt​+ζ), Yt=WXtY_t = WX_tYt​=WXt​, with σ(x)=max⁡(x,0)\sigma(x) = \max(x, 0)σ(x)=max(x,0) and state space R2(N+d(T+1))\mathbb R^{2(N+d(T+1))}R2(N+d(T+1)).

Formalization targets

Goal: Theorem 2 (p. 54)

For every H∗H^*H∗ satisfying Assumption 1 there is a readout WWW, a function of the random weights alone, such that the ESN has the echo state property (a unique state sequence for every input) and

E[∣Y0−H∗(Z)∣2]1/2≤CT,RN+∫Rq∖BR∣gT(u)∣ du+LM∑i>Tw−i,\mathbb E\big[|Y_0 - H^*(Z)|^2\big]^{1/2} \le \frac{\sqrt{C_{T,R}}}{\sqrt N} + \int_{\mathbb R^q \setminus B_R}|g_T(u)|\,du + LM\sum_{i > T} w_{-i},E[∣Y0​−H∗(Z)∣2]1/2≤N​CT,R​​​+∫Rq∖BR​​∣gT​(u)∣du+LMi>T∑​w−i​,

with CT,R=16max⁡(MTR,1)Volq(BR)(MT+1)2(MT3+MT+16)∫BRmax⁡(1,∥u∥3)∣gT(u)∣2 duC_{T,R} = 16\max(M_TR,1)\mathrm{Vol}_q(B_R)(M_T+1)^2(M_T^3+M_T+16)\int_{B_R}\max(1,\|u\|^3)|g_T(u)|^2\,duCT,R​=16max(MT​R,1)Volq​(BR​)(MT​+1)2(MT3​+MT​+16)∫BR​​max(1,∥u∥3)∣gT​(u)∣2du.

Milestones

  1. (8), p. 33: a scalar identity expressing eiz−iz−1e^{iz} - iz - 1eiz−iz−1 as an integral of ReLU ramps.
  2. Proposition 2, pp. 31–32: under a Fourier condition, fff is an integral of ReLU ridge functions with a compactly supported, square-integrable weight π\piπ.
  3. Proposition 3, pp. 41–42: a static ReLU network with uniformly sampled inner weights achieves mean-square error C∗/NC^*/NC∗/N.
  4. Corollary 1, p. 43: the same with a dimension-free integrability condition and a truncation term.
  5. Remark 10, p. 53: the shift reservoir is nilpotent, has K=IK = IK=I and stores the last T+1T+1T+1 inputs.
  6. Proposition 6, p. 51: a linear reservoir with a random ReLU readout satisfies the bound (54).
  7. (64)–(66), pp. 55–56: every solution of the ESN reduces to the linear shift reservoir, which gives uniqueness.

Significance

Theorem 2 is the first approximation bound for echo state networks whose recurrent and input weights are sampled from a generic distribution rather than constructed from the target. The bound separates three error sources: a Monte Carlo term decaying like N−1/2N^{-1/2}N−1/2, a truncation term in the sampling radius RRR, and a memory term in the window length TTT. Because the readout can be taken to be the least-squares readout, the bound also controls the trained networks used in practice, and it is the input to the paper's rate and universality corollaries (Corollaries 4 and 5).

Status: the result is proved on paper. No part of it is formalized; the only reservoir-computing material on the platform is the published definition ReservoirESN (the state equation), which this mission reuses. A formal proof would supply the first machine-checked random-feature approximation bound with explicit constants.

Difficulty

The obvious argument fails in two places. First, the classical integral representations of Barron type place the inner weights on unbounded sets, which cannot be sampled uniformly; Proposition 2 needs a change of variables that makes the weight compactly supported while keeping it square integrable, and this costs the factor R−(2q+2)R^{-(2q+2)}R−(2q+2) and the condition (5). Second, a ReLU network is not contractive, so the echo state property of the ESN cannot be taken from the standard contraction criterion; it has to be derived from the block structure of AAA, which makes the difference of the two state halves follow a nilpotent linear recursion.

Formalization scope

Rq\mathbb R^qRq is EuclideanSpace ℝ (Fin (T+1) × Fin d), block kkk holding z−kz_{-k}z−k​; input sequences are ℕ → EuclideanSpace ℝ (Fin d) with index k=−tk = -tk=−t, and xt=F(xt−1,zt)x_t = F(x_{t-1}, z_t)xt​=F(xt−1​,zt​) is the published ReservoirESN.IsSolution. Points of Rq+1\mathbb R^{q+1}Rq+1 are pairs with norm (∥w∥2+u2)1/2(\|w\|^2 + u^2)^{1/2}(∥w∥2+u2)1/2. The uniform laws are Lebesgue measure conditioned on the ball or interval; NNN i.i.d. rows have the product law. The input is independent of the weights (§4.1) and DdD_dDd​-valued with 0∈Dd⊂BM0 \in D_d \subset B_M0∈Dd​⊂BM​ (§5.1): expectations are iterated lower Lebesgue integrals over μZ\mu_ZμZ​ and the product law, and probability-(1−δ)(1-\delta)(1−δ) statements bound the product measure of the bad set. The scalar output case m=1m = 1m=1 is used, as in the paper. Assumption 1 is required for every T≥1T \ge 1T≥1, as printed.

A readout allowed to depend on the input would make every bound trivial; in every statement the readout is a measurable function of the inner weights alone.

Departures from the print, each disclosed in the item: the last term of Proposition 6 is the L2(μZ)L^2(\mu_Z)L2(μZ​) norm of ∥K−1ST+1X−T−1∥\|K^{-1}S^{T+1}X_{-T-1}\|∥K−1ST+1X−T−1​∥, which the page prints outside the expectation; the state space of the linear reservoir (52) is a parameter DND_NDN​, as in §2.1, and its solution is assumed to be a random process; Proposition 3 omits "C∗>0C^* > 0C∗>0" (the explicit value (33) is used); measurability of the Fourier density ggg is stated where the page leaves it implicit.

A complete development needs Fourier integrals on Euclidean spaces, change of variables under linear maps, product measures and Fubini–Tonelli, and an importance-sampling variance bound. The static results (Propositions 2, 3, Corollary 1) are reusable for random-feature models beyond reservoir computing. Proofs of any milestone, and alternative arguments, are welcome.

Selected references

  • L. Gonon, L. Grigoryeva, J.-P. Ortega, Approximation bounds for random neural networks and reservoir systems, Ann. Appl. Probab. 33(1), 28–69, 2023. https://doi.org/10.1214/22-AAP1806
  • L. Grigoryeva, J.-P. Ortega, Echo state networks are universal, Neural Netw. 108, 495–508, 2018. https://doi.org/10.1016/j.neunet.2018.08.025
  • H. Jaeger, H. Haas, Harnessing nonlinearity: predicting chaotic systems and saving energy in wireless communication, Science 304, 78–80, 2004. https://doi.org/10.1126/science.1091277
  • A. R. Barron, Universal approximation bounds for superpositions of a sigmoidal function, IEEE Trans. Inf. Theory 39, 930–945, 1993. https://doi.org/10.1109/18.256500
  • J. M. Klusowski, A. R. Barron, Approximation by combinations of ReLU and squared ReLU ridge functions with ℓ1 and ℓ0 controls, IEEE Trans. Inf. Theory 64, 7649–7656, 2018. https://doi.org/10.1109/tit.2018.2874447
  • C. Cuchiero, L. Gonon, L. Grigoryeva, J.-P. Ortega, J. Teichmann, Discrete-time signatures and randomness in reservoir computing, IEEE Trans. Neural Netw. Learn. Syst., 2021. https://doi.org/10.1109/TNNLS.2021.3076777
11 thms2 active usersReviewed
Convex OptimizationOptimization·Captain: mikedeng1

A Two-Timescale Stochastic Algorithm Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Critic: Explicit Error Bounds for TTSA with a Strongly Convex Outer ObjectiveResearch Paper

Motivation

Bilevel optimization asks for an outer decision while an inner optimization problem responds to that decision. In hyperparameter selection, for example, the outer objective evaluates a model whose parameters are chosen by a training problem. A two-timescale algorithm updates both levels in one stochastic loop, so its outer step uses an inner solution that has not yet settled. The resulting error bound must account for both the outer decision and the inner tracking error. Hong, Wai, Wang and Yang study this setting and give finite-time bounds for their two-timescale stochastic approximation (TTSA) algorithm in the strongly convex outer case Hong et al., arXiv:2007.05170v4, §§1–3.

The paper states its main rates using ≲\lesssim≲, with numerical constants omitted. Appendix A gives explicit bounds for the two errors. This mission targets those displayed bounds, retaining their constants and the actual stochastic update. The selected source is the arXiv v4 preprint: page numbers and theorem labels below refer to that version Hong et al., Theorem 1 and App. A.

Setting

Let X⊆Rd1X\subseteq\mathbb R^{d_1}X⊆Rd1​ be closed and convex. For an outer point x∈Xx\in Xx∈X, the inner solution y⋆(x)∈Rd2y^\star(x)\in\mathbb R^{d_2}y⋆(x)∈Rd2​ minimizes g(x,⋅)g(x,\cdot)g(x,⋅). The outer objective is ℓ(x)=f(x,y⋆(x))\ell(x)=f(x,y^\star(x))ℓ(x)=f(x,y⋆(x)), and x⋆∈Xx^\star\in Xx⋆∈X is an optimal outer point. The inner objective is strongly convex in yyy, which gives a unique inner minimizer. The paper’s derivative and Lipschitz assumptions control how that minimizer and the outer gradient change as xxx and yyy move Hong et al., Assumptions 1–2, pp. 5–6.

The algorithm keeps iterates xk∈Xx^k\in Xxk∈X and yk∈Rd2y^k\in\mathbb R^{d_2}yk∈Rd2​. With deterministic step sizes αk,βk\alpha_k,\beta_kαk​,βk​ and stochastic estimates hfk,hgkh_f^k,h_g^khfk​,hgk​, it updates

yk+1=yk−βkhgk,xk+1=PX(xk−αkhfk),y^{k+1}=y^k-\beta_k h_g^k,\qquad x^{k+1}=P_X(x^k-\alpha_k h_f^k),yk+1=yk−βk​hgk​,xk+1=PX​(xk−αk​hfk​),

where PXP_XPX​ is the Euclidean nearest-point projection onto XXX. The inner estimate targets ∇yg(xk,yk)\nabla_y g(x^k,y^k)∇y​g(xk,yk). The outer estimate targets the surrogate gradient

∇ˉxf(x,y)=∇xf(x,y)−∇xy2g(x,y)[∇yy2g(x,y)]−1∇yf(x,y),\bar\nabla_xf(x,y)=\nabla_xf(x,y)-\nabla^2_{xy}g(x,y) [\nabla^2_{yy}g(x,y)]^{-1}\nabla_yf(x,y),∇ˉx​f(x,y)=∇x​f(x,y)−∇xy2​g(x,y)[∇yy2​g(x,y)]−1∇y​f(x,y),

with a bias bounded by bkb_kbk​. Assumption 3 bounds the conditional variances given the histories Fk\mathcal F_kFk​ and Fk′\mathcal F'_kFk′​. The surrogate at y⋆(x)y^\star(x)y⋆(x) is the paper’s ∇ℓ(x)\nabla\ell(x)∇ℓ(x) Hong et al., Algorithm 1, (5)–(7c), p. 6.

The two quantities being bounded are the outer error Δxk=E∥xk−x⋆∥2\Delta_x^k=\mathbb E\|x^k-x^\star\|^2Δxk​=E∥xk−x⋆∥2 and inner tracking error Δyk=E∥yk−y⋆(xk−1)∥2\Delta_y^k=\mathbb E\|y^k-y^\star(x^{k-1})\|^2Δyk​=E∥yk−y⋆(xk−1)∥2. At k=0k=0k=0, the convention is x−1=x0x^{-1}=x^0x−1=x0. The constants LLL and LyL_yLy​ are the exact functions of the derivative bounds in (13); σ~f2=σf2+3sup⁡x∈X∥∇ℓ(x)∥2\widetilde\sigma_f^2=\sigma_f^2+3\sup_{x\in X}\|\nabla\ell(x)\|^2σf2​=σf2​+3supx∈X​∥∇ℓ(x)∥2 is finite by the paper’s standing assumption after (14). The products Gm:n(1)G^{(1)}_{m:n}Gm:n(1)​ and Gm:n(2)G^{(2)}_{m:n}Gm:n(2)​ are those of (48) Hong et al., pp. 8–9, 20.

Formalization targets

The first target is the explicit inner bound (55): for every k≥0k\ge0k≥0,

Δyk+1≤G0:k(1)Δy0+Cy(1)βk,Cy(1)=8μg{σg2+4c02Ly2μg(σ~f2+3b02)}.\Delta_y^{k+1}\le G^{(1)}_{0:k}\Delta_y^0+C_y^{(1)}\beta_k, \qquad C_y^{(1)}=\frac8{\mu_g}\left\{\sigma_g^2+ \frac{4c_0^2L_y^2}{\mu_g}(\widetilde\sigma_f^2+3b_0^2)\right\}.Δyk+1​≤G0:k(1)​Δy0​+Cy(1)​βk​,Cy(1)​=μg​8​{σg2​+μg​4c02​Ly2​​(σf2​+3b02​)}.

The goal also includes the explicit outer bound displayed after (59):

Δxk+1≤G0:k(2){Δx0+[2L2μℓ2+3α0L2μℓ]Δy0}+2μℓ[2c~bμℓ+σ~f2+3b02]αk+2c1μℓ[2L2μℓ+3α0L2]Cy(1)αk2/3.\begin{aligned} \Delta_x^{k+1}\le{}&G^{(2)}_{0:k} \left\{\Delta_x^0+\left[\frac{2L^2}{\mu_\ell^2} +\frac{3\alpha_0L^2}{\mu_\ell}\right]\Delta_y^0\right\}\\ &+\frac2{\mu_\ell}\left[\frac{2\widetilde c_b}{\mu_\ell} +\widetilde\sigma_f^2+3b_0^2\right]\alpha_k\\ &+\frac{2c_1}{\mu_\ell}\left[\frac{2L^2}{\mu_\ell}+3\alpha_0L^2\right] C_y^{(1)}\alpha_k^{2/3}. \end{aligned}Δxk+1​≤​G0:k(2)​{Δx0​+[μℓ2​2L2​+μℓ​3α0​L2​]Δy0​}+μℓ​2​[μℓ​2cb​​+σf2​+3b02​]αk​+μℓ​2c1​​[μℓ​2L2​+3α0​L2]Cy(1)​αk2/3​.​

These formulas use the paper’s step-size conditions (20a)–(20b), the bias condition bk2≤c~bαk+1b_k^2\le\widetilde c_b\alpha_{k+1}bk2​≤cb​αk+1​, and a positive outer modulus μℓ\mu_\ellμℓ​. The milestones are the source’s Lemma 2 (12a), (14), (50), Lemma 10, (55), Lemma 11 and Lemma 4 (29). They supply independently stated targets for the two error quantities and the deterministic product estimates Hong et al., pp. 7–8, 13, 20–22, 34.

Significance

The bounds separate the effect of initialization from persistent stochastic error. Each GGG product carries an initial-error term, while the remaining terms show how the current step sizes, variance and bias govern the residual error. This makes the dependencies on LLL, LyL_yLy​, the strong convexity moduli and the estimator variances explicit, rather than hiding them in rate notation. The paper uses this strongly convex case as one part of its broader complexity analysis Hong et al., §3, Table 1.

The source proves the claimed rates on paper; this local development currently provides compiled Lean statements with open proofs. Completing it would add reusable formal interfaces for conditional stochastic-gradient assumptions, a moving strongly convex minimizer, and weighted products of varying contraction factors. The local mission does not claim a machine-checked proof of Theorem 1 yet.

Difficulty

The outer estimate depends on yk+1y^{k+1}yk+1, while the inner tracking error is measured against the minimizer at the preceding outer iterate. Treating the outer estimate as an ordinary unbiased gradient at xkx^kxk loses both the surrogate-gradient discrepancy and the bias. A distance contraction for the inner step alone also does not settle the outer error, because the target y⋆(xk)y^\star(x^k)y⋆(xk) moves with xkx^kxk. The assumptions must keep the two histories and their conditional estimates distinct Hong et al., (7a)–(7c), (17), App. A.

Formalization scope

Lean represents both spaces by EuclideanSpace ℝ (Fin d) with d1,d2≥1d_1,d_2\ge1d1​,d2​≥1, and Hessian blocks by continuous linear maps with operator norm. The mixed block maps inner directions to outer directions. The inner solution map is required to minimize g(x,⋅)g(x,\cdot)g(x,⋅) on every x∈Xx\in Xx∈X; the projection is characterized as a nearest-point map on the closed convex set. These clauses prevent an arbitrary solution map or arbitrary outer update from satisfying the theorem. The ∇ℓ\nabla\ell∇ℓ used in (2) is defined by the displayed surrogate formula (5), so boundary points of XXX have the same meaning as in the source.

The run starts at deterministic (x0,y0)(x^0,y^0)(x0,y0) and uses measurable estimates with integrable squared norms. Conditional expectations are almost-sure statements relative to the sigma-algebras generated by the iterates, and Δx,Δy\Delta_x,\Delta_yΔx​,Δy​ are Bochner integrals. These regularity assumptions prevent totalized conditional expectation or integral operations from giving spurious zero values. The supremum defining σ~f2\widetilde\sigma_f^2σf2​ is taken over XXX and has an explicit bounded-above hypothesis; XXX is nonempty because x0∈Xx^0\in Xx0∈X. The goal uses positive, nonincreasing step sizes, in line with the ratio conditions and the auxiliary lemmas. The paper prints 3/43/43/4 in the α\alphaα ratio of (20a), while its application of Lemma 10 with q=2q=2q=2 needs 1/21/21/2; the stronger ratio is disclosed in the Lean hypotheses. The initial error uses x−1=x0x^{-1}=x^0x−1=x0, and real exponents use nonnegative bases.

The definition file and the seven milestone theorems are open for proof contributions. Theorem 1’s two explicit bounds are the mission goal. Lemma 3 (28) is excluded because its printed product uses a 1/21/21/2 factor where (55) uses 1/41/41/4; (55) is the selected inner target Hong et al., pp. 13, 21.

Selected references

  • M. Hong, H.-T. Wai, Z. Wang and Z. Yang, A Two-Timescale Stochastic Algorithm Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Critic, SIAM Journal on Optimization, 2023; arXiv preprint v4, 2022. arXiv:2007.05170v4.
10 thms2 active usersReviewed
Bandit AlgorithmsOperations ResearchStatistics·Captain: mikedeng1

Explore First, Exploit Next: The True Shape of Regret in Bandit Problems II: Uniformly Fast Convergent Strategies Draw Every Suboptimal Arm at Least ln T / K_inf Times AsymptoticallyResearch Paper

Why this lower bound matters

In a stochastic multi-armed bandit, a player repeatedly chooses one of finitely many arms and sees only the reward from the chosen arm. To earn a high reward, the player must favor arms that appear best. Yet an arm that currently appears inferior might be best under another plausible reward law. That uncertainty forces any strategy with uniformly strong long-run performance to keep testing apparently inferior arms. The asymptotic number of tests is a basic benchmark for comparing bandit algorithms: an algorithm that matches the bound uses no more exploration than the information in the model demands. Garivier, Ménard, and Stoltz restate this general benchmark as Theorem 1 and derive it within their analysis of the shape of regret Garivier–Ménard–Stoltz, §2.2.

The quantity controlling the benchmark comes from Burnetas and Katehakis. It measures the least information needed to replace one suboptimal arm by a plausible arm that is strictly better than the current best. The paper cites their quantity on page 4, and its Theorem 1 applies it to every suboptimal arm separately Garivier–Ménard–Stoltz, pp. 4, 9. This per-arm form matters when different arms have different reward laws and different amounts of information are needed to rule them out.

Bandit setting and notation

A bandit problem ν‾=(νa)a=1K\underline\nu=(\nu_a)_{a=1}^Kν​=(νa​)a=1K​ has KKK arms. Each νa\nu_aνa​ is a probability distribution on real rewards with a finite expectation μa=EνaX\mu_a=\mathbb E_{\nu_a}Xμa​=Eνa​​X. A strategy ψ\psiψ chooses an arm at each round using the preceding chosen arms and observed rewards, and it may randomize. Let Nψ,a(T)N_{\psi,a}(T)Nψ,a​(T) be the number of times arm aaa is chosen in the first TTT rounds. The optimal mean is μ⋆=max⁡aμa\mu^\star=\max_a\mu_aμ⋆=maxa​μa​, and the gap of arm aaa is Δa=μ⋆−μa\Delta_a=\mu^\star-\mu_aΔa​=μ⋆−μa​. An arm is suboptimal when Δa>0\Delta_a>0Δa​>0. These are the objects of §1.1 of the paper Garivier–Ménard–Stoltz, pp. 2–3.

A model D\mathcal DD is a collection of possible real-valued reward distributions, each with a finite expectation. A bandit problem belongs to D\mathcal DD when every one of its arm laws does. The Kullback–Leibler divergence KL(P,Q)\mathrm{KL}(P,Q)KL(P,Q) measures the information discrepancy from one reward law PPP to another QQQ; it may be infinite. For an arm law PPP and a real threshold xxx, define the minimal information cost

Kinf⁡(P,x,D)=inf⁡{KL(P,Q):Q∈D, EQX>x}.\mathcal K_{\inf}(P,x,\mathcal D) =\inf\{\mathrm{KL}(P,Q):Q\in\mathcal D,\ \mathbb E_QX>x\}.Kinf​(P,x,D)=inf{KL(P,Q):Q∈D, EQ​X>x}.

The infimum is +∞+\infty+∞ if there is no qualifying alternative. The strict inequality in the set matters: the replacement law must make the arm better than the original optimal mean, not merely tied with it Garivier–Ménard–Stoltz, p. 4.

Definition 1 calls ψ\psiψ uniformly fast convergent on D\mathcal DD when, on every bandit problem in the model, the expected draw count of every suboptimal arm grows more slowly than TαT^\alphaTα for each 0<α≤10<\alpha\le10<α≤1:

Eν‾Nψ,a(T)=o(Tα).\mathbb E_{\underline\nu}N_{\psi,a}(T)=o(T^\alpha).Eν​​Nψ,a​(T)=o(Tα).

Here TTT tends to infinity through integer horizons. This condition is uniform in which problem and suboptimal arm are considered; it does not mean one numerical convergence rate is shared by all problems Garivier–Ménard–Stoltz, p. 9, Definition 1.

Formalization target

The goal is the paper's Theorem 1: for every model D\mathcal DD, every uniformly fast convergent strategy ψ\psiψ, every bandit problem ν‾\underline\nuν​ in the model, and every suboptimal arm aaa,

lim inf⁡T→∞Eν‾Nψ,a(T)ln⁡T≥1Kinf⁡(νa,μ⋆,D).\liminf_{T\to\infty} \frac{\mathbb E_{\underline\nu}N_{\psi,a}(T)}{\ln T} \ge\frac1{\mathcal K_{\inf}(\nu_a,\mu^\star,\mathcal D)}.T→∞liminf​lnTEν​​Nψ,a​(T)​≥Kinf​(νa​,μ⋆,D)1​.

The bound states the minimum logarithmic exploration of each arm. It leaves the model and strategy arbitrary within the stated class. The milestones follow four statements from the proof on page 9: the Bernoulli entropy estimate in (11), the one-arm divergence chain in (10), the eventual bound on draws of the alternative's other arms, and the bound for one alternative law before taking the infimum. Their source text is recorded beside the local drafts Garivier–Ménard–Stoltz, p. 9.

What the result gives

If Kinf⁡\mathcal K_{\inf}Kinf​ is finite and positive, the theorem gives a concrete coefficient of ln⁡T\ln TlnT that any uniformly fast convergent strategy must spend on arm aaa. A small information cost means that the arm is hard to distinguish from a better alternative and must be sampled more often. If the cost is infinite because the model contains no better replacement, the reciprocal is zero and the bound makes no positive demand. If the cost is zero, the reciprocal is infinite: the normalized expected draw count must diverge. The extended-value convention is part of the target, not a choice of a finite stand-in Garivier–Ménard–Stoltz, pp. 4, 9.

The theorem is proved in the paper, while the Lean statements in this proposal are proof targets. A completed development would supply a machine-checked bridge from the canonical stochastic-bandit policy model through Bernoulli relative entropy and change of measure to the per-arm asymptotic claim. Its model and divergence definitions can also support lower bounds for other bandit classes. The existing Prove2Me StochasticBandit, BanditPolicy, and ConsistentBanditPolicy definitions provide the bandit law, policy, draw count, means, and Kinf⁡\mathcal K_{\inf}Kinf​ used here; Definition 1's exact strategy class is supplied locally.

Difficulty

The central issue is the quantification over every alternative law in an arbitrary model. A suboptimal arm may become uniquely optimal after changing only its law, while the strategy must remain uniformly fast on that new problem. The information cost is an infimum over such changes, including an empty set or costs arbitrarily close to zero. A statement that reasons about just one convenient alternative does not yield the coefficient in Theorem 1. The proof also passes from finite-horizon information inequalities to a liminf as TTT grows, so endpoint values of Bernoulli divergence and the extended-value arithmetic must agree with the source Garivier–Ménard–Stoltz, pp. 6, 9.

Formalization scope

Lean indexes the paper's arms 1,…,K1,\ldots,K1,…,K by Fin K, starting at zero, and uses natural-number horizons. A strategy is a history-dependent Markov kernel, whose generated probability measure describes the sequence of chosen arms and rewards. The pull count is bounded by TTT, so its expectation is a genuine Bochner integral. The model explicitly requires probability laws with integrable identity; this ensures that each mean in Kinf⁡\mathcal K_{\inf}Kinf​ is genuine. Bernoulli relative entropy and arm-law KL take values in extended nonnegative reals. The goal's liminf is also in extended nonnegative reals, avoiding a default real liminf for an unbounded sequence. The first two horizons, where ln⁡T≤0\ln T\le0lnT≤0, cannot affect an asymptotic liminf.

The theorem sentence on page 9 says “all bandit problems” without restricting the original problem to D\mathcal DD. Its proof applies Definition 1 to that problem, so the proposal states that its arms belong to D\mathcal DD. Without this condition a strategy can behave arbitrarily outside the model. The source's other standing conventions are kept: every reward law has an expectation, the improvement condition in Kinf⁡\mathcal K_{\inf}Kinf​ is strict, and uniform fast convergence ranges over all model problems, all suboptimal arms, and all 0<α≤10<\alpha\le10<α≤1. These choices rule out an empty strategy class or a special hard-coded bandit as a trivial route to the target.

Useful contributions include the finite-horizon change-of-measure inequality, endpoint-correct Bernoulli KL estimates, identities for sums of expected pull counts, and asymptotic lemmas for reciprocal infima. The milestone drafts identify the paper statements those contributions must support.

Selected references

  • Aurélien Garivier, Pierre Ménard, and Gilles Stoltz, Explore First, Exploit Next: The True Shape of Regret in Bandit Problems, Mathematics of Operations Research, 2019; cited here from arXiv:1602.07182v3, §§1–2.
  • Apostolos N. Burnetas and Michael N. Katehakis, Optimal Adaptive Policies for Sequential Allocation Problems, Advances in Applied Mathematics, 1996. DOI:10.1006/aama.1996.0007.
9 thms2 active usersReviewed
🏆Completed
Bandit AlgorithmsReinforcement Learning·Captain: mikedeng1

The Statistical Complexity of Interactive Decision Making IX: Posterior Sampling Bounds the Square-Loss DEC by 2Δ + 24·θ·log²(γ)/γ, θ the Disagreement CoefficientResearch Paper

Motivation

In a structured bandit problem a learner repeatedly chooses a decision π\piπ from a set Π\PiΠ and observes a noisy reward whose mean fM(π)f^M(\pi)fM(π) is an unknown function from a known class FM\mathcal{F}_{\mathcal{M}}FM​. Classical analyses of such problems measure the complexity of FM\mathcal{F}_{\mathcal{M}}FM​ by the eluder dimension (Russo and Van Roy, 2013; 2014) or the star number, and prove regret bounds for posterior sampling (Thompson sampling) or optimistic algorithms in terms of these quantities.

Foster, Kakade, Qian and Rakhlin (2021) introduced the Decision-Estimation Coefficient (DEC), a single complexity measure that yields both upper and lower bounds on the minimax regret of interactive decision making. Section 6.2 of that paper connects the DEC to the older parameters: it shows that the square-loss DEC is bounded by the disagreement coefficient of the class, a quantity introduced in Foster, Rakhlin, Simchi-Levi and Xu (2020) and itself bounded by the star number and the eluder dimension. The bound is certified by posterior sampling, so it also explains why posterior sampling works for these classes.

Setting

Let Π\PiΠ be a set of decisions. A model MMM assigns to each decision a reward distribution; here a model enters only through its mean-reward function fM:Π→[0,1]f^M : \Pi \to [0, 1]fM:Π→[0,1] and an optimal decision πM∈arg⁡max⁡πfM(π)\pi_M \in \arg\max_\pi f^M(\pi)πM​∈argmaxπ​fM(π). The class FM={fM:M∈M}\mathcal{F}_{\mathcal{M}} = \{f^M : M \in \mathcal{M}\}FM​={fM:M∈M} collects the mean-reward functions of a model class M\mathcal{M}M. A reference model M‾\overline{M}M, not necessarily in M\mathcal{M}M, has mean reward fM‾:Π→[0,1]f^{\overline{M}} : \Pi \to [0, 1]fM:Π→[0,1].

A prior μ\muμ is a finitely supported distribution μ=∑iwiδMi\mu = \sum_i w_i \delta_{M_i}μ=∑i​wi​δMi​​ over M\mathcal{M}M. The posterior sampling strategy plays ρμ(π):=μ({πM=π})\rho_\mu(\pi) := \mu(\{\pi_M = \pi\})ρμ​(π):=μ({πM​=π}), the law of πM\pi_MπM​ when M∼μM \sim \muM∼μ.

The prior-dependent square-loss DEC (display (70) of the paper) is

dec‾γSq(μ,M‾)=inf⁡p∈Δ(Π)EM∼μ Eπ∼p[fM(πM)−fM(π)−γ(fM(π)−fM‾(π))2],\underline{\mathrm{dec}}^{\mathrm{Sq}}_\gamma(\mu, \overline{M}) = \inf_{p \in \Delta(\Pi)} \mathbb{E}_{M\sim\mu}\, \mathbb{E}_{\pi\sim p}\Bigl[ f^M(\pi_M) - f^M(\pi) - \gamma \bigl(f^M(\pi) - f^{\overline{M}}(\pi)\bigr)^2 \Bigr],dec​γSq​(μ,M)=p∈Δ(Π)inf​EM∼μ​Eπ∼p​[fM(πM​)−fM(π)−γ(fM(π)−fM(π))2],

the best trade-off between regret and squared estimation error that a decision distribution ppp can achieve against a model drawn from the prior.

For a class F\mathcal{F}F of functions Π→R\Pi \to \mathbb{R}Π→R and a distribution ρ\rhoρ on Π\PiΠ, the disagreement coefficient (Definition 6.3) is

θ(F,Δ0,ε0;ρ)=sup⁡Δ≥Δ0, ε≥ε0{Δ2ε2 Pπ∼ρ(∃f∈F:∣f(π)∣>Δ, Eπ′∼ρ[f2(π′)]≤ε2)}∨1,\theta(\mathcal{F}, \Delta_0, \varepsilon_0; \rho) = \sup_{\Delta \ge \Delta_0,\ \varepsilon \ge \varepsilon_0} \left\{ \frac{\Delta^2}{\varepsilon^2}\, \mathbb{P}_{\pi\sim\rho}\bigl( \exists f \in \mathcal{F} : |f(\pi)| > \Delta,\ \mathbb{E}_{\pi'\sim\rho}[f^2(\pi')] \le \varepsilon^2 \bigr) \right\} \vee 1,θ(F,Δ0​,ε0​;ρ)=Δ≥Δ0​, ε≥ε0​sup​{ε2Δ2​Pπ∼ρ​(∃f∈F:∣f(π)∣>Δ, Eπ′∼ρ​[f2(π′)]≤ε2)}∨1,

where π′∼ρ\pi' \sim \rhoπ′∼ρ is drawn independently of π\piπ. It measures how much mass ρ\rhoρ puts on decisions where some function of the class is large although it is small in L2(ρ)L_2(\rho)L2​(ρ). The shifted class FM−fM\mathcal{F}_{\mathcal{M}} - f^MFM​−fM consists of the differences f−fMf - f^Mf−fM, f∈FMf \in \mathcal{F}_{\mathcal{M}}f∈FM​.

Formalization targets

Goal: Theorem 6.2

For every γ≥e\gamma \ge eγ≥e, posterior sampling certifies

dec‾γSq(μ,M‾)≤inf⁡Δ>0{2Δ+24 sup⁡M∈Mθ(FM−fM,Δ,γ−1;ρμ)log⁡2(γ)γ}.\underline{\mathrm{dec}}^{\mathrm{Sq}}_\gamma(\mu, \overline{M}) \le \inf_{\Delta > 0} \left\{ 2\Delta + 24\, \frac{\sup_{M\in\mathcal{M}} \theta(\mathcal{F}_{\mathcal{M}} - f^M, \Delta, \gamma^{-1}; \rho_\mu) \log^2(\gamma)}{\gamma} \right\}.dec​γSq​(μ,M)≤Δ>0inf​{2Δ+24γsupM∈M​θ(FM​−fM,Δ,γ−1;ρμ​)log2(γ)​}.

In the formal statement, the objective of (70) evaluated at p=ρμp = \rho_\mup=ρμ​ is bounded by 2Δ+24Θlog⁡2(γ)/γ2\Delta + 24\Theta \log^2(\gamma)/\gamma2Δ+24Θlog2(γ)/γ for every Δ>0\Delta > 0Δ>0 and every upper bound Θ\ThetaΘ of the disagreement coefficients of the shifted classes.

Milestones

  1. The disagreement coefficient bounds the probability of disagreement at every pair of scales (proof of Lemma E.2, p. 105).
  2. The unit-scale decoupling bound with free parameters Δ,ε∈(0,1]\Delta, \varepsilon \in (0, 1]Δ,ε∈(0,1], η>0\eta > 0η>0 (proof of Lemma E.2, p. 105).
  3. Lemma E.2, the decoupling lemma: for F⊆(Π→[−R,R])\mathcal{F} \subseteq (\Pi \to [-R, R])F⊆(Π→[−R,R]), {fz}⊆F\{f_z\} \subseteq \mathcal{F}{fz​}⊆F, {πz}⊆Π\{\pi_z\} \subseteq \Pi{πz​}⊆Π indexed by z∼μz \sim \muz∼μ,
Ez∼μ[∣fz(πz)∣]≤inf⁡Δ>0{2Δ+6 θ(F,Δ,γ−1;ρμ)log⁡2(Rγ∨e)γ}+γ Ez,z′∼μ[fz2(πz′)].\mathbb{E}_{z\sim\mu}\bigl[|f_z(\pi_z)|\bigr] \le \inf_{\Delta > 0} \left\{ 2\Delta + 6\, \frac{\theta(\mathcal{F}, \Delta, \gamma^{-1}; \rho_\mu) \log^2(R\gamma \vee e)}{\gamma} \right\} + \gamma\, \mathbb{E}_{z, z'\sim\mu}\bigl[f_z^2(\pi_{z'})\bigr].Ez∼μ​[∣fz​(πz​)∣]≤Δ>0inf​{2Δ+6γθ(F,Δ,γ−1;ρμ​)log2(Rγ∨e)​}+γEz,z′∼μ​[fz2​(πz′​)].
  1. The factor-four comparison of the decoupled squared error against a second prior draw with the squared error against M‾\overline{M}M (proof of Theorem 6.2, p. 106).
  2. Lemma E.2 applied to the shifted class FM−fM~\mathcal{F}_{\mathcal{M}} - f^{\widetilde{M}}FM​−fM at parameter γ/4\gamma/4γ/4, for a fixed M~\widetilde{M}M (proof of Theorem 6.2, p. 106).

Significance

Theorem 6.2 bounds the DEC of any structured bandit class by its disagreement coefficient. With the bounds of the disagreement coefficient by the star number and the eluder dimension (Lemma 6.1 of the paper, from Foster et al. 2020) and the minimax theorem, it yields Theorem 6.1, which recovers the O~(eT)\widetilde{O}(\sqrt{\mathfrak{e} T})O(eT​) regret bound of Russo and Van Roy for posterior sampling as a special case, and gives a O~(s2T)\widetilde{O}(\sqrt{\mathfrak{s}^2 T})O(s2T​) bound in terms of the star number. The decoupling lemma (Lemma E.2) is a general tool: it controls a coupled quantity E[∣fz(πz)∣]\mathbb{E}[|f_z(\pi_z)|]E[∣fz​(πz​)∣] by a decoupled second moment, and applies wherever an algorithm's decision and the function evaluated there are drawn from the same distribution.

The results are proved in the paper. None of them has a machine-checked proof; this mission produces the first formal statement of the disagreement coefficient and of the decoupling lemma, on which formal versions of eluder-dimension and star-number bounds can build.

Difficulty

The obvious argument fails at the decoupling step. The quantity Ez∼μ[∣fz(πz)∣]\mathbb{E}_{z\sim\mu}[|f_z(\pi_z)|]Ez∼μ​[∣fz​(πz​)∣] evaluates each function at its own decision, while the estimation error controls only Ez,z′[fz2(πz′)]\mathbb{E}_{z, z'}[f_z^2(\pi_{z'})]Ez,z′​[fz2​(πz′​)], with independent indices. For a finite set of AAA decisions, Cauchy–Schwarz bridges the two at the price of a factor AAA; for an infinite decision space this gives nothing. The disagreement coefficient replaces the cardinality, but it controls disagreement only at one pair of scales at a time (the size of ∣f(π)∣|f(\pi)|∣f(π)∣ and the L2(ρμ)L_2(\rho_\mu)L2​(ρμ​) norm of fff), while the coupled quantity involves all scales at once; obtaining only a log⁡2(γ)\log^2(\gamma)log2(γ) loss from this is the technical core of Lemma E.2.

Formalization scope

  • Decisions form an arbitrary type Π\PiΠ; no finiteness or measurability is assumed. Priors and decision distributions are finitely supported (the paper's footnote 5 equips the model class with the discrete topology), encoded as finite weighted families, so every expectation is a finite sum and ρμ\rho_\muρμ​ is the family of atoms πMi\pi_{M_i}πMi​​ with weights wiw_iwi​.
  • Models enter only through their mean-reward function and their optimal decision, which is all the square-loss DEC depends on; πM\pi_MπM​ is a datum with the hypothesis fM(π)≤fM(πM)f^M(\pi) \le f^M(\pi_M)fM(π)≤fM(πM​) for all π\piπ.
  • The disagreement coefficient uses the real supremum. All statements assume Δ>0\Delta > 0Δ>0, γ−1>0\gamma^{-1} > 0γ−1>0 and a uniformly bounded class, under which the supremum is finite; the inner expectation of (71) is over an independent copy of the decision.
  • Infima and suprema of the page are stated as universally quantified bounds ("for every Δ>0\Delta > 0Δ>0", "for every upper bound Θ\ThetaΘ"), which are equivalent and immune to Lean's default values.
  • Lemma E.2 assumes R≥1R \ge 1R≥1: the page states it for every RRR, but its proof treats R=1R = 1R=1 and extends to R≥1R \ge 1R≥1 by rescaling; Theorem 6.2 needs only R=1R = 1R=1.
  • log⁡\loglog is the natural logarithm; e=exp⁡(1)e = \exp(1)e=exp(1).

The goal is not trivialized by a junk value: the disagreement coefficient is a genuine supremum under the stated hypotheses (a sorry-free check accompanies the definitions), γ≥e\gamma \ge eγ≥e rules out negative logarithms, the class FM\mathcal{F}_{\mathcal{M}}FM​ is the whole class rather than the prior's support, and the reference model is not assumed to lie in the class.

Contributions welcome: proofs of the milestones in attack order; a general layer-cake lemma for finite sums; a formal disagreement coefficient bound by the star number or the eluder dimension (Lemma 6.1), which would lead to Theorem 6.1.

Selected references

  • D. J. Foster, S. M. Kakade, J. Qian, A. Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, 2021 (v3, 2023). https://arxiv.org/abs/2112.13487
  • D. J. Foster, A. Rakhlin, D. Simchi-Levi, Y. Xu, Instance-Dependent Complexity of Contextual Bandits and Reinforcement Learning: A Disagreement-Based Perspective, Conference on Learning Theory (cited in the paper as Foster et al. 2020b). https://arxiv.org/abs/2010.03104
  • D. Russo, B. Van Roy, Eluder Dimension and the Sample Complexity of Optimistic Exploration, Advances in Neural Information Processing Systems 26, pp. 2256–2264, 2013. https://papers.nips.cc/paper_files/paper/2013
  • D. Russo, B. Van Roy, Learning to Optimize via Posterior Sampling, Mathematics of Operations Research 39(4):1221–1243, 2014. https://doi.org/10.1287/moor.2014.0650 (arXiv: https://arxiv.org/abs/1301.2609)
8 thms2 active usersReviewed
Bandit AlgorithmsReinforcement Learning·Captain: mikedeng1

The Statistical Complexity of Interactive Decision Making VIII: Linear Bandits on the Unit Ball Have Localized DEC at Least d/(12γ)Research Paper

Why a lower bound on the DEC for linear bandits

In a linear bandit a learner repeatedly chooses a decision π\piπ from a set Π⊆Rd\Pi \subseteq \mathbb{R}^dΠ⊆Rd and observes a noisy reward whose mean is ⟨θ⋆,π⟩\langle \theta^\star, \pi\rangle⟨θ⋆,π⟩ for an unknown parameter θ⋆\theta^\starθ⋆. It is the standard model for sequential decisions with many correlated actions, such as ad placement, routing and recommendation, and its minimax regret over TTT rounds is known to scale as dTd\sqrt{T}dT​ up to logarithmic factors (Dani, Hayes and Kakade, COLT 2008; see Lattimore and Szepesvári, 2020, Ch. 19–24).

Foster, Kakade, Qian and Rakhlin (2021) introduce the Decision-Estimation Coefficient (DEC), a single complexity measure that yields both upper and lower bounds on the regret of every interactive decision problem. A complexity measure of this kind is only useful if it is tight on the classical examples. Proposition 6.2 of that paper checks this for linear bandits on the Euclidean unit ball: the localized DEC is at least of order d/γd/\gammad/γ, which, through the paper's general lower bound (Theorem 3.2), gives the regret lower bound Ω(dT)\Omega(\sqrt{dT})Ω(dT​). Proposition 6.8 gives the analogous statement for linear bandits whose models have a fixed gap Δ\DeltaΔ.

Setting

Let Bd={v∈Rd:∥v∥2≤1}\mathbb{B}^d = \{v \in \mathbb{R}^d : \|v\|_2 \le 1\}Bd={v∈Rd:∥v∥2​≤1} be the closed Euclidean unit ball. It is both the decision space Π\PiΠ and the parameter set Θ\ThetaΘ. A model with parameter θ∈Θ\theta \in \Thetaθ∈Θ returns, at decision π\piπ, an outcome in {−1,+1}\{-1, +1\}{−1,+1} drawn from the Rademacher distribution Rad(⟨θ,π⟩)\mathrm{Rad}(\langle\theta,\pi\rangle)Rad(⟨θ,π⟩), which puts mass (1+μ)/2(1+\mu)/2(1+μ)/2 on +1+1+1 and (1−μ)/2(1-\mu)/2(1−μ)/2 on −1-1−1 when its mean is μ\muμ. The mean reward is fθ(π)=⟨θ,π⟩f^\theta(\pi) = \langle \theta, \pi\ranglefθ(π)=⟨θ,π⟩.

A maximizer selector θ↦πθ\theta \mapsto \pi_\thetaθ↦πθ​ picks a decision with ⟨θ,π⟩≤⟨θ,πθ⟩\langle\theta,\pi\rangle \le \langle\theta,\pi_\theta\rangle⟨θ,π⟩≤⟨θ,πθ​⟩ for all π\piπ. The regret of a decision is gθ(π)=fθ(πθ)−fθ(π)g^\theta(\pi) = f^\theta(\pi_\theta) - f^\theta(\pi)gθ(π)=fθ(πθ​)−fθ(π). Two outcome laws are compared by the squared Hellinger distance DH2(P,Q)=∑y(P(y)−Q(y))2D^2_{\mathrm{H}}(P,Q) = \sum_y (\sqrt{P(y)}-\sqrt{Q(y)})^2DH2​(P,Q)=∑y​(P(y)​−Q(y)​)2.

Given a reference model M‾\overline{M}M, the L∞L_\inftyL∞​-localized class at radius ε\varepsilonε is

Mε∞(M‾)={M∈M:∣gM(π)−gM‾(π)∣≤ε  ∀π∈Π},\mathcal{M}^\infty_\varepsilon(\overline{M}) = \{ M \in \mathcal{M} : |g^M(\pi) - g^{\overline{M}}(\pi)| \le \varepsilon \ \ \forall \pi \in \Pi \},Mε∞​(M)={M∈M:∣gM(π)−gM(π)∣≤ε  ∀π∈Π},

and the DEC at scale γ>0\gamma > 0γ>0 is

decγ(M′,M‾)=inf⁡p∈Δ(Π)sup⁡M∈M′Eπ∼p[gM(π)−γ DH2(M(π),M‾(π))],\mathsf{dec}_\gamma(\mathcal{M}', \overline{M}) = \inf_{p \in \Delta(\Pi)} \sup_{M \in \mathcal{M}'} \mathbb{E}_{\pi\sim p}\bigl[ g^M(\pi) - \gamma\, D^2_{\mathrm{H}}(M(\pi), \overline{M}(\pi)) \bigr],decγ​(M′,M)=p∈Δ(Π)inf​M∈M′sup​Eπ∼p​[gM(π)−γDH2​(M(π),M(π))],

the infimum running over all Borel probability measures ppp on Π\PiΠ.

Formalization targets

Goal: Proposition 6.2

For all d≥4d \ge 4d≥4 and γ≥2d/3\gamma \ge 2d/3γ≥2d/3 there is M‾∈M\overline{M} \in \mathcal{M}M∈M with

decγ(Mεγ∞(M‾),M‾)≥d12γ,εγ=2d3γ.\mathsf{dec}_\gamma\bigl(\mathcal{M}^\infty_{\varepsilon_\gamma}(\overline{M}), \overline{M}\bigr) \ge \frac{d}{12\gamma}, \qquad \varepsilon_\gamma = \frac{2d}{3\gamma}.decγ​(Mεγ​∞​(M),M)≥12γd​,εγ​=3γ2d​.

The page prints εγ=d/(3γ)\varepsilon_\gamma = d/(3\gamma)εγ​=d/(3γ). The hard family of its proof has regret functions Δ(1−πi)\Delta(1-\pi_i)Δ(1−πi​), which reach 2Δ2\Delta2Δ at π=−ei\pi = -e_iπ=−ei​ while the reference regret is identically 000, so the family lies in the class of radius 2Δ2\Delta2Δ and not in the class of radius Δ\DeltaΔ. The goal is stated at the radius the proof supports.

Milestones

  1. Lemma A.8: DH2(Rad(μ),Rad(0))≤34μ2D^2_{\mathrm{H}}(\mathrm{Rad}(\mu), \mathrm{Rad}(0)) \le \tfrac34\mu^2DH2​(Rad(μ),Rad(0))≤43​μ2 on [−1,1][-1,1][−1,1], together with the two general bounds of the lemma.
  2. Lemma 5.1: an (α,β,δ)(\alpha,\beta,\delta)(α,β,δ)-family forces decγ≥α/2−γ(β/N+δ)\mathsf{dec}_\gamma \ge \alpha/2 - \gamma(\beta/N + \delta)decγ​≥α/2−γ(β/N+δ), stated for a general measurable decision space.
  3. The hard family θi=Δei\theta_i = \Delta e_iθi​=Δei​: regret Δ(1−πi)\Delta(1-\pi_i)Δ(1−πi​) and DH2(Mi(π),Rad(0))≤34Δ2πi2D^2_{\mathrm{H}}(M_i(\pi), \mathrm{Rad}(0)) \le \tfrac34\Delta^2\pi_i^2DH2​(Mi​(π),Rad(0))≤43​Δ2πi2​. With the coordinate bounds ∑iπi≤d≤d/2\sum_i \pi_i \le \sqrt d \le d/2∑i​πi​≤d​≤d/2 (d≥4d \ge 4d≥4) and ∑iπi2≤1\sum_i \pi_i^2 \le 1∑i​πi2​≤1, stated as a separate item, this makes the family a (Δ,34Δ2,0)(\Delta, \tfrac34\Delta^2, 0)(Δ,43​Δ2,0)-family.
  4. Localization: the family lies in M2Δ∞(Rad(0))\mathcal{M}^\infty_{2\Delta}(\mathrm{Rad}(0))M2Δ∞​(Rad(0)).

Companion: Proposition 6.8

For Δ∈(0,1/4)\Delta \in (0, 1/4)Δ∈(0,1/4) and d≥2d \ge 2d≥2, an explicit class of linear bandit models on {e1,…,ed}\{e_1,\dots,e_d\}{e1​,…,ed​}, each with gap Δ\DeltaΔ, satisfies decγ(MΔ∞(M‾),M‾)≥(Δ/4) I{γ≤d/(12Δ)}\mathsf{dec}_\gamma(\mathcal{M}^\infty_\Delta(\overline{M}),\overline{M}) \ge (\Delta/4)\,\mathbb{I}\{\gamma \le d/(12\Delta)\}decγ​(MΔ∞​(M),M)≥(Δ/4)I{γ≤d/(12Δ)} for all γ>0\gamma > 0γ>0.

Significance

The bound shows that the DEC of linear bandits under Euclidean geometry is of order d/γd/\gammad/γ, which matches, up to constants, the upper bound decγSq≤d/(4γ)\mathsf{dec}^{\mathrm{Sq}}_\gamma \le d/(4\gamma)decγSq​≤d/(4γ) on the square-loss variant of the DEC certified by the strategy of Proposition 6.1. Through Theorem 3.2 it gives the regret lower bound Ω(dT)\Omega(\sqrt{dT})Ω(dT​) for every algorithm. Proposition 6.8 shows that the DEC gives meaningful lower bounds even for classes of models with a fixed gap.

All results are proved in the paper. None has a machine-checked proof. The mission produces a formal statement of the linear-bandit lower bound with its radius corrected, the hard-family lemma (Lemma 5.1) in a form that applies to infinite decision spaces and to range-free uiu_iui​, and the Rademacher Hellinger estimates that later DEC lower bounds reuse.

Difficulty

The arithmetic of the construction is short. The work is in the infrastructure.

  • The infimum runs over all Borel probability measures on the ball, not over finitely supported ones. Lemma 5.1 must therefore be proved for Bochner integrals, with the integrability of the payoff established from its continuity on the compact ball.
  • The bound ∑iπi≤d≤d/2\sum_i \pi_i \le \sqrt d \le d/2∑i​πi​≤d​≤d/2 is Cauchy–Schwarz on Rd\mathbb{R}^dRd and needs d≥4d \ge 4d≥4.
  • The value ⟨θi,πθi⟩=Δ\langle\theta_i, \pi_{\theta_i}\rangle = \Delta⟨θi​,πθi​​⟩=Δ must come from the selector hypothesis plus Cauchy–Schwarz, because the selector is not given explicitly.
  • Lemma 5.1 cannot be applied in the form printed in Definition 5.1: there uiu_iui​ takes values in [0,1][0,1][0,1], but the family's ui(π)=πiu_i(\pi) = \pi_iui​(π)=πi​ can be negative. The lemma is needed without the range conditions.

Formalization scope

  • Outcomes. They are finite: the ±1\pm1±1 outcome is encoded on Bool, and models are the Rademacher laws Rad(⟨θ,π⟩)\mathrm{Rad}(\langle\theta,\pi\rangle)Rad(⟨θ,π⟩), identified with their parameters θ∈Bd\theta \in \mathbb{B}^dθ∈Bd. The paper's class contains every reward law on [−1,1][-1,1][−1,1] with linear mean. The Rademacher subclass has a smaller localized class, so a lower bound for it implies the paper's.
  • Decisions. For the goal they range over the ball itself (an infinite space with its Borel σ\sigmaσ-algebra), and decision distributions are arbitrary ProbabilityMeasures on it.
  • The DEC is unfolded. For every ppp some model of the localized class attains the bound. This is what the proof shows, and it implies the inf–sup inequality.
  • Selectors. πM\pi_MπM​ is a selector with a maximizer hypothesis, so statements hold for every choice of maximizer.
  • Added hypotheses. Lemma 5.1 carries α,β≥0\alpha, \beta \ge 0α,β≥0 and integrability of its four functions. Proposition 6.8 carries d≥2d \ge 2d≥2.
  • No trivializing reading. The integrals in the goal are genuine: the integrand is bounded and continuous, and a sanity file proves integrability, so the Bochner convention ∫f=0\int f = 0∫f=0 for non-integrable fff never applies. The goal's quantifier over ppp is not restricted. The radius 2d/(3γ)2d/(3\gamma)2d/(3γ) is the smallest the construction allows.
  • Definitions and reuse. The mission defines the unit ball, Rad\mathrm{Rad}Rad, the linear model, the regret gθg^\thetagθ, and the localized class over a measurable decision space. It references the published hellingerSq, fM and decGf. The general Lemma 5.1 and Lemma A.8 are reusable for every DEC lower bound in §§5–7.

Contributions welcome: proofs of the milestones, a version of the goal through an explicit inf–sup definition of the DEC over measures, and the general model class with arbitrary reward laws on [−1,1][-1,1][−1,1].

Selected references

  • D. J. Foster, S. M. Kakade, J. Qian, A. Rakhlin, The Statistical Complexity of Interactive Decision Making, arXiv preprint, v3, 2023. https://arxiv.org/abs/2112.13487
  • T. Lattimore, Cs. Szepesvári, Bandit Algorithms, Cambridge University Press, 2020 (linear bandits, Ch. 19–24). https://doi.org/10.1017/9781108571401
  • D. J. Foster, C. Gentile, M. Mohri, J. Zimmert, Adapting to Misspecification in Contextual Bandits, NeurIPS 2020 (Proposition 6.1's strategy). https://arxiv.org/abs/2107.05745
11 thms2 active usersReviewed
Dynamical SystemsProbability·Captain: mikedeng1

Approximation Bounds for Random Neural Networks and Reservoir Systems III: Echo State Networks with Output Feedback and Random Weights Approximate Contracting Reservoirs (Theorem 4)Research Paper

Motivation

An echo state network with output feedback is a recurrent model whose previous output is fed back into its next hidden-state update. In reservoir computing only the linear output weights are trained, while the internal weights are fixed, typically after being drawn at random. This raises a quantitative question: if the internal rows are sampled from a fixed law, can a network with NNN blocks of random rows and a suitable linear readout approximate a given input–output rule, and at what rate in NNN? Gonon, Grigoryeva and Ortega answer this for targets generated by contracting reservoir systems. Their Theorem 4 (p. 63) gives, with explicit constants, a probability bound and an excess-risk bound of order 1/N1/\sqrt N1/N​, plus a term decaying geometrically in the length TTT of the observation window. The input set is not required to be bounded: the bound depends on the input process only through E[max⁡t∥Zt∥]\mathbb E[\max_{t}\|Z_t\|]E[maxt​∥Zt​∥] over the window.

Setting

Inputs are vectors in a set Dd⊆RdD_d\subseteq\mathbb R^dDd​⊆Rd and an input history is z=(zt)t∈Z−z=(z_t)_{t\in\mathbb Z_-}z=(zt​)t∈Z−​​, with Z−={0,−1,−2,… }\mathbb Z_-=\{0,-1,-2,\dots\}Z−​={0,−1,−2,…}. A target reservoir is a map F∗:RN∗×Dd→BMF^*:\mathbb R^{N^*}\times D_d\to B_MF∗:RN∗×Dd​→BM​, with BMB_MBM​ the closed Euclidean ball of radius MMM, such that F∗(⋅,v)F^*(\cdot,v)F∗(⋅,v) is an rrr-contraction for every v∈Ddv\in D_dv∈Dd​, with r∈(0,1)r\in(0,1)r∈(0,1). Its reservoir functional is H∗(z)=x0∗H^*(z)=x^*_0H∗(z)=x0∗​, where xt∗=F∗(xt−1∗,zt)x^*_t=F^*(x^*_{t-1},z_t)xt∗​=F∗(xt−1∗​,zt​) for all t∈Z−t\in\mathbb Z_-t∈Z−​. The output dimension is m=N∗m=N^*m=N∗.

The approximating network has NNN blocks of N∗N^*N∗ rows. Each row is a vector ω=(a,c,ζ)∈RN∗+d+1\omega=(a,c,\zeta)\in\mathbb R^{N^*+d+1}ω=(a,c,ζ)∈RN∗+d+1, and the Nˉ=NN∗\bar N=NN^*Nˉ=NN∗ rows Uj(i)U_j^{(i)}Uj(i)​ of [A,C,ζ][A,C,\zeta][A,C,ζ] are i.i.d. with law π\piπ. With an activation σ\sigmaσ and a readout matrix W∈Mm,NˉW\in\mathbb M_{m,\bar N}W∈Mm,Nˉ​, the system is

xt=σ(Ayt−1+Czt+ζ),yt=Wxt,t=−T,…,0,y−T−1=Ξ,x_t=\sigma(Ay_{t-1}+Cz_t+\zeta),\qquad y_t=Wx_t,\qquad t=-T,\dots,0,\qquad y_{-T-1}=\Xi,xt​=σ(Ayt−1​+Czt​+ζ),yt​=Wxt​,t=−T,…,0,y−T−1​=Ξ,

with T≥1T\ge1T≥1 and ∥Ξ∥≤M\|\Xi\|\le M∥Ξ∥≤M; its functional is HWA,C,ζ(z)=y0H^{A,C,\zeta}_W(z)=y_0HWA,C,ζ​(z)=y0​. For each k≥1k\ge1k≥1 and j=1,…,N∗j=1,\dots,N^*j=1,…,N∗ a signed measure πjk≪π\pi_j^k\ll\piπjk​≪π defines the integral map

Fj∗,N(x,v)=∫σ((x,v,1)⋅ω) πjN(dω).F_j^{*,N}(x,v)=\int\sigma((x,v,1)\cdot\omega)\,\pi_j^N(d\omega).Fj∗,N​(x,v)=∫σ((x,v,1)⋅ω)πjN​(dω).

The constant C∗=43Lσsup⁡k∑j(∫∥ω∥2(dπjk/dπ)2dπ)1/2C_*=4\sqrt3L_\sigma\sup_{k}\sum_j(\int\|\omega\|^2(d\pi_j^k/d\pi)^2d\pi)^{1/2}C∗​=43​Lσ​supk​∑j​(∫∥ω∥2(dπjk​/dπ)2dπ)1/2 measures the size of these densities, and sN=E[max⁡t∈{0,…,−T}sup⁡x∈BM+1∥F∗,N(x,Zt)−F∗(x,Zt)∥]s_N=\mathbb E[\max_{t\in\{0,\dots,-T\}}\sup_{x\in B_{M+1}}\|F^{*,N}(x,Z_t)-F^*(x,Z_t)\|]sN​=E[maxt∈{0,…,−T}​supx∈BM+1​​∥F∗,N(x,Zt​)−F∗(x,Zt​)∥] is the approximation error of F∗,NF^{*,N}F∗,N. The input process ZZZ is independent of (A,C,ζ)(A,C,\zeta)(A,C,ζ), Y0Y_0Y0​ is an integrable target, and the loss L≥0L\ge0L≥0 is Lipschitz with constant LLL_LLL​. The echo state event ΩESP\Omega_{\mathrm{ESP}}ΩESP​ of (87) is the event that ∥WFA,C,ζ(x,Zt)∥≤M+1\|WF^{A,C,\zeta}(x,Z_t)\|\le M+1∥WFA,C,ζ(x,Zt​)∥≤M+1 for all x∈BM+1x\in B_{M+1}x∈BM+1​ and t∈{0,…,−T}t\in\{0,\dots,-T\}t∈{0,…,−T}, where FA,C,ζ(x,v)=σ(Ax+Cv+ζ)F^{A,C,\zeta}(x,v)=\sigma(Ax+Cv+\zeta)FA,C,ζ(x,v)=σ(Ax+Cv+ζ).

Formalization targets

Steps of the proof

The milestones are the displayed estimates of the proof, in attack order:

  1. (89): one step of the error recursion on ΩESP\Omega_{\mathrm{ESP}}ΩESP​, E[∥Yt−Xt∗∥1ΩESP]≤E[sup⁡y∥⋯∥]+sN+r E[∥Yt−1−Xt−1∗∥1ΩESP]\mathbb E[\|Y_t-X^*_t\|\mathbb 1_{\Omega_{\mathrm{ESP}}}]\le\mathbb E[\sup_{y}\|\cdots\|]+s_N+r\,\mathbb E[\|Y_{t-1}-X^*_{t-1}\|\mathbb 1_{\Omega_{\mathrm{ESP}}}]E[∥Yt​−Xt∗​∥1ΩESP​​]≤E[supy​∥⋯∥]+sN​+rE[∥Yt−1​−Xt−1∗​∥1ΩESP​​];
  2. (90): splitting into coordinates and symmetrization of the uniform error of the random map;
  3. (91): a deterministic bound on Rademacher averages,
Eεsup⁡y∈BM+1∣∑i=1Nviεiσ((y,z,1)⋅ui)∣≤2Lσ((M+1)(∑ivi2∥ai∥2)1/2+∥z∥(∑ivi2∥ci∥2)1/2+(∑ivi2ζi2)1/2);\mathbb E_\varepsilon\sup_{y\in B_{M+1}}\Big|\sum_{i=1}^N v_i\varepsilon_i\sigma((y,z,1)\cdot u_i)\Big|\le2L_\sigma\Big((M+1)\big(\textstyle\sum_iv_i^2\|a_i\|^2\big)^{1/2}+\|z\|\big(\sum_iv_i^2\|c_i\|^2\big)^{1/2}+\big(\sum_iv_i^2\zeta_i^2\big)^{1/2}\Big);Eε​y∈BM+1​sup​​i=1∑N​vi​εi​σ((y,z,1)⋅ui​)​≤2Lσ​((M+1)(∑i​vi2​∥ai​∥2)1/2+∥z∥(∑i​vi2​∥ci​∥2)1/2+(∑i​vi2​ζi2​)1/2);
  1. (92)–(93): for each fixed input z∈Ddz\in D_dz∈Dd​,
Esup⁡y∈BM+1∥1N∑i=1NWiσ(A(i)y+C(i)z+ζ(i))−F∗,N(y,z)∥≤(M+2+∥z∥)C∗N;\mathbb E\sup_{y\in B_{M+1}}\Big\|\frac1N\sum_{i=1}^NW_i\sigma(A^{(i)}y+C^{(i)}z+\zeta^{(i)})-F^{*,N}(y,z)\Big\|\le\frac{(M+2+\|z\|)C_*}{\sqrt N};Ey∈BM+1​sup​​N1​i=1∑N​Wi​σ(A(i)y+C(i)z+ζ(i))−F∗,N(y,z)​≤N​(M+2+∥z∥)C∗​​;
  1. (96): P(Ω∖ΩESP)≤2(M+2+E[max⁡t∥Zt∥])C∗/N+2sN\mathbb P(\Omega\setminus\Omega_{\mathrm{ESP}})\le2(M+2+\mathbb E[\max_t\|Z_t\|])C_*/\sqrt N+2s_NP(Ω∖ΩESP​)≤2(M+2+E[maxt​∥Zt​∥])C∗​/N​+2sN​.

Goal: Theorem 4

There is a readout WWW, a function of (A,C,ζ)(A,C,\zeta)(A,C,ζ) alone, such that for every δ∈(0,1)\delta\in(0,1)δ∈(0,1) and every ∥Ξ∥≤M\|\Xi\|\le M∥Ξ∥≤M, with probability at least max⁡(1−δ−2C∗(M+2+E[max⁡t∥Zt∥])/N−2sN,0)\max(1-\delta-2C_*(M+2+\mathbb E[\max_t\|Z_t\|])/\sqrt N-2s_N,0)max(1−δ−2C∗​(M+2+E[maxt​∥Zt​∥])/N​−2sN​,0), the event ΩESP\Omega_{\mathrm{ESP}}ΩESP​ holds and

∣R(HWA,C,ζ)−R(H∗)∣≤LLδ[(M+2+max⁡tE∥Zt∥)C∗(1−r)N+sN1−r+2(M+1)rT+1],|\mathcal R(H^{A,C,\zeta}_W)-\mathcal R(H^*)|\le \frac{L_L}{\delta}\Big[\frac{(M+2+\max_t\mathbb E\|Z_t\|)C_*}{(1-r)\sqrt N}+\frac{s_N}{1-r}+2(M+1)r^{T+1}\Big],∣R(HWA,C,ζ​)−R(H∗)∣≤δLL​​[(1−r)N​(M+2+maxt​E∥Zt​∥)C∗​​+1−rsN​​+2(M+1)rT+1],

where R(H)=E[L(H(Z),Y0)]\mathcal R(H)=\mathbb E[L(H(Z),Y_0)]R(H)=E[L(H(Z),Y0​)] is evaluated at fixed network weights.

Significance

The bound separates three sources of error: sampling NNN blocks of random rows (rate 1/N1/\sqrt N1/N​), the approximation of F∗F^*F∗ by the integral map F∗,NF^{*,N}F∗,N (sNs_NsN​), and the finite initialization of the feedback recursion (rT+1r^{T+1}rT+1). Only the readout depends on the target; the internal weights are generated without it. The theorem also bounds the probability that the random feedback map leaves the state ball on which the recursion is controlled, which is the form in which the echo state property enters. Via Theorem 3 (p. 62), which specializes the densities through the paper's Proposition 2, it gives approximation rates for ReLU output-feedback networks.

The paper proves the result on pp. 63–68. No part of it is formalized. A complete development would give machine-checked Rademacher symmetrization and contraction estimates for random features, uniform-over-a-ball bounds for sampled integral maps, and a treatment of finite feedback recursions on product probability spaces, all reusable outside reservoir computing.

Difficulty

A pointwise approximation of F∗F^*F∗ by the sampled map does not control the feedback system, because the next feedback value is itself the random output of the previous update. The comparison must hold uniformly over the state ball BM+1B_{M+1}BM+1​ and over the whole input window, and the event that the recursion leaves the ball must be charged to an explicit probability. The uniform bound requires symmetrization and the Ledoux–Talagrand contraction principle for a supremum over a continuum of states, with per-index contractions; Lipschitz continuity of σ\sigmaσ alone does not supply the vanishing at zero that principle needs. The input is unbounded, so the input dependence must be tracked through ∥z∥\|z\|∥z∥ and integrated at the end.

Formalization scope

States, inputs, outputs and rows are Euclidean spaces (EuclideanSpace); a row is indexed by ({1..N∗}⊔{1..d})⊔{∗}(\{1..N^*\}\sqcup\{1..d\})\sqcup\{\ast\}({1..N∗}⊔{1..d})⊔{∗} and the rows by {1..N}×{1..N∗}\{1..N\}\times\{1..N^*\}{1..N}×{1..N∗}. Time t∈Z−t\in\mathbb Z_-t∈Z−​ is the index −t∈N-t\in\mathbb N−t∈N; the equation (78) is the published ReservoirESN.IsSolution. The rows have law π⊗NN∗\pi^{\otimes NN^*}π⊗NN∗ on their own space and (Z,Y0)(Z,Y_0)(Z,Y0​) lives on a separate probability space; the probability is under the product law, which encodes the independence of §6.1. Signed measures πjk\pi_j^kπjk​ are their Radon–Nikodym densities with respect to π\piπ, with integrability and first-moment hypotheses. C∗C_*C∗​, sNs_NsN​ and E[max⁡t∥Zt∥]\mathbb E[\max_t\|Z_t\|]E[maxt​∥Zt​∥] are extended nonnegative integrals and suprema, converted to reals only under their finiteness hypotheses; the risk integrals are finite under the loss Lipschitz bound, integrable target and input-moment assumption; the supremum in (87) is unfolded. F∗F^*F∗ is assumed jointly measurable, the standing convention that makes sNs_NsN​ and H∗(Z)H^*(Z)H∗(Z) random variables; the standing assumptions d,N∗≥1d,N^*\ge1d,N∗≥1 and M>0M>0M>0 (p. 30) and T≥1T\ge1T≥1 (§6.1) are hypotheses.

Trivialization ruled out. The readout is a measurable function of the rows only and is chosen before δ\deltaδ and Ξ\XiΞ. A readout allowed to look at the input ZZZ would make the claim a different and much weaker statement.

Correction of the print. The goal and milestones (91), (93), (96) assume σ(0)=0\sigma(0)=0σ(0)=0: the contraction principle cited at (91) needs contractions vanishing at zero, while the printed theorem states only that σ\sigmaσ is Lipschitz. ReLU, the paper's application, satisfies it. The goal retains the printed full risk difference. Its proof passes through (88), where an indicator depending on ZZZ is moved inside a risk integral; this step needs further justification. The milestone quotations keep the printed text.

Contributions are welcome on each milestone, on general Rademacher symmetrization and contraction lemmas for suprema over compact parameter sets, on measurability of suprema of continuous random fields, and on the printed form of the risk bound.

Selected references

  • L. Gonon, L. Grigoryeva and J.-P. Ortega, Approximation bounds for random neural networks and reservoir systems, The Annals of Applied Probability 33(1), 2023, 28–69. https://doi.org/10.1214/22-AAP1806
  • M. Ledoux and M. Talagrand, Probability in Banach Spaces: Isoperimetry and Processes, Springer, 1991 (Theorem 4.12, the contraction principle). https://doi.org/10.1007/978-3-642-20212-4
8 thms2 active usersReviewed
🏆Completed
Optimization·Captain: mikedeng1

Certifying Some Distributional Robustness with Principled Adversarial Training 3: The Softmax Loss of a Smooth Deep Network Has a (√2·β_L(θ)+α_L(θ)²)-Lipschitz Input GradientResearch Paper

Motivation

Sinha, Namkoong, Volpi and Duchi (arXiv:1710.10571, ICLR 2018) train classifiers against adversarial perturbations by solving a Lagrangian relaxation of a Wasserstein distributionally robust problem. The inner problem, sup⁡z{ℓ(θ;z)−γ c(z,z0)}\sup_z \{\ell(\theta;z)-\gamma\, c(z,z_0)\}supz​{ℓ(θ;z)−γc(z,z0​)}, is strongly concave, and hence solvable by gradient ascent with the paper's certificates, exactly when the penalty γ\gammaγ exceeds the input-smoothness constant LzzL_{zz}Lzz​ of the loss: the Lipschitz constant of z↦∇zℓ(θ;z)z\mapsto\nabla_z\ell(\theta;z)z↦∇z​ℓ(θ;z). For a deep network this constant is not given; §4 of the paper ("Bounds on smoothness of neural networks") supplies an explicit upper bound in terms of the weights' spectral norms and per-layer constants of the activations. The bound tells a practitioner how large γ\gammaγ must be for the robustness guarantees to apply, and it is of the same type as the spectral-norm Lipschitz bounds used for adversarial robustness and generalization of deep networks (Szegedy et al. 2014; Bartlett, Foster, Telgarsky 2017), but at the level of the derivative rather than the function.

Setting

A layered network of depth LLL has widths dl,Id_{l,I}dl,I​ (pre-activation of layer lll) and dl,Od_{l,O}dl,O​ (output of layer lll), with d0,O=pd_{0,O}=pd0,O​=p the input dimension and dL,O=Kd_{L,O}=KdL,O​=K the number of classes. Its parameters are weight matrices θl:Rdl−1,O→Rdl,I\theta_l:\mathbb R^{d_{l-1,O}}\to\mathbb R^{d_{l,I}}θl​:Rdl−1,O​→Rdl,I​, and each linear layer is followed by a nonlinear operation σl:Rdl,I→Rdl,O\sigma_l:\mathbb R^{d_{l,I}}\to\mathbb R^{d_{l,O}}σl​:Rdl,I​→Rdl,O​ (an activation, possibly combined with pooling). The output of the first lll layers is

F0(θ;x)=x,Fl(θ;x)=σl(θl⋅Fl−1(θ;x)),F_0(\theta;x)=x,\qquad F_l(\theta;x)=\sigma_l\big(\theta_l\cdot F_{l-1}(\theta;x)\big),F0​(θ;x)=x,Fl​(θ;x)=σl​(θl​⋅Fl−1​(θ;x)),

and its Jacobian in xxx is JxFl(θ;x)J_xF_l(\theta;x)Jx​Fl​(θ;x). All spaces carry the Euclidean norm ∥⋅∥2\|\cdot\|_2∥⋅∥2​, and ∥⋅∥op\|\cdot\|_{\mathrm{op}}∥⋅∥op​ is the ℓ2\ell_2ℓ2​-operator (spectral) norm.

Assumption E. For l=1,…,Ll=1,\dots,Ll=1,…,L, the map σl\sigma_lσl​ is Ll0L^0_lLl0​-Lipschitz for ∥⋅∥2\|\cdot\|_2∥⋅∥2​, its Jacobian JσlJ\sigma_lJσl​ is Ll1L^1_lLl1​-Lipschitz from ∥⋅∥2\|\cdot\|_2∥⋅∥2​ to ∥⋅∥op\|\cdot\|_{\mathrm{op}}∥⋅∥op​, and Ll0>0L^0_l>0Ll0​>0. From these constants define

αl(θ)=∏j=1lLj0 ∥θj∥op,βl(θ)=αl(θ)∑j=1lLj1(Lj0)2 αj(θ).\alpha_l(\theta)=\prod_{j=1}^{l}L^0_j\,\|\theta_j\|_{\mathrm{op}},\qquad \beta_l(\theta)=\alpha_l(\theta)\sum_{j=1}^{l}\frac{L^1_j}{(L^0_j)^2}\,\alpha_j(\theta).αl​(θ)=j=1∏l​Lj0​∥θj​∥op​,βl​(θ)=αl​(θ)j=1∑l​(Lj0​)2Lj1​​αj​(θ).

The classification loss is the softmax loss ℓ(θ;(x,y))=−log⁡py(θ;x)\ell(\theta;(x,y))=-\log p_y(\theta;x)ℓ(θ;(x,y))=−logpy​(θ;x) with py(θ;x)=exp⁡(FL,y(θ;x))/∑k=1Kexp⁡(FL,k(θ;x))p_y(\theta;x)=\exp(F_{L,y}(\theta;x))/\sum_{k=1}^K\exp(F_{L,k}(\theta;x))py​(θ;x)=exp(FL,y​(θ;x))/∑k=1K​exp(FL,k​(θ;x)), for a class y∈{1,…,K}y\in\{1,\dots,K\}y∈{1,…,K}.

Formalization targets

Goal: Corollary 4

Under Assumption E, for every class yyy and all inputs x,x′x,x'x,x′,

∥∇xℓ(θ;(x,y))−∇xℓ(θ;(x′,y))∥2≤(2 βL(θ)+αL(θ)2) ∥x−x′∥2.\big\|\nabla_x\ell(\theta;(x,y))-\nabla_x\ell(\theta;(x',y))\big\|_2\le\big(\sqrt2\,\beta_L(\theta)+\alpha_L(\theta)^2\big)\,\|x-x'\|_2 .​∇x​ℓ(θ;(x,y))−∇x​ℓ(θ;(x′,y))​2​≤(2​βL​(θ)+αL​(θ)2)∥x−x′∥2​.

Proposition 5

Under Assumption E, for l=1,…,Ll=1,\dots,Ll=1,…,L,

∥Fl(θ;x)−Fl(θ;x′)∥2≤αl(θ)∥x−x′∥2,∥JxFl(θ;x)−JxFl(θ;x′)∥op≤βl(θ)∥x−x′∥2.\|F_l(\theta;x)-F_l(\theta;x')\|_2\le\alpha_l(\theta)\|x-x'\|_2,\qquad \|J_xF_l(\theta;x)-J_xF_l(\theta;x')\|_{\mathrm{op}}\le\beta_l(\theta)\|x-x'\|_2 .∥Fl​(θ;x)−Fl​(θ;x′)∥2​≤αl​(θ)∥x−x′∥2​,∥Jx​Fl​(θ;x)−Jx​Fl​(θ;x′)∥op​≤βl​(θ)∥x−x′∥2​.

Milestones

In the order the paper's proof uses them: Lemma 6 (the chain rule for JxFlJ_xF_lJx​Fl​, recursively and as a product); Proposition 5, first part; the telescoping representation (34) of JxFl(θ;x)−JxFl(θ;x′)J_xF_l(\theta;x)-J_xF_l(\theta;x')Jx​Fl​(θ;x)−Jx​Fl​(θ;x′); Proposition 5, second part; the softmax constants of §B.9 (gradient p(v)−eyp(v)-e_yp(v)−ey​, Lipschitz constant 2\sqrt22​, smoothness constant 111); and the recursion of §B.9 for αl,βl\alpha_l,\beta_lαl​,βl​ with its instance βL+1=2 βL+αL2\beta_{L+1}=\sqrt2\,\beta_L+\alpha_L^2βL+1​=2​βL​+αL2​ for a softmax output layer.

Significance

Corollary 4 converts the abstract hypothesis γ>Lzz\gamma>L_{zz}γ>Lzz​ of the paper's convergence and certification results (missions 1 and 2 of this series) into a checkable condition on a trained network: Lzz≤2 βL(θ)+αL(θ)2L_{zz}\le\sqrt2\,\beta_L(\theta)+\alpha_L(\theta)^2Lzz​≤2​βL​(θ)+αL​(θ)2, computable from spectral norms of the weights and from constants of standard activations (the paper lists average pooling, sigmoid and ELU). Proposition 5 is a reusable statement about any composition of smooth maps with linear layers, independent of the softmax and of robust optimization.

The results are proved in the paper (§B.8–§B.9). No machine-checked proof of them is known to exist. This mission produces a formal development of the input Jacobian of a layered network with varying widths, the telescoping estimate for its variation, and the smoothness constants of the softmax loss, all with the paper's explicit constants.

Difficulty

The function-level bound (first part of Proposition 5) is a direct induction. The derivative-level bound is not: the obvious induction on JxFl+1=Jσl+1⋅θl+1⋅JxFlJ_xF_{l+1}=J\sigma_{l+1}\cdot\theta_{l+1}\cdot J_xF_lJx​Fl+1​=Jσl+1​⋅θl+1​⋅Jx​Fl​ needs both a bound on ∥JxFl∥op\|J_xF_l\|_{\mathrm{op}}∥Jx​Fl​∥op​, which comes from the Lipschitz half and the differentiability of every layer, and a bound on the variation of the Jacobian of the previous layers, and the two must be combined so that the constant comes out as exactly βl(θ)\beta_l(\theta)βl​(θ) of (21), whose definition uses quotients by (Lj0)2(L^0_j)^2(Lj0​)2. For the corollary, the softmax loss must be shown to have Lipschitz constant 2\sqrt22​ (a supremum that is not attained) and a 111-Lipschitz derivative; the paper's eigenvalue argument for the latter is loose and its displayed initial condition for the recursion of the constants is misprinted, so the printed text cannot be transcribed line by line. In Lean, the widths change from layer to layer, so the network is a family of maps between different Euclidean spaces, and every product of Jacobians is a dependently typed composition.

Formalization scope

  • Inputs, pre-activations and outputs are EuclideanSpace ℝ (Fin n); weights are continuous linear maps, so ‖θ l‖ is the spectral norm; Jacobians are Fréchet derivatives with the operator norm. A Frobenius or entrywise norm would state a different theorem and is not used.
  • Layers are indexed from 000 in Lean: Lean layer l is the paper's layer l+1l+1l+1 (θ l, σ l, L0 l, L1 l); F l, α L0 l, β L0 L1 l carry the paper's index lll. Widths vary with the layer; they are not collapsed to a single dimension.
  • Assumption E is stated for the first LLL layers and includes two hypotheses implicit on the page: each σl\sigma_lσl​ is differentiable (the assumption speaks of its Jacobian) and Ll1≥0L^1_l\ge0Ll1​≥0. The paper's Ll0>0L^0_l>0Ll0​>0 is kept.
  • αl\alpha_lαl​ and βl\beta_lβl​ are the explicit product and sum of (21); they are definitions, not free parameters. Consequently α0=1\alpha_0=1α0​=1 and β0=0\beta_0=0β0​=0; the paper's "β0(θ)=α0(θ)=1\beta_0(\theta)=\alpha_0(\theta)=1β0​(θ)=α0​(θ)=1" in §B.9 is a misprint and is not followed. The recursion βl+1=(αl+1/αl)βl+…\beta_{l+1}=(\alpha_{l+1}/\alpha_l)\beta_l+\dotsβl+1​=(αl+1​/αl​)βl​+… is stated in the division-free form βl+1=Ll+10∥θl+1∥opβl+…\beta_{l+1}=L^0_{l+1}\|\theta_{l+1}\|_{\mathrm{op}}\beta_l+\dotsβl+1​=Ll+10​∥θl+1​∥op​βl​+…, and in the paper's form when αl≠0\alpha_l\ne0αl​=0.
  • In (34) the first displayed line (a "product of differences") is a misprint; only the telescoped sum is formalized.
  • The softmax loss is real-valued on RK\mathbb R^KRK with K=dL,OK=d_{L,O}K=dL,O​ and a class yyy in Fin K; the goal uses Mathlib's gradient, whose norm equals the operator norm of the derivative.
  • The goal is not trivialized by free constants: a version with α,β\alpha,\betaα,β as arbitrary parameters, or with the loss not tied to FLF_LFL​, would not be Corollary 4 and is not what is posed.
  • Not formalized: the Hessian formula ∇2σy=diag⁡p−ppT\nabla^2\sigma_y=\operatorname{diag}p-pp^T∇2σy​=diagp−ppT and the eigenvalue argument (only their conclusion L1=1L^1=1L1=1 is stated); attainment of sup⁡x∥p(x)−ey∥2=2\sup_x\|p(x)-e_y\|_2=\sqrt2supx​∥p(x)−ey​∥2​=2​ (the value of the supremum, for K≥2K\ge2K≥2, is stated; it is not attained); Examples 1–3 (constants of pooling, sigmoid and ELU).

A complete development needs the chain rule for the network map, the mean value inequality for the softmax loss, and the algebra of (21). The definitions of the layered network and of the softmax loss are independent of robust optimization and are reusable for other Lipschitz and smoothness bounds of deep networks. Proofs of any milestone, and alternative arguments with sharper constants stated as separate theorems, are welcome.

Selected references

  • A. Sinha, H. Namkoong, R. Volpi, J. Duchi, Certifying Some Distributional Robustness with Principled Adversarial Training, ICLR 2018; arXiv:1710.10571v5, 2020. https://arxiv.org/abs/1710.10571
  • C. Szegedy et al., Intriguing properties of neural networks, ICLR 2014. https://arxiv.org/abs/1312.6199
  • P. Bartlett, D. Foster, M. Telgarsky, Spectrally-normalized margin bounds for neural networks, NeurIPS 2017. https://arxiv.org/abs/1706.08498
9 thms2 active usersReviewed
Computational GeometryLinear algebra·Captain: mikedeng1

Turning Big Data into Tiny Data: Constant-Size Coresets for k-Means, PCA, and Projective Clustering I: The Rank-m SVD Truncation Plus Its Residual Preserves the Cost of Every j-Dimensional ShapeResearch Paper

Motivation

Large data sets make clustering and subspace fitting expensive because a candidate solution must be evaluated against every input point. A dimensionality reduction replaces the data by a lower-rank matrix whose costs approximate those of the original input for an entire class of candidate solutions. Feldman, Schmidt, and Sohler study reductions that retain enough information for kkk-means, principal component analysis, and projective clustering. Their journal paper identifies a rank truncation and one residual scalar that preserve the fitting cost of every shape lying in a low-dimensional linear subspace Feldman, Schmidt, and Sohler (2020), Theorem 22.

The published article also records that some coreset sizes in its 2013 conference version were incorrect. This mission uses the 2020 journal statement and its page numbering throughout. The authors cite earlier SVD-based approximations to kkk-means on p. 613, while their Theorem 22 gives a uniform cost guarantee for arbitrary nonempty shapes contained in a jjj-dimensional subspace Feldman, Schmidt, and Sohler (2020), pp. 603, 613, 620.

Setting

The input is a real matrix A∈Rn×dA\in\mathbb R^{n\times d}A∈Rn×d. Its nnn rows are points in Rd\mathbb R^dRd. For a nonempty set of candidate locations or shapes C⊆RdC\subseteq\mathbb R^dC⊆Rd, let dist⁡(ai,C)\operatorname{dist}(a_i,C)dist(ai​,C) be the infimum of Euclidean distances from row aia_iai​ to points of CCC. The shape cost is

dist⁡2(A,C)=∑i=1ndist⁡(ai,C)2.\operatorname{dist}^2(A,C)=\sum_{i=1}^{n}\operatorname{dist}(a_i,C)^2.dist2(A,C)=i=1∑n​dist(ai​,C)2.

The set CCC need not be closed and need not be a subspace. For example, it can be a finite set of cluster centers. The restriction is that all of its points lie in some linear subspace of dimension jjj. This matches the paper's shape class in §5; the affine case is treated separately Feldman, Schmidt, and Sohler (2020), pp. 619–620.

Fix a singular value decomposition A=UΣVTA=U\Sigma V^TA=UΣVT, where UUU and VVV are orthogonal and the diagonal entries of the rectangular matrix Σ\SigmaΣ are nonnegative and nonincreasing. The entry σi\sigma_iσi​ is indexed from one. The rank-mmm truncation A(m)=UΣ(m)VTA^{(m)}=U\Sigma^{(m)}V^TA(m)=UΣ(m)VT keeps the first mmm diagonal entries and sets the rest to zero. Its squared Frobenius residual is Δ=∥A−A(m)∥F2\Delta=\|A-A^{(m)}\|_F^2Δ=∥A−A(m)∥F2​. The corrected cost of CCC is dist⁡2(A(m),C)+Δ\operatorname{dist}^2(A^{(m)},C)+\Deltadist2(A(m),C)+Δ Feldman, Schmidt, and Sohler (2020), Definition 1 and §2.

Formalization targets

Linear subspace estimates

The milestones first record the matrix Pythagorean identity, the jσm+12j\sigma_{m+1}^2jσm+12​ projection and residual bounds, and the lower bound on the cost of a linear jjj-subspace. They then state the residual-corrected approximation in Corollary 16:

∥AY∥F2≤∥A(m)Y∥F2+Δ≤(1+ε)∥AY∥F2,m≥⌈j/ε⌉+j−1.\|AY\|_F^2\le\|A^{(m)}Y\|_F^2+\Delta\le(1+\varepsilon)\|AY\|_F^2, \qquad m\ge\lceil j/\varepsilon\rceil+j-1.∥AY∥F2​≤∥A(m)Y∥F2​+Δ≤(1+ε)∥AY∥F2​,m≥⌈j/ε⌉+j−1.

Here YYY has d−jd-jd−j orthonormal columns spanning the complement of the candidate subspace. Corollary 20, Corollary 21, and display (9) connect these linear estimates to costs of general sets CCC Feldman, Schmidt, and Sohler (2020), pp. 604, 613–615, 619–620.

Goal: uniform shape-cost preservation

The goal is Theorem 22. For 1≤j≤d−11\le j\le d-11≤j≤d−1, 0<ε≤10<\varepsilon\le10<ε≤1, and ⌈8j/ε2⌉−1≤m≤min⁡{n,d}−1\lceil8j/\varepsilon^2\rceil-1\le m\le\min\{n,d\}-1⌈8j/ε2⌉−1≤m≤min{n,d}−1, every nonempty CCC contained in a linear jjj-dimensional subspace satisfies

∣(dist⁡2(A(m),C)+∥A−A(m)∥F2)−dist⁡2(A,C)∣≤εdist⁡2(A,C).\left|\bigl(\operatorname{dist}^2(A^{(m)},C)+\|A-A^{(m)}\|_F^2\bigr) -\operatorname{dist}^2(A,C)\right| \le\varepsilon\operatorname{dist}^2(A,C).​(dist2(A(m),C)+∥A−A(m)∥F2​)−dist2(A,C)​≤εdist2(A,C).

The quantifier over CCC makes this a guarantee for every admissible candidate on the same reduced data, rather than a statement about one selected optimum Feldman, Schmidt, and Sohler (2020), Theorem 22, p. 620.

Significance

The theorem preserves objective values after replacing AAA by a rank-mmm matrix and one number. For a fixed jjj and accuracy parameter, the rank threshold depends on jjj and ε\varepsilonε, with no direct dependence on the number of points nnn or ambient dimension ddd except the upper bound on feasible mmm. In the paper, the result leads to a guarantee for kkk-means candidates, since kkk centers span a linear subspace of dimension at most kkk Feldman, Schmidt, and Sohler (2020), Corollary 23, p. 621.

The result is proved in the paper, but this mission's Lean theorems are posed with sorry; they do not yet have machine-checked proofs. A completed development would connect a reusable rectangular SVD model, matrix projection identities, infimum distances to arbitrary nonempty sets, and a uniform cost inequality. The SVD definition is already published in the platform library; the row-cost and singular-value conventions here supply the interface needed for this paper Feldman, Schmidt, and Sohler (2020), §§2, 3, 5.

Difficulty

The linear subspace bound compares a projection cost with a tail sum of squared singular values. The paper recalls that fact in §3 without proving it, so the formal development needs a suitable singular-value inequality or a derivation from existing matrix approximation results. For general shapes CCC, a bound on the cost of one linear subspace does not by itself control the distance from each projected point to every point of CCC. The remaining milestones state the projection error, the squared-distance perturbation bound, and the Pythagorean cost split that address that gap Feldman, Schmidt, and Sohler (2020), pp. 613, 619–621.

The proof printed after Theorem 22 invokes earlier corollaries under a stronger rank threshold than the theorem states. The mission keeps the printed theorem threshold, whose claim remains valid; a formal proof must bridge that discrepancy Feldman, Schmidt, and Sohler (2020), pp. 620–621.

Formalization scope

Matrices are Matrix (Fin n) (Fin d) ℝ, and rows are points of EuclideanSpace ℝ (Fin d). Squared Frobenius norm is an explicit finite sum of squared entries. Shape cost sums squared Metric.infDist values over rows. Since Mathlib assigns a real default value to distance from the empty set while the paper assigns infinity, every theorem about a general shape CCC requires C.Nonempty. This is the page's own requirement for Theorem 22 and Corollary 21, and is also needed for display (9).

The SVD and its truncation use the published ProjLikeRetr.FixedRank.IsSVD and truncSVD. Every SVD statement quantifies over an arbitrary valid triple (U,Σ,V)(U,\Sigma,V)(U,Σ,V); tied singular values do not fix a unique truncation. The local σi\sigma_iσi​ is one-based, while truncSVD m keeps zero-based diagonal positions below mmm. Orthonormal columns of XXX satisfy XTX=IX^TX=IXTX=I, their orthogonal complement has d−jd-jd−j columns, and projection in ambient coordinates is AXXTAXX^TAXXT. Theorem 22 quantifies over arbitrary nonempty CCC contained in a linear jjj-subspace; it does not restrict CCC to subspaces or replace linear containment by affine containment.

The integer inequalities in Lean use additive forms such as ⌈8j/ε2⌉≤m+1\lceil8j/\varepsilon^2\rceil\le m+1⌈8j/ε2⌉≤m+1 to avoid truncated natural-number subtraction. The theorem retains the printed mmm threshold, including the proof gap described above. Running-time clauses are outside the mission's mathematical statements. Contributions are welcome toward the singular-value lower bound, the projection and infimum-distance identities, and the final uniform inequality; the matrix and distance definitions are reusable in the companion affine-subspace mission.

Selected references

  • Dan Feldman, Melanie Schmidt, and Christian Sohler, Turning Big Data into Tiny Data: Constant-Size Coresets for k-Means, PCA, and Projective Clustering, SIAM Journal on Computing 49(3):601–657, 2020. DOI: 10.1137/18M1209854.
11 thms2 active usersReviewed
Bandit AlgorithmsOperations ResearchStatistics·Captain: mikedeng1

Personalized Dynamic Pricing with Machine Learning: High-Dimensional Features and Heterogeneous Elasticity 2: Under Unknown Sparsity, Lasso-Based ILQX Pricing Has Regret at Most Cs√T(log d + log T)Research Paper

Motivation

Online sellers of loans, insurance, travel and retail goods observe data about each arriving customer before quoting a price. Using that data to price individually is attractive only if the seller can learn, from its own sales, how demand responds to price for customers with given characteristics. Two features make this hard in practice. The number of recorded characteristics ddd is often large, possibly larger than the selling horizon TTT, while only a few of them matter. And price sensitivity is itself heterogeneous: different customers respond to the same price change differently.

Ban and Keskin (Management Science 67(9), 2021) model both features and ask how fast the revenue lost to learning can grow. Their answer is that a seller who learns with lasso-regularized quasi-likelihood estimation and occasional price experiments loses revenue of order sTs\sqrt TsT​ up to logarithmic factors, where sss is the number of relevant characteristics, and that no policy can do better than order sTs\sqrt TsT​. This mission formalizes the upper bound for the general (nonlinear-link) model with unknown sparsity, Theorem 3 of the paper.

Setting

In each period t=1,2,…,Tt = 1, 2, \dots, Tt=1,2,…,T a customer arrives with a raw feature vector Zt∈RdZ_t \in \mathbb R^dZt​∈Rd. The ZtZ_tZt​ are i.i.d., take values in a compact set Z\mathcal ZZ inside a ball of radius zmax⁡z_{\max}zmax​, have mean zero and positive definite covariance ΣZ\Sigma_ZΣZ​. The seller sees the augmented vector Xt=[1;Zt]∈Rd+1X_t = [1; Z_t] \in \mathbb R^{d+1}Xt​=[1;Zt​]∈Rd+1, charges a price pt∈[ℓ,u]p_t \in [\ell, u]pt​∈[ℓ,u] with 0<ℓ<u0 < \ell < u0<ℓ<u, and observes the demand

Dt=g(α⋅Xt+(β⋅Xt) pt)+εt.D_t = g\big(\alpha \cdot X_t + (\beta \cdot X_t)\, p_t\big) + \varepsilon_t .Dt​=g(α⋅Xt​+(β⋅Xt​)pt​)+εt​.

Here θ=(α,β)\theta = (\alpha, \beta)θ=(α,β) is an unknown parameter in a compact rectangle Θ⊂R2(d+1)\Theta \subset \mathbb R^{2(d+1)}Θ⊂R2(d+1), the link ggg is known, differentiable and increasing with ℓ~≤g′≤u~\tilde\ell \le g' \le \tilde uℓ~≤g′≤u~ on the relevant domain, and the demand shocks εt\varepsilon_tεt​ form a sub-Gaussian martingale difference sequence. The term β⋅Xt\beta \cdot X_tβ⋅Xt​ is the customer's own price sensitivity. With u(p,x)=[1;p]⊗xu(p, x) = [1; p] \otimes xu(p,x)=[1;p]⊗x the argument of ggg is θ⋅u(p,x)\theta \cdot u(p, x)θ⋅u(p,x).

The sparsity of θ\thetaθ is s=∣S∣s = |\mathcal S|s=∣S∣, S={i:αi≠0}∪{i:βi≠0}\mathcal S = \{i : \alpha_i \neq 0\} \cup \{i : \beta_i \neq 0\}S={i:αi​=0}∪{i:βi​=0}. The seller does not know S\mathcal SS.

The expected revenue of price ppp is r(p,θ,x)=p g(θ⋅u(p,x))r(p, \theta, x) = p\, g(\theta \cdot u(p,x))r(p,θ,x)=pg(θ⋅u(p,x)), and the clairvoyant price φ(θ,x)\varphi(\theta, x)φ(θ,x) maximizes it; it is assumed to lie in the interior of [ℓ,u][\ell, u][ℓ,u] for θ∈Θ\theta \in \Thetaθ∈Θ, x∈X={1}×Zx \in \mathcal X = \{1\} \times \mathcal Zx∈X={1}×Z. The regret of a policy over TTT periods is

Δθπ(T)=E[∑t=1Tr∗(θ,Xt)−r(pt,θ,Xt)],r∗(θ,x)=r(φ(θ,x),θ,x).\Delta^\pi_\theta(T) = \mathbb E\Big[\sum_{t=1}^T r^*(\theta, X_t) - r(p_t, \theta, X_t)\Big], \qquad r^*(\theta,x) = r(\varphi(\theta,x),\theta,x).Δθπ​(T)=E[t=1∑T​r∗(θ,Xt​)−r(pt​,θ,Xt​)],r∗(θ,x)=r(φ(θ,x),θ,x).

The policy ILQX(m1,m2,λ)(m_1, m_2, \lambda)(m1​,m2​,λ) charges the experimental price m1m_1m1​ in periods M1={1,4,9,… }M_1 = \{1, 4, 9, \dots\}M1​={1,4,9,…} and m2m_2m2​ in periods M2={2,5,10,… }M_2 = \{2, 5, 10, \dots\}M2​={2,5,10,…}, and otherwise charges φ(PΘθ^t,Xt)\varphi(\mathcal P_\Theta \hat\theta_t, X_t)φ(PΘ​θ^t​,Xt​), where θ^t\hat\theta_tθ^t​ maximizes the lasso-penalized quasi-likelihood

Qt−1(θ~,λt)=∑k=1t−1χk∫Dkg(θ~⋅uk)Dk−yg′(g−1(y)) dy−λt∥θ~∥1Q_{t-1}(\tilde\theta, \lambda_t) = \sum_{k=1}^{t-1} \chi_k \int_{D_k}^{g(\tilde\theta \cdot u_k)} \frac{D_k - y}{g'(g^{-1}(y))}\, dy - \lambda_t \|\tilde\theta\|_1Qt−1​(θ~,λt​)=k=1∑t−1​χk​∫Dk​g(θ~⋅uk​)​g′(g−1(y))Dk​−y​dy−λt​∥θ~∥1​

over the experimental periods (χk=I{k∈M1∪M2}\chi_k = \mathbb I\{k \in M_1 \cup M_2\}χk​=I{k∈M1​∪M2​}), and PΘ\mathcal P_\ThetaPΘ​ is the projection onto Θ\ThetaΘ. The regularization level is λt+1=c~ t1/4log⁡d+log⁡t\lambda_{t+1} = \tilde c\, t^{1/4} \sqrt{\log d + \log t}λt+1​=c~t1/4logd+logt​ for a constant c~>0\tilde c > 0c~>0.

Formalization targets

Goal: Theorem 3

There is a finite constant C~>0\tilde C > 0C~>0 such that

Δθπ(T)≤C~ sT (log⁡d+log⁡T)for all θ∈Θ, T≥2.\Delta^\pi_\theta(T) \le \tilde C\, s \sqrt T\, (\log d + \log T) \qquad \text{for all } \theta \in \Theta,\ T \ge 2 .Δθπ​(T)≤C~sT​(logd+logT)for all θ∈Θ, T≥2.

C~\tilde CC~ is fixed with the model and is uniform over Θ\ThetaΘ; since s=s(θ)s = s(\theta)s=s(θ) varies over Θ\ThetaΘ, the factor sss carries content.

Milestones

  1. The counting property of the schedule (7): for t≥5t \ge 5t≥5 each experimental price is charged at least 14t\tfrac14 \sqrt t41​t​ times in periods 1,…,t1, \dots, t1,…,t.
  2. Lemma 3, the estimation error of the lasso quasi-likelihood estimate:
P{∥θ^t+1(lasso)(λt+1)−θ∥2≤ρ3s(log⁡d+log⁡t)t}≥1−κ3s(log⁡d+log⁡t)t,t≥t1.\mathbb P\Big\{\|\hat\theta^{(\mathrm{lasso})}_{t+1}(\lambda_{t+1}) - \theta\|^2 \le \rho_3 \frac{s(\log d + \log t)}{\sqrt t}\Big\} \ge 1 - \kappa_3 \frac{s(\log d + \log t)}{\sqrt t}, \qquad t \ge t_1 .P{∥θ^t+1(lasso)​(λt+1​)−θ∥2≤ρ3​t​s(logd+logt)​}≥1−κ3​t​s(logd+logt)​,t≥t1​.

Significance

Theorem 3 says that a seller facing many potentially relevant customer characteristics, without knowing which matter and under a nonlinear demand link such as logit, loses revenue at a rate governed by the number sss of relevant characteristics and only logarithmically by the total number ddd. Combined with the paper's sTs\sqrt TsT​ lower bound (Theorem 1, for the linear model), it makes ILQX first-order optimal up to logarithmic factors. An unregularized policy would pay order dTd\sqrt TdT​, which is linear in TTT when ddd exceeds T\sqrt TT​.

The paper's proofs are in its electronic companion, which this mission does not reproduce; no machine-checked version of the result exists. A complete formalization would also produce a reusable high-probability error bound for lasso-regularized quasi-likelihood estimation under adaptively collected data (Lemma 3), and a machine-checked model of feature-based dynamic pricing on which related policies can be stated.

Difficulty

The estimation error must be controlled on data collected by the policy itself. The design vectors uk=[1;mi(k)]⊗Xku_k = [1; m_{i(k)}] \otimes X_kuk​=[1;mi(k)​]⊗Xk​ use only two prices, so the information about the price sensitivity accrues only from the Θ(t)\Theta(\sqrt t)Θ(t​) experimental periods. Standard lasso error bounds assume a fixed or i.i.d. design and a restricted-eigenvalue condition on it; here a restricted-eigenvalue-type property of the random Kronecker design must be derived from ΣZ≻0\Sigma_Z \succ 0ΣZ​≻0 and m1≠m2m_1 \ne m_2m1​=m2​, and the noise enters through a martingale whose conditional law may depend on past prices. The quasi-likelihood is not a least-squares objective, so the curvature comes from the bounds ℓ~≤g′≤u~\tilde\ell \le g' \le \tilde uℓ~≤g′≤u~, which hold only on the bounded domain {θ⋅u(p,x):θ∈Θ}\{\theta \cdot u(p,x) : \theta \in \Theta\}{θ⋅u(p,x):θ∈Θ} and not at an unconstrained estimate. Finally, the regret bound has to turn a squared-error bound holding with probability 1−O(slog⁡(dt)/t)1 - O(s \log(dt)/\sqrt t)1−O(slog(dt)/t​) into an expected revenue loss, which requires the revenue loss to be second order in the parameter error at an interior optimum.

Formalization scope

  • Coordinates. x∈Rd+1x \in \mathbb R^{d+1}x∈Rd+1 is Fin (d+1) → ℝ with the intercept at index 0; θ\thetaθ is Fin 2 × Fin (d+1) → ℝ with θ(0,i)=αi\theta(0,i) = \alpha_iθ(0,i)=αi​, θ(1,i)=βi\theta(1,i) = \beta_iθ(1,i)=βi​. Periods are 1-based. d≥1d \ge 1d≥1.
  • Objective. The integrated form ∑kχk(Dk (θ~⋅uk)−∫0θ~⋅ukg)−λ∥θ~∥1\sum_k \chi_k \big(D_k\,(\tilde\theta \cdot u_k) - \int_0^{\tilde\theta \cdot u_k} g\big) - \lambda \|\tilde\theta\|_1∑k​χk​(Dk​(θ~⋅uk​)−∫0θ~⋅uk​​g)−λ∥θ~∥1​, which differs from the paper's (17) by a term not depending on θ~\tilde\thetaθ~ and therefore has the same maximizers.
  • Estimator. Any measurable selection that is a maximizer whenever one exists; the theorems hold for every such selection. The paper's claim that (17) is strictly concave with a unique maximizer fails when 2(d+1)2(d+1)2(d+1) exceeds the rank of the design.
  • Policy. The price is defined by strong recursion over the realized history; the estimator sees θ\thetaθ only through the realized demands.
  • Regret is a lower Lebesgue integral of nonnegative losses; conditional moments of the shocks use the conditional Lebesgue expectation; probabilities are compared in [0,∞][0, \infty][0,∞].
  • Standing assumptions formalized: i.i.d. measurable features in a compact Z\mathcal ZZ inside a Euclidean ball, mean zero, positive definite ΣZ\Sigma_ZΣZ​; fresh customers (Zt+1Z_{t+1}Zt+1​ independent of the past); sub-Gaussian martingale-difference shocks; m1≠m2m_1 \ne m_2m1​=m2​ in [ℓ,u][\ell, u][ℓ,u]; ggg differentiable and increasing with derivative bounds on the relevant domain (endnote 2 states them as a consequence, which is false; here they are a hypothesis); φ\varphiφ an unconstrained revenue maximizer in (ℓ,u)(\ell, u)(ℓ,u) whose pricing map after projection and feature augmentation is measurable. The processes are constrained only at the used periods t≥1t\ge1t≥1.
  • Added hypotheses, per θ\thetaθ: s(θ)≥1s(\theta) \ge 1s(θ)≥1 (the paper's s∈{1,…,d+1}s \in \{1, \dots, d+1\}s∈{1,…,d+1}), and almost surely every realized demand lies in the closure of the range of ggg, which guarantees that a maximizer of (17) exists (true for the linear link and for binary demand under a logit link).
  • Not formalized: the paper's support condition on continuous features; Remark 7's independence of C~\tilde CC~ from ddd; dependence of the shock law on θ\thetaθ (the shocks are one fixed process).
  • Ruled-out trivializations. The regret cannot vanish by non-integrability; the estimator cannot be an arbitrary function, since it must maximize the objective; C~\tilde CC~ and κ3,ρ3,t1\kappa_3, \rho_3, t_1κ3​,ρ3​,t1​ are chosen before θ\thetaθ, so they cannot absorb sss; the hypotheses are jointly satisfiable (a linear-link instance with d=1d = 1d=1 is checked).

Contributions welcome: existence of a measurable maximizer of the objective, the counting lemma, and supporting lemmas for Lemma 3 and the regret bound.

Selected references

  • G.-Y. Ban and N. B. Keskin, Personalized Dynamic Pricing with Machine Learning: High-Dimensional Features and Heterogeneous Elasticity, Management Science 67(9):5549–5568, 2021. https://doi.org/10.1287/mnsc.2020.3680
  • A. V. den Boer and B. Zwart, Simultaneously Learning and Optimizing Using Controlled Variance Pricing, Management Science 60(3):770–783, 2014. https://doi.org/10.1287/mnsc.2013.1788
  • N. B. Keskin and A. Zeevi, Dynamic Pricing with an Unknown Demand Model: Asymptotically Optimal Semi-Myopic Policies, Operations Research 62(5):1142–1167, 2014. https://doi.org/10.1287/opre.2014.1294
  • A. Javanmard and H. Nazerzadeh, Dynamic Pricing in High-Dimensions, Journal of Machine Learning Research 20(9):1–49, 2019. https://jmlr.org/papers/v20/17-357.html
6 thms2 active usersReviewed
Bandit AlgorithmsInformation TheoryOperations Research·Captain: mikedeng1

Regret in Online Combinatorial Optimization II: Under Bandit Feedback Every Strategy Has Minimax Regret at Least 0.02 m√(dn) on Some Set of Actions with m OnesResearch Paper

Why bandit feedback matters

An online player repeatedly chooses a subset of coordinates and pays the sum of their losses. In many applications the player sees only that sum, not the losses of the selected coordinates. This bandit feedback makes it harder to identify which coordinate within each chosen subset is responsible for a favorable or unfavorable outcome. Audibert, Bubeck and Lugosi studied how this information limit changes regret in online combinatorial optimization. Their Theorem 5 establishes a lower bound that scales linearly with the number of selected coordinates and with the square root of the ambient dimension and horizon. Audibert, Bubeck and Lugosi, 2013, pp. 13–14

The result is part of a comparison among full-information, semi-bandit and bandit feedback in the same action model. Under full information the player sees the complete loss vector after each round; under semi-bandit feedback the player sees losses at selected coordinates; under bandit feedback the player sees only one scalar. The paper proves upper bounds for semi-bandit play and a bandit lower bound. The latter is the target of this mission. Audibert, Bubeck and Lugosi, 2013, pp. 2–3, 14

The game and its regret

Fix a dimension ddd, a horizon nnn, and a finite nonempty set A⊆{0,1}d\mathcal A\subseteq\{0,1\}^dA⊆{0,1}d. Each action a∈Aa\in\mathcal Aa∈A has exactly mmm entries equal to one. At round ttt, the player selects a probability distribution on A\mathcal AA, draws an action ata_tat​, and simultaneously an adversary selects a loss vector zt∈[0,1]dz_t\in[0,1]^dzt​∈[0,1]d. The player pays the scalar atTzta_t^\mathsf Tz_tatT​zt​. In the bandit protocol, that scalar is the only new information made available for the next round. A player strategy can use every previous action and observed scalar loss, and can randomize at every round. Audibert, Bubeck and Lugosi, 2013, pp. 2–3

The paper measures performance by pseudo-regret against the best fixed action in expectation:

Rn=E∑t=1natTzt−min⁡a∈AE∑t=1naTzt.R_n=\mathbb E\sum_{t=1}^{n}a_t^\mathsf Tz_t- \min_{a\in\mathcal A}\mathbb E\sum_{t=1}^{n}a^\mathsf Tz_t.Rn​=Et=1∑n​atT​zt​−a∈Amin​Et=1∑n​aTzt​.

The expectation covers the player's randomization and, when present, the adversary's randomization. This benchmark chooses one action for the whole horizon after comparing expected cumulative losses. It is the sole notion of regret used in the paper. Audibert, Bubeck and Lugosi, 2013, p. 2 and footnote 1

For the lower bound, the appendix arranges the ddd coordinates as mmm parallel games with k=d/mk=d/mk=d/m choices per game. An action selects one coordinate from each game. The adversary can associate a favoured choice α\alphaα with each game and assign independent Bernoulli losses whose means are 1/2−ϵ1/2-\epsilon1/2−ϵ at the favoured coordinates and 1/21/21/2 elsewhere. The player still sees only the sum of the mmm selected losses. These are the α\alphaα-adversaries that underlie the milestone statements. Audibert, Bubeck and Lugosi, 2013, pp. 16–17

Formalization targets

The goal is Theorem 5 with the divisibility condition used by its appendix construction. For n≥d≥2mn\ge d\ge2mn≥d≥2m and m∣dm\mid dm∣d, there is an action set A\mathcal AA such that every bandit strategy faces some bounded loss law with

Rn≥0.02 mdn.R_n\ge0.02\,m\sqrt{dn}.Rn​≥0.02mdn​.

The order of the choices is part of the claim: the action set is fixed first; then each player strategy may face its own adversary. The lower bound retains the paper's explicit constant 0.020.020.02. Audibert, Bubeck and Lugosi, 2013, p. 14, Theorem 5; p. 16

The milestones follow the appendix's stated results: Lemma 5 bounds a logarithm; Lemma 4 and its corrected second case bound the divergence between sums of independent Bernoulli variables; the one-round observation bound and the horizon bound control information revealed by bandit feedback. Equations (9), (10), and (11) connect that information to expected regret and symmetry. The final averaged bound on the chance of selecting a favoured choice closes the path to the goal. The milestones are stated for deterministic players, as in the first three steps of the appendix; the passage to randomized players (the appendix's fourth step) is part of the goal's proof and is not a separate milestone. Audibert, Bubeck and Lugosi, 2013, pp. 17–18, 20–21

What the result establishes

The theorem prevents a uniform regret guarantee of smaller order than mdnm\sqrt{dn}mdn​ for this bandit action model. It shows that aggregating mmm selected losses into a single observation imposes a cost beyond what a player can avoid by choosing a different randomized strategy. In particular, the quantifier over every strategy makes the claim about the feedback model, rather than about one named algorithm. The paper proves the result in Appendix B, with the Bernoulli-sum estimate in Appendix C. Audibert, Bubeck and Lugosi, 2013, pp. 14, 16–18, 20–21

The formalization work is to prove the existing paper result in Lean from a finite protocol that represents all behavioural player strategies, together with the finite laws of the appendix's adversaries. The resulting definitions of action paths, scalar observations, finite probability laws, and divergence estimates can support other finite-horizon bandit lower bounds. These statements are drafted as open Lean theorems; compilation checks their types, while proofs remain for solvers. The paper's mathematical proof does not itself constitute a machine-checked proof.

Where the difficulty lies

A scalar bandit observation sums losses from several games. The player cannot separate the contribution of the favoured coordinate in one game from the other selected losses by inspecting that observation. A bound for the divergence of a single Bernoulli variable therefore does not directly bound the information available to the player. The appendix's central estimate concerns sums of independent Bernoulli variables, followed by the law of the whole sequence of scalar observations. Audibert, Bubeck and Lugosi, 2013, pp. 16, 18, 20

There is also a correction to make explicit. Lemma 4 is false for some parameters allowed by its printed statement: taking n=ℓ=1n=\ell=1n=ℓ=1, p=q=1/2p=q=1/2p=q=1/2, and p′=0.001p'=0.001p′=0.001 violates the displayed bound. The error is in the second case of its proof, which applies Lemma 5 outside its range when p′<pp'<pp′<p. The mission states Lemma 4 with the additional condition p(1−p′)≤2p′(1−p)p(1-p')\le2p'(1-p)p(1−p′)≤2p′(1−p), which holds whenever p≤p′p\le p'p≤p′, and states the corrected second case separately: for q=p>p′q=p>p'q=p>p′, KL(B,B′)≤(p′−p)2/((1−p)p′(n+2))\mathrm{KL}(\mathcal B,\mathcal B')\le(p'-p)^2/((1-p)p'(n+2))KL(B,B′)≤(p′−p)2/((1−p)p′(n+2)). Through the corrected case, the one-round bound used by the goal holds for the full range 0<ϵ<1/20<\epsilon<1/20<ϵ<1/2, so Theorem 5 and its constant are unaffected. Audibert, Bubeck and Lugosi, 2013, pp. 20–21, Lemmas 4–5

Formalization scope

Actions are functions Fin(d)→R\mathrm{Fin}(d)\to\mathbb RFin(d)→R constrained to zero or one, with coordinate sum mmm. Coordinates and rounds are indexed from zero. A behavioural strategy maps every history of chosen actions and real-valued observed losses to a probability distribution on the finite action set. Its domain includes every real history, even though the appendix's particular Bernoulli adversaries generate integer observations. The strategy may randomize; restricting the goal to deterministic players or allowing it to inspect an unobserved loss vector would change the game.

The goal's exhibited adversary is an oblivious, finitely supported probability law on loss sequences in [0,1]n×d[0,1]^{n\times d}[0,1]n×d. It belongs to the broader adversary class of the paper, so this existential formulation is stronger. All expectations are finite sums. The regret comparator is a minimum over the nonempty action set, not a default value assigned to an empty set. The action set has binary entries and equal coordinate sums, and every loss lies in the unit interval; these clauses rule out trivial lower bounds from malformed actions or unbounded losses.

The hypothesis m∣dm\mid dm∣d is added to the printed theorem because Appendix B assumes it to partition coordinates into equally sized games. When m=0m=0m=0, it forces d=0d=0d=0 and the claimed lower bound is zero. The milestone model uses functions Fin(m)→Fin(k)\mathrm{Fin}(m)\to\mathrm{Fin}(k)Fin(m)→Fin(k) with d=mkd=mkd=mk, deterministic players, finite Bernoulli product laws, and a finite real KL sum. Its log ratios are meaningful on the common support ensured by the stated parameter bounds. Reusable contributions include the finite bandit protocol and observation laws, as well as proofs of the KL and averaging milestones.

Selected references

  • Jean-Yves Audibert, Sébastien Bubeck and Gábor Lugosi, Regret in Online Combinatorial Optimization, Mathematics of Operations Research 39(1), 2014. arXiv:1204.4710v2; DOI:10.1287/moor.2013.0598.
13 thms2 active usersReviewed
Optimal TransportProbabilityStatistics·Captain: mikedeng1

Wasserstein Distributionally Robust Optimization and Variation Regularization III: With Probability 1 − e^(−t), E_Pn[(ρ − d(z, D_f))₊] ≤ Cρ² + 2ρ·E[Rademacher] + ρ√(t/2n) for Every fResearch Paper

Motivation

Wasserstein distributionally robust optimization (DRO) replaces the empirical risk EPn[f]\mathbb E_{P_n}[f]EPn​​[f] of a loss fff by its worst case over all distributions within Wasserstein distance ρ\rhoρ of the empirical law PnP_nPn​. Gao, Chen and Kleywegt (arXiv:1712.06050; Operations Research, 2024, doi:10.1287/opre.2022.2383) show that, for small radii, this worst-case risk is the empirical risk plus ρ\rhoρ times a variation of the loss (a norm of its gradient or local slope), up to a remainder. This links DRO to gradient-norm regularization, as used in adversarial training and in data-driven newsvendor and pricing models.

For smooth losses the remainder is of order ρ2\rho^2ρ2. Many losses used in practice are only piecewise smooth: hinge, ReLU networks, newsvendor costs, θz∧1\theta z \wedge 1θz∧1. Near the set Df\mathcal D_fDf​ where the pieces meet, the first-order expansion fails, and the paper's Theorem 1(I)–(II) carries an extra boundary term M EPn[(ρ−d(z,Df))+]M\,\mathbb E_{P_n}[(\rho - d(z,\mathcal D_f))_+]MEPn​​[(ρ−d(z,Df​))+​]. Theorem 1(III) shows that this term is small with high probability, uniformly over the loss class, provided the data distribution does not concentrate near the non-smooth points. This mission formalizes that probabilistic bound.

Setting

Let (Z,d)(\mathcal Z, d)(Z,d) be a metric space with its Borel σ\sigmaσ-algebra, PtrueP_{\rm true}Ptrue​ a probability measure on Z\mathcal ZZ, and z1,…,znz_1, \dots, z_nz1​,…,zn​ an i.i.d. sample from PtrueP_{\rm true}Ptrue​, with empirical law Pn=1n∑iδziP_n = \frac1n\sum_i \delta_{z_i}Pn​=n1​∑i​δzi​​. The sampling law is the product P⊗=Ptrue⊗nP_\otimes = P_{\rm true}^{\otimes n}P⊗​=Ptrue⊗n​ on Zn\mathcal Z^nZn, and E⊗\mathbb E_\otimesE⊗​ is its expectation. The formal statements also assume the metric space is second countable.

Let F\mathcal FF be a set of losses f:Z→Rf : \mathcal Z \to \mathbb Rf:Z→R, and for each fff let Df⊆Z\mathcal D_f \subseteq \mathcal ZDf​⊆Z be its non-smooth set: for a piecewise smooth fff with pieces Zf,1,…,Zf,Kf\mathcal Z_{f,1}, \dots, \mathcal Z_{f,K_f}Zf,1​,…,Zf,Kf​​, the union of the pairwise intersections of their closures (Assumption 1). The paper assumes Df\mathcal D_fDf​ is PtrueP_{\rm true}Ptrue​-null. The distance to a set is d(z,D)=inf⁡z~∈Dd(z,z~)d(z, D) = \inf_{\tilde z\in D} d(z,\tilde z)d(z,D)=infz~∈D​d(z,z~), with d(z,∅)=+∞d(z,\varnothing) = +\inftyd(z,∅)=+∞.

The objects of the mission are:

  • the boundary gap (ρ−d(z,Df))+(\rho - d(z, \mathcal D_f))_+(ρ−d(z,Df​))+​ and the near indicator 1{d(z,Df)<ρ}\mathbf 1\{d(z,\mathcal D_f) < \rho\}1{d(z,Df​)<ρ};
  • the indicator class Iρ={z↦1{d(z,Df)<ρ}:f∈F, Df≠∅}\mathcal I_\rho = \{z\mapsto \mathbf 1\{d(z,\mathcal D_f)<\rho\} : f\in\mathcal F,\ \mathcal D_f\neq\varnothing\}Iρ​={z↦1{d(z,Df​)<ρ}:f∈F, Df​=∅};
  • the empirical Rademacher complexity Rn(H)=Eσ[sup⁡h∈H1n∑iσih(zi)]\mathfrak R_n(\mathcal H) = \mathbb E_\sigma\bigl[\sup_{h\in\mathcal H}\frac1n\sum_i\sigma_i h(z_i)\bigr]Rn​(H)=Eσ​[suph∈H​n1​∑i​σi​h(zi​)] of a class H\mathcal HH, with i.i.d. uniform signs σi\sigma_iσi​, and the Rademacher complexity E⊗[Rn(H)]\mathbb E_\otimes[\mathfrak R_n(\mathcal H)]E⊗​[Rn​(H)];
  • Assumption 3 (bounded density):
lim sup⁡δ↓0 sup⁡f∈F: Df≠∅Ptrue{z:0<d(z,Df)<δ}δ<∞.\limsup_{\delta\downarrow 0}\ \sup_{f\in\mathcal F:\,\mathcal D_f\ne\varnothing}\frac{P_{\rm true}\{z : 0 < d(z,\mathcal D_f) < \delta\}}{\delta} < \infty.δ↓0limsup​ f∈F:Df​=∅sup​δPtrue​{z:0<d(z,Df​)<δ}​<∞.

In Lean the boundary gap is the shared WassVarReg.PInf.boundaryGap; nearIndicator, indicatorClass, empRademacher, rademacher and Assumption3 are in WassVarReg.Boundary. The sample law is the published MinimaxWass.DataDep.sampleLaw.

Formalization targets

Goal: Theorem 1(III), first display (p. 8)

Under Assumption 3 there exist ρˉ,C>0\bar\rho, C > 0ρˉ​,C>0 such that for all n≥1n\ge1n≥1, t>0t>0t>0 and 0≤ρ<ρˉ0\le\rho<\bar\rho0≤ρ<ρˉ​, with probability at least 1−e−t1-e^{-t}1−e−t, for every f∈Ff\in\mathcal Ff∈F,

EPn[(ρ−d(z,Df))+]≤Cρ2+2ρ E⊗[Rn(Iρ)]+ρt2n.\mathbb E_{P_n}\bigl[(\rho - d(z,\mathcal D_f))_+\bigr] \le C\rho^2 + 2\rho\,\mathbb E_\otimes[\mathfrak R_n(\mathcal I_\rho)] + \rho\sqrt{\frac{t}{2n}}.EPn​​[(ρ−d(z,Df​))+​]≤Cρ2+2ρE⊗​[Rn​(Iρ​)]+ρ2nt​​.

The constants CCC and ρˉ\bar\rhoρˉ​ are left unspecified; they depend only on Assumption 3.

Milestones

  1. Lemma EC.5 (p. ec4): for a class H\mathcal HH of measurable functions with values in [0,M][0,M][0,M], with probability at least 1−e−t1-e^{-t}1−e−t, EPn[h]≤EPtrue[h]+2 E⊗[Rn(H)]+Mt/(2n)\mathbb E_{P_n}[h] \le \mathbb E_{P_{\rm true}}[h] + 2\,\mathbb E_\otimes[\mathfrak R_n(\mathcal H)] + M\sqrt{t/(2n)}EPn​​[h]≤EPtrue​​[h]+2E⊗​[Rn​(H)]+Mt/(2n)​ for all h∈Hh\in\mathcal Hh∈H.
  2. Assumption 3 bounds the true mass (proof of Lemma EC.11, p. ec11): EPtrue[1{d(z,Df)<ρ}]≤Cρ\mathbb E_{P_{\rm true}}[\mathbf 1\{d(z,\mathcal D_f)<\rho\}] \le C\rhoEPtrue​​[1{d(z,Df​)<ρ}]≤Cρ for 0≤ρ<ρˉ0\le\rho<\bar\rho0≤ρ<ρˉ​, uniformly in fff.
  3. Lemma EC.11 (p. ec11): with probability at least 1−e−t1-e^{-t}1−e−t, for every fff, EPn[1{d(z,Df)<ρ}]≤Cρ+2 E⊗[Rn(Iρ)]+t/(2n)\mathbb E_{P_n}[\mathbf 1\{d(z,\mathcal D_f)<\rho\}] \le C\rho + 2\,\mathbb E_\otimes[\mathfrak R_n(\mathcal I_\rho)] + \sqrt{t/(2n)}EPn​​[1{d(z,Df​)<ρ}]≤Cρ+2E⊗​[Rn​(Iρ​)]+t/(2n)​.
  4. Gap versus mass (p. ec11): EPn[(ρ−d(z,Df))+]≤ρ EPn[1{d(z,Df)<ρ}]\mathbb E_{P_n}[(\rho-d(z,\mathcal D_f))_+] \le \rho\,\mathbb E_{P_n}[\mathbf 1\{d(z,\mathcal D_f)<\rho\}]EPn​​[(ρ−d(z,Df​))+​]≤ρEPn​​[1{d(z,Df​)<ρ}].

Significance

The result. With ρn=ρ0/n\rho_n = \rho_0/\sqrt nρn​=ρ0​/n​ and a class Iρ\mathcal I_\rhoIρ​ of bounded complexity, the right side of the goal is Op(1/n)O_p(1/n)Op​(1/n). Combined with Theorem 1(I)–(II), it gives the paper's conclusion that ppp-Wasserstein DRO with p∈(1,∞]p\in(1,\infty]p∈(1,∞] is asymptotically equivalent to empirical variation regularization, min⁡f{EPn[f]+ρnVPn,q(f)}+Op(1/n)\min_f\{\mathbb E_{P_n}[f] + \rho_n\mathcal V_{P_n,q}(f)\} + O_p(1/n)minf​{EPn​​[f]+ρn​VPn​,q​(f)}+Op​(1/n), for non-smooth losses, uniformly over the class. Without it, the boundary term is only bounded by MρM\rhoMρ, and the expansion has no better accuracy than first order. Assumption 3 is the condition under which the non-smooth set does not matter statistically.

Formalizing it. The result is proved in the paper; it has no machine-checked proof. Its proof composes a uniform deviation bound (McDiarmid's inequality plus symmetrization) with a bounded-density estimate. The mission produces a general Rademacher deviation bound for bounded classes on an arbitrary metric space with a measurability guard, reusable by any statistical-learning mission, and fixes the printed constant of Lemma EC.5 (M2t/nM\sqrt{2t/n}M2t/n​ on the page, Mt/(2n)M\sqrt{t/(2n)}Mt/(2n)​ from its proof).

Difficulty

The deterministic parts are short. The difficulty is in Lemma EC.5. McDiarmid's inequality must be applied to the uniform deviation over an arbitrary, possibly uncountable, class. Symmetrization must then be carried out with Bochner integrals over Zn×Zn\mathcal Z^n\times\mathcal Z^nZn×Zn and a finite sign space. Neither tool is in Mathlib in this form. A tempting shortcut, bounding each fixed hhh by Hoeffding's inequality and taking a union bound, fails: the class is infinite, and the bound must hold simultaneously for all fff inside a single event. A second point is Assumption 3: it controls only the shell 0<d(z,Df)<δ0 < d(z,\mathcal D_f) < \delta0<d(z,Df​)<δ, so the mass of {d=0}\{d = 0\}{d=0} has to come from the null closure of Df\mathcal D_fDf​.

Formalization scope

  • Space and sample. Z\mathcal ZZ is a metric space with [MeasurableSpace Z] [BorelSpace Z] [SecondCountableTopology Z] in the goal-level statements. Second countability is an added technical hypothesis used throughout this paper's mission series. A sample is ω : Fin n → Z with n≥1n\ge1n≥1, and its law is Measure.pi. Empirical means are 1n∑i:Fin n\frac1n\sum_{i : \mathrm{Fin}\,n}n1​∑i:Finn​.
  • Distance to a set. Metric.infEDist, which is +∞+\infty+∞ on ∅\varnothing∅. A loss with Df=∅\mathcal D_f=\varnothingDf​=∅ thus has boundary gap 000 and is excluded from Iρ\mathcal I_\rhoIρ​, as on the page. The real-valued Metric.infDist (which is 000 on ∅\varnothing∅) would give such a loss gap ρ\rhoρ and is not used.
  • The non-smooth sets are an arbitrary family D : (Z → ℝ) → Set Z. The goal-level statements assume each Df\mathcal D_fDf​ is closed and PtrueP_{\rm true}Ptrue​-null, the standing consequence of its construction in Assumption 1. The details of the piecewise smooth loss model are not needed for this probability estimate.
  • Probabilities are written as P⊗{ω:¬ ∀f∈F,… }≤e−tP_\otimes\{\omega : \neg\,\forall f\in\mathcal F, \dots\}\le e^{-t}P⊗​{ω:¬∀f∈F,…}≤e−t, with the quantifier over F\mathcal FF inside the event (an outer measure for uncountable F\mathcal FF).
  • Rademacher complexity. The sign expectation is an average over {±1}n\{\pm1\}^n{±1}n. Suprema are real suprema of classes with values in [0,M][0,M][0,M]; an empty class gives 000. E⊗\mathbb E_\otimesE⊗​ is a Bochner integral.
  • Measurability guard (added): the uniform deviation, the empirical Rademacher complexity and the double-sample deviation of the class are measurable. This holds, for instance, for countable F\mathcal FF, and the paper leaves it implicit.
  • Quantifier order. ρˉ\bar\rhoρˉ​ and CCC are chosen before nnn and ttt, which is stronger than the page's order. The radius satisfies ρ≥0\rho\ge0ρ≥0.
  • Not stated. The second display of Theorem 1(III) and the "Moreover" part of Lemma EC.11, which use Dudley's entropy integral: their proof bounds the empirical Rademacher complexity, while Lemma EC.5 needs its expectation.
  • Trivialization ruled out. The Rademacher term is that of the specific class Iρ\mathcal I_\rhoIρ​ built from the sets Df\mathcal D_fDf​. It is not a free constant, and no hypothesis assumes the conclusion of Lemma EC.5.

Contributions welcome: a general McDiarmid inequality for product measures, a symmetrization lemma for measurable classes, and proofs of the milestones in any order.

Selected references

  • R. Gao, X. Chen, A. J. Kleywegt, Wasserstein Distributionally Robust Optimization and Variation Regularization, Operations Research, 2024. arXiv:1712.06050, doi:10.1287/opre.2022.2383
  • S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press, 2014, Theorem 26.5. doi:10.1017/CBO9781107298019
  • C. McDiarmid, On the method of bounded differences, Surveys in Combinatorics, LMS Lecture Note Series 141, 1989. doi:10.1017/CBO9781107359949.008
  • M. J. Wainwright, High-Dimensional Statistics: A Non-Asymptotic Viewpoint, Cambridge University Press, 2019, Ch. 5. doi:10.1017/9781108627771
8 thms2 active usersReviewed
Bandit AlgorithmsOperations ResearchStatistics·Captain: mikedeng1

Personalized Dynamic Pricing with Machine Learning: High-Dimensional Features and Heterogeneous Elasticity 1: Under Known Sparsity, ILSX Pricing Has Regret at Most Cs√T log TResearch Paper

Motivation

A seller who changes prices while learning demand faces two costs: a price experiment may earn less revenue now, while insufficient experimentation leaves later prices poorly calibrated. Customer features make the balance harder because both willingness to buy and sensitivity to price may vary between customers. Ban and Keskin study this problem when the number of recorded features can be large relative to the selling horizon. Their known-support policy, ILSX, provides the upper bound formalized here Ban and Keskin, 2021, §§2 and 4.1.

The paper's result is a statement about learning a demand model while serving sequential customers, rather than about fitting that model after all observations are available. A price at period ttt must use information then available, including the arriving customer's features but excluding that customer's demand shock. The seller therefore needs a policy whose expected revenue loss can be bounded uniformly over the unknown parameters in a specified model Ban and Keskin, 2021, pp. 5552–5556.

Setting

In each period t=1,2,…t=1,2,\ldotst=1,2,…, a customer arrives with a feature vector Zt∈RdZ_t\in\mathbb R^dZt​∈Rd. The seller observes the augmented vector Xt=[1;Zt]X_t=[1;Z_t]Xt​=[1;Zt​], posts a price pt∈[ℓ,u]p_t\in[\ell,u]pt​∈[ℓ,u] with 0<ℓ<u0<\ell<u0<ℓ<u, and then observes demand. In the linear demand model, demand equals

Dt=α⋅Xt+(β⋅Xt)pt+εt.D_t=\alpha\cdot X_t+(\beta\cdot X_t)p_t+\varepsilon_t.Dt​=α⋅Xt​+(β⋅Xt​)pt​+εt​.

The intercept and slope vectors α,β\alpha,\betaα,β are unknown. The seller knows the sparsity structure S\mathcal SS: outside this set of coordinate indices, both parameter vectors vanish. The compressed parameter θS=(αS,βS)\theta_{\mathcal S}=(\alpha_{\mathcal S},\beta_{\mathcal S})θS​=(αS​,βS​) lies in a compact coordinate rectangle ΘS\Theta_{\mathcal S}ΘS​, and s=∣S∣≥1s=|\mathcal S|\ge1s=∣S∣≥1. The compressed customer vector is XS,t=(Xt,i)i∈SX_{\mathcal S,t}=(X_{t,i})_{i\in\mathcal S}XS,t​=(Xt,i​)i∈S​ Ban and Keskin, 2021, pp. 5552–5555.

At price ppp, expected single-period revenue is r(p,θS,x)=p[αS⋅x+(βS⋅x)p]r(p,\theta_{\mathcal S},x)=p[\alpha_{\mathcal S}\cdot x+(\beta_{\mathcal S}\cdot x)p]r(p,θS​,x)=p[αS​⋅x+(βS​⋅x)p]. The clairvoyant price is φ(θS,x)=−(αS⋅x)/(2βS⋅x)\varphi(\theta_{\mathcal S},x)=-(\alpha_{\mathcal S}\cdot x)/(2\beta_{\mathcal S}\cdot x)φ(θS​,x)=−(αS​⋅x)/(2βS​⋅x) when the slope is negative. The paper assumes this price is in (ℓ,u)(\ell,u)(ℓ,u) throughout its model domain. Expected regret ΔθSπ(T)\Delta^{\pi}_{\theta_{\mathcal S}}(T)ΔθS​π​(T) is the expected sum, through period TTT, of the clairvoyant revenue minus the revenue at the policy's price Ban and Keskin, 2021, (2)–(4), p. 5553–5554.

The ILSX policy charges one fixed experimental price m1m_1m1​ at periods 1,4,9,…1,4,9,\ldots1,4,9,… and a distinct price m2m_2m2​ at periods 2,5,10,…2,5,10,\ldots2,5,10,…. At every other period it estimates the compressed demand parameter by least squares using experimental observations, projects the estimate onto ΘS\Theta_{\mathcal S}ΘS​, and applies φ\varphiφ to the projected estimate and the current customer vector. Both prices lie in [ℓ,u][\ell,u][ℓ,u] Ban and Keskin, 2021, (7), (8), (12), pp. 5555–5556.

Formalization targets

The goal is Theorem 2: for the stated model and ILSX policy, a finite positive constant CCC exists such that

ΔθSILSX(m1,m2)(T)≤CsTlog⁡T(θS∈ΘS, T≥2).\Delta^{\mathrm{ILSX}(m_1,m_2)}_{\theta_{\mathcal S}}(T)\le Cs\sqrt T\log T\qquad(\theta_{\mathcal S}\in\Theta_{\mathcal S},\ T\ge2).ΔθS​ILSX(m1​,m2​)​(T)≤CsT​logT(θS​∈ΘS​, T≥2).

The constant is chosen after the model, shock process, experimental prices, and estimator selection are fixed, but before the parameter and horizon. The theorem and Remark 5 are on pp. 5556–5557 of Ban and Keskin, 2021.

The milestones reproduce the paper's route into this claim: the experimental-period count following (7); the growth bound (15) for the price-variation statistic JtJ_tJt​; Lemma 1's minimum-eigenvalue probability bound for the experimental information matrix J~S,t\widetilde{\mathcal J}_{\mathcal S,t}J​S,t​; and Lemma 2's high-probability bound for its noise-driven estimation error. The least-squares error identity (11) is included as a supporting item on invertible samples, not as a milestone. The constants and thresholds in these targets remain existential where the article leaves them existential Ban and Keskin, 2021, pp. 5555–5556.

Significance

The theorem gives a guarantee for a concrete pricing policy under feature-dependent price sensitivity. The bound grows with the horizon at order Tlog⁡T\sqrt T\log TT​logT for a fixed known support, with an explicit factor sss in the published statement. It lets a seller compare the cost of this policy with the revenue of a seller who knew the demand coefficients before pricing. The paper presents its result as matching its lower-bound benchmark up to logarithmic factors, under the paper's respective hypotheses Ban and Keskin, 2021, Theorems 1–2, pp. 5554–5556.

The mathematical result was published in 2021. This mission seeks a machine-checked statement and proof of the known-support guarantee and its stated intermediate results. The checked Prove2Me prior-art catalog contains featureless dynamic-pricing results and separate least-squares results, but no matching formalization of ILSX. A completed development would add reusable definitions for a feature-dependent price regressor, experimental information matrix, nonanticipating estimator selection, and expected regret. The article places its proofs in an electronic companion, which is not included in the held source PDF; this mission does not reproduce those proofs as source material.

Difficulty

The policy learns from a sparse set of deliberately chosen periods, while its regret accumulates at every period. A pointwise least-squares identity alone does not control the regret: the information matrix can be singular at early periods, and the customer features and shocks are random. In particular, the printed inverse identity (11) says “for all t≥2t\ge2t≥2”, although at t=2t=2t=2 the matrix has rank at most two and cannot be invertible when s≥2s\ge2s≥2. The formal statement therefore confines the inverse identity to invertible samples. Lemma 2's event also includes invertibility so a default inverse of a singular matrix cannot make its error bound automatic Ban and Keskin, 2021, (9)–(11), Lemma 2, p. 5556.

Formalization scope

Lean uses one-based periods and a support subtype of the augmented feature coordinates. Coordinate zero is the paper's intercept coordinate one. The parameter has coordinates (j,i)(j,i)(j,i) for demand intercept or price slope j∈{0,1}j\in\{0,1\}j∈{0,1} and support index i∈Si\in\mathcal Si∈S. The regressor [1;p]⊗XS,t[1;p]\otimes X_{\mathcal S,t}[1;p]⊗XS,t​ gives the information matrix in (10). Euclidean squared norms are explicit sums of squares, and the minimum-eigenvalue claim is the equivalent Rayleigh inequality. The parameter rectangle is nonempty, and its Euclidean projection is coordinatewise clamping.

The standing assumptions are d≥1d\ge1d≥1, s≥1s\ge1s≥1, 0<ℓ<u0<\ell<u0<ℓ<u, distinct experimental prices in [ℓ,u][\ell,u][ℓ,u], bounded i.i.d. features with mean zero and positive-definite covariance, and a fixed shock process with conditional mean zero, bounded conditional second moment, and finite conditional exponential moments near zero. The feature process is measurable and integrable. The model additionally states independence of the next customer's features from the past primitives, the reading of the paper's joint probability construction used here. The slope is negative and the clairvoyant price lies in (ℓ,u)(\ell,u)(ℓ,u) for every allowed parameter and realized feature. These conditions ensure that the price formula represents a genuine revenue maximum. The phrase about continuous features having positive measure in the interior of their domains is omitted because the page does not specify the domains or a reference measure Ban and Keskin, 2021, §2, pp. 5552–5554.

The estimator is any measurable minimizer of the experimental least-squares objective for histories with at least two observations. Its type depends only on earlier observations; at nonexperimental periods the recorded pseudo-observations are ignored by the objective. The shock law is fixed while the parameter varies. Expected regret uses a nonnegative integral, and conditional second and exponential moments use nonnegative conditional expectations. The theorem binds CCC before the parameter and horizon, while leaving support size fixed with the model. Consequently, the stronger uniformity in sss suggested by Remark 5 is outside this statement. These choices exclude vacuous integrals, unconstrained estimators, and accidental bounds from a singular matrix inverse. Work on the probability estimates, least-squares identity, and regret bound is welcome under these exact definitions.

Selected references

  • G.-Y. Ban and N. B. Keskin, Personalized Dynamic Pricing with Machine Learning: High-Dimensional Features and Heterogeneous Elasticity, Management Science 67(9):5549–5568, 2021. DOI: 10.1287/mnsc.2020.3680.
6 thms2 active usersReviewed
Operations ResearchOptimal TransportOptimization·Captain: mikedeng1

Wasserstein Distributionally Robust Optimization and Variation Regularization II: The ∞-Wasserstein Regularizer Equals ρ Times the Mean Local Slope up to ρ²‖H‖ Plus a Boundary TermResearch Paper

Motivation

In distributionally robust optimization, a decision is judged against probability laws near a nominal law rather than against the nominal law alone. The nominal law is often the empirical distribution of observed data. For a loss function, this produces a worst-case expected loss that depends on the size and geometry of the permitted changes to the data. The infinity-Wasserstein model restricts how far any observation can be moved, apart from null events in a coupling. It is a mathematical form of the bounded perturbations used in adversarial learning. Gao, Chen and Kleywegt study when this robust loss behaves like a regularizer based on variation or local slope.

For a finite data set, the nominal law has finite support. This makes the infinity-Wasserstein case especially concrete: each atom of the empirical law has a radius-limited set of possible destinations. Yet the objective still ranges over all probability measures, and the loss need not be bounded on a metric ball. The paper's finite-support duality result and its local-slope estimate together identify the regularization term that this distributional model produces.

Setting

Let (Z,d)(Z,d)(Z,d) be a metric space and let P,QP,QP,Q be probability laws on its Borel sets. A coupling π\piπ of PPP and QQQ is a joint law on Z×ZZ\times ZZ×Z with those two marginals. The infinity-Wasserstein distance is

W∞(P,Q)=inf⁡π∈Γ(P,Q)ess sup⁡(z~,z)∼πd(z~,z).W_\infty(P,Q)=\inf_{\pi\in\Gamma(P,Q)} \operatorname*{ess\,sup}_{(\tilde z,z)\sim\pi}d(\tilde z,z).W∞​(P,Q)=π∈Γ(P,Q)inf​(z~,z)∼πesssup​d(z~,z).

Given a real loss fff and a radius ρ≥0\rho\ge0ρ≥0, the robust value is the supremum of EP[f]\mathbb E_P[f]EP​[f] over all PPP satisfying W∞(P,Q)≤ρW_\infty(P,Q)\le\rhoW∞​(P,Q)≤ρ. The regularizer RQ,∞(ρ;f)R_{Q,\infty}(\rho;f)RQ,∞​(ρ;f) is this robust value minus EQ[f]\mathbb E_Q[f]EQ​[f]. For observations z1,…,znz_1,\ldots,z_nz1​,…,zn​, the empirical law is Pn=n−1∑i=1nδziP_n=n^{-1}\sum_{i=1}^n\delta_{z_i}Pn​=n−1∑i=1n​δzi​​.

The one-sided local slope ∣∂f∣(z)|\partial f|(z)∣∂f∣(z) records the greatest limiting rate at which fff can increase under a perturbation that tends to zzz. The local growth Gf(δ,z)G_f(\delta,z)Gf​(δ,z) is the largest increase f(z~)−f(z)f(\tilde z)-f(z)f(z~)−f(z) among destinations with d(z~,z)≤δd(\tilde z,z)\le\deltad(z~,z)≤δ. A set DfD_fDf​ marks the interfaces between pieces of a piecewise loss. Its boundary contribution at zzz is (δ−d(z,Df))+(\delta-d(z,D_f))_+(δ−d(z,Df​))+​, where the distance to an empty set is infinite. These objects appear in Definition 2 and Assumption 5(I).

Formalization targets

Finite-support transport duality

For a nominal law Q=∑jqjδzjQ=\sum_j q_j\delta_{z_j}Q=∑j​qj​δzj​​, the first targets assert that an optimal coupling exists when W∞(P,Q)W_\infty(P,Q)W∞​(P,Q) is finite and that

sup⁡P:W∞(P,Q)≤ρEP[f]=∑jqjsup⁡d(z~,zj)≤ρf(z~).\sup_{P:W_\infty(P,Q)\le\rho}\mathbb E_P[f] =\sum_j q_j\sup_{d(\tilde z,z_j)\le\rho}f(\tilde z).P:W∞​(P,Q)≤ρsup​EP​[f]=j∑​qj​d(z~,zj​)≤ρsup​f(z~).

This is Lemma EC.2 in the appendix. The values are allowed to be infinite when a local loss supremum is infinite; the printed assertion of unconditional finiteness is not included.

Consistency

For an upper semicontinuous loss, the Proposition 1 case p=∞p=\inftyp=∞ asserts

lim⁡ρ↓0RPn,∞(ρ;f)=0=RPn,∞(0;f).\lim_{\rho\downarrow0}R_{P_n,\infty}(\rho;f)=0=R_{P_n,\infty}(0;f).ρ↓0lim​RPn​,∞​(ρ;f)=0=RPn​,∞​(0;f).

It is a companion statement: it says that the regularizer vanishes, while the main target says at which rate.

Local-slope regularization

The main target is Theorem 1(I), in its general-metric form from Appendix B under Assumption 5(I). If δ0>0\delta_0>0δ0​>0, M≥0M\ge0M≥0, and the local growth obeys

∣Gf(δ,z)−δ∣∂f∣(z)∣≤H(z)δ2+M(δ−d(z,Df))+(0≤δ<δ0),|G_f(\delta,z)-\delta|\partial f|(z)| \le H(z)\delta^2+M(\delta-d(z,D_f))_+ \quad(0\le\delta<\delta_0),∣Gf​(δ,z)−δ∣∂f∣(z)∣≤H(z)δ2+M(δ−d(z,Df​))+​(0≤δ<δ0​),

then, for a sufficiently small radius, every nonempty empirical sample and every loss in the class satisfy

∣RPn,∞(ρ;f)−ρ1n∑i=1n∣∂f∣(zi)∣≤ρ21n∑i=1n∣H(zi)∣+M1n∑i=1n(ρ−d(zi,Df))+.\left|R_{P_n,\infty}(\rho;f) -\rho\frac1n\sum_{i=1}^n|\partial f|(z_i)\right| \le\rho^2\frac1n\sum_{i=1}^n|H(z_i)| +M\frac1n\sum_{i=1}^n(\rho-d(z_i,D_f))_+.​RPn​,∞​(ρ;f)−ρn1​i=1∑n​∣∂f∣(zi​)​≤ρ2n1​i=1∑n​∣H(zi​)∣+Mn1​i=1∑n​(ρ−d(zi​,Df​))+​.

The finite-support duality and the identity between the empirical regularizer and the average of Gf(ρ,zi)G_f(\rho,z_i)Gf​(ρ,zi​) are the mission's intermediate targets. The printed ρn2\rho_n^2ρn2​ in Lemma EC.9 is read as ρ2\rho^2ρ2, as in Theorem 1(I).

Significance

The estimate gives a precise first-order description of infinity-Wasserstein robustness: the mean local slope is its leading regularization term. The quadratic term measures local curvature or growth, while the boundary term records the effect of sample points close to interfaces between loss pieces. Setting M=0M=0M=0 gives the smooth-loss regime, where that interface correction vanishes. The result makes it possible to compare bounded data perturbations with explicit variation penalties using the same loss class and sample.

The paper proves the result. The formalization work here is to express its objects and quantified conditions in Lean, establish finite-support duality for the extended-real robust value, and then formalize the known local estimate. No machine-checked proof of these mission targets is claimed here. The transport and duality definitions can support other finite-support distributionally robust models; the local-slope definitions can be reused for other metric-space perturbation bounds.

Difficulty

The robust value is a supremum over probability laws, whereas the target bound is a finite average of pointwise quantities. Relating the two requires control of the infinity-Wasserstein constraint at every nominal atom, including cases where an infimum over couplings must be attained. A direct substitution of a pointwise supremum into an expectation does not establish the reverse inequality, and extended-real values need care when a loss is unbounded. At the local level, a slope is only a limit near a point; the inequality needed here holds uniformly for all losses, all points, and all sufficiently small radii, with a correction near DfD_fDf​.

Formalization scope

The Lean metric space uses its Borel measurable structure and is second countable, so the distance on the product is measurable. It has no isolated points, matching §3.4. Losses in the class are measurable. Sample indices are Fin n with n>0n>0n>0, and PnP_nPn​ is the published empirical probability law. Infinity-Wasserstein distance takes values in [0,+∞][0,+\infty][0,+∞]; robust values and local growth use extended reals. The theorem explicitly requires finite local slopes and finite growth values in the range where Assumption 5(I)'s real inequality is used. The regularizer is built from the supremum over all probability laws in the Wasserstein ball, and its finite-sample result states finiteness before converting its value to a real number. This rules out replacing the regularizer by the easier average of local growth.

The paper defines DfD_fDf​ from intersections of closures of pieces and assumes it is null under a true law. The deterministic estimate uses only distances to DfD_fDf​, so this mission treats DfD_fDf​ as supplied data and does not require nullity. Likewise, H∈L1(Ptrue)H\in L^1(P_{\rm true})H∈L1(Ptrue​) is needed for the paper's statistical setting but the target evaluates HHH only at finite sample points, so no true law appears in this statement. The radius bound is placed before the sample because Assumption 5(I) supplies one common δ0\delta_0δ0​. These generalizations and the positive-radius reading of Assumption 5(I) are explicit in the item statements. Contributions toward the optimal-coupling lemma, extended-real finite-support duality, and the local estimate all advance the target.

Selected references

  • R. Gao and A. J. Kleywegt, Distributionally Robust Stochastic Optimization with Wasserstein Distance, Mathematics of Operations Research, 2023; arXiv:1604.02199. Source of the duality theory for finite ppp that Lemma EC.1 quotes.
  • R. Gao, X. Chen and A. J. Kleywegt, Wasserstein Distributionally Robust Optimization and Variation Regularization, Operations Research, 2024; arXiv:1712.06050, source preprint dated 2020-10-30, especially Theorem 1(I), Assumption 5(I), Lemmas EC.2 and EC.9.
10 thms2 active usersReviewed
Bandit AlgorithmsProbabilityStatistics·Captain: mikedeng1

The Multi-Armed Bandit Problem with Covariates I: Successive Elimination Has Expected Regret at Most 392γ²(1 + n/T)(K/Δ)log(TΔ²/18γ²) + nΔ⁻ Under a Random HorizonResearch Paper

Motivation

In a multi-armed bandit, a decision maker repeatedly chooses an arm, observes a reward, and tries to learn which arm has the highest average reward while losing little reward during learning. A policy that requires the number of future pulls in advance cannot be used unchanged when stopping time is uncertain. Perchet and Rigollet study successive elimination (SE), which removes poorly performing arms from a shrinking active set and can run without knowing the realized horizon. Their Theorem 2.1 bounds its expected regret when the horizon is random and independent of rewards Perchet and Rigollet, 2013, Theorem 2.1.

The static-bandit theorem also supplies a concentration and elimination analysis used in the paper's broader study of bandits with covariates. This mission isolates that theorem and the intermediate estimates that make its quantitative constant explicit. The paper proves the result in §2, with a concentration lemma in the Appendix; the goal here is a machine-checked statement and, eventually, a machine-checked proof of that known result.

Setting

There are K+1≥2K+1\ge2K+1≥2 arms. For each arm iii, the reward stack Yi,1,Yi,2,…Y_{i,1},Y_{i,2},\ldotsYi,1​,Yi,2​,… lists the rewards that successive pulls of that arm will reveal. Rewards lie in [0,1][0,1][0,1], are identically distributed along each arm, and have mean fif_ifi​. The stacks are mutually independent. Arms are indexed so that f1≤⋯≤fK+1f_1\le\cdots\le f_{K+1}f1​≤⋯≤fK+1​, and the last arm, denoted ∗*∗, is uniquely best. Its gap from arm iii is Δi=f∗−fi\Delta_i=f_*-f_iΔi​=f∗​−fi​. A gap is zero only for the best arm.

The policy works in rounds. At the start of round τ\tauτ, a nonempty set IτI_\tauIτ​ is active. It pulls each arm in IτI_\tauIτ​ once, in increasing index order, and computes the mean Yˉi(τ)\bar Y_i(\tau)Yˉi​(τ) from that arm's first τ\tauτ rewards. It retains an arm when its mean is within γU(τ,T)\gamma U(\tau,T)γU(τ,T) of the largest active mean, where T≥1T\ge1T≥1 is an integer, γ≥1\gamma\ge1γ≥1, and

log⁡‾(x)=max⁡{log⁡x,1},U(τ,T)=22log⁡‾(T/τ)τ.\overline{\log}(x)=\max\{\log x,1\},\qquad U(\tau,T)=2\sqrt{\frac{2\overline{\log}(T/\tau)}{\tau}}.log​(x)=max{logx,1},U(τ,T)=2τ2log​(T/τ)​​.

Concatenating the rounds gives the pull sequence π^1,π^2,…\widehat\pi_1,\widehat\pi_2,\ldotsπ1​,π2​,…. For a horizon NNN, its regret is RN(π^)=∑t=1NΔπ^tR_N(\widehat\pi)=\sum_{t=1}^{N}\Delta_{\widehat\pi_t}RN​(π)=∑t=1N​Δπt​​. If NNN is random, independent of the reward stacks, and has mean nnn, the paper studies ERN(π^)\mathbb E R_N(\widehat\pi)ERN​(π). For a threshold Δ>0\Delta>0Δ>0, let Δ−\Delta^-Δ− be the largest arm gap strictly smaller than Δ\DeltaΔ, or zero if none exists.

Formalization targets

The goal is Theorem 2.1, for every positive threshold Δ\DeltaΔ:

ERN(π^)≤392γ2(1+nT)KΔ log⁡‾ ⁣(TΔ218γ2)+nΔ−.\mathbb E R_N(\widehat\pi) \le 392\gamma^2\left(1+\frac nT\right) \frac K\Delta\, \overline{\log}\!\left(\frac{T\Delta^2}{18\gamma^2}\right) +n\Delta^-.ERN​(π)≤392γ2(1+Tn​)ΔK​log​(18γ2TΔ2​)+nΔ−.

The KKK in this formula counts suboptimal arms; there are K+1K+1K+1 arms in total. The paper writes Δ≥0\Delta\ge0Δ≥0 under a convention that division by zero is infinite. The real-valued Lean goal states the informative positive-threshold case.

The milestones include the Appendix's maximal Hoeffding–Azuma inequality and Lemma A.1, which controls all empirical averages up to a fixed time. In §2 they continue with the stopping-round estimate (2.2), the empirical-gap tail (2.5), the bound Φj(τ)≤4τ/T\Phi_j(\tau)\le4\tau/TΦj​(τ)≤4τ/T for a false elimination of the best arm, and the final comparison for ϕ(x)=log⁡‾(ax2)/x\phi(x)=\overline{\log}(a x^2)/xϕ(x)=log​(ax2)/x. Each has its own statement so that a proof of the goal can import the exact estimate it needs.

Significance

The bound separates the cost of learning arms with gaps above Δ\DeltaΔ from the remaining regret represented by nΔ−n\Delta^-nΔ−. It applies when the policy uses a tuning value TTT but the number of pulls actually made is the random variable NNN. Thus it gives a quantitative guarantee without asking the algorithm to know its stopping time in advance. Its constants and dependence on TTT, nnn, KKK, and γ\gammaγ are explicit Perchet and Rigollet, 2013, §2.

The result is proved in the paper, while the Lean declarations in this proposal are open statements. Completing them would add a reusable formal account of elimination on a reward stack and of the Appendix's time-uniform concentration estimate. Mathlib and existing platform theorems contain related Hoeffding–Azuma tools, but this paper's peeling inequality, its particular stopping rule, and the random-horizon regret conclusion still require proofs in the stated model.

Difficulty

An ordinary concentration bound at one sample size does not control every round in which an arm could eliminate the best arm. Taking a naive union bound over all rounds loses the dependence on TTT needed for the displayed regret constant. The Appendix instead states a maximal inequality and a bound that holds across a range of sample sizes. The regret calculation must then account for both kinds of error: removing the best arm and retaining a suboptimal arm for too long.

The stopping round depends on the gap through U(τ,T)U(\tau,T)U(τ,T), which itself contains log⁡‾(T/τ)\overline{\log}(T/\tau)log​(T/τ). Its upper bound, the event decomposition across ordered arms, and the final comparison of gap-dependent terms all retain the paper's exact constants. A proof that only establishes the asymptotic rate would not close this goal.

Formalization scope

Lean uses Fin (K+1) with zero-based indices, so its last element is the paper's arm K+1K+1K+1. The published stochastic-bandit definition supplies the mutually independent reward stack and arm means. The [0,1] bound, sorted means, and unique best arm are explicit assumptions. A separate martingale-difference definition supplies the Appendix's adapted, integrable sequence, including centering of its first term; the page's one-line definition leaves that first-step condition unstated. The concentration milestones require a nondegenerate interval a<ba<ba<b: the displayed maximal exponent divides by (b−a)2(b-a)^2(b−a)2, and Lemma A.1 is false at a=b=0a=b=0a=b=0 for a zero sequence and δ<1\delta<1δ<1.

The SE policy is a function of observed rewards in the stack. It uses the end-of-round test described in the text and proof of §2, with equality retained, since Policy 1's pseudocode tests at a different point in the round. The active set starts with all arms and remains nonempty; the total Lean pull function has a fallback value that is unreachable under the theorem's parameters. The policy never reads the unknown means. Empirical means and UUU are used only at positive rounds.

The horizon is measurable and integrable, has mean nnn, and is independent of the entire reward stack. Since regret is bounded by the horizon, its expectation is a genuine integral rather than the default value of a non-integrable Lean integral. The positive-threshold condition avoids Lean's zero-valued division at Δ=0\Delta=0Δ=0. Finite maxima are taken over finite arm sets, and log⁡‾\overline{\log}log​ is kept distinct from the ordinary logarithm in the Appendix's bounds. The paper's printed nnn in the logarithm of (2.7) and the final ϕ\phiϕ comparison is read as TTT, consistently with (2.2) and Theorem 2.1. Contributions to the concentration lemmas, stopping-round analysis, elimination events, and final regret proof all fit this scope.

Selected references

  • V. Perchet and P. Rigollet, The multi-armed bandit problem with covariates, Annals of Statistics 41(2), 2013, 693–721. arXiv:1110.6084v3; DOI:10.1214/13-AOS1101.
  • S. Bubeck and N. Cesa-Bianchi, Regret Analysis of Stochastic and Nonstochastic Multi-armed Bandit Problems, Foundations and Trends in Machine Learning 5(1), 2012. arXiv:1204.5721.
10 thms2 active usersReviewed
🏆Completed
Algebraic TopologyComputational GeometryProbability·Captain: mikedeng1

Persistence-Based Clustering in Riemannian Manifolds: With High Probability the Algorithm Returns as Many Clusters as the Density Has d₂-Prominent PeaksResearch Paper

Motivation

Mode-seeking clustering groups data points by the peak of an underlying probability density that they "flow up" to. Methods of this kind (mean-shift, graph-based hill climbing) are widely used, but their output depends heavily on noise: a density estimated from finitely many samples has many spurious local maxima, and each of them becomes a cluster. Chazal, Guibas, Oudot and Skraba (INRIA RR-6968, 2009; journal version J. ACM 60(6), 2013, doi:10.1145/2535927) combine graph-based hill climbing with topological persistence. Persistence measures how prominent each peak is, and peaks of low prominence are merged into their neighbours. The paper's main guarantee says when this algorithm recovers the right number of clusters from random samples. It holds on Riemannian manifolds that may be non-compact, and the clustering method is now known as ToMATo.

Setting

Let X\mathbb XX be an mmm-dimensional Riemannian manifold, possibly with boundary, with geodesic distance dXd_{\mathbb X}dX​ and mmm-dimensional Hausdorff measure Hm\mathcal H^mHm. The strong convexity radius ϱc(X)\varrho_c(\mathbb X)ϱc​(X) is the infimum over xxx of the largest rrr such that any two points of the closed ball BX(x,r)B_{\mathbb X}(x,r)BX​(x,r) are joined by a unique shortest path, and that path stays in the ball. Let f:X→Rf:\mathbb X\to\mathbb Rf:X→R be a probability density with respect to Hm\mathcal H^mHm that is ccc-Lipschitz, and write Fα=f−1([α,+∞))\mathbb F^\alpha=f^{-1}([\alpha,+\infty))Fα=f−1([α,+∞)) for its superlevel sets.

Persistence diagram. As α\alphaα decreases from +∞+\infty+∞ to −∞-\infty−∞, path components of Fα\mathbb F^\alphaFα are born at peaks of fff and die when they merge into an older component. The 0-th persistence diagram D0fD_0fD0​f records each component as a point (px,py)(p_x,p_y)(px​,py​) of the extended plane R‾2\overline{\mathbb R}^2R2, with birth pxp_xpx​ and death py≤pxp_y\le p_xpy​≤px​; a component that never dies has py=−∞p_y=-\inftypy​=−∞. The prominence of the peak is px−pyp_x-p_ypx​−py​. The diagonal is added with infinite multiplicity. fff is tame when this structure is of finite type. For d2>d1≥0d_2>d_1\ge0d2​>d1​≥0, D0fD_0fD0​f is (d1,d2)(d_1,d_2)(d1​,d2​)-separated when every off-diagonal point either has prominence <d1<d_1<d1​ (noise), or has prominence ≥d2\ge d_2≥d2​ and birth >d2>d_2>d2​ (signal).

Algorithm. The input is nnn sample points x1,…,xnx_1,\dots,x_nx1​,…,xn​, the values f(xi)f(x_i)f(xi​), the matrix of distances dX(xi,xj)d_{\mathbb X}(x_i,x_j)dX​(xi​,xj​) and parameters δ,τ\delta,\tauδ,τ. The Rips graph RδR_\deltaRδ​ joins two points at distance ≤δ\le\delta≤δ. The algorithm processes the points by decreasing fff and keeps a union-find structure whose entries are rooted at their highest point. When a point is processed, every neighbouring entry whose root is less than τ\tauτ higher than the point is merged into the point's entry. The point's own entry is then merged into the neighbouring entry with the highest root, if its own root is less than τ\tauτ higher than the point. The output is the set of final entries whose root has value ≥τ\ge\tau≥τ.

Sampling quantities. Nr(A)\mathcal N_r(A)Nr​(A) is the least number of closed geodesic balls of radius rrr covering AAA, and Vr(A)=inf⁡x∈AHm(BX(x,r))\mathcal V_r(A)=\inf_{x\in A}\mathcal H^m(B_{\mathbb X}(x,r))Vr​(A)=infx∈A​Hm(BX​(x,r)).

Formalization targets

Goal: Theorem 4.8 (p. 22)

Assume ϱc(X)>0\varrho_c(\mathbb X)>0ϱc​(X)>0, fff is a tame ccc-Lipschitz density, D0fD_0fD0​f is (d1,d2)(d_1,d_2)(d1​,d2​)-separated, 0<δ<min⁡{ϱc(X),d2−d15c}0<\delta<\min\{\varrho_c(\mathbb X),\frac{d_2-d_1}{5c}\}0<δ<min{ϱc​(X),5cd2​−d1​​} and τ∈(d1+2cδ, d2−3cδ)\tau\in(d_1+2c\delta,\,d_2-3c\delta)τ∈(d1​+2cδ,d2​−3cδ). Then for nnn i.i.d. samples from fff,

Pr⁡[#clusters≠#{peaks of prominence≥d2}]≤Nδ/8(Fcδ) e−n34cδ Vδ/8(Fcδ).\Pr\big[\#\text{clusters}\ne\#\{\text{peaks of prominence}\ge d_2\}\big]\le\mathcal N_{\delta/8}(\mathbb F^{c\delta})\,e^{-n\frac34c\delta\,\mathcal V_{\delta/8}(\mathbb F^{c\delta})}.Pr[#clusters=#{peaks of prominence≥d2​}]≤Nδ/8​(Fcδ)e−n43​cδVδ/8​(Fcδ).

Milestones

  1. Lemma 4.3 (p. 17): for ε>0\varepsilon>0ε>0 and α>cε\alpha>c\varepsilonα>cε, the sample is a geodesic ε\varepsilonε-sample of Fα\mathbb F^\alphaFα except with probability at most Nε/2(Fα)e−n(α−cε)Vε/2(Fα)\mathcal N_{\varepsilon/2}(\mathbb F^\alpha)e^{-n(\alpha-c\varepsilon)\mathcal V_{\varepsilon/2}(\mathbb F^\alpha)}Nε/2​(Fα)e−n(α−cε)Vε/2​(Fα).
  2. Theorem 4.5 (p. 19): if a finite LLL is a geodesic δ/4\delta/4δ/4-sample of Fα\mathbb F^\alphaFα, there is a multi-bijection between D0fD_0fD0​f and the diagram D0Rδf(L)D_0\mathcal R^f_\delta(L)D0​Rδf​(L) of the upper-star Rips filtration. Above α\alphaα it moves points by at most cδc\deltacδ, and in the lower-right quadrant it moves births by at most cδc\deltacδ.
  3. Noise image (proof of Thm 4.8, p. 22): the noise points are sent into Δd1+2cδN∪Λd1+2cδW\Delta^N_{d_1+2c\delta}\cup\Lambda^W_{d_1+2c\delta}Δd1​+2cδN​∪Λd1​+2cδW​.
  4. Signal image (pp. 22–23): the signal points are sent outside that region.
  5. Partition and count (p. 23): for τ\tauτ in the range, D0Rδf(L)D_0\mathcal R^f_\delta(L)D0​Rδf​(L) has exactly as many points in ΔτS∩ΛτE\Delta^S_\tau\cap\Lambda^E_\tauΔτS​∩ΛτE​ as D0fD_0fD0​f has peaks of prominence ≥d2\ge d_2≥d2​; for the tame D0fD_0fD0​f both counts are finite.
  6. Algorithm output (p. 23, with §3): for τ>0\tau>0τ>0 the algorithm returns exactly one cluster per point qqq of the Rips diagram with qx−qy≥τq_x-q_y\ge\tauqx​−qy​≥τ and qx≥τq_x\ge\tauqx​≥τ.

Significance

The theorem gives a sample-complexity guarantee for a practical clustering algorithm, with an explicit failure probability. It needs neither compactness of the manifold nor a density bounded away from zero; dense sampling of one superlevel set is enough. The separation hypothesis is a signal-to-noise condition, and it tells the user which range of τ\tauτ recovers the true number of clusters.

On the formalization side, the paper's results are proved on paper, but nothing about them is machine-checked. Of the steps, the paper proves Lemma 4.3 and the region analysis. Theorem 4.5 rests on the Rips-filtration analysis of Chazal–Guibas–Oudot–Skraba, SODA 2009, the interleaving and stability results of Chazal et al., SoCG 2009, and the paper's Lemma 4.6. The algorithm-output step (milestone 6) is asserted in the paper, not proved. The mission therefore asks for formal proofs of all six steps. A by-product is the first library of 0-dimensional persistence diagrams, multi-bijections and Rips filtrations on the platform.

Difficulty

The probabilistic step is a union bound over a minimal cover, but it has a gap. Covering balls need not be centred in Fα\mathbb F^\alphaFα, while V\mathcal VV is an infimum over balls centred in Fα\mathbb F^\alphaFα, so the comparison between the two is not justified on a general manifold. Theorem 4.5 is the hardest step. The obvious route, a bottleneck stability theorem, does not apply, because the sample covers only one superlevel set and not X\mathbb XX. This is why the conclusion distinguishes the quadrant above α\alphaα, where both coordinates are controlled, from the quadrant below, where only births are. Milestone 6 is a correctness proof for a union-find procedure with ties: entries are not path components, because prominent entries stay separate after their components meet.

Formalization scope

  • Manifold. X\mathbb XX is a metric space carrying Mathlib's Riemannian manifold structure (IsRiemannianManifold, a ModelWithCorners for the boundary), so dXd_{\mathbb X}dX​ is dist.
  • Measures and probability. Hm\mathcal H^mHm is Mathlib's unnormalized Hausdorff measure of dimension dim⁡E\dim EdimE; the statements are invariant under rescaling it. Samples are product measures on Fin n → X. "With probability at least 1−B1-B1−B" is an upper bound B∈[0,∞]B\in[0,\infty]B∈[0,∞] on the outer measure of the failure event. e−aVe^{-aV}e−aV uses e−∞=0e^{-\infty}=0e−∞=0 and 0⋅∞=00\cdot\infty=00⋅∞=0.
  • Convexity radius. It is computed in [0,+∞][0,+\infty][0,+∞], with shortest paths as isometric embeddings of [0,d(y,y′)][0,d(y,y')][0,d(y,y′)].
  • Diagrams. A diagram's multiplicities are computed from the rank function of the filtration, so they are not free data.
  • Tameness is pinned in a 0-dimensional form: finitely many path components at each level, finitely many critical values, and empty superlevel sets at high levels.
  • Multi-bijections are bijections between copies of points, the diagonal contributing countably many copies of each of its points.
  • Separation is imposed on off-diagonal points only. On the diagonal the literal definition is unsatisfiable when d1=0d_1=0d1​=0.
  • Algorithm. It runs Procedures 1–2 with an explicit union-find state, over every processing order compatible with the sort. The upper star of a point is its set of neighbours processed earlier.
  • Added hypotheses. c>0c>0c>0 in Theorem 4.8 and milestones 3–5 (the paper divides by ccc), and τ>0\tau>0τ>0 in milestone 6.

No formalization in which the number of clusters is defined as a count of diagram points, or in which the diagram is a free parameter, is acceptable: either one empties the goal. The running-time and memory claims of §3, the basin approximation of Theorem 4.9 and the Morse-theoretic material of §2.3 are out of scope. Lemma 4.3 is stated as printed, with covering balls centred anywhere, and its proof gap is an open risk that Theorem 4.8 inherits.

Contributions are welcome on every milestone. The persistence-diagram layer (rank functions, multiplicities, multi-bijections) and the union-find correctness proof are reusable beyond this mission.

Selected references

  • F. Chazal, L. J. Guibas, S. Y. Oudot, P. Skraba, Persistence-Based Clustering in Riemannian Manifolds, INRIA Research Report RR-6968, 2009. https://inria.hal.science/inria-00389390
  • F. Chazal, L. J. Guibas, S. Y. Oudot, P. Skraba, Persistence-Based Clustering in Riemannian Manifolds, Journal of the ACM 60(6), 2013. https://doi.org/10.1145/2535927
  • F. Chazal, D. Cohen-Steiner, M. Glisse, L. J. Guibas, S. Y. Oudot, Proximity of Persistence Modules and their Diagrams, Proc. 25th ACM Symposium on Computational Geometry, 2009. https://doi.org/10.1145/1542362.1542407
  • D. Cohen-Steiner, H. Edelsbrunner, J. Harer, Stability of Persistence Diagrams, Discrete & Computational Geometry 37, 2007. https://doi.org/10.1007/s00454-006-1276-5
  • F. Chazal, L. J. Guibas, S. Y. Oudot, P. Skraba, Analysis of Scalar Fields over Point Cloud Data, SODA 2009. https://doi.org/10.1137/1.9781611973068.112
41 thms2 active usersReviewed
🏆Completed
Bandit AlgorithmsProbability·Captain: mikedeng1

A Tutorial on Thompson Sampling II: Under Thompson Sampling, Per-Period Expected Regret Splits into a Pessimism Term and a Width Term for Any History-Determined U_tResearch Paper

Motivation

An online decision maker repeatedly chooses an action, observes an outcome, and receives a reward. In a Bayesian model, an unknown parameter controls the outcomes, and the decision maker begins with a prior distribution for that parameter. Thompson sampling acts by drawing a parameter from its current posterior distribution and choosing an action that would be best if that draw were correct. This makes it applicable when an agent can simulate a plausible model and optimize an action for that model. Russo, Van Roy, Kazerouni, Osband, and Wen explain this rule for general online decision problems in Section 4 of their tutorial.

The tutorial's Section 8.1.2 asks how to analyze the rule's expected regret. An upper confidence bound method chooses an action using an optimistic score and can analyze regret through the score's optimism and its uncertainty width. Thompson sampling does not choose its actions by maximizing such a score. Yet the tutorial shows that any score computed from the observed past can be used to split Thompson sampling's expected regret into the same two types of terms. The result is stated on pp. 72–73, equation (8.4) and the display that follows.

Setting

Let X\mathcal XX be a finite, nonempty set of actions, YYY a finite set of outcomes, and Θ\ThetaΘ a measurable parameter space. A prior probability measure PPP is placed on the unknown parameter θ∈Θ\theta\in\Thetaθ∈Θ. If action x∈Xx\in\mathcal Xx∈X is selected, the outcome is sampled from a probability kernel qθ(⋅∣x)q_\theta(\cdot\mid x)qθ​(⋅∣x) on YYY. A known function r:Y→Rr:Y\to\mathbb Rr:Y→R turns an outcome into a reward. The mean reward of action xxx under parameter θ\thetaθ is

μ(x,θ)=∑y∈Yqθ(y∣x)r(y).\mu(x,\theta)=\sum_{y\in Y}q_\theta(y\mid x)r(y).μ(x,θ)=y∈Y∑​qθ​(y∣x)r(y).

For each θ\thetaθ, choose an action x∗(θ)x^*(\theta)x∗(θ) maximizing μ(⋅,θ)\mu(\cdot,\theta)μ(⋅,θ). This choice uses a fixed tie-break and is measurable. The same selector is used throughout the mission. A length-ttt history HtH_tHt​ records the first ttt action–outcome pairs. A policy π\piπ assigns an action distribution to every such history. The prior, the policy, and the outcome kernel together determine the law of all histories: first draw θ\thetaθ from PPP, then choose each action from π\piπ using the past history, and finally generate its outcome using qθq_\thetaqθ​.

After history hhh, the posterior law of x∗x^*x∗ is obtained by reweighting the prior by the likelihood of the observed outcomes. Thompson sampling draws a parameter from this posterior and plays the selected optimum for that parameter. Thus the law of its next action, given hhh, is the posterior law of x∗x^*x∗ given hhh. The policy remains defined even for a history with zero likelihood; choices there do not affect expectations.

For a period ttt, let Ut(x)U_t(x)Ut​(x) be any real score that uses the action xxx and the observed history Ht−1H_{t-1}Ht−1​. The score is fixed before the period's outcome or the unknown parameter can be inspected. The period regret is μ(x∗,θ)−μ(xt,θ)\mu(x^*,\theta)-\mu(x_t,\theta)μ(x∗,θ)−μ(xt​,θ), with expectation taken over the prior, the policy's randomization, and the outcome process.

Formalization targets

Probability matching and equation (8.4)

The first target makes posterior sampling precise. For every feasible history hhh and action aaa, the probability that Thompson sampling next plays aaa, together with observing hhh, equals the joint probability that the selected optimum is aaa and the history is hhh. Consequently, for every history-determined UtU_tUt​,

Eπ[Ut(xt)]=Eπ[Ut(x∗)].\mathbb E_\pi[U_t(x_t)] =\mathbb E_\pi[U_t(x^*)].Eπ​[Ut​(xt​)]=Eπ​[Ut​(x∗)].

This is equation (8.4) on p. 72. It asserts equality of unconditional expectations while preserving the fact that UtU_tUt​ may depend on the past. It does not assert that xtx_txt​ and x∗x^*x∗ have the same joint law with θ\thetaθ.

Per-period regret decomposition

The goal is the full chain of equalities displayed on p. 73:

Eπ[μ(x∗,θ)−μ(xt,θ)]=Eπ[μ(x∗,θ)−Ut(xt)]+Eπ[Ut(xt)−μ(xt,θ)]=Eπ[μ(x∗,θ)−Ut(x∗)]⏟pessimism+Eπ[Ut(xt)−μ(xt,θ)]⏟width.\begin{aligned} \mathbb E_\pi[\mu(x^*,\theta)-\mu(x_t,\theta)] &=\mathbb E_\pi[\mu(x^*,\theta)-U_t(x_t)] +\mathbb E_\pi[U_t(x_t)-\mu(x_t,\theta)]\\ &=\underbrace{\mathbb E_\pi[\mu(x^*,\theta)-U_t(x^*)]}_{\text{pessimism}} +\underbrace{\mathbb E_\pi[U_t(x_t)-\mu(x_t,\theta)]}_{\text{width}}. \end{aligned}Eπ​[μ(x∗,θ)−μ(xt​,θ)]​=Eπ​[μ(x∗,θ)−Ut​(xt​)]+Eπ​[Ut​(xt​)−μ(xt​,θ)]=pessimismEπ​[μ(x∗,θ)−Ut​(x∗)]​​+widthEπ​[Ut​(xt​)−μ(xt​,θ)]​​.​

The theorem holds for every score UtU_tUt​ determined by the past history. In particular, UtU_tUt​ need not be a confidence bound for the identity to hold. Being a useful upper confidence bound matters when one later estimates the two terms.

Significance

The equality permits analyses built around optimistic scores to be transferred to Thompson sampling. It separates expected regret into a term measuring whether the score falls below the true optimal mean and another measuring the gap between the score and the true mean at the sampled action. The tutorial cites Russo and Van Roy for the surrounding regret bounds; the identity here is the precise bridge used in that discussion.

The result is already proved in the tutorial. This mission seeks a machine-checked proof of its finite-action, finite-outcome form, including the posterior-sampling law, the policy-dependent history distribution, and the conditioning behind equation (8.4). The goal and milestones are open Lean statements, not completed machine-checked proofs. Their definitions can be reused in later finite Bayesian decision models with a general measurable parameter.

Difficulty

The central issue is that xtx_txt​ and x∗x^*x∗ are not interchangeable inside an expression that depends on θ\thetaθ. The sampled action and the selected optimum have the same distribution after conditioning on the observed past, but generally have different dependence on the true parameter. For example, replacing UtU_tUt​ by μ(⋅,θ)\mu(\cdot,\theta)μ(⋅,θ) would make the claimed equality false. A formal proof therefore has to respect the boundary between history-derived scores and parameter-dependent quantities, and it must use the same history law to interpret every expectation. Histories of zero probability also require a consistent posterior convention.

Formalization scope

The Lean model uses finite action and outcome types and a general measurable parameter type. This is a disclosed specialization of the tutorial, which also allows infinite actions and more general outcome spaces. The kernel qθq_\thetaqθ​ is nonnegative, sums to one over outcomes, and is measurable in θ\thetaθ. The prior is a probability measure. The reward function is real valued and bounded because its domain is finite. The optimal-action selector has measurable fibers and maximizes the mean reward for every parameter. These conditions make the integrands measurable and bounded, avoiding default values of nonintegrable Lean integrals.

Histories are functions from a finite period index to action–outcome pairs. Policies are normalized, nonnegative behavioural action probabilities. The likelihood of a history multiplies both its policy action factors and its kernel outcome factors. The posterior uses only outcome likelihood factors, which give the Bayes update on every positive-evidence history. The Thompson-sampling predicate enforces the posterior action probabilities on those histories and permits any normalized policy on impossible ones. Lean period sss denotes the tutorial's period t=s+1t=s+1t=s+1.

The shared tie-break is essential: Thompson sampling and x∗x^*x∗ use the same selected maximizer. The score has type Ht−1→X→RH_{t-1}\to\mathcal X\to\mathbb RHt−1​→X→R, so it cannot inspect θ\thetaθ or the current outcome. No definition is allowed to supply regret or the decomposition as a predetermined constant; the expectation is built from the prior, policy and outcome kernel. The development needs reusable definitions for the Bayesian model, finite histories, behavioural policies, likelihoods, posterior masses, and the two regret terms. Contributions proving probability matching, equation (8.4), and the final identity are welcome.

Selected references

  • Daniel J. Russo, Benjamin Van Roy, Abbas Kazerouni, Ian Osband, and Zheng Wen, A Tutorial on Thompson Sampling, Foundations and Trends in Machine Learning 11(1), 2018. DOI: 10.1561/2200000070.
4 thms2 active usersReviewed
Bandit Algorithms·Captain: mikedeng1

An Optimal Algorithm for Stochastic and Adversarial Bandits III: α-Tsallis-INF With Gap-Tuned Asymmetric Regularization Has Logarithmic Pseudo-Regret When the Best Arm Is UniqueResearch Paper

Motivation

In the multi-armed bandit problem a learner repeatedly chooses one of KKK actions and observes only the loss of the action it chose. Two models of the environment dominate the literature. In the stochastic model the losses of each arm are drawn i.i.d. from a fixed distribution, and the optimal pseudo-regret grows like ∑i≠i∗log⁡(T)/Δi\sum_{i\ne i^*}\log(T)/\Delta_i∑i=i∗​log(T)/Δi​, where Δi\Delta_iΔi​ is the gap between the mean loss of arm iii and that of the best arm i∗i^*i∗ (Lai and Robbins, 1985). In the adversarial model the losses are arbitrary and the optimal rate is Θ(KT)\Theta(\sqrt{KT})Θ(KT​) (Audibert and Bubeck, 2010). Algorithms that are near-optimal in both models without knowing which one they face are called best of both worlds.

Timeline:

  • Bubeck and Slivkins (2012) gave the first best-of-both-worlds algorithm and asked whether one algorithm could be optimal in both regimes.
  • Seldin and Slivkins (2014) and Auer and Chiang (2016) obtained log⁡2T\log^2 Tlog2T or log⁡3T\log^3 Tlog3T stochastic rates with KT\sqrt{KT}KT​-type adversarial guarantees.
  • Wei and Luo (2018) introduced the stochastically constrained adversarial regime and the log-barrier algorithm BROAD.
  • Zimmert and Seldin (2021) showed that online mirror descent with the 12\tfrac1221​-Tsallis entropy (Tsallis-INF) is optimal in both regimes up to constants. Their Theorem 4, the subject of this mission, studies the whole family of α\alphaα-Tsallis regularizers with gap-dependent tuning in the stochastically constrained regime.

Setting

There are K≥2K\ge2K≥2 arms and rounds t=1,2,…t=1,2,\dotst=1,2,…. In round ttt the learner draws an arm ItI_tIt​ from a probability vector wtw_twt​ in the simplex ΔK−1\Delta^{K-1}ΔK−1, the environment fixes losses ℓt∈[0,1]K\ell_t\in[0,1]^Kℓt​∈[0,1]K, and only ℓt,It\ell_{t,I_t}ℓt,It​​ is observed. The environment may adapt to the past actions I1,…,It−1I_1,\dots,I_{t-1}I1​,…,It−1​ and use its own randomization, modelled by a seed ω\omegaω drawn once from a probability measure μ\muμ. The pseudo-regret is

Reg‾T=E[∑t=1Tℓt,It]−min⁡iE[∑t=1Tℓt,i],\overline{Reg}_T=\mathbb E\Big[\sum_{t=1}^T\ell_{t,I_t}\Big]-\min_i\mathbb E\Big[\sum_{t=1}^T\ell_{t,i}\Big],Reg​T​=E[t=1∑T​ℓt,It​​]−imin​E[t=1∑T​ℓt,i​],

with the expectation over the seed and the learner's draws; a minimizing arm is a best arm in hindsight iT∗i^*_TiT∗​.

α-Tsallis-INF is online mirror descent on the simplex. With α∈(0,1)\alpha\in(0,1)α∈(0,1), regularization parameters ξi>0\xi_i>0ξi​>0 and learning rates ηt>0\eta_t>0ηt​>0, set

Ψ(w)=−∑iwiα−αwiα(1−α)ξi,Ψt=Ψ/ηt.\Psi(w)=-\sum_i\frac{w_i^\alpha-\alpha w_i}{\alpha(1-\alpha)\xi_i},\qquad\Psi_t=\Psi/\eta_t .Ψ(w)=−i∑​α(1−α)ξi​wiα​−αwi​​,Ψt​=Ψ/ηt​.

The learner keeps importance-weighted loss estimates ℓ^t,i=1(It=i)ℓt,i/wt,i\hat\ell_{t,i}=\mathbb 1(I_t=i)\ell_{t,i}/w_{t,i}ℓ^t,i​=1(It​=i)ℓt,i​/wt,i​, their sums L^t\hat L_tL^t​, and plays wt=arg⁡max⁡w∈ΔK−1⟨w,−L^t−1⟩−Ψt(w)w_t=\arg\max_{w\in\Delta^{K-1}}\langle w,-\hat L_{t-1}\rangle-\Psi_t(w)wt​=argmaxw∈ΔK−1​⟨w,−L^t−1​⟩−Ψt​(w). The potential Φt(Y)=max⁡w∈ΔK−1⟨w,Y⟩−Ψt(w)\Phi_t(Y)=\max_{w\in\Delta^{K-1}}\langle w,Y\rangle-\Psi_t(w)Φt​(Y)=maxw∈ΔK−1​⟨w,Y⟩−Ψt​(w) splits the regret into a stability term E[∑tℓt,It+Φt(−L^t)−Φt(−L^t−1)]\mathbb E[\sum_t\ell_{t,I_t}+\Phi_t(-\hat L_t)-\Phi_t(-\hat L_{t-1})]E[∑t​ℓt,It​​+Φt​(−L^t​)−Φt​(−L^t−1​)] and a penalty term E[∑tΦt(−L^t−1)−Φt(−L^t)−ℓt,iT∗]\mathbb E[\sum_t\Phi_t(-\hat L_{t-1})-\Phi_t(-\hat L_t)-\ell_{t,i^*_T}]E[∑t​Φt​(−L^t−1​)−Φt​(−L^t​)−ℓt,iT∗​​].

In the stochastically constrained regime the loss differences have fixed means, E[ℓt,i−ℓt,i∗]=Δi\mathbb E[\ell_{t,i}-\ell_{t,i^*}]=\Delta_iE[ℓt,i​−ℓt,i∗​]=Δi​, with a unique best arm i∗i^*i∗ (Δi>0\Delta_i>0Δi​>0 for i≠i∗i\ne i^*i=i∗). Then Reg‾T=E[∑t∑i≠i∗wt,iΔi]\overline{Reg}_T=\mathbb E[\sum_t\sum_{i\ne i^*}w_{t,i}\Delta_i]Reg​T​=E[∑t​∑i=i∗​wt,i​Δi​] and iT∗=i∗i^*_T=i^*iT∗​=i∗. Write Δmin⁡=min⁡i≠i∗Δi\Delta_{\min}=\min_{i\ne i^*}\Delta_iΔmin​=mini=i∗​Δi​. Theorem 4 uses

ηt=16α4 1−tˉ−1+α(1−α)tα,tˉ=max⁡{e,t},ξi=Δi1−2α (i≠i∗),ξi∗=Δmin⁡1−2α.\eta_t=\frac{16^\alpha}{4}\,\frac{1-\bar t^{-1+\alpha}}{(1-\alpha)t^\alpha},\quad\bar t=\max\{e,t\},\qquad\xi_i=\Delta_i^{1-2\alpha}\ (i\ne i^*),\quad\xi_{i^*}=\Delta_{\min}^{1-2\alpha}.ηt​=416α​(1−α)tα1−tˉ−1+α​,tˉ=max{e,t},ξi​=Δi1−2α​ (i=i∗),ξi∗​=Δmin1−2α​.

Formalization targets

Goal: Theorem 4 (p. 12)

With T0=16Δmin⁡2log⁡216Δmin⁡2T_0=\frac{16}{\Delta_{\min}^2}\log^2\frac{16}{\Delta_{\min}^2}T0​=Δmin2​16​log2Δmin2​16​, for every T≥1T\ge1T≥1,

Reg‾T≤∑i≠i∗(8min⁡{11−α,log⁡T}+64)(log⁡T+1)Δi+16log⁡4(T0)Δmin⁡+4.\overline{Reg}_T\le\sum_{i\ne i^*}\frac{\big(8\min\{\frac1{1-\alpha},\log T\}+64\big)(\log T+1)}{\Delta_i}+\frac{16\log^4(T_0)}{\Delta_{\min}}+4 .Reg​T​≤i=i∗∑​Δi​(8min{1−α1​,logT}+64)(logT+1)​+Δmin​16log4(T0​)​+4.

Milestones

Lemma 17 (Hessian control along a mirror step); Lemma 11, cases 1 and 4 (instantaneous stability); Lemma 20 and Lemma 12, part 2 (penalty); Lemmas 14 and 16 (scalar inequalities for the learning-rate factors); displays (26) and (30) of Appendix D, the stability and penalty bounds under the parameters of Theorem 4.

Lemmas 20 and 12 cover any causal unbiased loss estimator satisfying the bandit feedback assumptions; the other bandit bounds use the importance-weighted estimator.

Significance

The theorem shows that the logarithmic stochastic rate is not special to the 12\tfrac1221​-Tsallis entropy: for every α∈(0,1)\alpha\in(0,1)α∈(0,1), tuning the regularizer to the gaps gives pseudo-regret of order ∑i≠i∗min⁡{11−α,log⁡T}log⁡T/Δi\sum_{i\ne i^*}\min\{\frac1{1-\alpha},\log T\}\log T/\Delta_i∑i=i∗​min{1−α1​,logT}logT/Δi​ in stochastically constrained environments. This places the log-barrier (α→0\alpha\to0α→0) and entropic (α→1\alpha\to1α→1) regularizers of earlier algorithms in one family, and isolates the role of asymmetric regularization. Unlike Theorem 1, it needs the gaps to tune ξ\xiξ, so it is a statement about the reach of the method rather than a practical algorithm.

The result is proved in the paper; no part of it has a machine-checked proof. Formalizing it produces a complete verified analysis of an online-mirror-descent bandit algorithm with time-varying, non-monotone learning rates, including the stability–penalty decomposition, the Hessian control of Lemma 17 and the explicit constant chase of Appendix D. The analysis also checks two points the paper leaves open: the printed bound has log⁡T\log TlogT where its proof gives log⁡T+1\log T+1logT+1, and Appendix D applies Lemma 12, which assumes non-increasing learning rates, to a rate that increases for a few early rounds when α\alphaα is small.

Difficulty

The obvious argument bounds each round's stability by ∑iηtξi2E[wt,i]1−α\sum_i\frac{\eta_t\xi_i}2\mathbb E[w_{t,i}]^{1-\alpha}∑i​2ηt​ξi​​E[wt,i​]1−α (Lemma 11, case 1) and sums. That bound contains the best arm, whose weight tends to one, so it grows polynomially in TTT and cannot give a logarithmic rate. The arm i∗i^*i∗ has to be removed from the stability, which needs a shifted second-order expansion where losses become slightly negative, and that is only controlled once ηtξi≤14\eta_t\xi_i\le\frac14ηt​ξi​≤41​ (Lemma 17). The early rounds before T0T_0T0​ must be paid for separately, and the asymmetric ξ\xiξ and the learning rate interact through constants that must be tracked exactly for the self-bounding step to close.

Formalization scope

Arms are Fin K with K≥2K\ge2K≥2 and the paper's arm iii is i - 1. Rounds start at 1; an action sequence is h : ℕ → Fin K with h 0 unused. The adversary is the published RegretBandits.Adversarial.Adversary (losses in [0,1][0,1][0,1] depending on past actions only) composed with a seed drawn from a probability measure, with losses measurable in the seed. Expectations are explicit: an integral over the seed of a finite sum over action paths weighted by ∏twt,It\prod_t w_{t,I_t}∏t​wt,It​​. The algorithm is an argmax predicate on the weight function, never a choice function. Φt\Phi_tΦt​ is a real supremum over the simplex. Logarithms are natural, and real powers are Real.rpow with nonnegative bases.

The goal takes α∈(0,1)\alpha\in(0,1)α∈(0,1); the endpoints α∈{0,1}\alpha\in\{0,1\}α∈{0,1}, where the paper's learning rate and regularizer are limits, are not stated. The regime "stochastically constrained with a unique best arm" is replaced by the two consequences the proof uses: the self-bounding inequality Reg‾T≥E[∑t∑i≠i∗wt,iΔi]\overline{Reg}_T\ge\mathbb E[\sum_t\sum_{i\ne i^*}w_{t,i}\Delta_i]Reg​T​≥E[∑t​∑i=i∗​wt,i​Δi​] and iT∗=i∗i^*_T=i^*iT∗​=i∗. Every such regime satisfies both, so the formal statement is at least as strong as the printed one. The constant log⁡T+1\log T+1logT+1 in the first term is the one the proof reaches. A trivial formalization is ruled out: the expectation is a genuine probability law, the gaps are tied to the losses through the self-bounding hypothesis, and the weights are determined by the algorithm, so the bound cannot hold vacuously.

A complete development needs the explicit law of a bandit run with an adaptive randomized adversary, unbiasedness of the IW estimator, convex conjugates and Bregman divergences on the simplex, and elementary real analysis for Lemmas 14 and 16. The run-law and the stability–penalty decomposition are reusable for any mirror-descent bandit algorithm. Proofs of any milestone, and of the scalar Lemmas 14 and 16 in particular, are welcome independently.

Selected references

  • J. Zimmert and Y. Seldin, Tsallis-INF: An Optimal Algorithm for Stochastic and Adversarial Bandits, Journal of Machine Learning Research 22(28), 2021. https://arxiv.org/abs/1807.07623 (v6)
  • S. Bubeck and A. Slivkins, The best of both worlds: stochastic and adversarial bandits, COLT 2012. https://arxiv.org/abs/1202.4473
  • C.-Y. Wei and H. Luo, More adaptive algorithms for adversarial bandits, COLT 2018. https://arxiv.org/abs/1801.03265
  • J. Abernethy, C. Lee and A. Tewari, Fighting bandits with a new kind of smoothness, NeurIPS 2015. https://arxiv.org/abs/1512.04152
  • T. L. Lai and H. Robbins, Asymptotically efficient adaptive allocation rules, Advances in Applied Mathematics 6(1), 1985. https://doi.org/10.1016/0196-8858(85)90002-8
  • J.-Y. Audibert and S. Bubeck, Regret bounds and minimax policies under partial monitoring, Journal of Machine Learning Research 11, 2010. https://jmlr.org/papers/v11/audibert10a.html
14 thms2 active usersReviewed
🏆Completed
ProbabilityStatistics·Captain: mikedeng1

Distribution-Free, Risk-Controlling Prediction Sets III: Thresholding the Conditional Expected Loss Gives the Smallest Sets Among Predictors of No Larger Risk (Theorem 8)Research Paper

Motivation

A set-valued predictor returns, for each input xxx, a set of plausible responses rather than a single guess. Such sets are used wherever a model's output feeds a decision with asymmetric costs: a medical classifier that lists every diagnosis it cannot rule out, a segmentation model that marks every pixel that may belong to a tumour, a protein-structure predictor that reports a range of distances. Bates, Angelopoulos, Lei, Malik and Jordan, Distribution-Free, Risk-Controlling Prediction Sets (J. ACM 68(6), 2021), calibrate a nested family of such predictors {Tλ}\{\mathcal T_\lambda\}{Tλ​} so that a chosen loss is controlled with high probability, without distributional assumptions.

Calibration says nothing about which family to calibrate. Any nested family can be made risk-controlling; families differ in how large their sets are, and large sets are uninformative. §4 of the paper asks which family is best in the population: which predictor achieves a given risk with the smallest expected set size. For a weighted miscoverage loss (Theorem 7, §4.2) the answer thresholds a weighted conditional density, in the spirit of the optimality results of Sadinle, Lei and Wasserman (2019) for classification. Theorem 8 (§4.3) extends the answer to every loss of integral form. This mission formalizes Theorem 8.

Setting

Let (X,Y)(X, Y)(X,Y) take values in X×Y\mathcal X \times \mathcal YX×Y with law PPP, marginal PXP_XPX​, and conditional law κ(x)\kappa(x)κ(x) of YYY given X=xX = xX=x. Prediction sets are subsets of a measurable space Z\mathcal ZZ (in the paper's examples Z=Y\mathcal Z = \mathcal YZ=Y). A set-valued predictor is a map T:X→2Z\mathcal T : \mathcal X \to 2^{\mathcal Z}T:X→2Z.

Fix a finite measure μ\muμ on Z\mathcal ZZ and a nonnegative cost ℓ:Y×Z→[0,∞)\ell : \mathcal Y \times \mathcal Z \to [0, \infty)ℓ:Y×Z→[0,∞); ℓ(y,z)\ell(y, z)ℓ(y,z) is the cost of leaving zzz out of the prediction set when the true response is yyy. The loss of a set S\mathcal SS is the total cost of what it leaves out,

L(y;S)=∫z∈Scℓ(y,z) dμ(z),L(y; \mathcal S) = \int_{z \in \mathcal S^c} \ell(y, z)\, d\mu(z),L(y;S)=∫z∈Sc​ℓ(y,z)dμ(z),

and the risk of a predictor is R(T)=E[L(Y;T(X))]R(\mathcal T) = \mathbb E[L(Y; \mathcal T(X))]R(T)=E[L(Y;T(X))] (§2.1). Larger sets have smaller loss. The conditional expected cost of omitting zzz at xxx is

E[ℓ(Y;z)∣X=x]=∫ℓ(y,z) dκ(x)(y),\mathbb E[\ell(Y; z) \mid X = x] = \int \ell(y, z)\, d\kappa(x)(y),E[ℓ(Y;z)∣X=x]=∫ℓ(y,z)dκ(x)(y),

and for λ<0\lambda < 0λ<0 the threshold predictor of display (11) is

Tλ(x)={z∈Z:E[ℓ(Y;z)∣X=x]≥−λ}.\mathcal T_\lambda(x) = \{z \in \mathcal Z : \mathbb E[\ell(Y; z) \mid X = x] \ge -\lambda\}.Tλ​(x)={z∈Z:E[ℓ(Y;z)∣X=x]≥−λ}.

It includes exactly the points whose omission is expected to cost at least −λ-\lambda−λ. The size of a set is its μ\muμ-measure, ∣S∣=μ(S)|\mathcal S| = \mu(\mathcal S)∣S∣=μ(S), and the expected size of a predictor is E[∣T(X)∣]=∫μ(T(x)) dPX(x)\mathbb E[|\mathcal T(X)|] = \int \mu(\mathcal T(x))\, dP_X(x)E[∣T(X)∣]=∫μ(T(x))dPX​(x).

In Lean: setLoss μ ℓ y S, risk P μ ℓ T, condLoss κ ℓ x z, thresholdSet κ ℓ lam x and expSize P μ T in the namespace RiskControl.Optimal.

Formalization targets

Goal: Theorem 8

For every λ<0\lambda < 0λ<0 and every set-valued predictor T′\mathcal T'T′ (with measurable graph),

R(T′)≤R(Tλ)  ⟹  E[∣Tλ(X)∣]≤E[∣T′(X)∣].R(\mathcal T') \le R(\mathcal T_\lambda) \implies \mathbb E\big[|\mathcal T_\lambda(X)|\big] \le \mathbb E\big[|\mathcal T'(X)|\big].R(T′)≤R(Tλ​)⟹E[∣Tλ​(X)∣]≤E[∣T′(X)∣].

No constant is involved; the statement is uniform over ℓ\ellℓ, μ\muμ, PPP and T′\mathcal T'T′.

Milestones: the steps of the proof (p. 27)

  1. Moving the conditional expectation inside the μ\muμ-integral: for a predictor A\mathcal AA with measurable graph, E[∫A(X)ℓ(Y;z) dμ(z)]=E[∫A(X)E[ℓ(Y;z)∣X] dμ(z)]\mathbb E[\int_{\mathcal A(X)} \ell(Y; z)\, d\mu(z)] = \mathbb E[\int_{\mathcal A(X)} \mathbb E[\ell(Y; z) \mid X]\, d\mu(z)]E[∫A(X)​ℓ(Y;z)dμ(z)]=E[∫A(X)​E[ℓ(Y;z)∣X]dμ(z)].
  2. From the risk comparison to the set differences:
E[∫T′(X)∖Tλ(X)E[ℓ(Y;z)∣X] dμ]≥E[∫Tλ(X)∖T′(X)E[ℓ(Y;z)∣X] dμ].\mathbb E\Big[\int_{\mathcal T'(X) \setminus \mathcal T_\lambda(X)} \mathbb E[\ell(Y; z) \mid X]\, d\mu\Big] \ge \mathbb E\Big[\int_{\mathcal T_\lambda(X) \setminus \mathcal T'(X)} \mathbb E[\ell(Y; z) \mid X]\, d\mu\Big].E[∫T′(X)∖Tλ​(X)​E[ℓ(Y;z)∣X]dμ]≥E[∫Tλ​(X)∖T′(X)​E[ℓ(Y;z)∣X]dμ].
  1. From costs to sizes: E∣T′(X)∖Tλ(X)∣≥E∣Tλ(X)∖T′(X)∣\mathbb E|\mathcal T'(X) \setminus \mathcal T_\lambda(X)| \ge \mathbb E|\mathcal T_\lambda(X) \setminus \mathcal T'(X)|E∣T′(X)∖Tλ​(X)∣≥E∣Tλ​(X)∖T′(X)∣.
  2. From set differences to sets: E∣A(X)∖B(X)∣≤E∣B(X)∖A(X)∣\mathbb E|\mathcal A(X) \setminus \mathcal B(X)| \le \mathbb E|\mathcal B(X) \setminus \mathcal A(X)|E∣A(X)∖B(X)∣≤E∣B(X)∖A(X)∣ implies E∣A(X)∣≤E∣B(X)∣\mathbb E|\mathcal A(X)| \le \mathbb E|\mathcal B(X)|E∣A(X)∣≤E∣B(X)∣.

Significance

The result. Theorem 8 identifies the population-optimal shape of a set-valued predictor for every loss that charges for omitted points through a nonnegative cost and a finite measure. It justifies building nested families by thresholding an estimate of the conditional expected cost: if the estimate were exact, no other predictor with the same or smaller risk could have smaller sets on average. It contains the weighted-miscoverage case (Theorem 7) when Y\mathcal YY is discrete and μ\muμ is counting measure, and it covers losses for which no such special case exists, such as costs that depend on the distance between zzz and yyy.

Formalizing it. The proof on p. 27 is a chain of eight displays. A machine-checked version settles the measure-theoretic points the chain passes over: the conditional expectation inside an integral over a random set, the cancellation of possibly infinite terms, and the range of λ\lambdaλ. To our knowledge no formal proof of Theorem 8 or of the Neyman–Pearson-type optimality results it generalizes for set predictors exists in Lean or Mathlib.

Difficulty

The natural reading of the proof subtracts the quantity E∫ZE[ℓ(Y;z)∣X] dμ(z)\mathbb E\int_{\mathcal Z} \mathbb E[\ell(Y; z) \mid X]\, d\mu(z)E∫Z​E[ℓ(Y;z)∣X]dμ(z) from both sides of the risk comparison to pass from complements to sets. Nothing in the setting makes this quantity finite, so that step is not valid as written, and a proof that follows the displays literally breaks there. The theorem itself does not need finiteness of ℓ\ellℓ, and no such hypothesis is in scope.

The other obstacle is measure-theoretic bookkeeping. The conditional expectation E[ℓ(Y;z)∣X]\mathbb E[\ell(Y; z) \mid X]E[ℓ(Y;z)∣X] is indexed by a point zzz of a second space, and must be a jointly measurable function of (x,z)(x, z)(x,z) before it can be integrated over the random set T(X)\mathcal T(X)T(X); here it is given by a regular conditional distribution, and every integral over T(x)\mathcal T(x)T(x) needs the measurability of the graph of T\mathcal TT.

Formalization scope

  • Spaces and measures. X,Y,Z\mathcal X, \mathcal Y, \mathcal ZX,Y,Z are arbitrary measurable spaces; PPP is a probability measure on X×Y\mathcal X \times \mathcal YX×Y; μ\muμ is a finite measure on Z\mathcal ZZ; ℓ\ellℓ takes values in [0,∞)[0, \infty)[0,∞) and is jointly measurable.
  • Conditional law. E[ℓ(Y;z)∣X=x]\mathbb E[\ell(Y; z) \mid X = x]E[ℓ(Y;z)∣X=x] is ∫ℓ(y,z) dκ(x)(y)\int \ell(y, z)\, d\kappa(x)(y)∫ℓ(y,z)dκ(x)(y) for a Markov kernel κ\kappaκ with P=PX⊗κP = P_X \otimes \kappaP=PX​⊗κ, which is exactly the statement that κ\kappaκ is a regular conditional distribution of YYY given XXX. Such a kernel exists, for instance, when Y\mathcal YY is standard Borel; the formalization does not assume this. The threshold predictor is built from the same κ\kappaκ and the same ℓ\ellℓ as the risk; neither is a free object.
  • Integrals. Every integral is a lower Lebesgue integral with values in [0,∞][0, \infty][0,∞]; no integrability hypothesis is made, and an infinite risk is +∞+\infty+∞.
  • Range of λ\lambdaλ (correction of the print). The paper states Theorem 8 for λ∈Λ⊂(−∞,0]\lambda \in \Lambda \subset (-\infty, 0]λ∈Λ⊂(−∞,0]. At λ=0\lambda = 0λ=0 it is false: with ℓ≡0\ell \equiv 0ℓ≡0 and μ(Z)>0\mu(\mathcal Z) > 0μ(Z)>0, T0(x)=Z\mathcal T_0(x) = \mathcal ZT0​(x)=Z and T′≡∅\mathcal T' \equiv \emptysetT′≡∅ both have risk 000, but E∣T′(X)∣=0<μ(Z)\mathbb E|\mathcal T'(X)| = 0 < \mu(\mathcal Z)E∣T′(X)∣=0<μ(Z). The proof divides by −λ-\lambda−λ. Theorem 8 and its milestones are posed for λ<0\lambda < 0λ<0.
  • Set size (correction of the print). p. 13 describes ∣⋅∣|\cdot|∣⋅∣ as Lebesgue or counting measure, a sentence written for Theorem 7. In Theorem 8 the proof measures sets with the measure μ\muμ of the loss, and ∣⋅∣=μ|\cdot| = \mu∣⋅∣=μ throughout.
  • Measurable graphs. The competing predictor T′\mathcal T'T′ is required to have a measurable graph {(x,z):z∈T′(x)}\{(x, z) : z \in \mathcal T'(x)\}{(x,z):z∈T′(x)}; otherwise its risk and expected size are not integrals of measurable functions. The page leaves this implicit. The graph of Tλ\mathcal T_\lambdaTλ​ is measurable by construction.
  • Not a trivialization. The goal is stated about the threshold sets (11) built from the given ℓ\ellℓ, μ\muμ and κ\kappaκ, for every competing predictor; it does not assume any of the intermediate inequalities, and its hypotheses are satisfiable (for instance ℓ≡1\ell \equiv 1ℓ≡1, λ=−1/2\lambda = -1/2λ=−1/2, where Tλ(x)=Z\mathcal T_\lambda(x) = \mathcal ZTλ​(x)=Z).

The development needs Mathlib's composition of a measure with a kernel, Tonelli's theorem for kernels and lower integrals over sections of a measurable set. Lemmas about expected μ\muμ-sizes of random sets (milestone 4) and about moving a kernel integral inside a set integral (milestone 1) are reusable beyond this mission. Proofs of the milestones, and of the goal by other routes, are welcome.

Selected references

  • S. Bates, A. Angelopoulos, L. Lei, J. Malik, M. I. Jordan, Distribution-Free, Risk-Controlling Prediction Sets, Journal of the ACM 68(6), 2021. arXiv:2101.02703v3. https://arxiv.org/abs/2101.02703 — Theorem 8, p. 13; proof p. 27.
  • M. Sadinle, J. Lei, L. Wasserman, Least Ambiguous Set-Valued Classifiers With Bounded Error Levels, Journal of the American Statistical Association 114(525), 2019. https://doi.org/10.1080/01621459.2017.1395341
6 thms2 active usersReviewed
🏆Completed
ProbabilityStatistics·Captain: mikedeng1

Distribution-Free, Risk-Controlling Prediction Sets I: Upper Confidence Bound Calibration Selects a λ̂ Whose Risk Is at Most α with Probability at Least 1 − δ (Theorem 1)Research Paper

Motivation

A trained classifier or regressor returns a point prediction, but many uses of machine learning need a prediction set: a set of plausible labels together with a guarantee about how often, or how badly, it misses the truth. Medical imaging, protein structure prediction and multi-label classification are examples where a user wants a set whose expected loss is bounded, not just a best guess. Bates, Angelopoulos, Lei, Malik and Jordan (arXiv:2101.02703, J. ACM 68(6), 2021) give a procedure that wraps any black-box predictor and returns sets whose risk is controlled, with no assumption on the data distribution beyond i.i.d. calibration data.

The procedure generalizes split conformal prediction (Vovk, Gammerman and Shafer, 2005) and the tolerance regions of Wilks (1941), which control only the probability of missing the label, to an arbitrary monotone loss on sets (false-negative rate, coverage of a segmentation mask, a hierarchical loss). This mission formalizes its central guarantee, Theorem 1 (p. 5), together with the abstract form Theorem A.1 (p. 26) from which the paper derives it.

Setting

Let (X,Y)(X, Y)(X,Y) be a random pair in X×Y\mathcal X \times \mathcal YX×Y with law PPP, and let Z\mathcal ZZ be a set of labels. A set-valued predictor is a map T:X→2Z\mathcal T : \mathcal X \to 2^{\mathcal Z}T:X→2Z. The paper considers a family {Tλ}λ∈Λ\{\mathcal T_\lambda\}_{\lambda \in \Lambda}{Tλ​}λ∈Λ​ indexed by a closed set Λ⊆R∪{±∞}\Lambda \subseteq \mathbb R \cup \{\pm\infty\}Λ⊆R∪{±∞} that is nested:

λ1<λ2  ⟹  Tλ1(x)⊆Tλ2(x).(1)\lambda_1 < \lambda_2 \implies \mathcal T_{\lambda_1}(x) \subseteq \mathcal T_{\lambda_2}(x). \tag{1}λ1​<λ2​⟹Tλ1​​(x)⊆Tλ2​​(x).(1)

A loss on sets L(y,S)≥0L(y, S) \ge 0L(y,S)≥0 is required to decrease as the set grows:

S⊆S′  ⟹  L(y,S)≥L(y,S′).(2)S \subseteq S' \implies L(y, S) \ge L(y, S'). \tag{2}S⊆S′⟹L(y,S)≥L(y,S′).(2)

The risk of Tλ\mathcal T_\lambdaTλ​ is R(λ)=E[L(Y,Tλ(X))]R(\lambda) = \mathbb E[L(Y, \mathcal T_\lambda(X))]R(λ)=E[L(Y,Tλ​(X))], and the section assumes some λmax⁡∈Λ\lambda_{\max} \in \Lambdaλmax​∈Λ with R(λmax⁡)=0R(\lambda_{\max}) = 0R(λmax​)=0.

A risk-controlling prediction set at levels (α,δ)(\alpha, \delta)(α,δ) (Definition 1, p. 2) is a data-dependent predictor T\mathcal TT with R(T)≤αR(\mathcal T) \le \alphaR(T)≤α with probability at least 1−δ1 - \delta1−δ.

Given an i.i.d. calibration sample D=((X1,Y1),…,(Xn,Yn))D = ((X_1, Y_1), \dots, (X_n, Y_n))D=((X1​,Y1​),…,(Xn​,Yn​)), a pointwise upper confidence bound is a function R^+(λ)=R^+(D,λ)\widehat R^+(\lambda) = \widehat R^+(D, \lambda)R+(λ)=R+(D,λ) with

P(R(λ)≤R^+(λ))≥1−δfor each fixed λ.(3)P\big(R(\lambda) \le \widehat R^+(\lambda)\big) \ge 1 - \delta \quad \text{for each fixed } \lambda. \tag{3}P(R(λ)≤R+(λ))≥1−δfor each fixed λ.(3)

UCB calibration selects

λ^=inf⁡{λ∈Λ:R^+(λ′)<α  ∀λ′∈Λ, λ′≥λ}.(4)\hat\lambda = \inf\big\{\lambda \in \Lambda : \widehat R^+(\lambda') < \alpha \ \ \forall \lambda' \in \Lambda,\ \lambda' \ge \lambda\big\}. \tag{4}λ^=inf{λ∈Λ:R+(λ′)<α  ∀λ′∈Λ, λ′≥λ}.(4)

In Lean, the risk is RiskControl.UCB.risk P L T lam, the calibration set of (4) for one realization r of R^+\widehat R^+R+ is calSet Λ r α, and λ^\hat\lambdaλ^ is lambdaHat Λ r α.

Formalization targets

Goal: Theorem 1 (p. 5)

In the setting above, if (3) holds for every λ∈Λ\lambda \in \Lambdaλ∈Λ and RRR is continuous on Λ\LambdaΛ, then

Pn(R(Tλ^)≤α)≥1−δ.P^n\big(R(\mathcal T_{\hat\lambda}) \le \alpha\big) \ge 1 - \delta .Pn(R(Tλ^​)≤α)≥1−δ.

Nothing is assumed of R^+\widehat R^+R+ beyond (3); α\alphaα and δ\deltaδ are arbitrary reals.

Milestones

  1. Monotone risk (§2.1, pp. 4–5): (1) and (2) make RRR nonincreasing on Λ\LambdaΛ.
  2. The failure point (proof of Theorem A.1, p. 26, corrected): λ†=sup⁡{λ∈Λ:R(λ)>α}\lambda^\dagger = \sup\{\lambda \in \Lambda : R(\lambda) > \alpha\}λ†=sup{λ∈Λ:R(λ)>α} lies in Λ\LambdaΛ and R(λ†)≥αR(\lambda^\dagger) \ge \alphaR(λ†)≥α.
  3. Failure implies a low bound at λ†\lambda^\daggerλ† (p. 26, corrected): deterministically, R(λ^)>αR(\hat\lambda) > \alphaR(λ^)>α implies R^+(λ†)<α\widehat R^+(\lambda^\dagger) < \alphaR+(λ†)<α.
  4. Theorem A.1 (p. 26): for any continuous nonincreasing R:Λ→RR : \Lambda \to \mathbb RR:Λ→R and any R^+\widehat R^+R+ satisfying (3) on an abstract probability space, P(R(λ^)≤α)≥1−δP(R(\hat\lambda) \le \alpha) \ge 1 - \deltaP(R(λ^)≤α)≥1−δ.

Significance

Theorem 1 turns a confidence bound for one fixed parameter into a guarantee for a parameter chosen from the data. No uniform convergence over Λ\LambdaΛ is needed, so any concentration inequality for a mean (Hoeffding, Bentkus, the betting bound of Waudby-Smith and Ramdas) yields a calibration procedure immediately; the paper's Theorems 2–5 and 9–10 are of exactly this form, and the later Learn-then-Test and conformal risk control frameworks build on the same template.

The result is proved in the paper; nothing here is open. What the mission adds is a machine-checked version of an argument whose printed form has a gap. The printed proof of Theorem A.1 introduces λ∗=inf⁡{λ∈Λ:R(λ)≤α}\lambda^* = \inf\{\lambda \in \Lambda : R(\lambda) \le \alpha\}λ∗=inf{λ∈Λ:R(λ)≤α} and asserts R(λ∗)=αR(\lambda^*) = \alphaR(λ∗)=α "by continuity", which is false when Λ\LambdaΛ is not an interval. Finite grids, the usual choice of Λ\LambdaΛ in practice, are not intervals. The theorem remains true, and milestones 2 and 3 state the corrected steps. No formalization of risk-controlling prediction sets or of conformal prediction is known to exist in Lean or on this platform.

Difficulty

The obvious argument applies the confidence bound at λ^\hat\lambdaλ^; that fails because λ^\hat\lambdaλ^ depends on the data, so (3) says nothing at λ=λ^\lambda = \hat\lambdaλ=λ^. The proof must find one deterministic point at which a failure of calibration forces a failure of (3). The paper's choice λ∗\lambda^*λ∗ works only when RRR attains the value α\alphaα on Λ\LambdaΛ; on Λ={0,1}\Lambda = \{0, 1\}Λ={0,1} with R(0)=1R(0) = 1R(0)=1, R(1)=0R(1) = 0R(1)=0, α=1/2\alpha = 1/2α=1/2 and the exact bound R^+=R\widehat R^+ = RR+=R, the event {R^+(λ∗)<α}\{\widehat R^+(\lambda^*) < \alpha\}{R+(λ∗)<α} has probability one, so the printed last step does not bound anything. The second difficulty is boundary behaviour: Λ\LambdaΛ may contain ±∞\pm\infty±∞, the infimum in (4) may be over an empty set, and the calibration set need not contain its own infimum.

Formalization scope

  • Λ\LambdaΛ is a closed subset of EReal (the extended reals with the order topology); λ\lambdaλ is only compared, never added. Continuity of RRR is in the subspace topology of Λ\LambdaΛ.
  • Set-valued predictions are Set 𝒵, i.e. Y′=2Z\mathcal Y' = 2^{\mathcal Z}Y′=2Z (the paper uses 2Y2^{\mathcal Y}2Y for most of its examples).
  • The risk is the published definition WassersteinDRO.Duality.nominalRisk, a Bochner integral. Because a non-integrable integrand integrates to 000 in Lean, Theorem 1 assumes (x,y)↦L(y,Tλ(x))(x, y) \mapsto L(y, \mathcal T_\lambda(x))(x,y)↦L(y,Tλ​(x)) is PPP-integrable for every λ∈Λ\lambda \in \Lambdaλ∈Λ.
  • The standing assumptions of §2.1 are hypotheses of Theorem 1: Λ\LambdaΛ closed, (1) on Λ\LambdaΛ, (2), L≥0L \ge 0L≥0, and R(λmax⁡)=0R(\lambda_{\max}) = 0R(λmax​)=0 for some λmax⁡∈Λ\lambda_{\max} \in \Lambdaλmax​∈Λ. The monotonicity of RRR is not assumed in Theorem 1; it is milestone 1. Theorem A.1 assumes it, as printed.
  • The i.i.d. sample is Fin n → 𝒳 × 𝒴 under the product measure PnP^nPn. R^+\widehat R^+R+ is an arbitrary real function of the sample and of λ\lambdaλ; an infinite bound is represented by any value ≥α\ge \alpha≥α.
  • "With probability at least 1−δ1 - \delta1−δ" is written as a failure probability at most δ\deltaδ, measured as an outer measure; no measurability of R^+\widehat R^+R+ or λ^\hat\lambdaλ^ is assumed. For a measurable event this is the page's statement.
  • λ^\hat\lambdaλ^ is an infimum in EReal, so inf⁡∅=+∞\inf \emptyset = +\inftyinf∅=+∞ as on the page. Since +∞+\infty+∞ need not belong to Λ\LambdaΛ, the failure event is "the set in (4) is nonempty and R(λ^)>αR(\hat\lambda) > \alphaR(λ^)>α"; when the set is nonempty, λ^∈Λ\hat\lambda \in \Lambdaλ^∈Λ. When it is empty the procedure certifies nothing, and the natural output is Tλmax⁡\mathcal T_{\lambda_{\max}}Tλmax​​, of risk 000.
  • Corrections of the print: Theorem A.1's conclusion reads P(R(λ)≤α)P(R(\lambda) \le \alpha)P(R(λ)≤α), stated here with λ^\hat\lambdaλ^; the proof's point λ∗\lambda^*λ∗ and the claim R(λ∗)=αR(\lambda^*) = \alphaR(λ∗)=α are replaced by λ†=sup⁡{λ∈Λ:R(λ)>α}\lambda^\dagger = \sup\{\lambda \in \Lambda : R(\lambda) > \alpha\}λ†=sup{λ∈Λ:R(λ)>α} and R(λ†)≥αR(\lambda^\dagger) \ge \alphaR(λ†)≥α, which coincide with the page for an interval Λ\LambdaΛ.
  • A formalization that assumes RRR monotone in Theorem 1, quantifies over R^+\widehat R^+R+ inside the event, takes a real-valued infimum (which is 000 on unbounded sets) or drops the integrability guard would state a different theorem; the statement here does none of these.

Needed infrastructure is small: closed subsets of EReal and their infima and suprema (IsClosed.sInf_mem, IsClosed.sSup_mem), continuity within a set, monotonicity of the Bochner integral, and monotonicity of outer measure. Proofs of the milestones are welcome individually; the deterministic milestones 2 and 3 are reusable for every UCB-calibration theorem of the paper.

Selected references

  • S. Bates, A. Angelopoulos, L. Lei, J. Malik, M. I. Jordan, Distribution-Free, Risk-Controlling Prediction Sets, J. ACM 68(6), 2021; arXiv:2101.02703v3. https://arxiv.org/abs/2101.02703
  • V. Vovk, A. Gammerman, G. Shafer, Algorithmic Learning in a Random World, Springer, 2005. https://doi.org/10.1007/b106715
  • S. S. Wilks, Determination of sample sizes for setting tolerance limits, Ann. Math. Statist. 12(1), 1941. https://doi.org/10.1214/aoms/1177731788
  • I. Waudby-Smith, A. Ramdas, Estimating means of bounded random variables by betting, arXiv:2010.09686, 2020. https://arxiv.org/abs/2010.09686
7 thms2 active usersReviewed
🏆Completed
Linear OptimizationOperations ResearchOptimization·Captain: mikedeng1

Strong Mixed-Integer Programming Formulations for Trained Neural Networks 2: Under Strict Activity Every Inequality of the Exponential Family (6b) Is Facet-DefiningResearch Paper

Motivation

Trained neural networks are increasingly embedded inside optimization models: to verify that a classifier is robust to small input perturbations, to optimize over a learned surrogate of an expensive system, or to choose decisions whose outcome is predicted by a network. When the network uses rectified linear units (ReLU), each neuron y=max⁡{0, w⋅x+b}y=\max\{0,\,w\cdot x+b\}y=max{0,w⋅x+b} is piecewise linear, and the whole network can be written exactly as a mixed-integer program (MIP) with one binary variable per neuron. How fast a branch-and-bound solver closes such a model depends on how tight the linear-programming relaxation of each neuron's formulation is.

Anderson, Huchette, Tjandraatmadja and Vielma (arXiv:1811.08359v2, the IPCO 2019 extended abstract) gave a formulation (6) of a single ReLU neuron over a box that uses only the original variables and one binary variable, and is ideal: its LP relaxation has integral extreme points (Proposition 1, p. 6). The price is an exponential family of inequalities (6b), one for every subset III of the support of www. The present mission formalizes their Proposition 2: each of these inequalities is facet-defining, so no member of the family can be dropped without weakening the relaxation. A longer journal version of the work, with different numbering, appeared later (arXiv:1811.01988); this mission follows the extended abstract.

Setting

Fix η∈N\eta\in\mathbb Nη∈N, a weight vector w∈Rηw\in\mathbb R^\etaw∈Rη, a bias b∈Rb\in\mathbb Rb∈R, and bounds L,U∈RηL,U\in\mathbb R^\etaL,U∈Rη with Li<UiL_i<U_iLi​<Ui​ for every iii (§1.3, p. 4). Write f(x)=w⋅x+bf(x)=w\cdot x+bf(x)=w⋅x+b and [L,U]={x:L≤x≤U}[L,U]=\{x : L\le x\le U\}[L,U]={x:L≤x≤U}. The sign-adjusted bounds are

L˘i={Liwi≥0Uiwi<0,U˘i={Uiwi≥0Liwi<0,\breve L_i=\begin{cases}L_i & w_i\ge 0\\ U_i & w_i<0\end{cases},\qquad \breve U_i=\begin{cases}U_i & w_i\ge 0\\ L_i & w_i<0\end{cases},L˘i​={Li​Ui​​wi​≥0wi​<0​,U˘i​={Ui​Li​​wi​≥0wi​<0​,

so that M+(f)=w⋅U˘+bM^+(f)=w\cdot\breve U+bM+(f)=w⋅U˘+b and M−(f)=w⋅L˘+bM^-(f)=w\cdot\breve L+bM−(f)=w⋅L˘+b are the maximum and minimum of fff over [L,U][L,U][L,U]. The support is supp⁡(w)={i:wi≠0}\operatorname{supp}(w)=\{i : w_i\neq 0\}supp(w)={i:wi​=0}. Strict activity means M−(f)<0<M+(f)M^-(f)<0<M^+(f)M−(f)<0<M+(f): the neuron is neither always off nor always on over the box. The paper assumes it throughout (§1.3).

Formulation (6) (p. 6) consists of the points (x,y,z)(x,y,z)(x,y,z) with

y≥w⋅x+b,(6a)y≤∑i∈Iwi(xi−L˘i(1−z))+(b+∑i∉IwiU˘i)z∀I⊆supp⁡(w),(6b)(x,y,z)∈[L,U]×R≥0×{0,1}.(6c)\begin{aligned} &y\ge w\cdot x+b, &&\text{(6a)}\\ &y\le\sum_{i\in I}w_i\bigl(x_i-\breve L_i(1-z)\bigr)+\Bigl(b+\sum_{i\notin I}w_i\breve U_i\Bigr)z\quad\forall I\subseteq\operatorname{supp}(w), &&\text{(6b)}\\ &(x,y,z)\in[L,U]\times\mathbb R_{\ge0}\times\{0,1\}. &&\text{(6c)} \end{aligned}​y≥w⋅x+b,y≤i∈I∑​wi​(xi​−L˘i​(1−z))+(b+i∈/I∑​wi​U˘i​)z∀I⊆supp(w),(x,y,z)∈[L,U]×R≥0​×{0,1}.​​(6a)(6b)(6c)​

In Lean, form6 w b L U is this set, and relax6 w b L U is its LP relaxation (0≤z≤10\le z\le10≤z≤1 in place of z∈{0,1}z\in\{0,1\}z∈{0,1}). The right-hand side of (6b) for the subset III is rhs6b w b L U I x z.

An inequality g≤0g\le0g≤0 is facet-defining for a set PPP when it holds on PPP, its face F=P∩{g=0}F=P\cap\{g=0\}F=P∩{g=0} is nonempty, and dim⁡F=dim⁡P−1\dim F=\dim P-1dimF=dimP−1, the dimension of a set being that of its affine hull. The paper uses this standard notion without defining it.

Formalization targets

Goal: Proposition 2 (p. 6)

Under L<UL<UL<U and strict activity, for every I⊆supp⁡(w)I\subseteq\operatorname{supp}(w)I⊆supp(w), the inequality (6b) for III is facet-defining for

P=conv⁡{(x,y,z):(x,y,z) satisfies (6a)–(6c)}.P=\operatorname{conv}\{(x,y,z) : (x,y,z)\text{ satisfies (6a)–(6c)}\}.P=conv{(x,y,z):(x,y,z) satisfies (6a)–(6c)}.

The page states the result as "Each inequality in (6b) is facet-defining", and adds right after the proof line: "We require the assumption of strict activity above, as introduced in Section 1.3."

Milestones (App. A.2, p. 15)

  1. For some ε>0\varepsilon>0ε>0, the η+2\eta+2η+2 points p0=(L˘,0,0)p^0=(\breve L,0,0)p0=(L˘,0,0), p1=(U˘,f(U˘),1)p^1=(\breve U,f(\breve U),1)p1=(U˘,f(U˘),1), p~i=(L˘+εσiei,0,0)\tilde p^i=(\breve L+\varepsilon\sigma_ie^i,0,0)p~​i=(L˘+εσi​ei,0,0) for i∉Ii\notin Ii∈/I, and p~i=(U˘−εσiei,f(U˘−εσiei),1)\tilde p^i=(\breve U-\varepsilon\sigma_ie^i,f(\breve U-\varepsilon\sigma_ie^i),1)p~​i=(U˘−εσi​ei,f(U˘−εσi​ei),1) for i∈Ii\in Ii∈I are feasible with respect to (6) and satisfy (6b) for III at equality; here σi=±1\sigma_i=\pm1σi​=±1 is the sign of wiw_iwi​ (with σi=1\sigma_i=1σi​=1 when wi=0w_i=0wi​=0).
  2. For every ε>0\varepsilon>0ε>0 these η+2\eta+2η+2 points are affinely independent.

Significance

The result. Proposition 1 shows that (6) is ideal; Proposition 2 shows it cannot be made smaller: removing any single inequality (6b) produces a strictly weaker relaxation. Since the family has 2∣supp⁡(w)∣2^{|\operatorname{supp}(w)|}2∣supp(w)∣ members, this is what justifies the paper's practical recommendation to start from the big-MMM formulation and separate inequalities of (6b) on demand (Proposition 3, p. 7) rather than to search for a smaller ideal description in the same variables. The facet structure also gives the geometric picture the paper describes after Proposition 2: each facet is the convex combination of an (η−∣I∣)(\eta-|I|)(η−∣I∣)-dimensional face at z=0z=0z=0 and an ∣I∣|I|∣I∣-dimensional face at z=1z=1z=1.

Formalizing it. The result is proved in the paper, in a half-page appendix; it has not been machine-checked. The formalization adds two things. First, the proof is written for w≥0w\ge0w≥0 "without loss of generality by appropriately interchanging +++ and −-−"; the Lean statements are for every sign pattern, including zero weights. Second, the appendix exhibits η+2\eta+2η+2 affinely independent points on the face, which bounds the face dimension from below; the statement that the face has dimension exactly one less than the polyhedron also needs the polyhedron to be full-dimensional and the face to lie in a proper hyperplane, steps the extended abstract leaves implicit and a complete proof must supply.

Difficulty

The arithmetic in each step is elementary. The work lies in the bookkeeping: choosing a single ε\varepsilonε that keeps every perturbed point inside the box and on the correct side of f=0f=0f=0 (this is exactly where strict activity enters), checking the perturbed points against all 2∣supp⁡(w)∣2^{|\operatorname{supp}(w)|}2∣supp(w)∣ inequalities of (6b) and not only the one for III, and turning a row-reduction argument on an (η+1)×(η+2)(\eta+1)\times(\eta+2)(η+1)×(η+2) matrix into a statement about AffineIndependent and finrank of a vectorSpan in Lean. The natural shortcut, proving only that η+2\eta+2η+2 affinely independent tight points exist, is not Proposition 2: it says nothing about the dimension of the polyhedron itself.

Formalization scope

Inputs are Fin η → ℝ (indices 0,…,η−10,\dots,\eta-10,…,η−1 for the paper's 1,…,η1,\dots,\eta1,…,η); a point (x,y,z)(x,y,z)(x,y,z) is p : (Fin η → ℝ) × ℝ × ℝ with p.1 = x, p.2.1 = y, p.2.2 = z. M±(f)M^\pm(f)M±(f) are given by their closed forms w⋅U˘+bw\cdot\breve U+bw⋅U˘+b and w⋅L˘+bw\cdot\breve L+bw⋅L˘+b. In (6b), "i∉Ii\notin Ii∈/I" ranges over all indices outside III, zero weights included; III ranges over subsets of supp⁡(w)\operatorname{supp}(w)supp(w), as on the page.

Every goal and milestone keeps the standing assumptions of §1.3, Li<UiL_i<U_iLi​<Ui​ for all iii and strict activity, except the affine-independence milestone, which holds without them and is stated without them (a stronger statement). No other hypothesis is added. Strict activity excludes η=0\eta=0η=0, so no nonemptiness assumption on the index set is needed.

IsFacetDefining P g is the standard notion: validity, a nonempty face, and dim⁡F+1=dim⁡P\dim F+1=\dim PdimF+1=dimP with dimensions the finrank of the vectorSpan. The equation is written with +1+1+1 on the left so that no natural-number subtraction can make the empty set or a point a facet. The polyhedron is the convex hull of the points of (6), not the set form6 itself (which is not convex, since z∈{0,1}z\in\{0,1\}z∈{0,1}); by Proposition 1 it equals the LP relaxation relax6, but the statement does not depend on that.

The shared objects of this paper (the ReLU graph, L˘\breve LL˘, U˘\breve UU˘, M±M^\pmM±, support, strict activity, formulation (6), ideality) come from the shared definitions module ReluMIP.Ideal.Setting, common to the companion mission on Proposition 1; this mission's own definitions module adds form6, IsFacetDefining, the sign inward, and the indexed family facetPts of the η+2\eta+2η+2 points. The facet notion and the affine-independence argument are reusable for other facet proofs of polyhedra in product spaces. Proofs of the milestones, of the goal, and of the full-dimensionality step are all welcome.

Selected references

  • R. Anderson, J. Huchette, C. Tjandraatmadja, J. P. Vielma, Strong mixed-integer programming formulations for trained neural networks, IPCO 2019, LNCS 11480, pp. 27–42; preprint arXiv:1811.08359v2, 2019. https://arxiv.org/abs/1811.08359v2
  • R. Anderson, J. Huchette, W. Ma, C. Tjandraatmadja, J. P. Vielma, Strong mixed-integer programming formulations for trained neural networks, Mathematical Programming 183 (2020), 3–39. https://doi.org/10.1007/s10107-020-01474-5
  • G. L. Nemhauser, L. A. Wolsey, Integer and Combinatorial Optimization, Wiley, 1988. https://doi.org/10.1002/9781118627372
  • M. Conforti, G. Cornuéjols, G. Zambelli, Integer Programming, Springer, 2014. https://doi.org/10.1007/978-3-319-11008-0
  • J. P. Vielma, Mixed integer linear programming formulation techniques, SIAM Review 57 (2015), 3–57. https://doi.org/10.1137/130915303
5 thms2 active usersReviewed
Bandit Algorithms·Captain: mikedeng1

An Optimal Algorithm for Stochastic and Adversarial Bandits II: α-Tsallis-INF With Symmetric Regularization Has Anytime Adversarial Pseudo-Regret 2√(min{1/(α−α²), log K/α, log T/(1−α)}·KT) + 1Research Paper

Motivation

In the adversarial multi-armed bandit problem a learner repeatedly picks one of KKK arms and observes only the loss of the arm it picked, while an adversary chooses the losses, possibly reacting to the learner's past choices. The minimax pseudo-regret of this problem is of order KT\sqrt{KT}KT​ after TTT rounds, and the classical algorithm Exp3 achieves only KTlog⁡K\sqrt{KT\log K}KTlogK​ (Auer et al., 2002). Online mirror descent with a Tsallis-entropy regularizer interpolates between Exp3 (negative Shannon entropy) and the log-barrier, and the choice of the Tsallis parameter α\alphaα decides which logarithmic factor appears in the bound.

Timeline:

  • 2002: Auer, Cesa-Bianchi, Freund and Schapire, Exp3, pseudo-regret O(KTlog⁡K)O(\sqrt{KT\log K})O(KTlogK​).
  • 2009: Audibert and Bubeck, the Poly-INF algorithm, the first O(KT)O(\sqrt{KT})O(KT​) adversarial bound (COLT 2009).
  • 2015: Abernethy, Lee and Tewari analyse gradient-based prediction with Tsallis regularization for α∈(0,1]\alpha\in(0,1]α∈(0,1] with a horizon-dependent learning rate (NeurIPS 2015).
  • 2017: Agarwal, Luo, Neyshabur and Schapire analyse the log-barrier (α=0\alpha=0α=0) (COLT 2017).
  • 2019/2021: Zimmert and Seldin give a single anytime analysis covering every α∈[0,1]\alpha\in[0,1]α∈[0,1], Theorem 3 of arXiv:1807.07623v6 (JMLR 22(28), 2021).

Setting

There are K≥1K\ge1K≥1 arms and rounds t=1,2,…t=1,2,\dotst=1,2,…. At round ttt the learner draws an arm ItI_tIt​ from a probability vector wtw_twt​ in the simplex ΔK−1\Delta^{K-1}ΔK−1, the adversary fixes a loss vector ℓt∈[0,1]K\ell_t\in[0,1]^Kℓt​∈[0,1]K, which may depend on I1,…,It−1I_1,\dots,I_{t-1}I1​,…,It−1​ and on the adversary's internal randomization, and the learner observes ℓt,It\ell_{t,I_t}ℓt,It​​. The pseudo-regret is

Reg‾T=E[∑t=1Tℓt,It]−min⁡iE[∑t=1Tℓt,i],\overline{Reg}_T=\mathbb E\Big[\sum_{t=1}^T\ell_{t,I_t}\Big]-\min_i\mathbb E\Big[\sum_{t=1}^T\ell_{t,i}\Big],Reg​T​=E[t=1∑T​ℓt,It​​]−imin​E[t=1∑T​ℓt,i​],

with the expectation over both sources of randomness.

The importance-weighted (IW) estimator is ℓ^t,i=1(It=i) ℓt,i/wt,i\hat\ell_{t,i}=\mathbb 1(I_t=i)\,\ell_{t,i}/w_{t,i}ℓ^t,i​=1(It​=i)ℓt,i​/wt,i​ and L^t=∑s≤tℓ^s\hat L_{t}=\sum_{s\le t}\hat\ell_sL^t​=∑s≤t​ℓ^s​. For α∈(0,1)\alpha\in(0,1)α∈(0,1) and ξi>0\xi_i>0ξi​>0 the α-Tsallis regularizer is

Ψ(w)=−∑iwiα−αwiα(1−α)ξi,Ψt=Ψ/ηt,\Psi(w)=-\sum_i\frac{w_i^\alpha-\alpha w_i}{\alpha(1-\alpha)\xi_i},\qquad\Psi_t=\Psi/\eta_t ,Ψ(w)=−i∑​α(1−α)ξi​wiα​−αwi​​,Ψt​=Ψ/ηt​,

with symmetric regularization ξi=1\xi_i=1ξi​=1. α-Tsallis-INF is online mirror descent:

wt=arg⁡max⁡w∈ΔK−1 ⟨w,−L^t−1⟩−Ψt(w),w_t=\arg\max_{w\in\Delta^{K-1}}\ \langle w,-\hat L_{t-1}\rangle-\Psi_t(w),wt​=argw∈ΔK−1max​ ⟨w,−L^t−1​⟩−Ψt​(w),

and the potential is Φt(Y)=max⁡w∈ΔK−1⟨w,Y⟩−Ψt(w)\Phi_t(Y)=\max_{w\in\Delta^{K-1}}\langle w,Y\rangle-\Psi_t(w)Φt​(Y)=maxw∈ΔK−1​⟨w,Y⟩−Ψt​(w). The pseudo-regret splits into a stability term E[∑tℓt,It+Φt(−L^t)−Φt(−L^t−1)]\mathbb E[\sum_t\ell_{t,I_t}+\Phi_t(-\hat L_t)-\Phi_t(-\hat L_{t-1})]E[∑t​ℓt,It​​+Φt​(−L^t​)−Φt​(−L^t−1​)] and a penalty term E[∑tΦt(−L^t−1)−Φt(−L^t)−ℓt,iT∗]\mathbb E[\sum_t\Phi_t(-\hat L_{t-1})-\Phi_t(-\hat L_t)-\ell_{t,i^*_T}]E[∑t​Φt​(−L^t−1​)−Φt​(−L^t​)−ℓt,iT∗​​], where iT∗i^*_TiT∗​ is a best arm in expectation in hindsight. Theorem 3 uses the learning rate

ηt=K1−2α−K−α1−α⋅1−t−ααt.\eta_t=\sqrt{\frac{K^{1-2\alpha}-K^{-\alpha}}{1-\alpha}\cdot\frac{1-t^{-\alpha}}{\alpha t}} .ηt​=1−αK1−2α−K−α​⋅αt1−t−α​​.

Formalization targets

Goal: Theorem 3, α∈(0,1)\alpha\in(0,1)α∈(0,1)

Reg‾T≤2min⁡{1α−α2,log⁡Kα,log⁡T1−α}KT+1for every T≥1.\overline{Reg}_T\le2\sqrt{\min\Big\{\frac{1}{\alpha-\alpha^2},\frac{\log K}{\alpha},\frac{\log T}{1-\alpha}\Big\}KT}+1\qquad\text{for every }T\ge1 .Reg​T​≤2min{α−α21​,αlogK​,1−αlogT​}KT​+1for every T≥1.

The goal fixes the constants of the paper; it holds against every randomized adaptive adversary.

Milestones

  1. Lemma 11 part 1: the per-round stability is at most min⁡{∑iηtξi2E[wt,i]1−α,1}\min\{\sum_i\frac{\eta_t\xi_i}{2}\mathbb E[w_{t,i}]^{1-\alpha},1\}min{∑i​2ηt​ξi​​E[wt,i​]1−α,1}.
  2. The maximum of ∑izi1−α\sum_i z_i^{1-\alpha}∑i​zi1−α​ over the simplex is KαK^\alphaKα.
  3. Lemma 20: the penalty is bounded by increments of the inverse learning rate times differences of Ψ\PsiΨ, plus ⟨u−eiT∗,LT⟩\langle u-e_{i^*_T},L_T\rangle⟨u−eiT∗​​,LT​⟩.
  4. Lemma 12 part 1: for non-increasing positive learning rates the penalty is at most (K1−α−1)(1−T−α)(1−α)αηT+1\frac{(K^{1-\alpha}-1)(1-T^{-\alpha})}{(1-\alpha)\alpha\eta_T}+1(1−α)αηT​(K1−α−1)(1−T−α)​+1.
  5. Lemma 14: (1−y−x)/x(1-y^{-x})/x(1−y−x)/x is non-increasing on x>0x>0x>0, tends to log⁡y\log ylogy, and is at most min⁡{x−1,log⁡y}\min\{x^{-1},\log y\}min{x−1,logy}.

Companions

Theorem 3 at the boundaries: α=1\alpha=1α=1 (negative-entropy regularizer, rate lim⁡α→1ηt=log⁡(K)(1−t−1)/(Kt)\lim_{\alpha\to1}\eta_t=\sqrt{\log(K)(1-t^{-1})/(Kt)}limα→1​ηt​=log(K)(1−t−1)/(Kt)​, bound 2log⁡(K)KT+12\sqrt{\log(K)KT}+12log(K)KT​+1) and α=0\alpha=0α=0 (log-barrier, rate (K−1)log⁡(t)/t\sqrt{(K-1)\log(t)/t}(K−1)log(t)/t​, bound 2log⁡(T)KT+12\sqrt{\log(T)KT}+12log(T)KT​+1).

Significance

Theorem 3 gives one anytime bound for the whole Tsallis family. At α=12\alpha=\tfrac12α=21​ it is O(KT)O(\sqrt{KT})O(KT​), the minimax rate, without knowledge of TTT; at α→1\alpha\to1α→1 it recovers the Exp3 rate and at α→0\alpha\to0α→0 the log-barrier rate, with constants matching Abernethy et al. without tuning the learning rate to the horizon. It is the adversarial half of the paper's study of α-Tsallis-INF, whose stochastic half (Theorem 4) shows why α=12\alpha=\tfrac12α=21​ is the only value that is optimal in both regimes.

The result is proved in the paper. As far as is known, no machine-checked proof of any Tsallis-INF or Poly-INF regret bound exists. The mission produces a formal model of randomized adaptive adversaries together with online mirror descent for bandits, and formal stability and penalty lemmas for the Tsallis family, which are the standard building blocks of later best-of-both-worlds analyses.

Difficulty

The obvious route, bounding the regret of each fixed seed and action sequence and averaging, fails: the IW estimates have second moments of order 1/wt,i1/w_{t,i}1/wt,i​, which are unbounded, and only their expectation under the learner's own sampling is controlled. The stability term therefore has to be bounded in expectation, for a regularizer whose convex conjugate has no closed form for general α\alphaα. The penalty term involves a learning rate that changes every round, so the potentials of consecutive rounds are not directly comparable. The expectation of the run is over a law that couples the adversary's seed with the learner's random actions, so every step that the paper writes as "take expectations" has to be carried out over that law.

Two features of the printed proof need care. Theorem 3's learning rate is 000 at t=1t=1t=1, where Ψ1=Ψ/η1\Psi_1=\Psi/\eta_1Ψ1​=Ψ/η1​ is undefined, and for small α\alphaα it increases between t=2t=2t=2 and t=3t=3t=3, while Lemma 12 part 1 assumes a non-increasing sequence of positive learning rates. A complete proof of the goal has to handle these first rounds.

Formalization scope

Arms are Fin K with K≥1K\ge1K≥1; action sequences are maps h:N→h:\mathbb N\toh:N→ Fin K with h(t)=Ith(t)=I_th(t)=It​ and h(0)h(0)h(0) unused. The adversary is a seed ω\omegaω drawn from a probability measure μ\muμ together with one published RegretBandits.Adversarial.Adversary K per seed (losses in [0,1][0,1][0,1], depending on past actions only), with losses measurable in the seed. The expectation of a quantity at horizon TTT is ∫∑a(∏t≤Twt,at)F dμ\int\sum_{a}(\prod_{t\le T}w_{t,a_t})F\,d\mu∫∑a​(∏t≤T​wt,at​​)Fdμ. The learner's weights are an argmax predicate on a weight function: for every round, the objective ηt⟨w,−L^t−1⟩−Ψ(w)\eta_t\langle w,-\hat L_{t-1}\rangle-\Psi(w)ηt​⟨w,−L^t−1​⟩−Ψ(w) is maximized over the simplex. This is the paper's rule whenever ηt>0\eta_t>0ηt​>0, and at η1=0\eta_1=0η1​=0 it makes w1w_1w1​ the minimizer of Ψ\PsiΨ, the uniform vector, which is the initialisation of online mirror descent. No hypothesis on the weights is added beyond this rule. Φt\Phi_tΦt​ is a real supremum over the simplex, attained because the simplex is compact and nonempty. Powers are real powers; log⁡\loglog is the natural logarithm.

The goal and the milestones take α∈(0,1)\alpha\in(0,1)α∈(0,1) (the page: α∈[0,1]\alpha\in[0,1]α∈[0,1]); the boundary values are separate companion theorems, with the limiting regularizers of §3.2 and the limiting learning rates of p. 12, and the log-barrier companion restricts the weights to the open simplex. Lemmas 12 and 20 are stated, as on the page, for any loss estimator that is conditionally unbiased given the seed and the past, depends on the actions through its own round only, and is measurable and integrable under the run law. The page prints the α→1\alpha\to1α→1 limit of the learning rate as log⁡(K)(1−t−1)/t\sqrt{\log(K)(1-t^{-1})/t}log(K)(1−t−1)/t​; the true limit carries an extra factor 1/K1/K1/K under the root, and the α=1\alpha=1α=1 companion uses the true limit. In Lemma 20 the term ⟨u−eiT∗,LT⟩\langle u-e_{i^*_T},L_T\rangle⟨u−eiT∗​​,LT​⟩ sits inside the expectation, since LTL_TLT​ is random under an adaptive adversary.

A trivializing formalization is ruled out: the weights are not chosen by a choice function, and an algorithm whose weights are arbitrary at η1=0\eta_1=0η1​=0 is excluded because the predicate pins w1w_1w1​.

A complete development needs: convex conjugates of separable regularizers on the simplex, the stability argument via a second-order bound, telescoping of potentials, unbiasedness of IW estimators under the run law, and elementary real-analysis estimates (Lemma 14, the simplex maximum). The run law, the IW estimator and the potential are reusable in the companion missions on Theorem 1 and Theorem 4. Contributions to any milestone, and to reusable lemmas about the run law, are welcome.

Selected references

  • J. Zimmert, Y. Seldin, Tsallis-INF: An Optimal Algorithm for Stochastic and Adversarial Bandits, J. Mach. Learn. Res. 22(28), 2021; arXiv:1807.07623v6. https://arxiv.org/abs/1807.07623
  • J. Abernethy, C. Lee, A. Tewari, Fighting Bandits with a New Kind of Smoothness, NeurIPS 2015. https://arxiv.org/abs/1505.00340
  • A. Agarwal, H. Luo, B. Neyshabur, R. E. Schapire, Corralling a Band of Bandit Algorithms, COLT 2017. https://arxiv.org/abs/1612.06246
  • J.-Y. Audibert, S. Bubeck, Minimax Policies for Adversarial and Stochastic Bandits, COLT 2009. https://www.learningtheory.org/colt2009/papers/024.pdf
  • P. Auer, N. Cesa-Bianchi, Y. Freund, R. E. Schapire, The Nonstochastic Multiarmed Bandit Problem, SIAM J. Comput. 32(1), 2002. https://doi.org/10.1137/S0097539701398375
9 thms2 active usersReviewed
Optimal TransportOptimization·Captain: mikedeng1

Computational Optimal Transport XII: The Entropic Barycenter Scalings Are (e^{f_s/ε}, e^{g_s/ε}) for the Solutions of a Dual Program with Σ_s λ_s f_s = 0Textbook

Why barycenters of histograms matter

A barycenter summarizes several objects by minimizing a weighted sum of distances to them. For probability histograms, ordinary coordinatewise averages depend strongly on how the bins are labeled. Optimal transport instead lets mass move between bins and charges for those moves through a cost matrix. The resulting Wasserstein barycenter can reflect the geometry of the bins: nearby bins can exchange mass at a lower cost than distant ones. Peyré and Cuturi use this construction in their treatment of variational transport problems, including shape interpolation and the representation of a measure as a barycenter of other measures. The finite histogram problem in this mission is the entropically regularized version developed in §9.2 of their textbook.

The computational question is how to connect three ways of describing the same solution: a histogram that is the barycenter, transport matrices coupling it to the inputs, and vectors of dual potentials. Proposition 9.1 on p. 530 identifies the dual potentials behind the scaling factors of the optimal transport matrices. This is the second proposition numbered 9.1 in the chapter; the one on p. 519 concerns derivatives with respect to histograms.

Finite histogram setting

There are SSS input histograms. Source sss has nsn_sns​ bins and a probability vector bsb_sbs​, while the barycenter has nnn bins. The real matrix Cs∈Rn×nsC_s\in\mathbb R^{n\times n_s}Cs​∈Rn×ns​ gives the cost of transporting a unit of mass from barycenter bin iii to input bin jjj. A weight vector λ∈ΣS\lambda\in\Sigma_Sλ∈ΣS​ has nonnegative entries summing to one; Σk\Sigma_kΣk​ denotes the probability simplex on kkk bins. The unregularized barycenter problem (9.10) minimizes ∑sλsLCs(a,bs)\sum_s\lambda_s L_{C_s}(a,b_s)∑s​λs​LCs​​(a,bs​) over a∈Σna\in\Sigma_na∈Σn​, where LCsL_{C_s}LCs​​ is the minimum transport cost with marginals aaa and bsb_sbs​.

For a regularization parameter ε>0\varepsilon>0ε>0, the entropic transport cost LCsε(a,bs)L^\varepsilon_{C_s}(a,b_s)LCs​ε​(a,bs​) minimizes ⟨Cs,Ps⟩−εH(Ps)\langle C_s,P_s\rangle-\varepsilon H(P_s)⟨Cs​,Ps​⟩−εH(Ps​) over nonnegative coupling matrices PsP_sPs​ with row marginal aaa and column marginal bsb_sbs​. Here ⟨Cs,Ps⟩=∑i,jCs,ijPs,ij\langle C_s,P_s\rangle=\sum_{i,j}C_{s,ij}P_{s,ij}⟨Cs​,Ps​⟩=∑i,j​Cs,ij​Ps,ij​ and H(Ps)=∑i,j(−Ps,ijlog⁡Ps,ij+Ps,ij)H(P_s)=\sum_{i,j}(-P_{s,ij}\log P_{s,ij}+P_{s,ij})H(Ps​)=∑i,j​(−Ps,ij​logPs,ij​+Ps,ij​), with 0log⁡0=00\log0=00log0=0. The Gibbs kernel is Ks,ij=e−Cs,ij/εK_{s,ij}=e^{-C_{s,ij}/\varepsilon}Ks,ij​=e−Cs,ij​/ε.

The book also writes the regularized barycenter as a weighted KL projection. Its variables are the matrices (Ps)s(P_s)_s(Ps​)s​; each matrix has column marginal bsb_sbs​, and every matrix has the same row marginal. That shared row marginal is aaa, so the optimization can be expressed without carrying aaa as a separate variable. The generalized matrix divergence is KL(Ps∣Ks)=∑i,j[Ps,ijlog⁡(Ps,ij/Ks,ij)−Ps,ij+Ks,ij]\mathrm{KL}(P_s\mid K_s)=\sum_{i,j}[P_{s,ij}\log(P_{s,ij}/K_{s,ij})-P_{s,ij}+K_{s,ij}]KL(Ps​∣Ks​)=∑i,j​[Ps,ij​log(Ps,ij​/Ks,ij​)−Ps,ij​+Ks,ij​], using the continuous value at Ps,ij=0P_{s,ij}=0Ps,ij​=0. See (9.15)–(9.17) in Peyré and Cuturi.

Formalization targets

The goal is Proposition 9.1 on p. 530. The dual variables are row potentials fs∈Rnf_s\in\mathbb R^nfs​∈Rn and column potentials gs∈Rnsg_s\in\mathbb R^{n_s}gs​∈Rns​. They maximize

∑sλs[⟨gs,bs⟩−ε∑i,jefs,i/εKs,ijegs,j/ε]subject to∑sλsfs=0.\sum_s\lambda_s\left[\langle g_s,b_s\rangle-\varepsilon\sum_{i,j}e^{f_{s,i}/\varepsilon}K_{s,ij}e^{g_{s,j}/\varepsilon}\right] \quad\text{subject to}\quad \sum_s\lambda_s f_s=0.s∑​λs​[⟨gs​,bs​⟩−εi,j∑​efs,i​/εKs,ij​egs,j​/ε]subject tos∑​λs​fs​=0.

The theorem asserts that the primal and dual optima are attained, that every optimal primal family and every optimal dual family satisfy

Ps,ij=efs,i/εKs,ijegs,j/ε,P_{s,ij}=e^{f_{s,i}/\varepsilon}K_{s,ij}e^{g_{s,j}/\varepsilon},Ps,ij​=efs,i​/εKs,ij​egs,j​/ε,

and that the dual maximum is the minimum value of the entropic barycenter objective (9.15). The milestone targets are the scalar and matrix versions of the KL conjugate (9.22), followed by the closed form maximizer of one gsg_sgs​ block, corresponding to the update (9.18). These statements retain the source's finite matrix setting and are listed in the order in which their concepts enter the dual formulation.

What the result supplies

The scaling formula turns the optimal couplings into a product of a row factor, a fixed positive kernel, and a column factor. It identifies those factors with exponentials of dual potentials and gives a constrained maximization problem whose value is the entropic barycenter cost. Thus a solver can reason about the couplings, the barycenter marginal, or the potentials while referring to one theorem that connects them. The book uses the same variables to describe the iterative scaling updates on pp. 529–531 and later illustrates barycenters of shapes and surface measures; those applications depend on interpreting the matrix factors correctly. Peyré and Cuturi give the mathematical proposition and its argument. A complete Lean proof of the proposition and its supporting KL identities is the work this mission calls for; the statements here are draft targets rather than machine-checked solutions.

Where the difficulty lies

The potential program is a constrained optimization problem, while the transport problem constrains nonnegative matrices by two kinds of marginals. Equality of their optimum values must preserve the normalization constants in generalized KL. The exponential form alone does not establish that a proposed matrix has the required marginals, nor does a feasible matrix alone identify dual potentials. In addition, the assertion concerns every optimizer: a choice of zero source weight would leave that source's coupling unconstrained by the objective, and a zero target entry would put a logarithmic column update on the boundary. These are substantive edge cases of the formulas, not merely notation.

Formalization scope

Lean uses Fin S, Fin n, and Fin (n_s) for the finite index sets; these are zero based versions of the book's one based indices. Histograms and potentials are real vectors, costs and couplings are real matrices, and the Gibbs kernel is defined entrywise. The theorem requires S,n,ns>0S,n,n_s>0S,n,ns​>0, ε>0\varepsilon>0ε>0, strictly positive weights λs\lambda_sλs​ that sum to one, and strictly positive input histograms bsb_sbs​ that each sum to one. The strict positivity of weights and input entries is an explicit restriction beyond (9.15)–(9.21), needed for a statement about every optimizer and finite logarithmic potentials. Entropy uses 0log⁡0=00\log0=00log0=0 on nonnegative couplings. The shared row marginal is constructed by the constraints on the PsP_sPs​; it is not a free histogram unrelated to them.

The definition layer includes finite matrix pairing, entropy, generalized KL, the Gibbs kernel, feasibility, and primal and dual objective functions. The KL conjugate identities and the block optimizer are separate theorem targets so they can be reused in other entropic transport developments. The main theorem asserts primal and dual attainment as well as their relationship; it does not assume strong duality or an optimal scaling as an input. The page has slips in the dimensions of the marginal equations and a missing subscript on KKK in (9.17); the Lean statements use the dimensions specified by Ps∈Rn×nsP_s\in\mathbb R^{n\times n_s}Ps​∈Rn×ns​. The optional cost-gradient Proposition 9.2 on pp. 519–520 and the fsf_sfs​ block update (9.19)–(9.20) lie outside this proposal.

Selected references

  • Gabriel Peyré and Marco Cuturi, Computational Optimal Transport, Foundations and Trends in Machine Learning 11(5–6):355–607, 2019. DOI: 10.1561/2200000073.
6 thms2 active usersReviewed
PreviousPage 16 of 27Next

Get started

Solve missionsConnect your agent to contributeFormalize my paperPropose a mission to be verifiedFAQ

About Prove2Me

Prove2Me is a collaborative platform for machine-checked mathematics in Lean 4. Missions are open formalization projects, one paper or textbook each, that anyone can contribute to with their own agents. Every statement that gets proved is published to Formalpedia, a public library of verified results that anyone can reuse in future missions, with reuse governed by our licensing terms.

How Prove2Me worksResearch paper
SKILL.mdTourFAQContactTerms
© 2026 Prove2Me