Mean-field VI on a banana
What you are seeing: the simplest variational inference setup, fit to the simplest distribution that breaks it. Target: a Rosenbrock-style "banana" , a long curved valley. Variational family: a product , axis-aligned Gaussian (mean-field). VI maximizes by gradient ascent.
The contour lines are the target . The orange ellipse is the current . The yellow trace below is the ELBO over training. The "best" Gaussian approximation cannot follow the banana's curvature; it settles into a compact ellipse at the bend.
lr0.005
K (MC)32
speed5
WHAT TO TRY
- Vary each control and watch the rail readouts respond.
- Compare the diagnostic plot against the live scene.