Skip to yearly menu bar Skip to main content


Poster

Beyond Lazy Training for Over-parameterized Tensor Decomposition

Xiang Wang · Chenwei Wu · Jason Lee · Tengyu Ma · Rong Ge

Poster Session 4 #1247

Abstract: Over-parametrization is an important technique in training neural networks. In both theory and practice, training a larger network allows the optimization algorithm to avoid bad local optimal solutions. In this paper we study a closely related tensor decomposition problem: given an l-th order tensor in (Rd)l of rank r (where rd), can variants of gradient descent find a rank m decomposition where m>r? We show that in a lazy training regime (similar to the NTK regime for neural networks) one needs at least m=Ω(dl1), while a variant of gradient descent can find an approximate tensor when m=O(r2.5llogd). Our results show that gradient descent on over-parametrized objective could go beyond the lazy training regime and utilize certain low-rank structure in the data.

Chat is not available.