中文

浅层多项式网络的几何与优化

机器学习 2025-12-18 v2 代数几何

摘要

我们研究具有单项式激活函数且输出维度为一的浅层神经网络。这些模型的函数空间可识别为具有有界秩的对称张量集合。我们描述了这些网络的一般特征,聚焦于宽度与优化之间的关系。随后我们考虑教师-学生问题,可视为针对由数据分布诱导的非标准内积的低秩张量逼近问题。在此setting下,我们引入教师-度量数据判别符,以编码优化在训练数据分布函数中的定性行为。最后,我们聚焦于具有二次激活函数的网络,深入分析了优化景观。特别是,我们提出了Eckart-Young定理的一种变体,刻画了具有高斯训练数据的二次网络教师-学生问题中所有临界点及其Hessian特征值符号。

关键词

引用

@article{arxiv.2501.06074,
  title  = {Geometry and Optimization of Shallow Polynomial Networks},
  author = {Yossi Arjevani and Joan Bruna and Joe Kileel and Elzbieta Polak and Matthew Trager},
  journal= {arXiv preprint arXiv:2501.06074},
  year   = {2025}
}

备注

39 pages, 3 figures; accepted to SIAM Journal on Applied Algebra and Geometry (SIAGA)