论文

轻量化 Tucker 低秩张量学习方法:内存最高降 900 倍以上

Lightweight Representation Learning for Low Tucker Rank Tensors

精选理由

张量计算的人看过来,这篇没改模型,纯工程优化就把 Tucker 学习内存砍了 900 倍,轨迹还能对到 10^-16 精度。

arXiv 论文提出一套针对低 Tucker 秩张量学习的轻量化实现,不改变表示目标或参数化。方法包含三个部分:只提取任务所需谱分量的顺序抽取、避免构造大型 Kronecker 算子的逐因子多重线性收缩、以及不存储完整环境张量的紧凑潜状态更新。在 5 个低 Tucker 秩张量估计器上,实现与原算法轨迹一致(相对误差 10^-16 到 10^-13),单次迭代运行时间最高降至 3.12 倍快,n=100 的 order-3 基准上常驻内存最多降低 616 倍(99.8%),整套评测内存降低超过 900 倍。

原文 · arXiv cs.LG

Lightweight Representation Learning for Low Tucker Rank Tensors

Low Tucker rank tensors provide compact representations of multidimensional dependencies, but standard learning pipelines can incur substantial computational and memory overhead from redundant intermediate quantities. We reduce this overhead without modifying the underlying representation objective or parameterization. Our approach combines three complementary components: (i) sequential extraction of only task-required spectral components, avoiding unnecessary full decompositions; (ii) factor-wise multilinear contractions without materializing large Kronecker operators; and (iii) compact latent-state updates without storing full ambient tensor iterates. These components preserve the algebraic operations and learning trajectories of the original algorithms while reducing their computational and memory footprints. Across five low Tucker rank tensor estimators, our implementations match the original trajectories to numerical precision (relative discrepancies of $10^{-16}$ to $10^{-13}$), while reducing per-iteration runtime by up to $3.12\times$ and resident memory by up to $616\times$ ($99.8\%$) on the main order-3 benchmark at $n=100$, with memory reductions exceeding $900\times$ across the full evaluation suite. These results show that substantial computational redundancy can be eliminated from Tucker representation learning without changing the underlying model or learned representation.