We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Statistics > Machine Learning

Title: Learning Curves for SGD on Structured Features

Abstract: The generalization performance of a machine learning algorithm such as a neural network depends in a non-trivial way on the structure of the data distribution. Models of generalization in machine learning theory often ignore the low-dimensional structure of natural signals, either by considering data-agnostic bounds or by studying the performance of the algorithm when trained on uncorrelated features. To analyze the influence of data structure on test loss dynamics, we study an exactly solveable model of stochastic gradient descent (SGD) which predicts test loss when training on features with arbitrary covariance structure. We solve the theory exactly for both Gaussian features and arbitrary features and we show that the simpler Gaussian model accurately predicts test loss of nonlinear random-feature models and deep neural networks trained with SGD on real datasets such as MNIST and CIFAR-10. We show that modeling the geometry of the data in the induced feature space is indeed crucial to accurately predict the test error throughout learning.
Comments: Fixed Typo in A.7
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:2106.02713 [stat.ML]
  (or arXiv:2106.02713v2 [stat.ML] for this version)

Submission history

From: Blake Bordelon [view email]
[v1] Fri, 4 Jun 2021 20:48:20 GMT (1827kb,D)
[v2] Wed, 9 Jun 2021 14:44:31 GMT (1827kb,D)
[v3] Fri, 15 Oct 2021 12:57:28 GMT (2039kb,D)
[v4] Fri, 4 Feb 2022 19:53:26 GMT (2465kb,D)
[v5] Tue, 15 Mar 2022 01:53:28 GMT (2494kb,D)

Link back to: arXiv, form interface, contact.