arXiv:2603.21062stat.MLcs.LG2026-03

用投影梯度下降训练两层神经网络,高效学习低次球面多项式。

Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate

  • 提出投影梯度下降算法,突破传统核方法的样本瓶颈。
  • 样本复杂度仅需 d^k₀/ε,接近最优下界,且对δ有良好依赖。
  • 首次证明带ReLU的过参数化网络可实现近最优泛化率,适合高维学习任务。

本文研究在单位球面上学习低次球面多项式(次数 $k_0 = Θ(1) o 1$)的问题,通过训练带有增强特征的过参数化两层神经网络实现。主结果表明:对于任意回归风险 $\ힶ \\in (0, Θ(d^{-k_0})]$,采用新型投影梯度下降(GDP)训练的网络,以概率 $1-δ$ 实现样本复杂度 $n \asymp Θ( \log(4/δ) \cdot d^{k_0}/\eps)$,显著优于传统方法 $Θ(d^{k_0} \max\{\eps^{-2},\log d\})$。该样本复杂度近乎不可改进,因网络达到非参数回归风险 $\log(4/δ) \cdot Θ(d^{k_0}/n)$,接近最小最大最优率 $Θ(d^{k_0}/n)$。当真实次数 $k_0$ 未知时,还提出可证明的自适应度选择算法,能准确识别 $k_0$ 并保持近最优率。此为首次在使用主流 ReLU 激活函数与算法保证下,实现低次球面多项式学习的近最优风险界。由于 GDP 具备特征学习能力,结果超越经典 Neural Tangent Kernel (NTK) 极限。

原文摘要 · Abstract (English)

We study the problem of learning a low-degree spherical polynomial of degree $k_0 = Θ(1) \ge 1$ defined on the unit sphere in $\RR^d$ by training an over-parameterized two-layer neural network with augmented feature in this paper. Our main result is the significantly improved sample complexity for learning such low-degree polynomials. We show that, for any regression risk $\eps \in (0, Θ(d^{-k_0})]$, an over-parameterized two-layer neural network trained by a novel Gradient Descent with Projection (GDP) requires a sample complexity of $n \asymp Θ( \log(4/δ) \cdot d^{k_0}/\eps)$ with probability $1-δ$ for $δ\in (0,1)$, in contrast with the representative sample complexity $Θ(d^{k_0} \max\set{\eps^{-2},\log d})$. Moreover, such sample complexity is nearly unimprovable since the trained network renders a nearly optimal rate of the nonparametric regression risk of the order $\log({4}/δ) \cdot Θ(d^{k_0}/{n})$ with probability at least $1-δ$. On the other hand, the minimax optimal rate for the regression risk with a kernel of rank $Θ(d^{k_0})$ is $Θ(d^{k_0}/{n})$, so that the rate of the nonparametric regression risk of the network trained by GDP is nearly minimax optimal. In the case that the ground truth degree $k_0$ is unknown, we present a novel and provable adaptive degree selection algorithm which identifies the true degree and achieves the same nearly optimal regression rate. To the best of our knowledge, this is the first time that a nearly optimal risk bound is obtained by training an over-parameterized neural network with a popular activation function (ReLU) and algorithmic guarantee for learning low-degree spherical polynomials. Due to the feature learning capability of GDP, our results are beyond the regular Neural Tangent Kernel (NTK) limit.

神经网络泛化理论优化算法低次多项式

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