揭示神经网络特征学习的相变阈值,解释为何某些数据量下学习会失败。
Phase Transitions for Feature Learning in Neural Networks
- 通过梯度下降分析两层网络在高维数据中的特征学习机制。
- 发现存在临界样本比δ_NN,低于此值则无法有效学习。
- 揭示训练后期海森矩阵谱变化导致的学习停滞现象,适合研究理论深度者阅读。
根据主流观点,神经网络通过先识别低维表示,再在此空间中拟合最优模型来学习。近期工作形式化了多指标模型下的这一现象:给定n个独立同分布的样本对(𝐱ᵢ, 𝑦ᵢ),其中𝐱ᵢ ∈ ℝᵈ为各向同性协变量,响应𝑦ᵢ仅依赖于𝐱ᵢ在𝑘维投影Θ*ᵀ𝐱ᵢ上的值。特征学习即学习由Θ*张成的隐空间。本文研究在比例渐近n,d→∞、n/d→δ,且隐空间维度𝑘和隐藏神经元数𝑚固定条件下,两层神经网络的梯度下降动力学。已有研究表明,当δ > δ_alg时,多项式时间算法可实现特征学习;而δ < δ_alg时则不可能(在特定算法类中)。本文推导出两层网络对应的阈值δ_NN。δ_NN的刻画揭示了学习动态对网络结构与训练算法的依赖关系。该阈值对应于训练进入第二阶段后海森矩阵谱的相变现象:初期梯度大,学习梯度方向;后期梯度减小,系统受海森负方向主导,此时出现谱相变。
原文摘要 · Abstract (English)
According to a popular viewpoint, neural networks learn from data by first identifying low-dimensional representations, and subsequently fitting the best model in this space. Recent works provide a formalization of this phenomenon when learning multi-index models. In this setting, we are given $n$ i.i.d. pairs $({\boldsymbol x}_i,y_i)$, where the covariate vectors ${\boldsymbol x}_i\in\mathbb{R}^d$ are isotropic, and responses $y_i$ only depend on ${\boldsymbol x}_i$ through a $k$-dimensional projection ${\boldsymbol Θ}_*^{\sf T}{\boldsymbol x}_i$. Feature learning amounts to learning the latent space spanned by ${\boldsymbol Θ}_*$. In this context, we study the gradient descent dynamics of two-layer neural networks under the proportional asymptotics $n,d\to\infty$, $n/d\toδ$, while the dimension of the latent space $k$ and the number of hidden neurons $m$ are kept fixed. Earlier work establishes that feature learning via polynomial-time algorithms is possible if $δ> δ_{\text{alg}}$, for $δ_{\text{alg}}$ a threshold depending on the data distribution, and is impossible (within a certain class of algorithms) below $δ_{\text{alg}}$. Here we derive an analogous threshold $δ_{\text{NN}}$ for two-layer networks. Our characterization of $δ_{\text{NN}}$ opens the way to study the dependence of learning dynamics on the network architecture and training algorithm. The threshold $δ_{\text{NN}}$ is determined by the following scenario. Training first visits points for which the gradient of the empirical risk is large and learns the directions spanned by these gradients. Then the gradient becomes smaller and the dynamics becomes dominated by negative directions of the Hessian. The threshold $δ_{\text{NN}}$ corresponds to a phase transition in the spectrum of the Hessian in this second phase.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。