3D Gradient Descent

Machine Learning Theory
Visualization
React
Plotly
JavaScript
CSS
web
Published

November 30, 2022

A React web app that visualizes gradient descent in 3D.

With this app you can:

  • manually fit a quadratic function to some data,
  • use a Stepper to automatically step the parameters towards the local minimum,
  • have a look at the trace the loss left as it walked through the loss surface,
  • choose from different critics (mae,mse,rmse) to “see” how changing the way we measure the loss effects gradient descent,
  • intuitively interact with the 3D-plot by dragging and zoom, to keep the interesting parts in focus.