← All problems
Unverified

How Much Overparameterization Is Needed for ALS in Tensor Decomposition?

Let

T=∑j=1raj⊗bj⊗cj, T=\sum_{j=1}^{r} a_j\otimes b_j\otimes c_j,

where A=[a1 ⋯ ar]A=[a_1\ \cdots\ a_r], B=[b1 ⋯ br]B=[b_1\ \cdots\ b_r], and C=[c1 ⋯ cr]C=[c_1\ \cdots\ c_r] are n×rn\times r factor matrices and r≪nr\ll n. Consider the following smoothed-analysis setting. Start from arbitrary well-conditioned matrices Aˉ,Bˉ,Cˉ\bar A,\bar B,\bar C. Independently perturb every factor by Gaussian noise:

ai−aˉi, bi−bˉi, ci−cˉi∼N ⁣(0,ρ2nIn),ρ≈1poly⁡(r). a_i-\bar a_i,\ b_i-\bar b_i,\ c_i-\bar c_i \sim \mathcal{N}\!\left(0,\frac{\rho^2}{n}I_n\right), \qquad \rho\approx\frac{1}{\operatorname{poly}(r)}.

Given TT, run ALS, gradient descent, or another iterative algorithm from random initialization on

min⁡xi,yi,zi∈Rn, i∈[k]∥T−∑i=1kxi⊗yi⊗zi∥F2. \min_{x_i,y_i,z_i\in\mathbb{R}^n,\ i\in[k]} \left\|T-\sum_{i=1}^{k}x_i\otimes y_i\otimes z_i\right\|_F^2.
  1. What is the smallest value k=k(r)k=k(r) for which ALS, or another iterative algorithm such as gradient descent, converges to the global optimum with high probability over the smoothed instance?

  2. Does subquadratic overparameterization k=o(r2)k=o(r^2) suffice to find, with high probability over the smoothed instance and in time poly⁡(n,r,log⁡(1/ϵ))\operatorname{poly}(n,r,\log(1/\epsilon)), a rank-kk decomposition satisfying

∥T−∑i=1kxi⊗yi⊗zi∥F≤ϵ∥T∥F? \left\|T-\sum_{i=1}^{k}x_i\otimes y_i\otimes z_i\right\|_F \leq \epsilon\|T\|_F?
  1. Conversely, can one prove that there is a universal constant c>0c>0 such that ALS, gradient descent, and related iterative methods require k=r1+ck=r^{1+c}? More precisely, for k≤r1+ck\leq r^{1+c}, does the algorithm converge to a solution of positive objective value with constant probability on a smoothed instance?

A particularly compelling lower bound would hold for random or smoothed rank-rr tensors rather than only for a worst-case pathological tensor. It is also of interest to determine whether the lower bound can approach k=r2−ck=r^{2-c} for a small constant c>0c>0.

Coming soon

Organizer

Boyuan Wang portraitBoyuan Wang
Minghan Wang portraitMinghan Wang
Bochao Li portraitBochao Li
Hongwei Hu portraitHongwei Hu