2022

Implicit Regularization Towards Rank Minimization in ReLU Networks

Timor, Nadav, Vardi, Gal, Shamir, Ohad

Understand

We study the conjectured relationship between the implicit regularization in neural networks, trained with gradient-based methods, and rank minimization of their weight matrices.

  • Previously, it was proved that for linear networks (of depth 2 and vector-valued outputs), gradient flow (GF) w.r.t.
  • the square loss acts as a rank minimization heuristic.
  • However, understanding to what extent this generalizes to nonlinear networks is an open problem.

Reading the bibliography…