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 · 431 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

Open222Completed431All653
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
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
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
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
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
Bandit Algorithms·Captain: mikedeng1

An Optimal Algorithm for Stochastic and Adversarial Bandits I: Tsallis-INF Has Square-Root Adversarial Pseudo-Regret and Logarithmic Pseudo-Regret Under a Self-Bounding ConstraintResearch Paper

Motivation

A multi-armed bandit learner repeatedly chooses one of several actions, observes only the loss of the chosen action, and tries to perform nearly as well as a single fixed action selected in hindsight. Two common loss models make different demands on the learner. In a stochastic model, each arm has a stable mean loss and the learner can gradually distinguish the best arm. In an adversarial model, losses may change with time and even respond to previous choices. A useful bandit algorithm should handle both without first being told which model generated the losses.

Zimmert and Seldin's Tsallis-INF paper analyzes one algorithm in both regimes. Its Theorem 1 gives a square-root guarantee against arbitrary adversarial losses and a logarithmic guarantee when the regret satisfies a self-bounding constraint. The latter condition includes the usual stochastic case with a unique best arm and also permits some adversarial deviations. This mission formalizes the reduced-variance (RV) estimator part of that theorem and six supporting statements from its analysis.

Setting

There are K≥1K\ge1K≥1 arms and rounds t=1,2,…t=1,2,\ldotst=1,2,…. At round ttt, the environment supplies a loss ℓt,i∈[0,1]\ell_{t,i}\in[0,1]ℓt,i​∈[0,1] for each arm iii. The learner draws an arm ItI_tIt​ from a probability vector wtw_twt​ and observes only ℓt,It\ell_{t,I_t}ℓt,It​​. The environment can base its current losses on the actions from earlier rounds. It may also randomize: a seed ω\omegaω records its internal random choices, while its response to a history remains adaptive. The learner's random actions are then drawn successively from its history-dependent weight vectors. Expectations below average over both sources of randomness.

The pseudo-regret at horizon TTT compares the learner's expected total loss with the lowest expected total loss of a fixed arm:

Reg⁡T=E ⁣[∑t=1Tℓt,It]−min⁡iE ⁣[∑t=1Tℓt,i].\operatorname{Reg}_T =\mathbb E\!\left[\sum_{t=1}^{T}\ell_{t,I_t}\right] -\min_i\mathbb E\!\left[\sum_{t=1}^{T}\ell_{t,i}\right].RegT​=E[t=1∑T​ℓt,It​​]−imin​E[t=1∑T​ℓt,i​].

For the algorithm studied here, L^t−1\widehat L_{t-1}Lt−1​ is the vector of estimated losses accumulated before round ttt. The symmetric half-Tsallis regularizer is

Ψt(w)=−4ηt−1∑i=1K(wi−12wi),ηt=4t.\Psi_t(w)=-4\eta_t^{-1}\sum_{i=1}^{K}\left(\sqrt{w_i}-\tfrac12w_i\right), \qquad \eta_t=\frac4{\sqrt t}.Ψt​(w)=−4ηt−1​i=1∑K​(wi​​−21​wi​),ηt​=t​4​.

The vector wtw_twt​ maximizes ⟨w,−L^t−1⟩−Ψt(w)\langle w,-\widehat L_{t-1}\rangle-\Psi_t(w)⟨w,−Lt−1​⟩−Ψt​(w) over the probability simplex. With the RV loss estimator, arm iii has baseline Bt(i)=121{wt,i≥ηt2}B_t(i)=\tfrac12\mathbf 1\{w_{t,i}\ge\eta_t^2\}Bt​(i)=21​1{wt,i​≥ηt2​} and estimate ℓ^t,i=1{It=i}(ℓt,i−Bt(i))/wt,i+Bt(i)\widehat\ell_{t,i}=\mathbf 1\{I_t=i\}(\ell_{t,i}-B_t(i))/w_{t,i}+B_t(i)ℓt,i​=1{It​=i}(ℓt,i​−Bt​(i))/wt,i​+Bt​(i). This is an estimator based on the one observed loss, even though its update is a vector.

The self-bounding condition chooses gaps Δi∈[0,1]\Delta_i\in[0,1]Δi​∈[0,1] with exactly one zero, at i∗i^*i∗, and a constant C≥0C\ge0C≥0, as defined in Section 2. It requires

Reg⁡T≥E ⁣[∑t=1T∑i≠i∗wt,iΔi]−C.\operatorname{Reg}_T\ge \mathbb E\!\left[\sum_{t=1}^{T}\sum_{i\ne i^*}w_{t,i}\Delta_i\right]-C.RegT​≥E​t=1∑T​i=i∗∑​wt,i​Δi​​−C.

The gaps measure how costly it is to place weight on each non-best arm. In a stochastic bandit with a unique best arm, the usual expected-loss gaps give this relation with C=0C=0C=0 Zimmert and Seldin, §2 and §4.1.

Formalization targets

Adversarial guarantee

The first target is an upper bound for every bounded, randomized adaptive adversary and every T≥1T\ge1T≥1:

Reg⁡T≤2KT+14Klog⁡T+16.\operatorname{Reg}_T\le 2\sqrt{KT}+14K\log T+16.RegT​≤2KT​+14KlogT+16.

The paper prints 10Klog⁡T10K\log T10KlogT in Theorem 1's display. Its proof yields 14Klog⁡T14K\log T14KlogT, the coefficient formalized here; the difference is recorded in the moderation notes. The theorem's other displayed bounds retain their printed constants.

Self-bounding guarantee

For K≥2K\ge2K≥2, set Δmin⁡=min⁡i≠i∗Δi\Delta_{\min}=\min_{i\ne i^*}\Delta_iΔmin​=mini=i∗​Δi​ and

X=∑i≠i∗log⁡T+3Δi+2Δmin⁡.X=\sum_{i\ne i^*}\frac{\log T+3}{\Delta_i}+\frac2{\Delta_{\min}}.X=i=i∗∑​Δi​logT+3​+Δmin​2​.

Under the self-bounding condition, the same algorithm must satisfy

Reg⁡T≤X+28Klog⁡T+32K+32+C.\operatorname{Reg}_T\le X+28K\log T+\tfrac32\sqrt K+32+C.RegT​≤X+28KlogT+23​K​+32+C.

When C>XC>XC>X, it must also satisfy Reg⁡T≤2XC+28Klog⁡T+32K+32\operatorname{Reg}_T\le2\sqrt{XC}+28K\log T+\tfrac32\sqrt K+32RegT​≤2XC​+28KlogT+23​K​+32. The goal states both clauses together with the adversarial guarantee. Its six milestones give a pathwise stability bound from Lemma 19, two per-round stability bounds from Lemma 11, two accumulated penalty bounds from Lemma 12, and a finite tail-sum estimate from Lemma 15.

Significance

The adversarial clause limits the cost of learning even when no arm has a persistent advantage. The self-bounding clause gives gap-sensitive logarithmic growth when the learner receives enough information to identify a unique best arm, while allowing the additive deviation CCC. The two guarantees apply to the same weight rule and the same RV estimator Zimmert and Seldin, Theorem 1.

The paper proves its RV claims in prose and equations. A complete Lean proof would make the algorithm's randomized adaptive loss model, its path law, the estimator and the exact constants explicit in machine-checked form. The local theorem statements and definitions in this proposal compile with placeholder theorem proofs; the mission's mathematical results are therefore targets for formalization, not results already machine checked here. The setting definitions may also support formal analyses of other bandit algorithms with adaptive randomized losses.

Difficulty

The learner sees only one loss at each round, so the weight vector depends on estimates whose individual coordinates can be much larger than the observed losses. Bounding the change in the regularized potential by a simple worst-case estimate loses the dependence on the arm weights that the gap-sensitive result needs. The RV estimator reduces this difficulty but can take negative values, which changes the stability calculation. The self-bounding guarantee also compares the theorem's zero-gap arm with a best arm chosen in expectation at horizon TTT; for C>0C>0C>0 these need not coincide. That comparator issue is documented for audit.

Formalization scope

The arms are Fin K, with the paper's first arm represented by Lean index zero. Histories are functions on natural-number rounds; entry zero is unused. A referenced published adversarial protocol supplies bounded losses and dependence on past actions. A probability measure on adversary seeds and a finite sum over action paths define the run law. The algorithm is an argmax predicate whose estimates are computed from its own weights and the observed loss. The real-valued simplex supremum defines Φt\Phi_tΦt​; K≥1K\ge1K≥1 makes that simplex nonempty, and its compactness bounds the objective. The goal requires measurable seeded losses, a probability measure, T≥1T\ge1T≥1, C≥0C\ge0C≥0 in the self-bounding regime, and K≥2K\ge2K≥2 where Δmin⁡\Delta_{\min}Δmin​ is used. The penalty milestones take conditionally unbiased estimators that depend only on actions through their own round, are measurable, and have finite expected absolute size, as required for a sequential run and genuine expectations. Positive, non-increasing learning rates are explicit.

The argmax is taken over every simplex weight vector, and the run law includes every action path with its product probability. These choices rule out an arbitrary weight rule or a zero-valued integral masquerading as the algorithm. The development needs reusable facts about the simplex, measurability and integrability of the finite-path law, positivity of Tsallis-INF weights, and bounds on the real potential. Contributions that establish those facts or close the six milestone theorems fit this mission.

Selected references

  • Julian Zimmert and Yevgeny Seldin, Tsallis-INF: An Optimal Algorithm for Stochastic and Adversarial Bandits, Journal of Machine Learning Research 22(28), 2021; formalization follows arXiv:1807.07623v6, pp. 5–10, 18–21.
9 thms2 active usersReviewed
OptimizationProbability·Captain: mikedeng1

Convergence and Dynamical Behavior of the ADAM Algorithm for Nonconvex Stochastic Optimization 6: Decreasing-Step Adam Iterates Are Almost Surely BoundedResearch Paper

Motivation

Adam (Kingma and Ba, arXiv:1412.6980) is the default optimizer for training neural networks. It combines a stochastic gradient step with two exponential moving averages: a momentum term mnm_nmn​ and a coordinatewise second-moment estimate vnv_nvn​ that rescales each coordinate of the step. Its convergence theory is delicate: Reddi, Kale and Kumar (arXiv:1904.09237) exhibited convex problems on which Adam with constant hyperparameters fails to converge.

Barakat and Bianchi (arXiv:1810.02263, SIAM J. Math. Data Sci. 3(1), 2021) study Adam on nonconvex objectives through the ODE method of stochastic approximation. In the decreasing-stepsize regime, their Theorem 5.2 proves that the iterates converge almost surely to the critical points of the objective — provided the iterates are almost surely bounded. A stability hypothesis of this kind is standard in stochastic approximation (Benaïm, Dynamics of stochastic approximation algorithms, Séminaire de Probabilités XXXIII, 1999, doi:10.1007/BFb0096509), and it is often the hardest part to verify. Theorem 5.4 of the paper gives sufficient conditions, on the objective and the stepsizes only, under which almost sure boundedness holds. This mission formalizes that theorem.

Setting

Let (Ω,F,P)(\Omega,\mathcal F,\mathbb P)(Ω,F,P) be a probability space and (Ξ,S)(\Xi,\mathfrak S)(Ξ,S) a measurable space carrying a probability measure μ\muμ. A loss f:Rd×Ξ→Rf:\mathbb R^d\times\Xi\to\mathbb Rf:Rd×Ξ→R is given, with gradient ∇f(x,ξ)\nabla f(x,\xi)∇f(x,ξ) in xxx, and the objective is

F(x)=Ef(x,ξ)=∫Ξf(x,ξ) μ(dξ).F(x)=\mathbb E f(x,\xi)=\int_\Xi f(x,\xi)\,\mu(d\xi).F(x)=Ef(x,ξ)=∫Ξ​f(x,ξ)μ(dξ).

The samples (ξn)n≥1(\xi_n)_{n\ge1}(ξn​)n≥1​ are iid with law μ\muμ (Assumption 4.1). Vector operations below are coordinatewise.

Fix ε>0\varepsilon>0ε>0 and real sequences (γn)(\gamma_n)(γn​) (stepsizes), (αn)(\alpha_n)(αn​), (βn)(\beta_n)(βn​) (moment parameters). Algorithm 5.1 (Adam with decreasing stepsize) starts from x0∈Rdx_0\in\mathbb R^dx0​∈Rd, m0=v0=0m_0=v_0=0m0​=v0​=0, r0=rˉ0=0r_0=\bar r_0=0r0​=rˉ0​=0, and for n≥1n\ge1n≥1 sets

mn=αnmn−1+(1−αn)∇f(xn−1,ξn),rn=αnrn−1+(1−αn),vn=βnvn−1+(1−βn)∇f(xn−1,ξn)⊙2,rˉn=βnrˉn−1+(1−βn),xn=xn−1−γn m^n/(ε+v^n),m^n=mn/rn,v^n=vn/rˉn.\begin{aligned} m_n&=\alpha_nm_{n-1}+(1-\alpha_n)\nabla f(x_{n-1},\xi_n), & r_n&=\alpha_nr_{n-1}+(1-\alpha_n),\\ v_n&=\beta_nv_{n-1}+(1-\beta_n)\nabla f(x_{n-1},\xi_n)^{\odot2}, & \bar r_n&=\beta_n\bar r_{n-1}+(1-\beta_n),\\ x_n&=x_{n-1}-\gamma_n\,\hat m_n/(\varepsilon+\sqrt{\hat v_n}), & \hat m_n&=m_n/r_n,\quad \hat v_n=v_n/\bar r_n . \end{aligned}mn​vn​xn​​=αn​mn−1​+(1−αn​)∇f(xn−1​,ξn​),=βn​vn−1​+(1−βn​)∇f(xn−1​,ξn​)⊙2,=xn−1​−γn​m^n​/(ε+v^n​​),​rn​rˉn​m^n​​=αn​rn−1​+(1−αn​),=βn​rˉn−1​+(1−βn​),=mn​/rn​,v^n​=vn​/rˉn​.​

The weights rn,rˉnr_n,\bar r_nrn​,rˉn​ implement the bias correction of Adam.

The hypotheses are: Assumption 2.2 (f(x,⋅)f(x,\cdot)f(x,⋅) measurable, f(⋅,ξ)f(\cdot,\xi)f(⋅,ξ) continuously differentiable for a.e. ξ\xiξ, integrability at one point, and E∥∇f(x,ξ)−∇f(y,ξ)∥2≤LK2∥x−y∥2\mathbb E\|\nabla f(x,\xi)-\nabla f(y,\xi)\|^2\le L_K^2\|x-y\|^2E∥∇f(x,ξ)−∇f(y,ξ)∥2≤LK2​∥x−y∥2 on each compact KKK); Assumption 2.3 (FFF coercive); Assumption 4.2 i) with p=4p=4p=4 (sup⁡x∈KE∥∇f(x,ξ)∥4<∞\sup_{x\in K}\mathbb E\|\nabla f(x,\xi)\|^4<\inftysupx∈K​E∥∇f(x,ξ)∥4<∞ on compacts); Assumption 5.1 (γn>0\gamma_n>0γn​>0, γn+1/γn→1\gamma_{n+1}/\gamma_n\to1γn+1​/γn​→1, ∑γn=∞\sum\gamma_n=\infty∑γn​=∞, ∑γn2<∞\sum\gamma_n^2<\infty∑γn2​<∞, αn,βn∈[0,1]\alpha_n,\beta_n\in[0,1]αn​,βn​∈[0,1], (1−αn)/γn→a(1-\alpha_n)/\gamma_n\to a(1−αn​)/γn​→a, (1−βn)/γn→b(1-\beta_n)/\gamma_n\to b(1−βn​)/γn​→b with 0<b<4a0<b<4a0<b<4a); and Assumption 5.3 (∇F\nabla F∇F Lipschitz, E∥∇f(x,ξ)∥2≤C(1+F(x))\mathbb E\|\nabla f(x,\xi)\|^2\le C(1+F(x))E∥∇f(x,ξ)∥2≤C(1+F(x)), and lim sup⁡n(γn−1−1−αn+21−αn+1γn+1−1)<2(a−b/4)\limsup_n\big(\gamma_n^{-1}-\frac{1-\alpha_{n+2}}{1-\alpha_{n+1}}\gamma_{n+1}^{-1}\big)<2(a-b/4)limsupn​(γn−1​−1−αn+1​1−αn+2​​γn+1−1​)<2(a−b/4)).

Formalization targets

Goal: Theorem 5.4

Under Assumptions 2.2, 2.3, 4.1, 5.1, 5.3 and 4.2 i) with p=4p=4p=4,

P(sup⁡n∈N∥(xn,mn,vn)∥<∞)=1.\mathbb P\Big(\sup_{n\in\mathbb N}\|(x_n,m_n,v_n)\|<\infty\Big)=1 .P(n∈Nsup​∥(xn​,mn​,vn​)∥<∞)=1.

Milestones (in the order the proof of §9.2 uses them)

  1. Lemma 9.1 i)–ii): rn=1−∏i=1nαir_n=1-\prod_{i=1}^n\alpha_irn​=1−∏i=1n​αi​, and (rn)(r_n)(rn​) is nondecreasing with rn→1r_n\to1rn​→1; likewise for rˉn\bar r_nrˉn​.
  2. (9.1): with an=(1−αn+1)/γna_n=(1-\alpha_{n+1})/\gamma_nan​=(1−αn+1​)/γn​ and Pn=12anrn⟨mn⊙2,1/(ε+v^n)⟩P_n=\frac1{2a_nr_n}\langle m_n^{\odot2},1/(\varepsilon+\sqrt{\hat v_n})\ranglePn​=2an​rn​1​⟨mn⊙2​,1/(ε+v^n​​)⟩,
F(xn)≤F(xn−1)−γn⟨∇F(xn),m^nε+v^n⟩+Cγn2Pn.F(x_n)\le F(x_{n-1})-\gamma_n\Big\langle\nabla F(x_n),\frac{\hat m_n}{\varepsilon+\sqrt{\hat v_n}}\Big\rangle+C\gamma_n^2P_n .F(xn​)≤F(xn−1​)−γn​⟨∇F(xn​),ε+v^n​​m^n​​⟩+Cγn2​Pn​.
  1. (9.3): coordinatewise, v^n−v^n+1≤cn+1v^n+1\sqrt{\hat v_n}-\sqrt{\hat v_{n+1}}\le c_{n+1}\sqrt{\hat v_{n+1}}v^n​​−v^n+1​​≤cn+1​v^n+1​​ with an explicit cn+1∼bγn/2c_{n+1}\sim b\gamma_n/2cn+1​∼bγn​/2.
  2. (9.4): with un=1−an+1/anu_n=1-a_{n+1}/a_nun​=1−an+1​/an​, for every δ>0\delta>0δ>0 and nnn large,
En(F(xn)+Pn+1)≤F(xn−1)+Pn+unEnPn+1−2(a−b+δ4)γnPn+Cγn2(1+F(xn)+Pn).\mathbb E_n\big(F(x_n)+P_{n+1}\big)\le F(x_{n-1})+P_n+u_n\mathbb E_nP_{n+1}-2\Big(a-\frac{b+\delta}4\Big)\gamma_nP_n+C\gamma_n^2\big(1+F(x_n)+P_n\big).En​(F(xn​)+Pn+1​)≤F(xn−1​)+Pn​+un​En​Pn+1​−2(a−4b+δ​)γn​Pn​+Cγn2​(1+F(xn​)+Pn​).
  1. The Robbins–Siegmund inequality: Vn=(1−Cγn−12)F(xn−1)+(1−un−1)PnV_n=(1-C\gamma_{n-1}^2)F(x_{n-1})+(1-u_{n-1})P_nVn​=(1−Cγn−12​)F(xn−1​)+(1−un−1​)Pn​ satisfies En(Vn+1)≤(1+C′γn2)Vn+Cγn2\mathbb E_n(V_{n+1})\le(1+C'\gamma_n^2)V_n+C\gamma_n^2En​(Vn+1​)≤(1+C′γn2​)Vn​+Cγn2​ for nnn large.

Significance

Theorem 5.2 of the paper, the almost sure convergence of decreasing-step Adam to the critical set, is conditional on almost sure boundedness. Theorem 5.4 removes that condition for objectives with Lipschitz gradient and quadratic gradient-noise growth, and so turns Theorem 5.2 into an unconditional convergence result in that class. The Lyapunov function F(xn−1)+PnF(x_{n-1})+P_nF(xn−1​)+Pn​ — the objective plus a weighted kinetic energy of the momentum — is the discrete counterpart of the energy used for the continuous-time Adam ODE in the same paper; the condition b<4ab<4ab<4a that keeps it decreasing is the discrete trace of the condition b≤4ab\le4ab≤4a of the ODE analysis.

The result is proved in the paper; nothing about Adam is formalized on the platform. A formal proof produces a machine-checked stability theorem for an adaptive, momentum-based stochastic algorithm, together with reusable pieces: the Robbins–Siegmund almost-supermartingale theorem in the conditional-expectation form used here, and the bias-correction lemma, which applies to every exponential moving average with vanishing forgetting rate.

Difficulty

For plain stochastic gradient descent, boundedness follows from the descent lemma applied to FFF alone: F(xn)F(x_n)F(xn​) is an almost supermartingale. For Adam this fails. The step m^n/(ε+v^n)\hat m_n/(\varepsilon+\sqrt{\hat v_n})m^n​/(ε+v^n​​) is not a descent direction for FFF: mnm_nmn​ averages past gradients, so ⟨∇F(xn),m^n/(ε+v^n)⟩\langle\nabla F(x_n),\hat m_n/(\varepsilon+\sqrt{\hat v_n})\rangle⟨∇F(xn​),m^n​/(ε+v^n​​)⟩ has no sign. The momentum must therefore enter the Lyapunov function, through PnP_nPn​, and the coordinatewise preconditioner 1/(ε+v^n)1/(\varepsilon+\sqrt{\hat v_n})1/(ε+v^n​​) changes from step to step, so Pn+1P_{n+1}Pn+1​ and PnP_nPn​ are measured in different weighted norms. Controlling that change is what (9.3) does, and it costs a term of size b2γnPn\frac b2\gamma_nP_n2b​γn​Pn​ that must be absorbed by the friction 2aγnPn2a\gamma_nP_n2aγn​Pn​ of the momentum update; the margin is exactly b<4ab<4ab<4a. The time-varying weights ana_nan​ and rnr_nrn​ produce the further term unPn+1u_nP_{n+1}un​Pn+1​, which is why Assumption 5.3 iii) is needed. Finally, the conditional expectations involve the next sample through both mn+1m_{n+1}mn+1​ and v^n+1\hat v_{n+1}v^n+1​, so the second-moment growth condition of Assumption 5.3 ii) has to be used with the right weights.

Formalization scope

Points of Rd\mathbb R^dRd are EuclideanSpace ℝ (Fin d); the state (x,m,v)(x,m,v)(x,m,v) lives in the product E×E×EE\times E\times EE×E×E, whose Mathlib norm is the maximum of the three Euclidean norms (the same bounded sets). The gradient of fff is a function gf : E → Ξ → E that equals the true gradient for μ\muμ-almost every ξ\xiξ. The run of Algorithm 5.1 is defined along a sample path and then pathwise in ω\omegaω. Moments are lower Lebesgue integrals, so no hypothesis is vacuous through the convention that a non-integrable Bochner integral is 000. Assumption 5.3 iii) is stated as "there is c<2(a−b/4)c<2(a-b/4)c<2(a−b/4) bounding the bracket for all large nnn", which avoids a real limsup. The algorithm divides by rnr_nrn​ and rˉn\bar r_nrˉn​; the theorem therefore assumes α1<1\alpha_1<1α1​<1 and β1<1\beta_1<1β1​<1, which Algorithm 5.1 needs for m^1,v^1\hat m_1,\hat v_1m^1​,v^1​ to be defined. In the milestones, En\mathbb E_nEn​ applied to a function of the next state is the integral of that function over the next sample, ∫G(Tn+1(zn,ξ)) μ(dξ)\int G(T_{n+1}(z_n,\xi))\,\mu(d\xi)∫G(Tn+1​(zn​,ξ))μ(dξ), a version of the conditional expectation under Assumption 4.1. In (9.4) the term printed as unPn+1u_nP_{n+1}un​Pn+1​ is unEnPn+1u_n\mathbb E_nP_{n+1}un​En​Pn+1​, the reading the derivation produces.

The goal asserts a single bound for the whole trajectory almost surely; finiteness of each iterate, boundedness of a projected variant of Adam, and boundedness under an almost-sure bound on ∇f(x,ξ)\nabla f(x,\xi)∇f(x,ξ) are all weaker statements and are not this theorem. The "without loss of generality F≥0F\ge0F≥0" of the proof appears only as a hypothesis of the last milestone.

A complete development needs the Robbins–Siegmund theorem (open on the platform as AdaptiveProtection.Convergence.lemma2_robbins_siegmund), differentiation under the integral sign for FFF, the descent lemma for Lipschitz gradients, and conditional expectations given σ(ξ1,…,ξn)\sigma(\xi_1,\dots,\xi_n)σ(ξ1​,…,ξn​) for functions of an independent sample. Proofs of any milestone, and a proof of Robbins–Siegmund in the form used here, are welcome.

Selected references

  • A. Barakat, P. Bianchi, Convergence and Dynamical Behavior of the ADAM Algorithm for Nonconvex Stochastic Optimization, SIAM J. Math. Data Sci. 3(1), 2021; arXiv:1810.02263v4. https://arxiv.org/abs/1810.02263
  • D. P. Kingma, J. Ba, Adam: A Method for Stochastic Optimization, ICLR 2015. https://arxiv.org/abs/1412.6980
  • S. J. Reddi, S. Kale, S. Kumar, On the Convergence of Adam and Beyond, ICLR 2018. https://arxiv.org/abs/1904.09237
  • H. Robbins, D. Siegmund, A convergence theorem for non negative almost supermartingales and some applications, in Optimizing Methods in Statistics, Academic Press, 1971. https://doi.org/10.1016/B978-0-12-604550-5.50015-8
  • M. Benaïm, Dynamics of stochastic approximation algorithms, Séminaire de Probabilités XXXIII, LNM 1709, Springer, 1999. https://doi.org/10.1007/BFb0096509
12 thms2 active usersReviewed
Dynamical SystemsOptimizationProbability·Captain: mikedeng1

Convergence and Dynamical Behavior of the ADAM Algorithm for Nonconvex Stochastic Optimization 5: Decreasing-Step Adam Converges Almost Surely to the Critical Points of FResearch Paper

Motivation

Adam (Kingma and Ba, 2015) is the default optimizer for training neural networks: a stochastic gradient method that keeps an exponential moving average mnm_nmn​ of past gradients and an exponential moving average vnv_nvn​ of their coordinatewise squares, and divides the first by the square root of the second. Its practical success came well before any convergence theory for nonconvex objectives. Reddi, Kale and Kumar (2018) showed that Adam with constant momentum parameters can fail to converge even on convex problems, which made the question of under which parameter schedules Adam provably converges a live one.

Barakat and Bianchi (arXiv:1810.02263, SIAM J. Math. Data Sci. 2021) answer it through the ODE method of stochastic approximation (Benaïm, 1999). This mission takes the paper's almost-sure convergence theorem for Adam with decreasing stepsizes, Theorem 5.2. Under a stepsize schedule in which 1−αn1 - \alpha_n1−αn​ and 1−βn1 - \beta_n1−βn​ shrink in proportion to the stepsize γn\gamma_nγn​, the iterates converge almost surely to the critical points of the objective, provided they stay bounded.

Setting

Let ξ\xiξ be a random variable on a measurable space Ξ\XiΞ with law μ\muμ, let f:Rd×Ξ→Rf : \mathbb R^d \times \Xi \to \mathbb Rf:Rd×Ξ→R be an integrand with gradient ∇f(x,ξ)\nabla f(x, \xi)∇f(x,ξ) in xxx, and set

F(x)=Ef(x,ξ),S(x)=E(∇f(x,ξ)⊙2),S=∇F−1({0}),F(x) = \mathbb E f(x, \xi), \qquad S(x) = \mathbb E\big(\nabla f(x,\xi)^{\odot 2}\big), \qquad \mathcal S = \nabla F^{-1}(\{0\}),F(x)=Ef(x,ξ),S(x)=E(∇f(x,ξ)⊙2),S=∇F−1({0}),

where ⊙2\odot 2⊙2 is the coordinatewise square and S\mathcal SS is the critical set. Let ξ1,ξ2,…\xi_1, \xi_2, \dotsξ1​,ξ2​,… be iid copies of ξ\xiξ on a probability space (Ω,F,P)(\Omega, \mathcal F, \mathbb P)(Ω,F,P).

Algorithm 5.1 takes stepsizes γn>0\gamma_n > 0γn​>0, weights αn,βn∈[0,1]\alpha_n, \beta_n \in [0,1]αn​,βn​∈[0,1] and a constant ε>0\varepsilon > 0ε>0. Starting from x0∈Rdx_0 \in \mathbb R^dx0​∈Rd, m0=v0=0m_0 = v_0 = 0m0​=v0​=0 and r0=rˉ0=0r_0 = \bar r_0 = 0r0​=rˉ0​=0, it computes for n≥1n \ge 1n≥1, with all vector operations coordinatewise:

mn=αnmn−1+(1−αn)∇f(xn−1,ξn),vn=βnvn−1+(1−βn)∇f(xn−1,ξn)⊙2,rn=αnrn−1+(1−αn),rˉn=βnrˉn−1+(1−βn),xn=xn−1−γnm^nε+v^n,m^n=mn/rn,v^n=vn/rˉn.\begin{aligned} m_n &= \alpha_n m_{n-1} + (1-\alpha_n)\nabla f(x_{n-1}, \xi_n), & v_n &= \beta_n v_{n-1} + (1-\beta_n)\nabla f(x_{n-1}, \xi_n)^{\odot 2},\\ r_n &= \alpha_n r_{n-1} + (1 - \alpha_n), & \bar r_n &= \beta_n \bar r_{n-1} + (1 - \beta_n),\\ x_n &= x_{n-1} - \gamma_n \frac{\hat m_n}{\varepsilon + \sqrt{\hat v_n}}, & \hat m_n &= m_n / r_n,\quad \hat v_n = v_n / \bar r_n . \end{aligned}mn​rn​xn​​=αn​mn−1​+(1−αn​)∇f(xn−1​,ξn​),=αn​rn−1​+(1−αn​),=xn−1​−γn​ε+v^n​​m^n​​,​vn​rˉn​m^n​​=βn​vn−1​+(1−βn​)∇f(xn−1​,ξn​)⊙2,=βn​rˉn−1​+(1−βn​),=mn​/rn​,v^n​=vn​/rˉn​.​

The divisions by rnr_nrn​, rˉn\bar r_nrˉn​ are the bias correction.

The stepsizes satisfy Assumption 5.1: γn+1/γn→1\gamma_{n+1}/\gamma_n \to 1γn+1​/γn​→1, ∑nγn=+∞\sum_n \gamma_n = +\infty∑n​γn​=+∞, ∑nγn2<+∞\sum_n \gamma_n^2 < +\infty∑n​γn2​<+∞, and there are a,ba, ba,b with 0<b<4a0 < b < 4a0<b<4a, (1−αn)/γn→a(1-\alpha_n)/\gamma_n \to a(1−αn​)/γn​→a and (1−βn)/γn→b(1-\beta_n)/\gamma_n \to b(1−βn​)/γn​→b. The limiting dynamics is the autonomous ODE z˙=h∞(z)\dot z = h_\infty(z)z˙=h∞​(z) on Z+=Rd×Rd×[0,∞)d\mathcal Z_+ = \mathbb R^d \times \mathbb R^d \times [0,\infty)^dZ+​=Rd×Rd×[0,∞)d, written (ODE∞)(\mathrm{ODE}_\infty)(ODE∞​), with

h∞(x,m,v)=(−mε+v, a(∇F(x)−m), b(S(x)−v)).h_\infty(x, m, v) = \Big( -\frac{m}{\varepsilon + \sqrt v},\ a(\nabla F(x) - m),\ b(S(x) - v) \Big).h∞​(x,m,v)=(−ε+v​m​, a(∇F(x)−m), b(S(x)−v)).

Formalization targets

Goal: Theorem 5.2

Assume Assumption 2.2 (regularity and moments of fff), FFF coercive (2.3), S(x)>0S(x) > 0S(x)>0 coordinatewise (2.4), iid samples (4.1), Assumption 5.1, sup⁡x∈KE∥∇f(x,ξ)∥4<∞\sup_{x\in K}\mathbb E\|\nabla f(x,\xi)\|^4 < \inftysupx∈K​E∥∇f(x,ξ)∥4<∞ on compacts (4.2 i) with p=4p = 4p=4), that F(S)F(\mathcal S)F(S) has empty interior, and that (xn,mn,vn)(x_n, m_n, v_n)(xn​,mn​,vn​) is bounded with probability one. Then, almost surely,

d(xn,S)→0,mn→0,S(xn)−vn→0,d(x_n, \mathcal S) \to 0, \qquad m_n \to 0, \qquad S(x_n) - v_n \to 0,d(xn​,S)→0,mn​→0,S(xn​)−vn​→0,

and if S\mathcal SS is countable, almost surely (xn,mn,vn)→(x∗,0,S(x∗))(x_n, m_n, v_n) \to (x^*, 0, S(x^*))(xn​,mn​,vn​)→(x∗,0,S(x∗)) for some x∗∈Sx^* \in \mathcal Sx∗∈S.

Milestones

  1. Lemma 9.1 i)–ii): rn=1−∏i=1nαir_n = 1 - \prod_{i=1}^n \alpha_irn​=1−∏i=1n​αi​, and rnr_nrn​ increases to 111.
  2. §9.1: in the decomposition zˉn+1=zˉn+γn+1h∞(zˉn)+γn+1χn+1+γn+1ςn+1\bar z_{n+1} = \bar z_n + \gamma_{n+1} h_\infty(\bar z_n) + \gamma_{n+1}\chi_{n+1} + \gamma_{n+1}\varsigma_{n+1}zˉn+1​=zˉn​+γn+1​h∞​(zˉn​)+γn+1​χn+1​+γn+1​ςn+1​ of zˉn=(xn−1,mn,vn)\bar z_n = (x_{n-1}, m_n, v_n)zˉn​=(xn−1​,mn​,vn​), the remainder satisfies ςn→0\varsigma_n \to 0ςn​→0 almost surely.
  3. §9.1: the piecewise-affine interpolation of (zˉn)(\bar z_n)(zˉn​) on the times τn=∑k≤nγk\tau_n = \sum_{k\le n}\gamma_kτn​=∑k≤n​γk​ is almost surely a bounded asymptotic pseudotrajectory of the semiflow of (ODE∞)(\mathrm{ODE}_\infty)(ODE∞​).
  4. Proposition 7.13: (ODE∞)(\mathrm{ODE}_\infty)(ODE∞​) is well posed on Z+\mathcal Z_+Z+​ and defines a semiflow Φ\PhiΦ.
  5. Proposition 7.14 (after Benaïm): the limit set of a relatively compact asymptotic pseudotrajectory of a semiflow with a strict Lyapunov function lies in the set of equilibria.
  6. Proposition 7.15: on the closed hull of the orbits of a compact set, Wδ=V∞−δ⟨∇F(x),m⟩+δ∥S(x)−v∥2W_\delta = V_\infty - \delta\langle\nabla F(x), m\rangle + \delta\|S(x) - v\|^2Wδ​=V∞​−δ⟨∇F(x),m⟩+δ∥S(x)−v∥2 is a strict Lyapunov function for some δ>0\delta > 0δ>0.

Significance

The theorem identifies a regime of Adam's hyperparameters, 1−αn∼aγn1 - \alpha_n \sim a\gamma_n1−αn​∼aγn​ and 1−βn∼bγn1 - \beta_n \sim b\gamma_n1−βn​∼bγn​ with b<4ab < 4ab<4a, in which the algorithm behaves like a gradient method with vanishing noise: the iterates approach critical points, the momentum dies out, and the second-moment estimate converges to S(xn)S(x_n)S(xn​). The condition b<4ab < 4ab<4a ties the two averaging rates together. It is the condition under which the paper's Lyapunov function for the continuous dynamics decreases (Lemma 7.5). The argument structure is also a template for other adaptive methods: once the limiting ODE has a strict Lyapunov function, almost-sure convergence of the iterates follows from the same steps.

The paper proves the result on paper; none of it is machine-checked. A formal proof would also be the first formal instance on Prove2Me of the full ODE method for a concrete stochastic algorithm. That pipeline runs from the iterates to a perturbed Euler scheme, then to an asymptotic pseudotrajectory and a limit-set theorem, and finally to convergence. Benaïm's general results are being formalized separately on the platform (the asymptotic-pseudotrajectory definition this mission reuses comes from that effort), so the two developments meet in milestone 3.

Difficulty

The obvious approach is to treat Adam as stochastic gradient descent with a preconditioner and run a descent-lemma argument on F(xn)F(x_n)F(xn​). This fails because the update direction m^n/(ε+v^n)\hat m_n/(\varepsilon + \sqrt{\hat v_n})m^n​/(ε+v^n​​) is not a descent direction for FFF: mnm_nmn​ lags behind ∇F(xn)\nabla F(x_n)∇F(xn​). Any Lyapunov function has to involve the momentum, and FFF alone does not decrease. The continuous-time energy V∞V_\inftyV∞​ decreases only weakly, and only its perturbation WδW_\deltaWδ​ is strict, and only on compact sets. A second difficulty is that the stepsize-dependent coefficients (1−αn+1)/γn+1(1-\alpha_{n+1})/\gamma_{n+1}(1−αn+1​)/γn+1​ and the bias corrections rnr_nrn​, rˉn\bar r_nrˉn​ make the recursion a non-autonomous perturbation of the Euler scheme of h∞h_\inftyh∞​. The remainder must be shown to vanish along almost every path before the general theory applies. Finally, the abstract limit-set theorem needs the semiflow to be well defined on the closed set Z+\mathcal Z_+Z+​, where h∞h_\inftyh∞​ involves v\sqrt vv​ and is not Lipschitz at v=0v = 0v=0.

Formalization scope

Rd\mathbb R^dRd is EuclideanSpace ℝ (Fin d). The state space Z\mathcal ZZ is the product of three copies with Mathlib's max-of-factors norm, which is equivalent to the Euclidean norm of R3d\mathbb R^{3d}R3d and leaves boundedness and limits unchanged. FFF and SSS are Bochner integrals against the law μ\muμ. Assumption 2.2 makes the integrands integrable, so they are the true expectations. Moments are lower Lebesgue integrals. Algorithm 5.1 is a pathwise recursion adamIter, driven by a random sequence ξ:N→Ω→Ξ\xi : \mathbb N \to \Omega \to \Xiξ:N→Ω→Ξ whose index 000 is unused. The iid hypothesis is iIndepFun with every ξn+1\xi_{n+1}ξn+1​ of law μ\muμ.

Two hypotheses are left implicit on the page and are added explicitly. The first is ε>0\varepsilon > 0ε>0. The second is α1<1\alpha_1 < 1α1​<1 and β1<1\beta_1 < 1β1​<1: Assumption 5.1 iii) permits α1=1\alpha_1 = 1α1​=1, which gives r1=0r_1 = 0r1​=0, and Lean's division by zero returns 000. Lemma 9.1 is stated with γn>0\gamma_n > 0γn​>0 and ∑γn=∞\sum\gamma_n = \infty∑γn​=∞, which its convergence claim needs. Semiflows are Flow ℝ≥0 on the subtype Z+\mathcal Z_+Z+​, and the asymptotic pseudotrajectory and limit set are the published StochApproxDyn.LimitSet definitions. The ODE milestones 4–6 hold for abstract FFF, SSS under Assumptions 7.1–7.2 with 0<b≤4a0 < b \le 4a0<b≤4a, as in the paper's §7.

The goal concerns the iterates of Algorithm 5.1 itself. A statement about trajectories of (ODE∞)(\mathrm{ODE}_\infty)(ODE∞​), or one that replaces almost-sure boundedness by a bound uniform in ω\omegaω together with deterministic gradients, is a different and much weaker theorem and is ruled out. The interior condition on F(S)F(\mathcal S)F(S) and the countability of S\mathcal SS enter exactly as on the page.

A complete development needs the following:

  • well-posedness of the ODE on a closed set with a non-Lipschitz field;
  • the Benaïm machinery: martingale-noise control via Doob's convergence theorem, and the passage from vanishing perturbations to asymptotic pseudotrajectories;
  • the limit-set theorem for strict Lyapunov functions.

The last two are reusable for any stochastic approximation scheme, and contributions to them are welcome independently of Adam.

Selected references

  • A. Barakat, P. Bianchi, Convergence and Dynamical Behavior of the ADAM Algorithm for Nonconvex Stochastic Optimization, SIAM J. Math. Data Sci. 3(1), 2021; arXiv:1810.02263v4. https://arxiv.org/abs/1810.02263
  • M. Benaïm, Dynamics of stochastic approximation algorithms, Séminaire de Probabilités XXXIII, LNM 1709, Springer, 1999. https://doi.org/10.1007/BFb0096509
  • D. P. Kingma, J. Ba, Adam: A Method for Stochastic Optimization, ICLR 2015. https://arxiv.org/abs/1412.6980
  • S. J. Reddi, S. Kale, S. Kumar, On the Convergence of Adam and Beyond, ICLR 2018. https://arxiv.org/abs/1904.09237
16 thms2 active usersReviewed
Numerical AnalysisOptimal TransportOptimization·Captain: mikedeng1

Computational Optimal Transport VIII: Sinkhorn's Scalings Contract by λ(K)² per Iteration in Hilbert's Projective MetricTextbook

Why Sinkhorn's algorithm needs a convergence rate

Sinkhorn's algorithm is the standard method for computing entropically regularized optimal transport between two histograms. It is used throughout machine learning, statistics, imaging and graphics because each iteration is two matrix–vector products followed by entrywise divisions, which parallelize on GPUs and differentiate easily. The same iteration, under the names matrix scaling, RAS or iterative proportional fitting, has been used since the 1930s in transportation planning, economics and contingency-table analysis. Whether the iteration converges, and how fast, determines how many iterations a practitioner must run and which stopping criteria are reliable.

This mission formalizes the global convergence analysis of Sinkhorn's algorithm as presented in Chapter 4 (§4.2, Remarks 4.12–4.14) of G. Peyré and M. Cuturi, Computational Optimal Transport (2019).

Timeline. Sinkhorn (1964) proved that a positive matrix can be scaled to be doubly stochastic, and Sinkhorn and Knopp (1967) extended this to prescribed marginals. Birkhoff (1957) and Samelson (1957) independently showed that a positive linear map is a strict contraction of the positive cone for Hilbert's projective metric, with an explicit ratio. Franklin and Lorenz (1989) combined the two to prove that Sinkhorn's scalings converge linearly in Hilbert's metric at rate λ(K)2\lambda(K)^2λ(K)2. Linial, Samorodnitsky and Wigderson (1998) related the iteration to the permanent.

Setting

Fix n,m≥1n, m \ge 1n,m≥1, a matrix K∈Rn×mK \in \mathbb{R}^{n\times m}K∈Rn×m with positive entries (in the book, the Gibbs kernel Ki,j=e−Ci,j/εK_{i,j} = e^{-C_{i,j}/\varepsilon}Ki,j​=e−Ci,j​/ε), and vectors a∈Rna \in \mathbb{R}^na∈Rn, b∈Rmb \in \mathbb{R}^mb∈Rm with positive entries. Products ⊙\odot⊙ and quotients of vectors are entrywise. The regularized transport plan has the form P=diag(u)Kdiag(v)P = \mathrm{diag}(u)K\mathrm{diag}(v)P=diag(u)Kdiag(v), where the scalings u,vu, vu,v solve

u⊙(Kv)=a,v⊙(K⊤u)=b.(4.14)u \odot (Kv) = a, \qquad v \odot (K^\top u) = b. \tag{4.14}u⊙(Kv)=a,v⊙(K⊤u)=b.(4.14)

Sinkhorn's algorithm alternately enforces each equation:

v(0)=1m,u(ℓ+1)=aKv(ℓ),v(ℓ+1)=bK⊤u(ℓ+1).(4.15)v^{(0)} = \mathbb{1}_m,\qquad u^{(\ell+1)} = \frac{a}{Kv^{(\ell)}},\qquad v^{(\ell+1)} = \frac{b}{K^\top u^{(\ell+1)}}. \tag{4.15}v(0)=1m​,u(ℓ+1)=Kv(ℓ)a​,v(ℓ+1)=K⊤u(ℓ+1)b​.(4.15)

For vectors with positive entries, Hilbert's projective metric is

dH(u,u′)=log⁡max⁡i,juiuj′ujui′,d_{\mathcal H}(u,u') = \log\max_{i,j}\frac{u_i u'_j}{u_j u'_i},dH​(u,u′)=logi,jmax​uj​ui′​ui​uj′​​,

which vanishes exactly when uuu and u′u'u′ are proportional, and equals the variation seminorm ∥log⁡u−log⁡u′∥var\|\log u - \log u'\|_{\mathrm{var}}∥logu−logu′∥var​ with ∥f∥var=max⁡ifi−min⁡ifi\|f\|_{\mathrm{var}} = \max_i f_i - \min_i f_i∥f∥var​=maxi​fi​−mini​fi​. Birkhoff's constants of KKK are

η(K)=max⁡i,j,k,ℓKi,kKj,ℓKj,kKi,ℓ,λ(K)=η(K)−1η(K)+1.\eta(K) = \max_{i,j,k,\ell}\frac{K_{i,k}K_{j,\ell}}{K_{j,k}K_{i,\ell}},\qquad \lambda(K) = \frac{\sqrt{\eta(K)}-1}{\sqrt{\eta(K)}+1}.η(K)=i,j,k,ℓmax​Kj,k​Ki,ℓ​Ki,k​Kj,ℓ​​,λ(K)=η(K)​+1η(K)​−1​.

Write P(ℓ)=diag(u(ℓ))Kdiag(v(ℓ))P^{(\ell)} = \mathrm{diag}(u^{(\ell)})K\mathrm{diag}(v^{(\ell)})P(ℓ)=diag(u(ℓ))Kdiag(v(ℓ)) and P⋆=diag(u⋆)Kdiag(v⋆)P^\star = \mathrm{diag}(u^\star)K\mathrm{diag}(v^\star)P⋆=diag(u⋆)Kdiag(v⋆) for a positive solution (u⋆,v⋆)(u^\star, v^\star)(u⋆,v⋆) of (4.14).

Formalization targets

Goal: Theorem 4.2 (Franklin–Lorenz)

For every ℓ\ellℓ,

dH(v(ℓ),v⋆)≤λ(K)2ℓ dH(v(0),v⋆),dH(u(ℓ+1),u⋆)≤λ(K)2ℓ+1 dH(v(0),v⋆),d_{\mathcal H}(v^{(\ell)}, v^\star) \le \lambda(K)^{2\ell}\, d_{\mathcal H}(v^{(0)}, v^\star),\qquad d_{\mathcal H}(u^{(\ell+1)}, u^\star) \le \lambda(K)^{2\ell+1}\, d_{\mathcal H}(v^{(0)}, v^\star),dH​(v(ℓ),v⋆)≤λ(K)2ℓdH​(v(0),v⋆),dH​(u(ℓ+1),u⋆)≤λ(K)2ℓ+1dH​(v(0),v⋆),

so both distances tend to 000; for ℓ≥1\ell \ge 1ℓ≥1,

dH(u(ℓ),u⋆)≤dH(P(ℓ)1m,a)1−λ(K)2,max⁡i,j∣log⁡Pi,j(ℓ)−log⁡Pi,j⋆∣≤dH(u(ℓ),u⋆)+dH(v(ℓ),v⋆),d_{\mathcal H}(u^{(\ell)}, u^\star) \le \frac{d_{\mathcal H}(P^{(\ell)}\mathbb{1}_m, a)}{1-\lambda(K)^2},\qquad \max_{i,j}\big|\log P^{(\ell)}_{i,j} - \log P^\star_{i,j}\big| \le d_{\mathcal H}(u^{(\ell)},u^\star) + d_{\mathcal H}(v^{(\ell)},v^\star),dH​(u(ℓ),u⋆)≤1−λ(K)2dH​(P(ℓ)1m​,a)​,i,jmax​​logPi,j(ℓ)​−logPi,j⋆​​≤dH​(u(ℓ),u⋆)+dH​(v(ℓ),v⋆),

and for ℓ≥0\ell \ge 0ℓ≥0 the mirrored bound for v(ℓ)v^{(\ell)}v(ℓ) with the half-step coupling diag(u(ℓ+1))Kdiag(v(ℓ))\mathrm{diag}(u^{(\ell+1)})K\mathrm{diag}(v^{(\ell)})diag(u(ℓ+1))Kdiag(v(ℓ)). The book writes the rate as O(λ(K)2ℓ)O(\lambda(K)^{2\ell})O(λ(K)2ℓ); the constants stated are those its proof yields.

Milestones, in attack order

  1. (4.21): dH(u,u′)=∥log⁡u−log⁡u′∥vard_{\mathcal H}(u,u') = \|\log u - \log u'\|_{\mathrm{var}}dH​(u,u′)=∥logu−logu′∥var​.
  2. Remark 4.12: dHd_{\mathcal H}dH​ is a distance on rays (triangle inequality; zero exactly on proportional vectors).
  3. Invariances: dH(v,v′)=dH(v/v′,1)=dH(1/v,1/v′)d_{\mathcal H}(v,v') = d_{\mathcal H}(v/v',\mathbb{1}) = d_{\mathcal H}(\mathbb{1}/v,\mathbb{1}/v')dH​(v,v′)=dH​(v/v′,1)=dH​(1/v,1/v′) and invariance under positive diagonal scaling.
  4. Theorem 4.1 (Birkhoff–Hopf): dH(Kv,Kv′)≤λ(K) dH(v,v′)d_{\mathcal H}(Kv,Kv') \le \lambda(K)\,d_{\mathcal H}(v,v')dH​(Kv,Kv′)≤λ(K)dH​(v,v′) and λ(K)<1\lambda(K) < 1λ(K)<1.
  5. One half-step contracts by λ(K)\lambda(K)λ(K): dH(u(ℓ+1),u⋆)≤λ(K) dH(v(ℓ),v⋆)d_{\mathcal H}(u^{(\ell+1)},u^\star) \le \lambda(K)\,d_{\mathcal H}(v^{(\ell)},v^\star)dH​(u(ℓ+1),u⋆)≤λ(K)dH​(v(ℓ),v⋆), and the vvv-analogue.
  6. The display behind (4.23): dH(u(ℓ),u⋆)≤dH(a,u(ℓ)⊙Kv(ℓ))+λ(K)2dH(u(ℓ),u⋆)d_{\mathcal H}(u^{(\ell)},u^\star) \le d_{\mathcal H}(a, u^{(\ell)}\odot Kv^{(\ell)}) + \lambda(K)^2 d_{\mathcal H}(u^{(\ell)},u^\star)dH​(u(ℓ),u⋆)≤dH​(a,u(ℓ)⊙Kv(ℓ))+λ(K)2dH​(u(ℓ),u⋆), and its mirror.

Significance

The result. Theorem 4.2 gives a global linear rate that holds from the first iteration, independent of the starting point, with a ratio computable from KKK alone. Through (4.21) the rate transfers to the dual potentials εlog⁡u(ℓ)\varepsilon\log u^{(\ell)}εlogu(ℓ), and the bounds (4.23) turn the marginal violations P(ℓ)1mP^{(\ell)}\mathbb{1}_mP(ℓ)1m​ versus aaa — which are computed anyway — into certified stopping criteria. Theorem 4.1 alone also yields a quantitative Perron–Frobenius theorem (Remark 4.13).

Formalizing it. All results here are classical and proved in the literature (Birkhoff 1957; Franklin–Lorenz 1989); Theorem 4.1 is quoted by the book without proof. To our knowledge neither Hilbert's projective metric on the positive cone, the Birkhoff–Hopf theorem, nor the convergence rate of Sinkhorn's algorithm has a machine-checked proof in Mathlib. The mission produces a reusable development of dHd_{\mathcal H}dH​ and Birkhoff's contraction theorem, and a certified rate and stopping rule for the most widely used optimal-transport solver.

Difficulty

Milestones 1–3 and 5–6 are finite manipulations of maxima of ratios. The central difficulty is Theorem 4.1. The obvious argument — bound (Kv)i/(Kv′)i(Kv)_i/(Kv')_i(Kv)i​/(Kv′)i​ entrywise by max⁡kvk/vk′\max_k v_k/v'_kmaxk​vk​/vk′​ — only shows that KKK is non-expansive (λ≤1\lambda \le 1λ≤1); it loses the strict contraction entirely. The sharp ratio (η−1)/(η+1)(\sqrt{\eta}-1)/(\sqrt{\eta}+1)(η​−1)/(η​+1) depends on how KKK acts on the two vectors jointly, which the entrywise argument does not see. The book quotes the theorem without proof, so the argument has to be supplied from Birkhoff's original paper or later expositions.

Formalization scope

Lean namespace CompOT.Sinkhorn. Indices are 000-based (Fin n, Fin m); vectors are functions Fin n → ℝ, matrices Matrix (Fin n) (Fin m) ℝ, and ⊙\odot⊙ and /// are the pointwise operations on functions. dHd_{\mathcal H}dH​, η\etaη and ∥⋅∥var\|\cdot\|_{\mathrm{var}}∥⋅∥var​ use real iSup/iInf over the finite index sets. Every theorem assumes 0<n0 < n0<n, 0<m0 < m0<m and positive entries for KKK, aaa, bbb and every vector passed to dHd_{\mathcal H}dH​, which excludes Lean's junk values for division by zero and logarithms of non-positive numbers. Sinkhorn's algorithm is a run predicate (IsSinkhornRun) stating the recursion (4.15) with v(0)=1mv^{(0)} = \mathbb{1}_mv(0)=1m​; u(0)u^{(0)}u(0) is not an iterate and is unconstrained. The solution (u⋆,v⋆)(u^\star, v^\star)(u⋆,v⋆) is any positive solution of (4.14), taken as a hypothesis (it exists iff ∑iai=∑jbj\sum_i a_i = \sum_j b_j∑i​ai​=∑j​bj​); the book's histograms are probability vectors, but only positivity is used. P⋆P^\starP⋆ in (4.24) is diag(u⋆)Kdiag(v⋆)\mathrm{diag}(u^\star)K\mathrm{diag}(v^\star)diag(u⋆)Kdiag(v⋆), the solution of the regularized problem (4.2) by Proposition 4.3.

Deviations from the print, all disclosed in the items: the O(⋅)O(\cdot)O(⋅) of (4.22) is replaced by the proof's explicit constants; the limit (u(ℓ),v(ℓ))→(u⋆,v⋆)(u^{(\ell)}, v^{(\ell)}) \to (u^\star, v^\star)(u(ℓ),v(ℓ))→(u⋆,v⋆) is stated as dH→0d_{\mathcal H} \to 0dH​→0, since scalings are determined only up to (ru,v/r)(ru, v/r)(ru,v/r); the second inequality of (4.23) is false as printed (its right side is 000 for ℓ≥1\ell \ge 1ℓ≥1) and is stated with the half-step coupling the mirrored proof gives.

A definition of dHd_{\mathcal H}dH​ without the logarithm, or as a maximum of single-entry ratios, would make the statements different theorems; the definition here is the book's, and (4.21) is a milestone checking it against the variation seminorm. A run predicate not tied to KKK, aaa, bbb, or an unsatisfiable one, is ruled out: the predicate is the recursion itself and is inhabited for every KKK, aaa, bbb.

Contributions welcome: the metric properties of dHd_{\mathcal H}dH​ (reusable for Perron–Frobenius and nonlinear consensus), a proof of the Birkhoff–Hopf theorem, and positivity lemmas for Sinkhorn iterates.

Selected references

  • G. Peyré, M. Cuturi, Computational Optimal Transport, Foundations and Trends in Machine Learning 11(5–6):355–607, 2019. https://doi.org/10.1561/2200000073 (§4.2, pp. 431–443)
  • J. Franklin, J. Lorenz, On the scaling of multidimensional matrices, Linear Algebra and its Applications 114/115:717–735, 1989. https://doi.org/10.1016/0024-3795(89)90490-4
  • G. Birkhoff, Extensions of Jentzsch's theorem, Transactions of the AMS 85(1):219–227, 1957. https://doi.org/10.1090/S0002-9947-1957-0087058-6
  • H. Samelson, On the Perron–Frobenius theorem, Michigan Mathematical Journal 4(1):57–59, 1957. https://doi.org/10.1307/mmj/1028990177
  • R. Sinkhorn, A relationship between arbitrary positive matrices and doubly stochastic matrices, Annals of Mathematical Statistics 35(2):876–879, 1964. https://doi.org/10.1214/aoms/1177703591
  • N. Linial, A. Samorodnitsky, A. Wigderson, A deterministic strongly polynomial algorithm for matrix scaling and approximate permanents, STOC 1998. https://doi.org/10.1145/276698.276880
8 thms2 active usersReviewed
Convex OptimizationOptimization·Captain: mikedeng1

Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization 3: Under Refined Dual Strong Convexity, SDCA Reaches Expected Dual Sub-optimality ε after 2(n/s)·log(2/ε) IterationsResearch Paper

Motivation

Regularized loss minimization covers support vector machines, logistic and ridge regression, and the absolute-deviation regression of robust statistics. Stochastic dual coordinate ascent (SDCA) solves these problems by maximizing the Fenchel dual one randomly chosen coordinate at a time. Shalev-Shwartz and Zhang (arXiv:1209.1873, JMLR 2013) gave the first analysis of SDCA in terms of the duality gap. For smooth losses their rate is linear; for Lipschitz but non-smooth losses such as the hinge loss it is of order 1/ϵ1/\epsilon1/ϵ, which is no better than stochastic gradient descent, although SDCA is observed to converge much faster than SGD on SVM problems asymptotically.

Section 5 of the paper explains that observation. Many non-smooth losses are smooth at most points: the hinge loss max⁡(0,1−uyi)\max(0,1-uy_i)max(0,1−uyi​) fails to be smooth only at uyi=1uy_i=1uyi​=1. If, at the optimum, most margins stay away from the kink, the dual is strongly concave in most coordinates, and SDCA converges linearly in the dual objective. Earlier linear-rate results for dual coordinate ascent on SVMs (Luo and Tseng, 1992) depend on the smallest nonzero eigenvalue of the data Gram matrix, which can be arbitrarily small when two data points are close; the condition of Section 5 avoids the Gram matrix.

Setting

Data x1,…,xn∈Rdx_1,\dots,x_n\in\mathbb R^dx1​,…,xn​∈Rd (Euclidean norm), scalar convex losses φ1,…,φn:R→R\varphi_1,\dots,\varphi_n:\mathbb R\to\mathbb Rφ1​,…,φn​:R→R and λ>0\lambda>0λ>0 define the primal problem

min⁡w∈RdP(w),P(w)=1n∑i=1nφi(w⊤xi)+λ2∥w∥2.\min_{w\in\mathbb R^d} P(w),\qquad P(w)=\frac1n\sum_{i=1}^n\varphi_i(w^\top x_i)+\frac\lambda2\|w\|^2 .w∈Rdmin​P(w),P(w)=n1​i=1∑n​φi​(w⊤xi​)+2λ​∥w∥2.

The convex conjugate of φi\varphi_iφi​ is φi∗(u)=sup⁡z(zu−φi(z))∈(−∞,+∞]\varphi_i^*(u)=\sup_z(zu-\varphi_i(z))\in(-\infty,+\infty]φi∗​(u)=supz​(zu−φi​(z))∈(−∞,+∞]. The dual problem is max⁡α∈RnD(α)\max_{\alpha\in\mathbb R^n}D(\alpha)maxα∈Rn​D(α) with

D(α)=1n∑i=1n−φi∗(−αi)−λ2∥w(α)∥2,w(α)=1λn∑i=1nαixi.D(\alpha)=\frac1n\sum_{i=1}^n-\varphi_i^*(-\alpha_i)-\frac\lambda2\|w(\alpha)\|^2,\qquad w(\alpha)=\frac{1}{\lambda n}\sum_{i=1}^n\alpha_ix_i .D(α)=n1​i=1∑n​−φi∗​(−αi​)−2λ​∥w(α)∥2,w(α)=λn1​i=1∑n​αi​xi​.

A dual variable is feasible when every φi∗(−αi)\varphi_i^*(-\alpha_i)φi∗​(−αi​) is finite. A function is LLL-Lipschitz if ∣φi(a)−φi(b)∣≤L∣a−b∣|\varphi_i(a)-\varphi_i(b)|\le L|a-b|∣φi​(a)−φi​(b)∣≤L∣a−b∣. The standing assumptions are ∥xi∥≤1\|x_i\|\le1∥xi​∥≤1, φi≥0\varphi_i\ge0φi​≥0 and φi(0)≤1\varphi_i(0)\le1φi​(0)≤1.

Procedure SDCA starts from α(0)=0\alpha^{(0)}=0α(0)=0. At iteration ttt it picks iii uniformly at random, independently of the past, chooses Δαi\Delta\alpha_iΔαi​ maximizing −φi∗(−(αi(t−1)+Δαi))−λn2∥w(α(t−1))+(λn)−1Δαixi∥2-\varphi_i^*(-(\alpha_i^{(t-1)}+\Delta\alpha_i))-\frac{\lambda n}{2}\|w(\alpha^{(t-1)})+(\lambda n)^{-1}\Delta\alpha_ix_i\|^2−φi∗​(−(αi(t−1)​+Δαi​))−2λn​∥w(α(t−1))+(λn)−1Δαi​xi​∥2, and sets α(t)=α(t−1)+Δαiei\alpha^{(t)}=\alpha^{(t-1)}+\Delta\alpha_ie_iα(t)=α(t−1)+Δαi​ei​. Let α∗\alpha^*α∗ be a dual optimum and w∗=w(α∗)w^*=w(\alpha^*)w∗=w(α∗).

The refined dual strong convexity condition (Definition 2, (4)) asks for functions γi(⋅)≥0\gamma_i(\cdot)\ge0γi​(⋅)≥0 with

φi∗(−a)−φi∗(−b)+u(a−b)≥γi(u)∣a−b∣2for all a,b and u∈∂φi∗(−b).\varphi_i^*(-a)-\varphi_i^*(-b)+u(a-b)\ge\gamma_i(u)|a-b|^2\qquad\text{for all }a,b\text{ and }u\in\partial\varphi_i^*(-b).φi∗​(−a)−φi∗​(−b)+u(a−b)≥γi​(u)∣a−b∣2for all a,b and u∈∂φi∗​(−b).

For the hinge loss one may take γi(u)=∣uyi−1∣\gamma_i(u)=|uy_i-1|γi​(u)=∣uyi​−1∣. With constants γi≥0\gamma_i\ge0γi​≥0, the dual strong convexity inequality (5) is

D(α∗)−D(α)≥1n∑i=1nγi∣αi−αi∗∣2+λ2∥w(α)−w∗∥2for every feasible α,D(\alpha^*)-D(\alpha)\ge\frac1n\sum_{i=1}^n\gamma_i|\alpha_i-\alpha_i^*|^2+\frac\lambda2\|w(\alpha)-w^*\|^2\quad\text{for every feasible }\alpha,D(α∗)−D(α)≥n1​i=1∑n​γi​∣αi​−αi∗​∣2+2λ​∥w(α)−w∗∥2for every feasible α,

and N(u)=#{i:γi<u}N(u)=\#\{i:\gamma_i<u\}N(u)=#{i:γi​<u} counts the coordinates with small modulus.

Formalization targets

Goal: Theorem 5 (p. 9)

Assume convex LLL-Lipschitz losses, the standing assumptions, and (5) with constants γi≥0\gamma_i\ge0γi​≥0. For every ϵD>0\epsilon_D>0ϵD​>0 and s∈(0,1]s\in(0,1]s∈(0,1] with ϵD≥8L2sλnN(sλn)/n\epsilon_D\ge 8L^2\frac{s}{\lambda n}N(\frac{s}{\lambda n})/nϵD​≥8L2λns​N(λns​)/n,

t≥2nslog⁡2ϵD⟹E[D(α∗)−D(α(t))]≤ϵD.t\ge2\frac ns\log\frac2{\epsilon_D}\quad\Longrightarrow\quad\mathbb E\bigl[D(\alpha^*)-D(\alpha^{(t)})\bigr]\le\epsilon_D .t≥2sn​logϵD​2​⟹E[D(α∗)−D(α(t))]≤ϵD​.

The constants 222 and 888 are the paper's.

Milestones

  1. Proposition 1 (pp. 8–9): condition (4), with γi=γi(w∗⊤xi)\gamma_i=\gamma_i(w^{*\top}x_i)γi​=γi​(w∗⊤xi​), implies (5), and ∣(w∗−w)⊤xi∣≥γi∣ai−αi∗∣|(w^*-w)^\top x_i|\ge\gamma_i|a_i-\alpha_i^*|∣(w∗−w)⊤xi​∣≥γi​∣ai​−αi∗​∣ whenever −ai∈∂φi(w⊤xi)-a_i\in\partial\varphi_i(w^\top x_i)−ai​∈∂φi​(w⊤xi​) (6).
  2. Lemma 5 (p. 19): under (5), one step raises the expected dual by at least s2n\frac{s}{2n}2ns​ times the dual sub-optimality, plus 3sλ4n∥w∗−w∥2\frac{3s\lambda}{4n}\|w^*-w\|^24n3sλ​∥w∗−w∥2, minus (sn)2G∗(s)/(2λ)(\frac sn)^2G_*(s)/(2\lambda)(ns​)2G∗​(s)/(2λ) with G∗(s)=1n∑i(∥xi∥2−γiλn/s)(αi∗−αi)2G_*(s)=\frac1n\sum_i(\|x_i\|^2-\gamma_i\lambda n/s)(\alpha_i^*-\alpha_i)^2G∗​(s)=n1​∑i​(∥xi​∥2−γi​λn/s)(αi∗​−αi​)2.
  3. Lemma 3 (p. 15): the conjugate of an LLL-Lipschitz function is +∞+\infty+∞ outside [−L,L][-L,L][−L,L].
  4. Lemma 6 (p. 20): G∗(s)≤4L2N(s/(λn))/nG_*(s)\le4L^2N(s/(\lambda n))/nG∗​(s)≤4L2N(s/(λn))/n.
  5. Dual recursion (§7.6, p. 20): ϵD(t)≤(1−s2n)ϵD(t−1)+(sn)2G∗(s)2λ\epsilon_D^{(t)}\le(1-\frac s{2n})\epsilon_D^{(t-1)}+(\frac sn)^2\frac{G_*(s)}{2\lambda}ϵD(t)​≤(1−2ns​)ϵD(t−1)​+(ns​)22λG∗​(s)​ and ϵD(t)≤(1−s2n)tϵD(0)+snG∗(s)λ\epsilon_D^{(t)}\le(1-\frac s{2n})^t\epsilon_D^{(0)}+\frac sn\frac{G_*(s)}\lambdaϵD(t)​≤(1−2ns​)tϵD(0)​+ns​λG∗​(s)​, with G∗(s)=4L2N(s/(λn))/nG_*(s)=4L^2N(s/(\lambda n))/nG∗​(s)=4L2N(s/(λn))/n.
  6. Lemma 2 (p. 13): D(α)≤P(w∗)≤P(0)≤1D(\alpha)\le P(w^*)\le P(0)\le1D(α)≤P(w∗)≤P(0)≤1 and D(0)≥0D(0)\ge0D(0)≥0.

Significance

Theorem 5 gives a linear rate for the dual sub-optimality of SDCA on non-smooth losses whenever few constants γi\gamma_iγi​ are small. If N(s0/(λn))=0N(s_0/(\lambda n))=0N(s0​/(λn))=0 for some s0>0s_0>0s0​>0 (for the SVM, λn∣w∗⊤xiyi−1∣≥s0\lambda n|w^{*\top}x_iy_i-1|\ge s_0λn∣w∗⊤xi​yi​−1∣≥s0​ for all iii), the iteration count is (2n/s0)log⁡(2/ϵD)(2n/s_0)\log(2/\epsilon_D)(2n/s0​)log(2/ϵD​), compared with order n+L2/(λϵ)n+L^2/(\lambda\epsilon)n+L2/(λϵ) from the Lipschitz analysis of Theorem 1 of the same paper. The paper uses Theorem 5 to explain the observed asymptotic advantage of SDCA over SGD, and its Theorem 6 converts it into a duality-gap bound.

The result is proved in the paper. It has not been formalized: no SDCA convergence result is on Prove2Me, and Mathlib has no conjugate-duality theory for this problem class. The mission produces a formal account of the dual objective in extended reals, of the exact coordinate maximization step, and of the averaging over uniformly random coordinate sequences, together with a checked proof of the rate. The companion missions of this series cover Theorem 1 (Lipschitz losses) and Theorem 2 (smooth losses).

Difficulty

Condition (5) gives a modulus γi\gamma_iγi​ per coordinate, and some γi\gamma_iγi​ may be zero, so the dual is not strongly concave and the standard linear-rate argument for strongly concave coordinate ascent does not apply. The per-step progress has a variance term with the sign of ∥xi∥2−γiλn/s\|x_i\|^2-\gamma_i\lambda n/s∥xi​∥2−γi​λn/s, which is positive exactly on the coordinates with small modulus; that term must be controlled by bounding the dual variables themselves, which requires the Lipschitz structure through the conjugate. The extended-real-valued conjugate is a second source of work: every quantity must be shown finite along the run before it can be manipulated as a real number.

Formalization scope

Lean works in EuclideanSpace ℝ (Fin d), with nnn data points indexed by Fin n, n>0n>0n>0. The conjugate is an EReal supremum and DDD is EReal-valued, −∞-\infty−∞ exactly at infeasible points. Real-valued expectations of dual values appear only together with a conjunct stating their finiteness. The SDCA step is a map Δ\DeltaΔ with the hypothesis that Δ(α,i)\Delta(\alpha,i)Δ(α,i) is an exact maximizer of the coordinate objective. The run is a fold over the sequence of chosen coordinates, and the expectation is the uniform average over all sequences of length ttt (SAGA.Convex.expectIdx, a published definition). The optimum α∗\alpha^*α∗ is a hypothesis D(β)≤D(α∗)D(\beta)\le D(\alpha^*)D(β)≤D(α∗) for all β\betaβ; in (5), w∗w^*w∗ is w(α∗)w(\alpha^*)w(α∗). Inequality (5) quantifies over feasible α\alphaα only. Lemmas 5 and 6 are stated for an arbitrary feasible current state, which is stronger than the history-averaged form printed. The paper's range s∈[0,1]s\in[0,1]s∈[0,1] becomes s∈(0,1]s\in(0,1]s∈(0,1], because at s=0s=0s=0 the iteration bound 2(n/s)log⁡(2/ϵD)2(n/s)\log(2/\epsilon_D)2(n/s)log(2/ϵD​) is undefined. A formalization that admits s=0s=0s=0 would evaluate it to 000 and claim the bound at t=0t=0t=0, which is false.

The development needs elementary facts about conjugates of scalar functions in EReal (convexity, the Fenchel–Young inequality, the Lipschitz domain bound), the weak duality inequality D≤PD\le PD≤P, and finite averages over coordinate sequences. The conjugate facts and the expectation calculus are reusable by the other two missions of the series. Proofs of individual milestones are welcome independently.

Selected references

  • S. Shalev-Shwartz and T. Zhang, Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization, Journal of Machine Learning Research 14 (2013), 567–599; preprint arXiv:1209.1873v2. https://arxiv.org/abs/1209.1873
  • Z.-Q. Luo and P. Tseng, On the convergence of coordinate descent method for convex differentiable minimization, Journal of Optimization Theory and Applications 72 (1992), 7–35. https://doi.org/10.1007/BF00939948
  • C.-J. Hsieh, K.-W. Chang, C.-J. Lin, S. S. Keerthi and S. Sundararajan, A dual coordinate descent method for large-scale linear SVM, ICML 2008. https://doi.org/10.1145/1390156.1390208
13 thms2 active usersReviewed
Operations ResearchOptimal TransportOptimization·Captain: mikedeng1

Regularization via Mass Transportation I: Wasserstein-Robust Linear Classification with Label Flipping Is a Finite Convex ProgramResearch Paper

Motivation

A classifier trained from finitely many examples can perform well on those examples while responding poorly to changes in the feature distribution or to mislabeled observations. Distributionally robust learning addresses this by choosing a classifier against every probability distribution within a specified distance of the empirical distribution. The resulting optimization appears to range over an infinite-dimensional family of distributions. Shafieezadeh-Abadeh, Kuhn, and Mohajerin Esfahani show that, for linear classification with a convex Lipschitz loss, this problem has an exact finite convex reformulation. That result is Theorem 3.11(ii) of the 2019 arXiv preprint, which is the source used here.

The authors' transport metric allows a data point's features to move and its binary label to flip, with separate prices for those two changes. This matters when robustness to inaccurate labels is part of the model. A label-preserving metric cannot express that choice. The paper builds on its Section 2 learning model and the transport duality developed in Appendix A.1; the present mission isolates the classification result and the intermediate statements used for it. The preprint's Corollary 3.14 specializes the result to logloss, while its Corollary 3.12 treats hinge loss. A logloss-specific reformulation and a label-preserving classification result already have proved statements on Prove2Me; the general convex-Lipschitz, finite label-cost theorem remains the target of this mission.

Setting

Let VVV be a finite-dimensional real vector space with a chosen norm ∥⋅∥\|\cdot\|∥⋅∥. Features are x∈Vx\in Vx∈V and labels are y∈{−1,+1}y\in\{-1,+1\}y∈{−1,+1}. A linear classifier is a continuous linear functional w:V→Rw:V\to\mathbb Rw:V→R, with score ⟨w,x⟩=w(x)\langle w,x\rangle=w(x)⟨w,x⟩=w(x). Its dual norm ∥w∥∗\|w\|_*∥w∥∗​ is the operator norm induced by ∥⋅∥\|\cdot\|∥⋅∥. Given N≥1N\ge1N≥1 training pairs (x^i,y^i)(\hat x_i,\hat y_i)(x^i​,y^​i​), their empirical distribution is P^N=N−1∑i=1Nδ(x^i,y^i)\hat P_N=N^{-1}\sum_{i=1}^N\delta_{(\hat x_i,\hat y_i)}P^N​=N−1∑i=1N​δ(x^i​,y^​i​)​.

For κ>0\kappa>0κ>0, the transport cost between pairs is

d((x,y),(x′,y′))=∥x−x′∥+κ1y≠y′.d((x,y),(x',y'))=\|x-x'\|+\kappa\mathbf1_{y\ne y'}.d((x,y),(x′,y′))=∥x−x′∥+κ1y=y′​.

The Wasserstein distance W(Q,P)W(Q,P)W(Q,P) is the least expected transport cost among joint distributions whose two marginals are QQQ and PPP. For a radius ρ≥0\rho\ge0ρ≥0, the Wasserstein ball Bρ(P^N)\mathbb B_\rho(\hat P_N)Bρ​(P^N​) contains probability distributions QQQ satisfying W(Q,P^N)≤ρW(Q,\hat P_N)\le\rhoW(Q,P^N​)≤ρ. The loss is L(y⟨w,x⟩)L(y\langle w,x\rangle)L(y⟨w,x⟩), where L:R→[0,∞)L:\mathbb R\to[0,\infty)L:R→[0,∞) is convex and Lipschitz continuous. The Lipschitz modulus lip⁡(L)\operatorname{lip}(L)lip(L) is the smallest Lipschitz bound for LLL, defined as the supremum of its difference quotients. These are the objects and standing conventions of Sections 1.1, 2, and 3.2 of the preprint.

Formalization targets

The goal is Theorem 3.11(ii): for every fixed www, the worst-case expected loss equals the value of program (18),

sup⁡Q∈Bρ(P^N)EQ[L(y⟨w,x⟩)]=inf⁡λ,s{λρ+1N∑i=1Nsi:L(y^i⟨w,x^i⟩)≤si,L(−y^i⟨w,x^i⟩)−κλ≤si(i=1,…,N),lip⁡(L)∥w∥∗≤λ}.\sup_{Q\in\mathbb B_\rho(\hat P_N)}\mathbb E^Q[L(y\langle w,x\rangle)] = \inf_{\lambda,s}\left\{\lambda\rho+\frac1N\sum_{i=1}^{N}s_i: \begin{array}{l} L(\hat y_i\langle w,\hat x_i\rangle)\le s_i,\\ L(-\hat y_i\langle w,\hat x_i\rangle)-\kappa\lambda\le s_i\quad(i=1,\ldots,N),\\ \operatorname{lip}(L)\|w\|_*\le\lambda \end{array}\right\}.Q∈Bρ​(P^N​)sup​EQ[L(y⟨w,x⟩)]=λ,sinf​⎩⎨⎧​λρ+N1​i=1∑N​si​:L(y^​i​⟨w,x^i​⟩)≤si​,L(−y^​i​⟨w,x^i​⟩)−κλ≤si​(i=1,…,N),lip(L)∥w∥∗​≤λ​⎭⎬⎫​.

Taking the infimum of both sides over www gives equality between problem (4) and the full program (18). The fixed-www identity is stated explicitly because it is the stronger assertion established in the proof. The milestone list follows the paper's appendix: Lemma A.1 converts the robust expectation into a scalar transport-price infimum; the label split isolates the two possible labels; Lemma A.3 evaluates the feature supremum; the ensuing display gives the program in terms of conjugate slopes; and the final identity identifies those slopes with lip⁡(L)\operatorname{lip}(L)lip(L). Each milestone is tied to its printed page in the preprint.

Significance

The theorem replaces a search over probability distributions by a finite convex program with one scalar λ\lambdaλ and one slack variable sis_isi​ per sample, in addition to the classifier www. It makes the price of label changes explicit through κλ\kappa\lambdaκλ and the price of feature changes through lip⁡(L)∥w∥∗\operatorname{lip}(L)\|w\|_*lip(L)∥w∥∗​. Consequently, different convex Lipschitz classification losses can be treated within the same transport model rather than by deriving a new distributional optimization problem for each loss. The preprint applies the result to hinge, smoothed hinge, and logloss examples in the paragraphs following Theorem 3.11.

The result is proved in the cited paper. Its general statement and the paper's appendix milestones are not yet machine-checked in this mission. A verified development would connect the published definitions of the Wasserstein ball and Lipschitz modulus to an exact program-value theorem that later work can import. The loss-specific proved statements already available on Prove2Me provide useful comparisons, but their narrower losses or label-preserving transport costs do not replace this theorem.

Difficulty

The visible finite constraints contain no probability distributions, while the left-hand side optimizes over all probability measures in a Wasserstein ball. Establishing that the two values coincide requires control of the transport budget and the inner worst-case loss at the same time. The straightforward comparison of feasible values gives only one inequality. It does not show that every improvement allowed by a distribution can be represented by the scalar transport price and the samplewise constraints. The case ρ=0\rho=0ρ=0 also needs care: Lemma A.1's strong-duality proof cites a result for strictly positive radius, whereas the theorem itself allows zero radius and states an infimum rather than an attained minimum.

Formalization scope

Lean represents VVV by an abstract finite-dimensional real normed space with its Borel measurable structure. This retains the arbitrary norm of Rn\mathbb R^nRn in the paper. Labels are Bool, interpreted as +1+1+1 and −1-1−1; the published DRLogReg_Reformulation_Core definition supplies their sign map, the metric, Wasserstein distance, ball, and empirical distribution. The published WassersteinDRO_Duality_lipschitzModulus definition supplies the exact Lipschitz modulus. The mission's own definitions add the general loss, the feasible set of (18), the fixed-classifier program value, and the effective domain Θ={θ:L∗(θ)<∞}\Theta=\{\theta:L^*(\theta)<\infty\}Θ={θ:L∗(θ)<∞} of the convex conjugate. The same objects and conventions are used across all milestones.

All potentially infinite robust values are represented in [0,∞][0,\infty][0,∞]; the unbounded feature suprema in Lemma A.3 and the label split use extended reals. The standing loss condition L≥0L\ge0L≥0 comes from the paper's Section 2.1 loss codomain. It also makes the lower Lebesgue integral and the finite program's extended-nonnegative objective faithful. The sample size is positive, κ>0\kappa>0κ>0, and the goal allows ρ=0\rho=0ρ=0. Lemma A.1 separately assumes ρ>0\rho>0ρ>0, matching the strong-duality condition used on page 29. The printed extra index j∈[J]j\in[J]j∈[J] in the page-35 classification display is carried in its verbatim milestone provenance but not in the Lean statement, because no jjj exists in part (ii). The formalization does not obtain the theorem by restricting LLL to a constant loss, by taking a singleton feature space, or by substituting an arbitrary Lipschitz bound for its modulus. Contributions that establish transport duality, the conjugate-domain slope identity, or the zero-radius case are directly reusable.

Selected references

  • S. Shafieezadeh-Abadeh, D. Kuhn, and P. Mohajerin Esfahani, Regularization via Mass Transportation, arXiv preprint arXiv:1710.10016v3, 2019. Preprint.
  • S. Shafieezadeh-Abadeh, P. Mohajerin Esfahani, and D. Kuhn, Distributionally Robust Logistic Regression, Advances in Neural Information Processing Systems 28, 2015. Published Prove2Me formalization.
9 thms2 active usersReviewed
Convex OptimizationOptimization·Captain: mikedeng1

Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization 2: With (1/γ)-Smooth Losses, SDCA Reaches Expected Duality Gap ε after (n + 1/(λγ))·log((n + 1/(λγ))/ε) IterationsResearch Paper

Motivation

Regularized loss minimization (training a linear predictor with a convex loss and an ℓ2\ell_2ℓ2​ penalty) covers support vector machines, ridge and logistic regression, and much of classical linear learning. A standard way to solve these problems at scale is dual coordinate ascent: optimize the dual problem one coordinate, i.e. one training example, at a time. Hsieh et al. (ICML 2008) reported that it can outperform stochastic gradient descent when moderately high accuracy is needed. Its theory, however, was unsatisfying (paper, §2): the known linear rates, adapted from Luo and Tseng's analysis of coordinate ascent, have a contraction factor that may be arbitrarily close to 111 (it depends on the smallest nonzero eigenvalue of X⊤XX^\top XX⊤X), and they bound only the dual sub-optimality, which translates into a much weaker primal guarantee.

Shai Shalev-Shwartz and Tong Zhang (JMLR 14, 2013; arXiv:1209.1873) analysed stochastic dual coordinate ascent (SDCA), in which the coordinate is drawn uniformly at random, and bounded the expected duality gap, hence the primal sub-optimality, with explicit constants. For (1/γ)(1/\gamma)(1/γ)-smooth losses the bound is linear: (n+1λγ)log⁡((n+1λγ)/ϵ)(n+\tfrac1{\lambda\gamma})\log\big((n+\tfrac1{\lambda\gamma})/\epsilon\big)(n+λγ1​)log((n+λγ1​)/ϵ) iterations. This mission formalizes that result, the paper's Theorem 2.

Timeline:

  • 1992: Luo and Tseng prove linear convergence of coordinate descent methods for a class of convex problems, with an unspecified rate parameter.
  • 2008: Hsieh, Chang, Lin, Keerthi and Sundararajan analyse dual coordinate descent for linear SVMs (bounds on the dual objective).
  • 2012: Le Roux, Schmidt and Bach (SAG) obtain a linear rate for a primal stochastic method on smooth strongly convex finite sums.
  • 2012–2013: Shalev-Shwartz and Zhang prove duality-gap rates for SDCA: for LLL-Lipschitz losses (Theorem 1) and the linear rate above for smooth losses (Theorem 2).

Setting

Let x1,…,xn∈Rdx_1,\dots,x_n\in\mathbb R^dx1​,…,xn​∈Rd, let φ1,…,φn:R→R\varphi_1,\dots,\varphi_n:\mathbb R\to\mathbb Rφ1​,…,φn​:R→R be convex, and let λ>0\lambda>0λ>0. The primal problem is to minimize

P(w)=1n∑i=1nφi(w⊤xi)+λ2∥w∥2.P(w)=\frac1n\sum_{i=1}^n\varphi_i(w^\top x_i)+\frac\lambda2\|w\|^2 .P(w)=n1​i=1∑n​φi​(w⊤xi​)+2λ​∥w∥2.

The convex conjugate of φi\varphi_iφi​ is φi∗(u)=sup⁡z (zu−φi(z))∈(−∞,+∞]\varphi_i^*(u)=\sup_z\,(zu-\varphi_i(z))\in(-\infty,+\infty]φi∗​(u)=supz​(zu−φi​(z))∈(−∞,+∞]. The dual problem is to maximize

D(α)=1n∑i=1n−φi∗(−αi)−λ2∥1λn∑i=1nαixi∥2,α∈Rn,D(\alpha)=\frac1n\sum_{i=1}^n-\varphi_i^*(-\alpha_i)-\frac\lambda2\Big\|\frac1{\lambda n}\sum_{i=1}^n\alpha_ix_i\Big\|^2,\qquad \alpha\in\mathbb R^n,D(α)=n1​i=1∑n​−φi∗​(−αi​)−2λ​​λn1​i=1∑n​αi​xi​​2,α∈Rn,

and the dual point α\alphaα determines the primal point w(α)=1λn∑iαixiw(\alpha)=\frac1{\lambda n}\sum_i\alpha_ix_iw(α)=λn1​∑i​αi​xi​. Weak duality gives D(α)≤P(w)D(\alpha)\le P(w)D(α)≤P(w) for all α,w\alpha,wα,w, so the duality gap P(w(α))−D(α)P(w(\alpha))-D(\alpha)P(w(α))−D(α) bounds the primal sub-optimality of w(α)w(\alpha)w(α).

A loss is (1/γ)(1/\gamma)(1/γ)-smooth if it is differentiable with a (1/γ)(1/\gamma)(1/γ)-Lipschitz derivative. Procedure SDCA starts from α(0)=0\alpha^{(0)}=0α(0)=0. At iteration ttt it draws iii uniformly from {1,…,n}\{1,\dots,n\}{1,…,n}, independently of the past, chooses Δαi\Delta\alpha_iΔαi​ to maximize

−φi∗(−(αi(t−1)+Δαi))−λn2∥w(t−1)+(λn)−1Δαixi∥2,-\varphi_i^*\big(-(\alpha_i^{(t-1)}+\Delta\alpha_i)\big)-\frac{\lambda n}2\big\|w^{(t-1)}+(\lambda n)^{-1}\Delta\alpha_i x_i\big\|^2,−φi∗​(−(αi(t−1)​+Δαi​))−2λn​​w(t−1)+(λn)−1Δαi​xi​​2,

and sets α(t)=α(t−1)+Δαiei\alpha^{(t)}=\alpha^{(t-1)}+\Delta\alpha_ie_iα(t)=α(t−1)+Δαi​ei​, w(t)=w(α(t))w^{(t)}=w(\alpha^{(t)})w(t)=w(α(t)). Its outputs are the last iterate, the average αˉ=1T−T0∑t=T0+1Tα(t−1)\bar\alpha=\frac1{T-T_0}\sum_{t=T_0+1}^T\alpha^{(t-1)}αˉ=T−T0​1​∑t=T0​+1T​α(t−1) with wˉ=w(αˉ)\bar w=w(\bar\alpha)wˉ=w(αˉ), or a random iterate from the same range. The paper's standing assumptions are ∥xi∥≤1\|x_i\|\le1∥xi​∥≤1, φi≥0\varphi_i\ge0φi​≥0 and φi(0)≤1\varphi_i(0)\le1φi​(0)≤1.

Formalization targets

Goal: Theorem 2 (p. 6)

For convex (1/γ)(1/\gamma)(1/γ)-smooth losses under the standing assumptions, and every ϵP>0\epsilon_P>0ϵP​>0,

T ≥ (n+1λγ)log⁡((n+1λγ)1ϵP) ⟹ E[P(w(T))−D(α(T))]≤ϵP,T\ \ge\ \Big(n+\frac1{\lambda\gamma}\Big)\log\Big(\Big(n+\frac1{\lambda\gamma}\Big)\frac1{\epsilon_P}\Big)\ \Longrightarrow\ \mathbb E\big[P(w^{(T)})-D(\alpha^{(T)})\big]\le\epsilon_P,T ≥ (n+λγ1​)log((n+λγ1​)ϵP​1​) ⟹ E[P(w(T))−D(α(T))]≤ϵP​,

and, for T>T0T>T_0T>T0​,

T0 ≥ (n+1λγ)log⁡((n+1λγ)1(T−T0)ϵP) ⟹ E[P(wˉ)−D(αˉ)]≤ϵPT_0\ \ge\ \Big(n+\frac1{\lambda\gamma}\Big)\log\Big(\Big(n+\frac1{\lambda\gamma}\Big)\frac1{(T-T_0)\epsilon_P}\Big)\ \Longrightarrow\ \mathbb E\big[P(\bar w)-D(\bar\alpha)\big]\le\epsilon_PT0​ ≥ (n+λγ1​)log((n+λγ1​)(T−T0​)ϵP​1​) ⟹ E[P(wˉ)−D(αˉ)]≤ϵP​

for the averaging and the random outputs. The constants are the paper's.

Milestones (in the order the proof uses them)

  1. (p. 2) (1/γ)(1/\gamma)(1/γ)-smooth φ\varphiφ ⇒\Rightarrow⇒ φ∗\varphi^*φ∗ is γ\gammaγ-strongly convex.
  2. (p. 13) The duality-gap identity P(w(α))−D(α)=1n∑i(φi(w⊤xi)+φi∗(−αi)+αiw⊤xi)P(w(\alpha))-D(\alpha)=\frac1n\sum_i\big(\varphi_i(w^\top x_i)+\varphi_i^*(-\alpha_i)+\alpha_iw^\top x_i\big)P(w(α))−D(α)=n1​∑i​(φi​(w⊤xi​)+φi∗​(−αi​)+αi​w⊤xi​).
  3. (10), p. 13: the one-coordinate dual increase.
  4. Lemma 1 (p. 12): the expected dual increase is at least sn\frac snns​ times the duality gap, minus a variance term.
  5. Lemma 2 (p. 13): D(α)≤P(w∗)≤P(0)≤1D(\alpha)\le P(w^*)\le P(0)\le1D(α)≤P(w∗)≤P(0)≤1 and D(0)≥0D(0)\ge0D(0)≥0.
  6. (p. 14) E[ϵD(t)]≤(1−s/n)t≤exp⁡(−λγt/(1+λγn))\mathbb E[\epsilon_D^{(t)}]\le(1-s/n)^t\le\exp(-\lambda\gamma t/(1+\lambda\gamma n))E[ϵD(t)​]≤(1−s/n)t≤exp(−λγt/(1+λγn)) for s=λnγ/(1+λnγ)s=\lambda n\gamma/(1+\lambda n\gamma)s=λnγ/(1+λnγ).
  7. (12), p. 14: E[P(w(t))−D(α(t))]≤nsE[ϵD(t)−ϵD(t+1)]≤nsE[ϵD(t)]\mathbb E[P(w^{(t)})-D(\alpha^{(t)})]\le\frac ns\mathbb E[\epsilon_D^{(t)}-\epsilon_D^{(t+1)}]\le\frac ns\mathbb E[\epsilon_D^{(t)}]E[P(w(t))−D(α(t))]≤sn​E[ϵD(t)​−ϵD(t+1)​]≤sn​E[ϵD(t)​].

Significance

Theorem 2 shows that a stochastic method whose per-iteration cost is a single example converges linearly in the primal objective, through the duality gap, with a condition number 1/(λγ)1/(\lambda\gamma)1/(λγ) that enters additively with nnn rather than multiplicatively. This additive n+1/(λγ)n+1/(\lambda\gamma)n+1/(λγ) dependence is the benchmark against which later variance-reduced and accelerated methods (SVRG, SAGA, accelerated proximal SDCA, Katyusha) are compared. The duality gap is computable, so the bound also yields a stopping criterion.

The result is proved in the paper. As far as the platform's catalogue shows, no part of it, nor the smoothness/strong-convexity duality for scalar conjugates, has been machine-checked. The mission produces a checked statement of the rate with explicit constants and checked versions of the lemmas reused by the series' other two missions (Lipschitz losses, almost-smooth losses).

Difficulty

The proof is short on paper, but each step crosses between objects that Lean keeps apart. The conjugates are extended-valued: φi∗\varphi_i^*φi∗​ is +∞+\infty+∞ off a bounded set for many smooth losses (logistic loss), so the dual objective, the coordinate step and the strong-convexity inequality all live in extended reals, and every passage to real arithmetic must be justified by finiteness along the run. The smoothness-to-strong-convexity step (milestone 1) is the classical duality, but Mathlib has no scalar conjugate calculus and no ready statement of it. The expectation is over the coordinate sequence, and the recursion compares expectations over histories of different lengths. The obvious attempt, a deterministic per-sequence bound, fails: the linear rate holds only in expectation.

Formalization scope

Data are x : Fin n → EuclideanSpace ℝ (Fin d), φ : Fin n → ℝ → ℝ (convex), lam : ℝ with 0 < lam, and 0 < n. (1/γ)(1/\gamma)(1/γ)-smoothness is a derivative function φ' with HasDerivAt everywhere and ∣φi′(a)−φi′(b)∣≤1γ∣a−b∣|φ'_i(a)-φ'_i(b)|\le\frac1\gamma|a-b|∣φi′​(a)−φi′​(b)∣≤γ1​∣a−b∣. The conjugate conj and the dual dual are EReal-valued (⨆, never a real sSup). Dual values are converted to reals only under conjuncts asserting their finiteness. The SDCA step is a map Δ : (Fin n → ℝ) → Fin n → ℝ constrained by the argmax predicate IsSDCAStep, which is satisfiable. The run is a fold over an index sequence js : Fin T → Fin n from α⁽⁰⁾ = 0, and E\mathbb EE is the uniform average over all nTn^TnT sequences (the published SAGA.Convex.expectIdx). Indices are 0-based. Lemma 1 and (10) are stated in the conditional one-step form at an arbitrary dual-feasible point, which implies the printed history-averaged form. The goal assumes no dual optimum; milestones 6–7 take a maximizer α∗\alpha^*α∗ of DDD as a hypothesis (it exists under the standing assumptions). The random output is stated over the iterates α(T0),…,α(T−1)\alpha^{(T_0)},\dots,\alpha^{(T-1)}α(T0​),…,α(T−1), which is what the proof bounds.

A trivializing formalization is ruled out: no statement takes the step, the conjugate or the expectation as an unconstrained variable, no real-valued toReal absorbs an infinite dual value, and the bounds are stated for the paper's exact iteration counts.

Needed infrastructure: scalar Fenchel conjugates in EReal (Fenchel–Young inequality and equality at subgradients), smooth/strongly convex duality, and averages over finite index sequences. The conjugate facts and the expectation bookkeeping are reusable beyond this mission, by the series' Lipschitz and almost-smooth missions in particular. Proofs of any milestone, and lemmas on EReal conjugates, are welcome.

Selected references

  • S. Shalev-Shwartz, T. Zhang, Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization, JMLR 14:567–599, 2013. arXiv:1209.1873v2
  • C.-J. Hsieh, K.-W. Chang, C.-J. Lin, S. S. Keerthi, S. Sundararajan, A Dual Coordinate Descent Method for Large-scale Linear SVM, ICML 2008. doi:10.1145/1390156.1390208
  • Z.-Q. Luo, P. Tseng, On the Convergence of the Coordinate Descent Method for Convex Differentiable Minimization, J. Optim. Theory Appl. 72:7–35, 1992. doi:10.1007/BF00939948
  • N. Le Roux, M. Schmidt, F. Bach, A Stochastic Gradient Method with an Exponential Convergence Rate for Finite Training Sets, NeurIPS 2012. arXiv:1202.6258
  • J.-B. Hiriart-Urruty, C. Lemaréchal, Fundamentals of Convex Analysis, Springer, 2001 (conjugacy of smooth and strongly convex functions, §E.4.2). doi:10.1007/978-3-642-56468-0
12 thms2 active usersReviewed
Convex OptimizationOptimization·Captain: mikedeng1

Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization 1: With L-Lipschitz Losses, SDCA Reaches Expected Duality Gap ε after ⌈n log(λn/(2L²))⌉ + n + 20L²/(λε) IterationsResearch Paper

Motivation

Regularized loss minimization over linear predictors is the training problem behind support vector machines, logistic regression, ridge regression and their many variants. Given examples x1,…,xn∈Rdx_1,\dots,x_n\in\mathbb R^dx1​,…,xn​∈Rd, one minimizes an average of convex losses of the predictions w⊤xiw^\top x_iw⊤xi​ plus a quadratic penalty. Dual coordinate ascent (DCA) solves the Fenchel dual of this problem by optimizing one dual variable at a time; it has been the workhorse of linear SVM solvers such as LIBLINEAR (Hsieh et al., ICML 2008). Before Shalev-Shwartz and Zhang's work, the available analyses of DCA either gave linear rates with constants depending on unknown problem quantities or did not control the duality gap, which is the quantity a practitioner can actually monitor as a stopping criterion.

Shalev-Shwartz and Zhang (2013) analysed stochastic dual coordinate ascent (SDCA), where the coordinate is picked uniformly at random, and proved explicit bounds on the expected duality gap. For Lipschitz losses (hinge loss, absolute loss) the bound is O~(n+L2/(λϵ))\tilde O(n+L^2/(\lambda\epsilon))O~(n+L2/(λϵ)) iterations, which matches stochastic gradient descent when the accuracy is low and improves on it when nnn is small compared with L2/(λϵ)L^2/(\lambda\epsilon)L2/(λϵ). This mission formalizes that result, Theorem 1 of the paper, together with the lemmas of its proof.

Timeline. Hsieh, Chang, Lin, Keerthi and Sundararajan (2008) proposed dual coordinate descent for linear SVMs and proved linear convergence with an unspecified constant. Shalev-Shwartz and Zhang (arXiv 2012, JMLR 2013) gave the first explicit duality-gap rates for SDCA, for Lipschitz and for smooth losses. Proximal and accelerated variants followed (Shalev-Shwartz and Zhang, Math. Program. 2016).

Setting

Fix n≥1n\ge1n≥1 examples x1,…,xn∈Rdx_1,\dots,x_n\in\mathbb R^dx1​,…,xn​∈Rd (Euclidean norm), convex losses ϕ1,…,ϕn:R→R\phi_1,\dots,\phi_n:\mathbb R\to\mathbb Rϕ1​,…,ϕn​:R→R and a regularization parameter λ>0\lambda>0λ>0. The primal problem (1) is to minimize

P(w)=1n∑i=1nϕi(w⊤xi)+λ2∥w∥2.P(w)=\frac1n\sum_{i=1}^n\phi_i(w^\top x_i)+\frac\lambda2\|w\|^2 .P(w)=n1​i=1∑n​ϕi​(w⊤xi​)+2λ​∥w∥2.

The convex conjugate of a scalar function is ϕ∗(u)=sup⁡z(zu−ϕ(z))∈(−∞,+∞]\phi^*(u)=\sup_z(zu-\phi(z))\in(-\infty,+\infty]ϕ∗(u)=supz​(zu−ϕ(z))∈(−∞,+∞]. The dual problem (2) is to maximize, over α∈Rn\alpha\in\mathbb R^nα∈Rn,

D(α)=1n∑i=1n−ϕi∗(−αi)−λ2∥w(α)∥2,w(α)=1λn∑i=1nαixi.D(\alpha)=\frac1n\sum_{i=1}^n-\phi_i^*(-\alpha_i)-\frac\lambda2\|w(\alpha)\|^2,\qquad w(\alpha)=\frac1{\lambda n}\sum_{i=1}^n\alpha_ix_i .D(α)=n1​i=1∑n​−ϕi∗​(−αi​)−2λ​∥w(α)∥2,w(α)=λn1​i=1∑n​αi​xi​.

Weak duality gives D(α)≤P(w)D(\alpha)\le P(w)D(α)≤P(w) for all α,w\alpha,wα,w, so the duality gap P(w(α))−D(α)P(w(\alpha))-D(\alpha)P(w(α))−D(α) bounds the primal sub-optimality of w(α)w(\alpha)w(α).

A loss ϕ\phiϕ is LLL-Lipschitz if ∣ϕ(a)−ϕ(b)∣≤L∣a−b∣|\phi(a)-\phi(b)|\le L|a-b|∣ϕ(a)−ϕ(b)∣≤L∣a−b∣ for all a,ba,ba,b. The paper's standing assumptions are ∥xi∥≤1\|x_i\|\le1∥xi​∥≤1, ϕi≥0\phi_i\ge0ϕi​≥0 and ϕi(0)≤1\phi_i(0)\le1ϕi​(0)≤1.

Procedure SDCA starts from α(0)=0\alpha^{(0)}=0α(0)=0. At iteration t=1,…,Tt=1,\dots,Tt=1,…,T it picks a coordinate iii uniformly at random, chooses Δαi\Delta\alpha_iΔαi​ maximizing

−ϕi∗(−(αi(t−1)+Δαi))−λn2∥w(t−1)+(λn)−1Δαixi∥2,-\phi_i^*\bigl(-(\alpha_i^{(t-1)}+\Delta\alpha_i)\bigr)-\frac{\lambda n}{2}\bigl\|w^{(t-1)}+(\lambda n)^{-1}\Delta\alpha_i x_i\bigr\|^2,−ϕi∗​(−(αi(t−1)​+Δαi​))−2λn​​w(t−1)+(λn)−1Δαi​xi​​2,

and sets α(t)=α(t−1)+Δαiei\alpha^{(t)}=\alpha^{(t-1)}+\Delta\alpha_ie_iα(t)=α(t−1)+Δαi​ei​ and w(t)=w(α(t))w^{(t)}=w(\alpha^{(t)})w(t)=w(α(t)). The averaging option outputs αˉ=1T−T0∑t=T0+1Tα(t−1)\bar\alpha=\frac1{T-T_0}\sum_{t=T_0+1}^T\alpha^{(t-1)}αˉ=T−T0​1​∑t=T0​+1T​α(t−1) and wˉ=w(αˉ)\bar w=w(\bar\alpha)wˉ=w(αˉ). Let α∗\alpha^*α∗ maximize DDD and write ϵD(t)=D(α∗)−D(α(t))\epsilon_D^{(t)}=D(\alpha^*)-D(\alpha^{(t)})ϵD(t)​=D(α∗)−D(α(t)) for the dual sub-optimality.

Formalization targets

Goal: Theorem 1 (p. 5)

For convex LLL-Lipschitz losses under the standing assumptions, if ϵP>0\epsilon_P>0ϵP​>0 and

T ≥ T0+n+4L2λϵP ≥ max⁡(0,⌈nlog⁡(0.5λnL−2)⌉)+n+20L2λϵP,T\ \ge\ T_0+n+\frac{4L^2}{\lambda\epsilon_P}\ \ge\ \max\bigl(0,\lceil n\log(0.5\lambda nL^{-2})\rceil\bigr)+n+\frac{20L^2}{\lambda\epsilon_P},T ≥ T0​+n+λϵP​4L2​ ≥ max(0,⌈nlog(0.5λnL−2)⌉)+n+λϵP​20L2​,

then

E[P(wˉ)−D(αˉ)]≤ϵP,andE[D(α∗)−D(α(t))]≤ϵP/2  for all t≥T0.\mathbb E\bigl[P(\bar w)-D(\bar\alpha)\bigr]\le\epsilon_P,\qquad\text{and}\qquad\mathbb E\bigl[D(\alpha^*)-D(\alpha^{(t)})\bigr]\le\epsilon_P/2\ \text{ for all }t\ge T_0 .E[P(wˉ)−D(αˉ)]≤ϵP​,andE[D(α∗)−D(α(t))]≤ϵP​/2  for all t≥T0​.

The constants 444 and 202020 are the paper's.

Milestones

In the order the proof uses them: the one-coordinate increase (10); the identity P(w(α))−D(α)=1n∑i(ϕi(w⊤xi)+ϕi∗(−αi)+αiw⊤xi)P(w(\alpha))-D(\alpha)=\frac1n\sum_i(\phi_i(w^\top x_i)+\phi_i^*(-\alpha_i)+\alpha_iw^\top x_i)P(w(α))−D(α)=n1​∑i​(ϕi​(w⊤xi​)+ϕi∗​(−αi​)+αi​w⊤xi​); Lemma 1 (expected dual increase at least sn\frac snns​ times the gap minus (sn)2G2λ(\frac sn)^2\frac{G}{2\lambda}(ns​)22λG​); Lemma 2 (D(α)≤P(w∗)≤P(0)≤1D(\alpha)\le P(w^*)\le P(0)\le1D(α)≤P(w∗)≤P(0)≤1, D(0)≥0D(0)\ge0D(0)≥0); Lemma 3 (ϕ∗=+∞\phi^*=+\inftyϕ∗=+∞ outside [−L,L][-L,L][−L,L]); Lemma 4 (G≤4L2G\le4L^2G≤4L2); the recursion after (13); the rate (14), E[ϵD(t)]≤2Gλ(2n+t−t0)\mathbb E[\epsilon_D^{(t)}]\le\frac{2G}{\lambda(2n+t-t_0)}E[ϵD(t)​]≤λ(2n+t−t0​)2G​; and the averaged duality-gap bound of p. 16.

Significance

Theorem 1 makes the duality gap of SDCA a certified stopping criterion with an explicit iteration count. Because the gap upper-bounds the primal sub-optimality, it also gives a primal guarantee for the averaged output wˉ\bar wwˉ. It covers non-smooth losses, the hinge loss in particular, where linear rates are not available. The analysis pattern, a lower bound on the expected dual increase by the duality gap (Lemma 1) followed by a recursion for the dual sub-optimality, was reused in the analyses of proximal, accelerated and mini-batch dual methods.

The result is proved in the paper; to the best of current knowledge it has no machine-checked proof. The mission produces a formal account of the primal–dual pair of regularized loss minimization with extended-real conjugates, a formal SDCA run with uniformly random coordinates, and the complete chain of lemmas. Lemma 1 is stated with a general strong-convexity modulus γ≥0\gamma\ge0γ≥0, as in the paper, so it serves the smooth-loss analysis as well.

Difficulty

The obvious first idea is to treat SDCA as stochastic coordinate ascent on a concave function and import a standard rate. That fails: DDD is neither smooth nor strongly concave for Lipschitz losses, and its domain {α:ϕi∗(−αi)<∞}\{\alpha:\phi_i^*(-\alpha_i)<\infty\}{α:ϕi∗​(−αi​)<∞} is a box, so the per-coordinate progress cannot be bounded by a gradient norm. The argument instead compares the exact coordinate maximizer with an explicit interpolation towards a sub-gradient, which needs the conjugate's value at a sub-gradient (Fenchel–Young equality) and its infinite values outside [−L,L][-L,L][−L,L].

The rate (14) is not a plain geometric recursion. It needs a burn-in phase and then an induction with a step size sss that changes with ttt, and the duality gap of the output has to be recovered from the dual progress by Jensen's inequality over the averaged iterates. Everything happens in expectation over the coordinate sequence, while the dual values have to be kept finite along every run.

Formalization scope

Lean conventions:

  • Rd\mathbb R^dRd is EuclideanSpace ℝ (Fin d), examples and losses are indexed by Fin n, and w⊤xw^\top xw⊤x is inner ℝ w x.
  • The conjugate is an EReal supremum, and DDD is EReal-valued. Real values of DDD are used only where the same statement asserts that they are finite.
  • The coordinate step is a free map Δ : (Fin n → ℝ) → Fin n → ℝ constrained by an arg-max predicate (IsSDCAStep). Any maximizer is admissible, and one exists for convex real losses.
  • The run is a function of the coordinate sequence js : Fin T → Fin n, started at α(0)=0\alpha^{(0)}=0α(0)=0, with w(t)=w(α(t))w^{(t)}=w(\alpha^{(t)})w(t)=w(α(t)).
  • Expectations are uniform averages over coordinate sequences (SAGA.Convex.expectIdx, a published definition).
  • α∗\alpha^*α∗ is any maximizer of DDD, and w∗w^*w∗ any minimizer of PPP.
  • The constant G=max⁡tG(t)G=\max_tG^{(t)}G=maxt​G(t) of the proof is pinned to its bound 4L24L^24L2. Theorem 1's burn-in max⁡(0,⌈nlog⁡(0.5λnL−2)⌉)\max(0,\lceil n\log(0.5\lambda nL^{-2})\rceil)max(0,⌈nlog(0.5λnL−2)⌉) is exactly the value of t0t_0t0​ obtained with G=4L2G=4L^2G=4L2 and ϵD(0)≤1\epsilon_D^{(0)}\le1ϵD(0)​≤1.
  • Lemmas 1 and 4 are stated for a fixed dual-feasible state. The paper's history-averaged form follows from this.

A real-valued conjugate, which returns 000 where the supremum is infinite, or an unconstrained step map would make the statements false or vacuous. The EReal conjugate and the arg-max predicate rule both out.

Infrastructure needed: Fenchel–Young for scalar convex functions, the existence of sub-gradients of convex real functions and their bound by the Lipschitz constant, Jensen's inequality for the convex PPP and the concave DDD, and the algebra of uniform averages over {1,…,n}t\{1,\dots,n\}^t{1,…,n}t. The conjugate and sub-differential definitions and the duality-gap identity can be reused beyond this mission. Proofs of any milestone are welcome, as are lemmas on EReal sums that keep the finiteness bookkeeping short.

Selected references

  • S. Shalev-Shwartz and T. Zhang, Stochastic Dual Coordinate Ascent Methods for Regularized Loss Minimization, arXiv:1209.1873v2, 2013; Journal of Machine Learning Research 14:567–599, 2013. https://arxiv.org/abs/1209.1873
  • C.-J. Hsieh, K.-W. Chang, C.-J. Lin, S. S. Keerthi and S. Sundararajan, A Dual Coordinate Descent Method for Large-scale Linear SVM, ICML 2008. https://doi.org/10.1145/1390156.1390208
  • S. Shalev-Shwartz and T. Zhang, Accelerated Proximal Stochastic Dual Coordinate Ascent for Regularized Loss Minimization, Mathematical Programming 155:105–145, 2016. https://doi.org/10.1007/s10107-014-0839-0
13 thms2 active usersReviewed
Convex OptimizationOptimization·Captain: mikedeng1

Making Gradient Descent Optimal for Strongly Convex Stochastic Optimization 2: For F Smooth at the Optimum, SGD's Last and Averaged Iterates Attain 2µG²/(λ²T) and 16µG²/(λ²T)Research Paper

Motivation

Stochastic gradient descent (SGD) is the standard method for minimizing an expected loss F(w)=Ez[ℓ(w,z)]F(w) = \mathbb E_z[\ell(w, z)]F(w)=Ez​[ℓ(w,z)] when only noisy gradients are available: in machine learning, www is a predictor, zzz a random example, and each step uses the gradient of the loss on one fresh example. When FFF is strongly convex, as for regularized learning problems such as the SVM, the minimax-optimal error after TTT oracle calls is of order 1/T1/T1/T. Rakhlin, Shamir and Sridharan (arXiv:1109.5647, ICML 2012) asked whether plain SGD attains this rate, or whether the O(log⁡T/T)O(\log T/T)O(logT/T) bound known for averaged SGD is the truth.

Their §3 settles the case where FFF is in addition smooth at its minimizer: both the last iterate and the average of all iterates attain O(1/T)O(1/T)O(1/T), with explicit constants. This mission formalizes that section.

Timeline of the relevant bounds:

  • 2007–2009. Hazan, Agarwal and Kale (doi:10.1007/s10994-007-5016-8) show O(log⁡T)O(\log T)O(logT) regret for online gradient descent on strongly convex losses, which by online-to-batch conversion gives O(log⁡T/T)O(\log T/T)O(logT/T) for the averaged SGD iterate.
  • 2009. Nemirovski, Juditsky, Lan and Shapiro (doi:10.1137/070704277) give an O(1/T)O(1/T)O(1/T) bound for the last iterate when FFF is strongly convex and smooth.
  • 2011. Hazan and Kale (arXiv:1006.2425) obtain O(1/T)O(1/T)O(1/T) for strongly convex, possibly non-smooth FFF with a modified algorithm.
  • 2012. Rakhlin, Shamir and Sridharan prove the O(1/T)O(1/T)O(1/T) bounds of this mission for SGD under smoothness at the optimum only, show the log⁡T\log TlogT is real for averaged SGD on non-smooth problems, and recover O(1/T)O(1/T)O(1/T) with suffix averaging.

Setting

Let W⊆RdW \subseteq \mathbb R^dW⊆Rd be a closed convex set containing 000, let F:Rd→RF : \mathbb R^d \to \mathbb RF:Rd→R, and let DDD be a probability distribution on a measurable space Z\mathcal ZZ.

  • FFF is λ\lambdaλ-strongly convex on WWW, λ>0\lambda > 0λ>0: for all w,w′∈Ww, w' \in Ww,w′∈W and every subgradient ggg of FFF at www, F(w′)≥F(w)+⟨g,w′−w⟩+λ2∥w′−w∥2F(w') \ge F(w) + \langle g, w' - w\rangle + \frac{\lambda}{2}\|w' - w\|^2F(w′)≥F(w)+⟨g,w′−w⟩+2λ​∥w′−w∥2 (equation (1)).
  • FFF attains its minimum over WWW at w∗∈Ww^* \in Ww∗∈W.
  • A stochastic subgradient oracle is a measurable map g:Rd×Z→Rdg : \mathbb R^d \times \mathcal Z \to \mathbb R^dg:Rd×Z→Rd such that, for every w∈Ww \in Ww∈W, the mean Ez∼D[g(w,z)]\mathbb E_{z \sim D}[g(w, z)]Ez∼D​[g(w,z)] is a subgradient of FFF at www relative to WWW, and Ez∼D∥g(w,z)∥2≤G2\mathbb E_{z\sim D}\|g(w,z)\|^2 \le G^2Ez∼D​∥g(w,z)∥2≤G2.
  • FFF is μ\muμ-smooth with respect to w∗w^*w∗ if
F(w)−F(w∗)≤μ2∥w−w∗∥2for all w∈W.(2)F(w) - F(w^*) \le \frac{\mu}{2}\|w - w^*\|^2 \qquad \text{for all } w \in W. \tag{2}F(w)−F(w∗)≤2μ​∥w−w∗∥2for all w∈W.(2)

Projected SGD with step sizes ηt=1/(λt)\eta_t = 1/(\lambda t)ηt​=1/(λt) starts at w1=0w_1 = 0w1​=0, draws z1,z2,⋯∼Dz_1, z_2, \dots \sim Dz1​,z2​,⋯∼D independently and sets

wt+1=ΠW(wt−ηt g(wt,zt)),w_{t+1} = \Pi_W\big(w_t - \eta_t\, g(w_t, z_t)\big),wt+1​=ΠW​(wt​−ηt​g(wt​,zt​)),

where ΠW\Pi_WΠW​ is the Euclidean projection onto WWW. The averaged iterate is wˉT=(w1+⋯+wT)/T\bar w_T = (w_1 + \dots + w_T)/TwˉT​=(w1​+⋯+wT​)/T. In Lean the run is UnderstandingML.sgdStrongIterates lam W g S and the average UnderstandingML.sgdStrongAverage lam W g S, for a sample S : Fin T → Z drawn from UnderstandingML.iidLaw D T.

Formalization targets

Goal: Theorem 2 (averaged iterate)

If FFF is λ\lambdaλ-strongly convex and μ\muμ-smooth with respect to w∗w^*w∗, then for every T≥1T \ge 1T≥1

E[F(wˉT)−F(w∗)]≤16 μG2λ2T.\mathbb E\big[F(\bar w_T) - F(w^*)\big] \le \frac{16\,\mu G^2}{\lambda^2 T}.E[F(wˉT​)−F(w∗)]≤λ2T16μG2​.

Milestones

  1. Lemma 2. If E∥g(w,z)∥2≤G2\mathbb E\|g(w,z)\|^2 \le G^2E∥g(w,z)∥2≤G2 at a point w∈Ww \in Ww∈W, then ∥w−w∗∥2≤4G2/λ2\|w - w^*\|^2 \le 4G^2/\lambda^2∥w−w∗∥2≤4G2/λ2.
  2. Lemma 1. For every T≥1T \ge 1T≥1, E∥wT−w∗∥2≤4G2/(λ2T)\mathbb E\|w_T - w^*\|^2 \le 4G^2/(\lambda^2 T)E∥wT​−w∗∥2≤4G2/(λ2T), without smoothness.
  3. Theorem 1 (last iterate). Under smoothness, E[F(wT)−F(w∗)]≤2μG2/(λ2T)\mathbb E[F(w_T) - F(w^*)] \le 2\mu G^2/(\lambda^2 T)E[F(wT​)−F(w∗)]≤2μG2/(λ2T).
  4. Averaged-iterate distance (App. B.3). For every T≥1T \ge 1T≥1, E∥wˉT−w∗∥2≤32G2/(λ2T)\mathbb E\|\bar w_T - w^*\|^2 \le 32G^2/(\lambda^2 T)E∥wˉT​−w∗∥2≤32G2/(λ2T), without smoothness.

The constants are the paper's. Theorem 1 is a companion of the goal rather than a step of its proof; both rest on Lemma 1.

Significance

The result. Theorem 2 shows that the log⁡T\log TlogT factor in the standard bound for averaged SGD is not needed when FFF grows at most quadratically away from its minimizer. Condition (2) holds for logistic and least-squares regression and, in general, for linear predictors with losses that have Lipschitz-continuous gradients. Together with the paper's Theorem 4, which shows that the log⁡T\log TlogT factor is unavoidable for averaged SGD on some non-smooth problems, it locates where the extra logarithm comes from. Lemma 1 is the paper's key lemma: an O(1/T)O(1/T)O(1/T) bound on the expected squared distance of the last iterate that needs no smoothness, and it is reused in the paper's suffix-averaging analysis.

Formalizing it. All results are proved in the paper; none is formalized. The platform has UnderstandingML.sgd_strongly_convex (Shalev-Shwartz and Ben-David, Theorem 14.11), the classical (ρ2/(2λT))(1+log⁡T)(\rho^2/(2\lambda T))(1 + \log T)(ρ2/(2λT))(1+logT) bound for the same averaged iterate without smoothness. This mission adds the O(1/T)O(1/T)O(1/T) bounds under (2), on the same SGD run, together with a reusable expected-distance bound for the projected iterates.

Difficulty

The obvious route to Theorem 2 is to apply smoothness to each iterate, use Lemma 1, and average: by Jensen, E[F(wˉT)]−F(w∗)≤1T∑t2μG2λ2t\mathbb E[F(\bar w_T)] - F(w^*) \le \frac{1}{T}\sum_t \frac{2\mu G^2}{\lambda^2 t}E[F(wˉT​)]−F(w∗)≤T1​∑t​λ2t2μG2​, which is of order log⁡T/T\log T/TlogT/T. Removing the logarithm requires controlling E∥wˉT−w∗∥2\mathbb E\|\bar w_T - w^*\|^2E∥wˉT​−w∗∥2 directly, where the iterates are dependent and the cross terms E⟨wˉt−w∗,wt+1−w∗⟩\mathbb E\langle \bar w_t - w^*, w_{t+1} - w^*\rangleE⟨wˉt​−w∗,wt+1​−w∗⟩ do not vanish. The resulting recursion is nonlinear in E∥wˉt−w∗∥2\mathbb E\|\bar w_t - w^*\|^2E∥wˉt​−w∗∥2, and an induction with the constant 323232 fails at its first step unless the base case uses the sharper bound of Lemma 2. On the formal side, the expectations are over the product of TTT independent samples, and each one-step argument needs the independence of ztz_tzt​ from wtw_twt​ and the measurability of the projected iterates.

Formalization scope

  • Space. The paper's Hilbert space is specialized to Rd\mathbb R^dRd with the Euclidean norm (UnderstandingML.Vec d).
  • Domain. Convex ℝ W, IsClosed W (so the projection exists; tacit in the paper) and 0 ∈ W (the paper's initialization).
  • Strong convexity is Mathlib's StrongConvexOn W lam F with 0 < lam. Combined with relative subgradients, it gives (1).
  • Oracle. OptimalSGD.Suffix.IsSubgradientOracleOn W F D g (the series' shared oracle predicate): for w∈Ww \in Ww∈W, g w is integrable and its mean is a subgradient of FFF at www relative to WWW. Measurable (Function.uncurry g) and a probability measure DDD are assumed. The second-moment bound holds at every w∈Ww \in Ww∈W, which is the paper's "regardless of how the iterates evolve".
  • Smoothness is (2) exactly (IsSmoothWrt W F μ wstar), with μ≥0\mu \ge 0μ≥0. It is not a Lipschitz-gradient assumption, and FFF need not be differentiable.
  • Indexing. Iterates are indexed from 000. The paper's wTw_TwT​ is sgdStrongIterates lam W g S (T - 1), and wˉT\bar w_TwˉT​ averages indices 0,…,T−10,\dots,T-10,…,T−1. Statements quantify over T≥1T \ge 1T≥1, including T=1T = 1T=1 (wˉ1=w1=0\bar w_1 = w_1 = 0wˉ1​=w1​=0).
  • Expectations. Bounds on ∥⋅∥2\|\cdot\|^2∥⋅∥2 use the lower Lebesgue integral. Bounds on FFF use the Bochner integral, and the integrability of F(wT)F(w_T)F(wT​) or F(wˉT)F(\bar w_T)F(wˉT​) is part of the conclusion.

A statement whose Bochner integral is not shown to be integrable would hold vacuously, because non-integrable functions integrate to 000; the integrability conjuncts rule this out. The run and the average are the published Understanding Machine Learning definitions, so the bounds here apply to the same object as UnderstandingML.sgd_strongly_convex.

Needed infrastructure: measurability of the projected iterates in the sample, independence of ztz_tzt​ from wtw_twt​ under the product measure, conditional unbiasedness of the oracle along the run, non-expansiveness of the projection (available as UnderstandingML.projection_lemma), and an elementary induction on the scalar recursions. Contributions of these as standalone lemmas are welcome, and they are reusable for mission 1 of this series (suffix averaging) and for other projected-SGD analyses.

Selected references

  • A. Rakhlin, O. Shamir, K. Sridharan, Making Gradient Descent Optimal for Strongly Convex Stochastic Optimization, ICML 2012; arXiv:1109.5647v7. https://arxiv.org/abs/1109.5647
  • E. Hazan, A. Agarwal, S. Kale, Logarithmic regret algorithms for online convex optimization, Machine Learning 69, 2007. https://doi.org/10.1007/s10994-007-5016-8
  • A. Nemirovski, A. Juditsky, G. Lan, A. Shapiro, Robust stochastic approximation approach to stochastic programming, SIAM J. Optim. 19(4), 2009. https://doi.org/10.1137/070704277
  • E. Hazan, S. Kale, Beyond the regret minimization barrier: an optimal algorithm for stochastic strongly-convex optimization, COLT 2011; arXiv:1006.2425. https://arxiv.org/abs/1006.2425
  • S. Shalev-Shwartz, S. Ben-David, Understanding Machine Learning: From Theory to Algorithms, Cambridge University Press, 2014, §14.4.4. https://doi.org/10.1017/CBO9781107298019
9 thms2 active usersReviewed
PreviousPage 4 of 9Next

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