Dying ReLU in Function Space
Train a small 24-neuron ReLU layer live and watch its hidden units die in 2D function space: every neuron's piecewise-linear hinge function is plotted directly, so you see the exact kink location freeze the moment a plain ReLU neuron's gradient goes to zero.
A 24-neuron hidden layer trains live, in your browser, to fit a wiggly target curve using ordinary gradient descent — the same network as the 3D bar version, seen through a genuinely different lens. Instead of a bar rising and falling per neuron, every hidden unit is drawn as its actual piecewise-linear hinge function v_i·ReLU(w_i·x + b_i), overlaid directly on the x–y plane the network is fitting. A small triangle marks each hinge's breakpoint on the x-axis — the point where the neuron switches from flat to sloped. Push the learning rate or initial weight scale up and watch a cluster of hinges snap flat and turn red, their breakpoints frozen in place, the instant their gradient goes to exactly zero; dial in a Leaky ReLU slope and see previously frozen hinges start tilting again as a small negative-side gradient starts flowing through them.
Train a 24-neuron hidden layer live and watch its ReLU units die in 2D function space: every neuron's own piecewise-linear hinge function is plotted directly on the curve it's fitting, so you see the exact breakpoint freeze the instant a plain ReLU neuron's gradient drops to zero.
2D · HTML5 Canvas 2D · 60 FPS target · runs fully client-side, no install