arXiv:2605.15082stat.MLcs.LG2026-05

用核回归的梯度外积,少样本就能发现数据隐藏结构。

Average Gradient Outer Product in kernel regression provably recovers the central subspace for multi-index models

论文配图:Average Gradient Outer Product in kernel regression provably recovers the central subspace for multi-index models
图 1 · 摘自论文原文
  • 通过核岭回归拟合后计算平均梯度外积,提取低维结构。
  • 只需约 $d^{p+δ}$ 个样本即可恢复核心子空间,远低于预测所需 $d^{p^*}$。
  • 解释了递归特征机器等方法为何在小样本下仍高效。

我们研究一种典型情形:学习器能在样本数少于准确预测所需时,发现数据中的有用低维结构。具体地,考虑从有限数据/标签对中恢复多指标多项式函数 $f^*(x)=h(Ux)$,其中 $U\in\mathbb{R}^{r\times d}$ 且 $r\ll d$,目标函数仅依赖于输入 $x$ 在未知 $r$ 维核心子空间上的投影。分析的算法极为简单:对数据拟合核岭回归(KRR),并从拟合结果中计算平均梯度外积(AGOP)。主要结果表明,在合理假设下,AGOP 的前 $r$ 维特征空间可严格恢复核心子空间,即使此时预测误差仍较大。若目标函数 $f^*$ 的次数为 $p^*$,则已知需 $n\asymp d^{p^*}$ 样本才能使 KRR 实现高精度预测;而我们证明,若 $f^*$ 的低阶 $p$ 项已包含所有预测相关方向,则在更小样本范围 $n\asymp d^{p+δ}$(任意 $δ\in(0,1)$)内即可完成子空间恢复。结果揭示了预测与表示之间的分离现象,为递归特征机器(RFM)等迭代核方法在实践中具备样本效率提供了理论解释。

原文摘要 · Abstract (English)

We study a prototypical situation when a learned predictor can discover useful low-dimensional structure in data, while using fewer samples than are needed for accurate prediction. Specifically, we consider the problem of recovering a multi-index polynomial $f^*(x)=h(Ux)$, with $U\in\mathbb{R}^{r\times d}$ and $r\ll d$, from finitely many data/label pairs. Importantly, the target function depends on input $x$ only through the projection onto an unknown $r$-dimensional central subspace. The algorithm we analyze is appealingly simple: fit kernel ridge regression (KRR) to the data and compute the Average Gradient Outer Product (AGOP) from the fitted predictor. Our main results show that under reasonable assumptions the top $r$-dimensional eigenspace of AGOP provably recovers the central subspace, even in regimes when the prediction error remains large. Specifically, if the target function $f^*$ has degree $p^*$, it is known that $n\asymp d^{p^*}$ samples are necessary for KRR to achieve accurate prediction. In contrast, we show that if a low degree $p$ component of $f^*$ already carries all relevant directions for prediction, subspace recovery occurs in the much lower sample regime $n\asymp d^{p+δ}$ for any $δ\in(0,1)$. Our results thus demonstrate a separation between prediction and representation, and provide an explanation for why iterative kernel methods such as Recursive Feature Machines (RFM) can be sample-efficient in practice.

核方法降维多指标模型样本效率

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