arXiv:2604.10208cs.LG2026-04

提出新方法在有限内存下实现高效不对称张量PCA学习

Mild Over-Parameterization Benefits Asymmetric Tensor PCA

  • 用矩阵参数化与三阶段交替更新,仅需d²内存
  • 样本复杂度接近最优,达d^k-2(k≥4为偶数)
  • 轻微过参数化提升适应性,适合结构有规律的问题

不对称张量主成分分析(ATPCA)是研究样本复杂度、计算与内存权衡的典型模型。现有算法通常需要至少d^{⌈k/2⌉}的存储空间来恢复信号,其中d为向量维度,k为张量阶数。本文聚焦于k≥4为偶数的情形,在有限内存预算下采用基于梯度下降的方法,允许模型存在轻微过参数化。提出一种矩阵参数化方法(内存开销d²),结合新颖的三阶段交替更新算法,证明了适度过参数化在两方面促进学习:(i) 提升样本效率,使方法在受限内存下实现近似最优的d^{k-2}样本复杂度;(ii) 增强对问题结构的自适应能力——当连续向量更对齐时,所需样本数自然下降,在对称极限下达到d^{k/2},匹配目前已知的最佳多项式时间复杂度。据我们所知,这是首个实现d^k无关内存开销的ATPCA可计算算法。

原文摘要 · Abstract (English)

Asymmetric Tensor PCA (ATPCA) is a prototypical model for studying the trade-offs between sample complexity, computation, and memory. Existing algorithms for this problem typically require at least $d^{\left\lceil\overline{k}/2\right\rceil}$ state memory cost to recover the signal, where $d$ is the vector dimension and $\overline{k}$ is the tensor order. We focus on the setting where $\overline{k} \geq 4$ is even and consider (stochastic) gradient descent-based algorithms under a limited memory budget, which permits only mild over-parameterization of the model. We propose a matrix-parameterized method (in $d^{2}$ state memory cost) using a novel three-phase alternating-update algorithm to address the problem and demonstrate how mild over-parameterization facilitates learning in two key aspects: (i) it improves sample efficiency, allowing our method to achieve \emph{near-optimal} $d^{\overline{k}-2}$ sample complexity in our limited memory setting; and (ii) it enhances adaptivity to problem structure, a previously unrecognized phenomenon, where the required sample size naturally decreases as consecutive vectors become more aligned, and in the symmetric limit attains $d^{\overline{k}/2}$, matching the \emph{best} known polynomial-time complexity. To our knowledge, this is the \emph{first} tractable algorithm for ATPCA with $d^{\overline{k}}$-independent memory costs.

张量分析优化算法机器学习理论

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。