Theorem 4.4, p. 305 — mirror prox with η = ρ/β on a convex β-smooth f satisfies f((1/t)Σ_{s=1}^t y_{s+1}) − f(x*) ≤ βR²/(ρt)
OpenConvexOptAlg.MirrorProx.theorem_4_4Fix an arbitrary norm on a finite-dimensional real space, a compact convex set , and a convex open set with and . Let be a mirror map on that is -strongly convex on with respect to , with . Let be convex and -smooth on with respect to , with , and let minimize over . Run mirror prox with from , and let satisfy
for instance . Then for every ,
Mirror prox, introduced by Nemirovski, attains the rate on smooth functions in non-Euclidean geometries, while mirror descent attains on Lipschitz functions (Theorem 4.2). The average is over the intermediate points , not over the .
Formalization Note The existence of the minimizer is the book's standing assumption (p. 242). The book does not specify in §4.5; it is taken in as in mirror descent (§4.2, p. 299), which is what makes bound . is any real with on ; the book's is the special case where the supremum is finite (when it is the book's bound is void). and are the implicit conditions for and the division by . The gradient of is required relative to (HasFDerivWithinAt), matching ; may lie on the boundary of .
import Mathlib import Definitions.Def_ConvexOptAlg_MirrorProx_Defs
namespace ConvexOptAlg.MirrorProx
/-- Theorem 4.4 (Bubeck, arXiv:1405.4980v2, §4.5, p. 305). Fix a norm on a finite-dimensional real
space `E`, a compact convex set `X`, and a convex open set `D` with `X ⊆ closure D` and
`X ∩ D ≠ ∅`. Let `Φ` be a mirror map on `D`, `ρ`-strongly convex on `X ∩ D` w.r.t. `‖·‖`
(`ρ > 0`), and let `f` be convex and `β`-smooth on `X` w.r.t. `‖·‖` (`β > 0`), with a minimizer
`x∗ ∈ X`. Let `(x_t, y_t, y'_t, x'_t)` be a run of mirror prox with `η = ρ/β` started at
`x₁ ∈ argmin_{X ∩ D} Φ`, and let `R` satisfy `Φ(w) − Φ(x₁) ≤ R²` for all `w ∈ X ∩ D` (the
book's `R² = sup_{X ∩ D} Φ − Φ(x₁)` is one such value). Then for every `t ≥ 1`,
`f((1/t) Σ_{s=1}^t y_{s+1}) − f(x∗) ≤ βR²/(ρt)`. -/
theorem theorem_4_4 {E : Type*} [NormedAddCommGroup E] [NormedSpace ℝ E] [FiniteDimensional ℝ E]
(X D : Set E) (hXc : IsCompact X) (hXconv : Convex ℝ X) (hXD : X ⊆ closure D)
(hXDne : (X ∩ D).Nonempty)
(Φ : E → ℝ) (Φ' : E → E →L[ℝ] ℝ) (hΦ : IsMirrorMap D Φ Φ')
(ρ : ℝ) (hρ : 0 < ρ) (hsc : IsStronglyConvexWRT (X ∩ D) Φ Φ' ρ)
(f : E → ℝ) (f' : E → E →L[ℝ] ℝ) (hf : ConvexOn ℝ X f)
(β : ℝ) (hβ : 0 < β) (hsm : IsSmoothWRT X f f' β)
(xstar : E) (hxstar : xstar ∈ X ∧ ∀ w ∈ X, f xstar ≤ f w)
(x y y' x' : ℕ → E) (hrun : IsMirrorProxRun X D Φ Φ' f' (ρ / β) x y y' x')
(hx1 : ∀ w ∈ X ∩ D, Φ (x 1) ≤ Φ w)
(R : ℝ) (hR : ∀ w ∈ X ∩ D, Φ w - Φ (x 1) ≤ R ^ 2)
(t : ℕ) (ht : 1 ≤ t) :
f ((1 / (t : ℝ)) • ∑ s ∈ Finset.Icc 1 t, y (s + 1)) - f xstar
≤ β * R ^ 2 / (ρ * t) := by sorry
end ConvexOptAlg.MirrorProx