Mathematics for AI and Machine Learning

Foundations for modern AI and machine learning

Langevin Step

Chapter 18 Dynamics & diffusion

Chapter 18: Langevin Dynamics and Sampling — Stationary Distribution Convergence

Animated preview (GIF)
Langevin Step — animated GIF preview
High-resolution PNG
Langevin Step — high-resolution mind-map icon

From the book

Chapter 18: Langevin Dynamics and Sampling. In the chapter mind map this icon labels Stationary Distribution Convergence. The discussion below is excerpted and lightly edited from § Example: Langevin Dynamics for a Simple Gaussian Distribution in Mathematics for AI and Machine Learning.

To illustrate how Langevin dynamics works, consider sampling from a 1D Gaussian distribution $p(x) = \mathcal{N}(x; \mu, \sigma^2)$ with $\mu = 0$ and $\sigma^2 = 1$.

Using the discrete-time update rule with step size $\eta = 0.01$:

Starting from $x_0 = 5$ (far from the mean), after 40,000 iterations:

  • The sample distribution converges to $\mathcal{N}(0, 1)$
  • The trajectory shows the particle moving toward the mean while being perturbed by noise
  • The algorithm successfully explores the entire distribution

the book figure demonstrates the convergence of Langevin dynamics to the $\mathcal{N}(0, 1)$ distribution. Starting from $x_0 = 5$ (far from the mean), the top panel shows the trajectory converging toward $\mu = 0$ with noise perturbations, with the starting point marked by a green dot. The bottom panel compares the high-resolution histogram of samples (all steps, 150 bins) with the true $\mathcal{N}(0, 1)$ distribution, confirming successful convergence.

What this drawing shows

What you see. Single Langevin update from $x_k$: dashed blue drift along $\nabla\log p$ versus red step that adds a noise kick beyond the drift target.

In the mind map. Chapter 18 — Stationary Distribution Convergence. See From the book above for definitions, figures, and worked examples.

Where to read next

Open Chapter 18 companion →

Read the full definitions, figures, and worked examples in Chapter 18: Langevin Dynamics and Sampling — see the mind-map node Stationary Distribution Convergence.