Assistant Professor Dominik Stöger will give a talk about: 'Which Sparse Solution Does Gradient Descent Find? Implicit Bias and Convergence Rates for Diagonal Linear Networks Under General Initialization'
Modern machine learning models are often overparametrized, with more parameters than training samples. The implicit bias of gradient descent determines which of the many solutions that fit the data is selected. Diagonal linear networks, in which a linear predictor is parametrized as an entrywise product of several weight vectors, are a tractable model for studying this bias in underdetermined linear regression. A well-known result states that gradient descent converges to the solution with minimum l1-norm as the scale of initialization tends to zero. This connects training dynamics to sparse recovery. However, the result assumes that all parameters are initialized to the same value. In this talk, we remove this assumption and address two questions: (a) What is the implicit bias when parameters are not initialized identically? (b) How fast is the implicit bias approached as the scale of initialization decreases?
For (a), we show that gradient flow converges to the minimizer of a weighted l1-type penalty whose weights depend on both the initialization and the sign of each coordinate of the solution. Our proof relies on a new characterization of the gradient dynamics in terms of a Bregman divergence. For (b), we derive tight convergence rates. In particular, we show that for two-layer networks the limit of gradient flow converges to this minimizer at a sublinear rate in the scale of initialization, whereas for more layers the rate of convergence is linear. Surprisingly, the constants governing these rates are connected to the null-space constants of compressed sensing. Our rates improve on previous results in the identical initialization setting and are, to our knowledge, the first in the general initialization setting.
Joint work with Hannes Matt (KU Eichstätt-Ingolstadt).