English

Non-Convex Tensor Recovery from Local Measurements

Machine Learning 2024-12-24 v1

Abstract

Motivated by the settings where sensing the entire tensor is infeasible, this paper proposes a novel tensor compressed sensing model, where measurements are only obtained from sensing each lateral slice via mutually independent matrices. Leveraging the low tubal rank structure, we reparameterize the unknown tensor X{\boldsymbol {\mathcal X}}^\star using two compact tensor factors and formulate the recovery problem as a nonconvex minimization problem. To solve the problem, we first propose an alternating minimization algorithm, termed \textsf{Alt-PGD-Min}, that iteratively optimizes the two factors using a projected gradient descent and an exact minimization step, respectively. Despite nonconvexity, we prove that \textsf{Alt-PGD-Min} achieves ϵ\epsilon-accuracy recovery with O(κ2log1ϵ)\mathcal O\left( \kappa^2 \log \frac{1}{\epsilon}\right) iteration complexity and O(κ6rn3logn3(κ2r(n1+n2)+n1log1ϵ))\mathcal O\left( \kappa^6rn_3\log n_3 \left( \kappa^2r\left(n_1 + n_2 \right) + n_1 \log \frac{1}{\epsilon}\right) \right) sample complexity, where κ\kappa denotes tensor condition number of X\boldsymbol{\mathcal X}^\star. To further accelerate the convergence, especially when the tensor is ill-conditioned with large κ\kappa, we prove \textsf{Alt-ScalePGD-Min} that preconditions the gradient update using an approximate Hessian that can be computed efficiently. We show that \textsf{Alt-ScalePGD-Min} achieves κ\kappa independent iteration complexity O(log1ϵ)\mathcal O(\log \frac{1}{\epsilon}) and improves the sample complexity to O(κ4rn3logn3(κ4r(n1+n2)+n1log1ϵ))\mathcal O\left( \kappa^4 rn_3 \log n_3 \left( \kappa^4r(n_1+n_2) + n_1 \log \frac{1}{\epsilon}\right) \right). Experiments validate the effectiveness of the proposed methods.

Keywords

Cite

@article{arxiv.2412.17281,
  title  = {Non-Convex Tensor Recovery from Local Measurements},
  author = {Tongle Wu and Ying Sun and Jicong Fan},
  journal= {arXiv preprint arXiv:2412.17281},
  year   = {2024}
}

Comments

The paper was accepted by AAAI 2025