Implicit bias of SGD in L_2-regularized linear DNNs: One-way jumps from high to low rank

05/25/2023
by   Zihan Wang, et al.
0

The L_2-regularized loss of Deep Linear Networks (DLNs) with more than one hidden layers has multiple local minima, corresponding to matrices with different ranks. In tasks such as matrix completion, the goal is to converge to the local minimum with the smallest rank that still fits the training data. While rank-underestimating minima can easily be avoided since they do not fit the data, gradient descent might get stuck at rank-overestimating minima. We show that with SGD, there is always a probability to jump from a higher rank minimum to a lower rank one, but the probability of jumping back is zero. More precisely, we define a sequence of sets B_1⊂ B_2⊂⋯⊂ B_R so that B_r contains all minima of rank r or less (and not more) that are absorbing for small enough ridge parameters λ and learning rates η: SGD has prob. 0 of leaving B_r, and from any starting point there is a non-zero prob. for SGD to go in B_r.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset