∇Continuous Function
FoundationsChecking saved investigationReading browser-local route memory before showing a continuation.

Foundation Lab

Natural Gradient & Riemannian Optimization

Natural gradient is coordinate-invariant—it gives the same update regardless of parameterization

Concept 56 of 100OptimizationPhase 10
Open DemoAll Foundations
#56Natural GradOptimization
key equation\tilde{\nabla} L = F(\theta)^{-1} \nabla_\theta L

Selected Foundation Object

Keep the equation fixed; move through the evidence.

Concept 56 of 100Natural GradOptimization / Phase 10: Mathematical foundations & information geometry
Current question

The gradient is a covector, not a vector—the metric turns it into a direction of steepest descent

\tilde{\nabla} L = F(\theta)^{-1} \nabla_\theta L
PredictionCommit before the demo.

Ask what should change when the equation is manipulated, then let the visualization test that expectation.

EvidenceCompare local witness and source.

Use the runnable panel, the key equation, and canonical papers as separate forms of evidence for the same object.

InvariantName what survives notation changes.

The useful learning product is the reusable mechanism you can carry into another model, paper, or engineering tradeoff.

Next moveContinue through the atlas.

Use prerequisites, dependents, and semantic links to repair the next gap without leaving the object behind.

Open demoInspect equationBrowse atlasNext foundation

Why It Matters for Modern Models

  • Natural gradient is coordinate-invariant—it gives the same update regardless of parameterization
  • TRPO and PPO are approximations to natural gradient updates for policy optimization
  • Adam can be viewed as a diagonal approximation to natural gradient with adaptive preconditioning

What Tutorials Skip

What is still poorly explained in textbooks and papers:

  • The gradient is a covector, not a vector—the metric turns it into a direction of steepest descent
  • Euclidean gradient depends on how you parameterize; natural gradient depends only on the distributions
  • Natural gradient avoids plateaus faster because it accounts for local curvature in distribution space

Interactive Visualization

Core Math (Optional Deep Dive)

If you want intuition first, start with the key equation and the visualization. Come back here for the full walkthrough.

Key Equation
∇~L=F(θ)−1∇θL\tilde{\nabla} L = F(\theta)^{-1} \nabla_\theta L

Natural gradient uses the Fisher metric instead of Euclidean:

∇~L=F(θ)−1∇θL\tilde{\nabla} L = F(\theta)^{-1} \nabla_\theta L

Update rule:

θt+1=θt−ηF(θt)−1∇L\theta_{t+1} = \theta_t - \eta F(\theta_t)^{-1} \nabla L

Variational characterization (why it's "natural"):

δ∗=arg⁡min⁡δ⟨∇L,δ⟩s.t.KL(pθ∥pθ+δ)≤ϵ\delta^* = \arg\min_\delta \langle \nabla L, \delta \rangle \quad \text{s.t.} \quad \text{KL}(p_\theta \| p_{\theta+\delta}) \leq \epsilon
⇒δ∗∝F−1∇L\Rightarrow \delta^* \propto F^{-1} \nabla L

Canonical Papers

Natural Gradient Works Efficiently in Learning

Amari1998Neural Computation
Read paper →

Trust Region Policy Optimization

Schulman et al.2015ICML
Read paper →

Connections

Prerequisites

ℐFisher Info∇Adam

Enables

🪞Bregman

Next Moves

Choose the next question to carry this object forward.