The Vanishing Gradient Plateau: Why Sigmoid Networks Saturate
Consider a single neuron with sigmoid activation , where .
Training uses gradient descent on the squared loss for a single training example .
(a) Show that the gradient contains the factor .
(b) Now suppose the network is initialized with a large weight and , . Argue clearly: even though the network is making a large error (since but ), learning is extremely slow. What is the maximum possible value of , and where is it achieved?
(c) This is the vanishing gradient / saturation phenomenon. Give a one-sentence intuition for why the sigmoid's shape causes this, and name one architectural change that mitigates it.
Answer: The Vanishing Gradient Plateau: Why Sigmoid Networks Saturate
Key Idea / Intuition
The sigmoid function "squashes" its input into , which is great for probability interpretation โ but the price is that its derivative is nearly zero whenever is large. So when a neuron is confidently wrong (output near 0 or 1), the gradient almost vanishes, and weights barely move. The network is stuck in a flat landscape of its own making.
Formal Proof / Solution
Part (a): Computing the gradient
By the chain rule:
Each factor:
- (standard sigmoid derivative)
So:
The factor appears explicitly.
Part (b): Large error yet slow learning
With , , : we have , so .
-
Error: . This is a large error (the neuron is outputting when the target is ).
-
Gradient factor: .
So the gradient . Learning is essentially frozen despite the large loss.
Maximum of :
Let . We maximize by AM-GM or calculus:
So the maximum gradient factor is , achieved when (the neuron is exactly at the decision boundary, outputting ).
This means even in the best case, back-propagating through a sigmoid layer multiplies the gradient by at most . With many layers, this compounds: through layers, the gradient can shrink by up to .
Part (c): Intuition and fix
Intuition: The sigmoid flattens out at both ends of its S-curve; a neuron that has made up its mind (output near 0 or 1) sits in a flat region where infinitesimal changes to the weight produce infinitesimal changes in output, killing the learning signal.
Fix: Replace sigmoid with ReLU (), whose derivative is either or โ it doesn't saturate for positive inputs, so gradients flow through cleanly (at the cost of "dying ReLU" for negative inputs, addressed by Leaky ReLU or ELU).
Summary Table
| Quantity | Value | |---|---| | Max of | at | | Value when | | | Gradient at large | despite large loss | | Per-layer gradient decay | up to factor |
Written to: questions/2026-06-17_am.md
Source: The Elements of Statistical Learning, Ch. 11 (Hastie, Tibshirani, Friedman)