Generalization for Least Squares Regression With Simple Spiked Covariances

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Jiping, Sonthalia, Rishi
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912076061999104
author Li, Jiping
Sonthalia, Rishi
author_facet Li, Jiping
Sonthalia, Rishi
contents Random matrix theory has proven to be a valuable tool in analyzing the generalization of linear models. However, the generalization properties of even two-layer neural networks trained by gradient descent remain poorly understood. To understand the generalization performance of such networks, it is crucial to characterize the spectrum of the feature matrix at the hidden layer. Recent work has made progress in this direction by describing the spectrum after a single gradient step, revealing a spiked covariance structure. Yet, the generalization error for linear models with spiked covariances has not been previously determined. This paper addresses this gap by examining two simple models exhibiting spiked covariances. We derive their generalization error in the asymptotic proportional regime. Our analysis demonstrates that the eigenvector and eigenvalue corresponding to the spike significantly influence the generalization error.
format Preprint
id arxiv_https___arxiv_org_abs_2410_13991
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generalization for Least Squares Regression With Simple Spiked Covariances
Li, Jiping
Sonthalia, Rishi
Statistics Theory
Machine Learning
Random matrix theory has proven to be a valuable tool in analyzing the generalization of linear models. However, the generalization properties of even two-layer neural networks trained by gradient descent remain poorly understood. To understand the generalization performance of such networks, it is crucial to characterize the spectrum of the feature matrix at the hidden layer. Recent work has made progress in this direction by describing the spectrum after a single gradient step, revealing a spiked covariance structure. Yet, the generalization error for linear models with spiked covariances has not been previously determined. This paper addresses this gap by examining two simple models exhibiting spiked covariances. We derive their generalization error in the asymptotic proportional regime. Our analysis demonstrates that the eigenvector and eigenvalue corresponding to the spike significantly influence the generalization error.
title Generalization for Least Squares Regression With Simple Spiked Covariances
topic Statistics Theory
Machine Learning
url https://arxiv.org/abs/2410.13991