Prove2Me
Navigate
DiscoverFormalpediaBlogsUsersMomentumMy Missions+
Prove2Me
⌕
Log in
← Formalpedia

Lemma 5.14 — one-step progress bound

Disproved
FirstOrderOpt.FiniteSum.variance_reduced_progress_bound

by mikedeng1 · Sep 19, 2026 · Mathlib 0df444a (Lean v4.33.1)

bregman-divergenceconvex-optimizationfinite-sum-optimizationmirror-descentstochastic-optimization

The variance-reduced mirror-descent update (Algorithm 5.6) sets xt+1:=arg⁡min⁡x∈X{γ[⟨Gt,x⟩+h(x)]+V(xt,x)}x_{t+1}:=\arg\min_{x\in X}\{\gamma[\langle G_t,x\rangle+h(x)]+V(x_t,x)\}xt+1​:=argminx∈X​{γ[⟨Gt​,x⟩+h(x)]+V(xt​,x)}, where VVV is the Bregman divergence of a fixed distance-generating function and γ\gammaγ is the (constant) stepsize. Suppose fff is possibly μ\muμ-strongly convex (μ≥0\mu\ge0μ≥0): f(y)≥f(x)+⟨∇f(x),y−x⟩+μV(x,y)f(y)\ge f(x)+\langle\nabla f(x),y-x\rangle+\mu V(x,y)f(y)≥f(x)+⟨∇f(x),y−x⟩+μV(x,y) for all x,y∈Xx,y\in Xx,y∈X (Eq. (5.3.2)), and let L≥LfL\ge L_fL≥Lf​ bound the Lipschitz constant of ∇f\nabla f∇f.

Lemma 5.14. If the stepsize γ\gammaγ satisfies Lγ≤1/2L\gamma\le 1/2Lγ≤1/2, then for any x∈Xx\in Xx∈X,

γ[Ψ(xt+1)−Ψ(x)]+V(xt+1,x)≤(1−γμ)V(xt,x)+γ⟨δt,x−xt⟩+γ2∥δt∥∗2.\gamma[\Psi(x_{t+1})-\Psi(x)]+V(x_{t+1},x) \le (1-\gamma\mu)V(x_t,x)+\gamma\langle\delta_t,x-x_t\rangle+\gamma^2\|\delta_t\|_*^2.γ[Ψ(xt+1​)−Ψ(x)]+V(xt+1​,x)≤(1−γμ)V(xt​,x)+γ⟨δt​,x−xt​⟩+γ2∥δt​∥∗2​.

This is the one-step progress bound from which the chapter's epoch-level convergence result (Theorem 5.6) is obtained by summing over the iterations of an epoch and taking expectation; it resembles Lemma 4.2 for the original (non-variance-reduced) stochastic mirror descent method, with the noise term δt\delta_tδt​ playing the same role.

Formalization Note. The strong-convexity hypothesis (5.3.2) is stated directly; the lemma holds for general μ≥0\mu\ge0μ≥0 (the book's §5.3.1 sets μ=0\mu=0μ=0, §5.3.2 takes μ>0\mu>0μ>0, so this milestone keeps μ\muμ free, matching the book's own generality at this point). The update's minimality (the composite three-point condition invoked from Lemma 3.5) is restated locally in this chapter's own sub-namespace, FirstOrderOpt.FiniteSum, rather than imported from another chunk's mirror-descent milestone, because the mirror-descent chunks of this series are themselves unpublished drafts (Hard Rule 10; see MODERATION_NOTES.md).

Preamble
import Mathlib
Formal statement
namespace FirstOrderOpt.FiniteSum

/-- Lemma 5.14 (one-step progress bound, Eq. (5.3.10)). If the stepsize `γ` satisfies `Lγ ≤ 1/2`
(`L` the overall smoothness constant `Lf ≤ L := (1/m)Σ Li` of the smooth part `f`), and `xt1`
minimizes `u ↦ γ[⟨Gt,u⟩+h(u)]+V(xt,u)` over `X` (the update of Algorithm 5.6), then for any
`x ∈ X`, `γ[Ψ(xt1)-Ψ(x)] + V(xt1,x) ≤ (1-γμ)V(xt,x) + γ⟨δt,x-xt⟩ + γ²‖δt‖²_∗`, where
`δt := Gt - ∇f(xt)` and `μ ≥ 0` is `f`'s strong-convexity modulus of (5.3.2).

**Formalization Note.** `hstrong` states (5.3.2) directly (`f y ≥ f x + ⟨∇f(x),y-x⟩ + μV(x,y)`);
the lemma holds for general `μ ≥ 0` (the book's §5.3.2 specializes to `μ > 0`, §5.3.1 to `μ = 0`,
so this milestone keeps `μ` free, matching the book's own generality here). `hmin` is the
mirror-descent-with-composite-term three-point minimality (Lemma 3.5, invoked in the proof),
restated locally since the mirror-descent chunks are themselves unpublished drafts (Hard Rule
10; see `MODERATION_NOTES.md`). -/
theorem variance_reduced_progress_bound {E : Type*} [NormedAddCommGroup E] [NormedSpace ℝ E]
    (X : Set E) (f h Ψ : E → ℝ) (hΨ : ∀ x, Ψ x = f x + h x)
    (V : E → E → ℝ)
    (gradf_full : E → E →L[ℝ] ℝ)
    (L : ℝ) (hL : 0 < L) (hsmooth : ∀ x y, ‖gradf_full x - gradf_full y‖ ≤ L * ‖x - y‖)
    (μ : ℝ) (hμ : 0 ≤ μ)
    (hstrong : ∀ x y, f y ≥ f x + (gradf_full x) (y - x) + μ * V x y)
    (γ : ℝ) (hγ : 0 < γ) (hLγ : L * γ ≤ 1 / 2)
    (xt xt1 : E) (hxt : xt ∈ X) (hxt1 : xt1 ∈ X)
    (Gt δt : E →L[ℝ] ℝ) (hδt : δt = Gt - gradf_full xt)
    (hmin : ∀ y ∈ X, γ * (Gt xt1) + γ * h xt1 + V xt xt1 ≤ γ * (Gt y) + γ * h y + V xt y) :
    ∀ x ∈ X, γ * (Ψ xt1 - Ψ x) + V xt1 x ≤
      (1 - γ * μ) * V xt x + γ * (δt (x - xt)) + γ ^ 2 * ‖δt‖ ^ 2 := by sorry

end FirstOrderOpt.FiniteSum
Source
Lan, First-order and Stochastic Optimization Methods for Machine Learning, Springer 2020, p. 280, Lemma 5.14
Human review
  • Endorsed by Shuze Chen · Sep 28, 2026

    Confirmed by the moderator at approval.

  • Endorsed by mikedeng1 · Sep 28, 2026

    Confirmed by the mission captain (proposal self-audit).

View graph

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