arXiv:2411.17201cs.LGcs.AI2024-11被引 5

三层网络可高效学习多非线性特征的分层多项式,突破传统方法样本瓶颈。

Learning Hierarchical Polynomials of Multiple Nonlinear Features with Three-Layer Networks

  • 用三层网络分层学习多个二次非线性特征的组合函数
  • 仅需约 d⁴ 个样本即可恢复特征空间并学习目标函数
  • 适用于复杂特征学习场景,为深度模型泛化提供理论支持

在深度学习理论中,理解神经网络如何学习分层特征是一个关键问题。本文研究三层数神经网络对多个非线性特征的分层多项式的学习能力。考虑一类形式为 $f^{ ext{⋆}}=g^{ ext{⋆}} abla p$ 的函数,其中 $p:bR^d o bR^r$ 表示 $r \\< d$ 个二次特征,$g^{ ext{⋆}}:bR^r o bR$ 是次数为 $p$ 的多项式。这可视为多指标模型的非线性推广,也拓展了以往仅针对单个非线性特征($r=1$)的研究。主要贡献表明,通过逐层梯度下降训练的三层数网络,可在 $ ilde{O}(d^4)$ 样本和多项式时间内:完全恢复非线性特征张成的空间;高效学习目标函数 $f^{ ext{⋆}} = g^{ ext{⋆}} abla p$,或实现不同链接函数下的迁移学习 $f = g abla p$。相比核方法 $Θ(d^{2p})$ 的样本复杂度,该结果显著降低,凸显了高效特征学习的优势。结果依赖全新技术,超越了单指标、多指标模型及仅依赖单一非线性特征的先前设定,推动了对深度学习特征学习更全面的理解。

原文摘要 · Abstract (English)

In deep learning theory, a critical question is to understand how neural networks learn hierarchical features. In this work, we study the learning of hierarchical polynomials of \textit{multiple nonlinear features} using three-layer neural networks. We examine a broad class of functions of the form $f^{\star}=g^{\star}\circ \bp$, where $\bp:\mathbb{R}^{d} \rightarrow \mathbb{R}^{r}$ represents multiple quadratic features with $r \ll d$ and $g^{\star}:\mathbb{R}^{r}\rightarrow \mathbb{R}$ is a polynomial of degree $p$. This can be viewed as a nonlinear generalization of the multi-index model \citep{damian2022neural}, and also an expansion upon previous work that focused only on a single nonlinear feature, i.e. $r = 1$ \citep{nichani2023provable,wang2023learning}. Our primary contribution shows that a three-layer neural network trained via layerwise gradient descent suffices for \begin{itemize}\item complete recovery of the space spanned by the nonlinear features \item efficient learning of the target function $f^{\star}=g^{\star}\circ \bp$ or transfer learning of $f=g\circ \bp$ with a different link function \end{itemize} within $\widetilde{\cO}(d^4)$ samples and polynomial time. For such hierarchical targets, our result substantially improves the sample complexity $Θ(d^{2p})$ of the kernel methods, demonstrating the power of efficient feature learning. It is important to highlight that{ our results leverage novel techniques and thus manage to go beyond all prior settings} such as single-index and multi-index models as well as models depending just on one nonlinear feature, contributing to a more comprehensive understanding of feature learning in deep learning.

深度学习理论特征学习三层数网络分层多项式

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