arXiv · 2501.06074
Geometry and Optimization of Shallow Polynomial Networks
Abstract
We study shallow neural networks with monomial activations and output dimension one. The function space for these models can be identified with a set of symmetric tensors with bounded rank. We describe general features of these networks, focusing on the relationship between width and optimization. We then consider teacher-student problems, which can be viewed as problems of low-rank tensor approximation with respect to non-standard inner products that are induced by the data distribution. In this setting, we introduce a teacher-metric data discriminant which encodes the qualitative behavior of the optimization as a function of the training data distribution. Finally, we focus on networks with quadratic activations, presenting an in-depth analysis of the optimization landscape. In particular, we present a variation of the Eckart-Young Theorem characterizing all critical points and their Hessian signatures for teacher-student problems with quadratic networks and Gaussian training data.
Explore related subjects
Keep this discovery
Yossi Arjevani, Joan Bruna, Joe Kileel, Elzbieta Polak, Matthew Trager. 2025-01-10. Geometry and Optimization of Shallow Polynomial Networks. https://arxiv.org/abs/2501.06074
Cite the original work for its findings. Save a collection to share your selection of sources.