advanced30 min studyLesson 49 in Mathematics

Mirror descent and geometry

Replace Euclidean distance by a chosen convex geometry and derive multiplicative updates on a probability simplex.

On this page

What you’ll understand

  • Compute a Bregman divergence from a differentiable convex generator.
  • Recover Euclidean gradient descent as a special mirror step.
  • Derive and normalize an entropy-based simplex update.

A Euclidean gradient step treats a change from 0.01 to 0.02 as the same absolute movement as a change from 0.50 to 0.51. For probability weights, those changes may deserve different treatment. Mirror descent lets the geometry of the update reflect the decision space. It still uses a first-order objective model, but penalizes movement with a divergence generated by a convex function rather than always using squared Euclidean distance.

Measure separation from a tangent plane

Choose a differentiable strictly convex generator ψ on the interior of its domain. The Bregman divergence from x to z is the gap between ψ(z) and the tangent plane to ψ at x. Convexity makes this nonnegative. It generally depends on argument order and need not obey a triangle inequality, so it is a divergence rather than a metric. Strong convexity relative to a chosen norm supplies the quantitative lower bound needed in many convergence analyses.

For ψ(x)=∥x∥2/2\psi(x)=\|x\|^2/2, the divergence is half the squared Euclidean distance. Other choices reshape how movement is penalized. The choice affects both the gradient's relevant dual norm and the distance to the comparator in regret or convergence bounds. Geometry is therefore not just a decorative coordinate transformation: it enters the assumptions and constants of the method. One must also ensure that iterates remain in the generator's domain.

Dψ(z,x)=ψ(z)−ψ(x)−∇ψ(x)T(z−x),x+=argmin⁡z∈C[ηgTz+Dψ(z,x)]D_\psi(z,x)=\psi(z)-\psi(x)-\nabla\psi(x)^{\mathsf T}(z-x),\qquad x^+=\underset{z\in C}{\operatorname{argmin}}\left[\eta g^{\mathsf T}z+D_\psi(z,x)\right]
A mirror step minimizes a linear objective model plus a Bregman movement cost. Existence, uniqueness, and rate guarantees require suitable domain, convexity, and regularity assumptions.↔ Scroll wide equations horizontally.

Entropy produces multiplicative weights

On the probability simplex, use negative entropy ψ(p)=∑ipilog⁡pi\psi(p)=\sum_i p_i\log p_i and start with strictly positive coordinates. The resulting divergence is ∑iqilog⁡(qi/pi)\sum_i q_i\log(q_i/p_i) for probability vectors, with the usual continuous convention for zero terms. The normalization constraint couples coordinates. Solving the mirror subproblem gives weights proportional to the old weights multiplied by exponentials of negative scaled gradient components.

A coordinate with a larger cost gradient loses relative weight, while all weights are normalized to sum to one. In exact arithmetic positive initial weights remain positive under finite gradients. A coordinate initialized at zero cannot generally be revived by the multiplicative formula, which is one reason positive initialization matters. Numerically, exponentials can underflow; stable implementations use shifted log-weights and log-sum-exp normalization rather than assuming symbolic positivity prevents floating-point failure.

pi+=piexp⁡(−ηgi)∑jpjexp⁡(−ηgj)p_i^+=\frac{p_i\exp(-\eta g_i)}{\sum_j p_j\exp(-\eta g_j)}
This entropic mirror update applies on the probability simplex with positive starting weights and finite gradient components. Its denominator enforces unit total mass.↔ Scroll wide equations horizontally.

Compare the geometry before comparing rates

Euclidean projection and entropic mirror descent can both maintain simplex feasibility, yet their paths and guarantees use different movement costs. On some high-dimensional probability problems, entropy aligns well with bounded coordinatewise gradients and yields favorable dimension dependence. This is conditional on the norm bounds and generator properties; mirror descent is not uniformly faster for every objective. Choosing a geometry requires identifying the feasible set, gradient scale, and computational cost of its subproblem.

The lab is the Euclidean special case on the real line with no simplex constraint. Its fixed quadratic objective and step-size control let you inspect the baseline recurrence, not an entropy-generated update. Predict a Euclidean step there, then compare it with the manually normalized probability update below. No change of the lab's initial position turns it into a probability vector, and its Play control does not execute multiplicative weights or adapt the underlying geometry.

Reweight two choices exactly

Start from p=(1/2,1/2)p=(1/2,1/2) with cost gradient g=(0,log⁡3)g=(0,\log3) and entropic mirror step η=1. Compute the next probability vector.

  1. The first unnormalized weight is (1/2)e0=1/2(1/2)e^0=1/2. The second is (1/2)e−log⁡3=1/6(1/2)e^{-\log3}=1/6. The higher-cost coordinate is reduced by a factor of three before normalization.

  2. The weights sum to 1/2+1/6=2/31/2+1/6=2/3. Dividing by this total gives p1+=(1/2)/(2/3)=3/4p_1^+=(1/2)/(2/3)=3/4 and p2+=(1/6)/(2/3)=1/4p_2^+=(1/6)/(2/3)=1/4.

  3. Both probabilities are positive and sum to one. The linear objective changes from (log⁡3)/2(\log3)/2 to (log⁡3)/4(\log3)/4. This update minimizes the specified linearized cost plus entropy divergence; it is not an exact minimizer of the linear cost alone, whose solution puts all mass on the first coordinate.

The result

The new distribution is (3/4,1/4). The divergence term limits movement, balancing improved cost against departure from the starting distribution.

Explore the model

Interactive lab

Interactive lab · optimization

The Euclidean mirror special case

For ψ(x)=x²/2 and fixed f(x)=x²/2, set η=0.5 and predict the next point from x₀=2. This lab displays the Euclidean unconstrained case only; it does not change ψ, hold a probability simplex, or perform entropic multiplicative updates.

The Euclidean mirror special caseGradient descent on f(x) = x²/2: xₙ = (1 − η)ⁿx₀. Playback advances two discrete iterations per second. Values outside |x| ≤ 4 are explicitly reported. Current readings: Position x0: 2; Objective x²/2: 2; Iteration behavior: Stable step. All parameters are adjustable with the labeled controls below.−4−202402.174.356.528.7Parameter xf(x) = x²/2n = 0
ObjectiveDiscrete iterate
Position x0
2
Objective x²/2
2
Iteration behavior
Stable step
n = 0 · t = 0 s

Gradient descent on f(x) = x²/2: xₙ = (1 − η)ⁿx₀. Playback advances two discrete iterations per second. Values outside |x| ≤ 4 are explicitly reported. Parameter changes pause playback and recompute the model at the current elapsed time. Reset restores both parameters and returns time to zero. Playback starts paused. Adjust either parameter, then choose Play or Step.

Look forMirror descent changes the movement geometry, while this particular displayed model retains ordinary Euclidean distance.

Check your understanding

If every gradient component in an entropic simplex step is increased by the same constant, what happens to the normalized update?

Try it yourself

Use an entropic step with η=1, p=(1/4,3/4)p=(1/4,3/4), and g=(log⁡3,0)g=(\log3,0). Find the unnormalized weights and normalized probabilities.

Take a moment to make a prediction or work it out on paper. Open a hint when you need one.

Give me a hint

The first exponential factor is one third and the second is one. Add the two unnormalized weights before dividing.

Compare with the solution

The unnormalized weights are (1/4)/3=1/12(1/4)/3=1/12 and 3/4=9/123/4=9/12. Their sum is 10/12=5/610/12=5/6, so the next distribution is (1/10,9/10)(1/10,9/10). The first coordinate remains positive but its relative weight is divided by three. Normalization changes absolute weights without changing this ratio calculation.

Take it with you

  • A Bregman divergence measures separation from a chosen convex generator's tangent plane.
  • Negative entropy on the simplex leads to normalized multiplicative updates.
  • Norm bounds, feasible geometry, and numerical domain handling determine whether a mirror method is appropriate.

Read further

Original sources and deeper explanations to follow at your own pace.

Optional progress, saved in this browser when available. Revisit any lesson at any time.

Lesson not marked complete.