SGD 在两层神经网络中寻找并调整特征,具有近乎最优的样本复杂度:以 XOR 问题为案例研究
💡
原文中文,约300字,阅读约需1分钟。
📝
内容提要
本文研究了使用SGD训练任意宽度的两层神经网络,证明了第一层权重将收敛于真实模型的k维主子空间,使用SGD训练的ReLU NNs可以通过恢复主方向来学习单指标目标,其样本复杂度与d成线性关系。
🎯
关键要点
- 研究使用随机梯度下降(SGD)训练任意宽度的两层神经网络(NN)。
- 输入 x 是高斯分布,目标 y 遵循多指数模型。
- 证明了NN的第一层权重将收敛于真实模型的k维主子空间。
- 建立了一个独立于NN宽度的一般化误差边界。
- 使用SGD训练的ReLU NNs可以通过恢复主方向来学习单指标目标。
- 样本复杂度与d成线性关系,而不是通过已知的p次多项式样本要求。
- 表明初始化时使用SGD训练的NNs可以胜过神经切向核。
➡️