Let the loss be the quadratic , where is symmetric with every eigenvalue between and . Gradient descent takes with a constant . Derive how the gap contracts in the worst direction, find the step that makes the worst-case contraction as fast as possible, and give the resulting contraction factor for and , to four decimal places.
Worked solution
Because is symmetric, write with orthogonal and diagonal, and set . The quadratic becomes and the gradient is . One step is therefore coordinate by coordinate. After steps, where . Each factor is largest at an endpoint of the interval , so .
For the term is the larger one and falls as grows, so the best step in that range is . Beyond the second term is , which grows with , while the first term still falls. The two are equal when At that step, For and the best step is and the factor is which rounds to . Compare the step used in section 17.4: there the factor is , so balancing the two curvatures is faster. The derivation also shows why the text's factor was not the end of the story: it is the best step among those at or below , and the balanced step sits above . To reach a gap smaller by a factor of the two factors need about steps and about steps respectively, so at this condition number the balanced step nearly halves the run.