Limit Theorems for Stochastic Gradient Descent in High-Dimensional Single-Layer Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Rangriz, Parsa
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914519032266752
author Rangriz, Parsa
author_facet Rangriz, Parsa
contents This paper studies the high-dimensional scaling limits of online stochastic gradient descent (SGD). Building on the recent work of Ben Arous, Gheissari, and Jagannath on the effective dynamics of SGD, we study the critical scaling regime of the step size for single-layer networks. Below this critical regime, the effective dynamics are governed by deterministic (ballistic) limits, whereas at the critical scale, a new correction term emerges that changes the phase diagram. In this regime, near fixed points, the corresponding diffusive (SDE) limits of the effective dynamics reduce to an Ornstein-Uhlenbeck process under certain conditions. These results highlight how the information exponent controls sample complexity and illustrate the limitations of deterministic scaling limits in capturing stochastic fluctuations in high-dimensional learning dynamics.
format Preprint
id arxiv_https___arxiv_org_abs_2511_02258
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Limit Theorems for Stochastic Gradient Descent in High-Dimensional Single-Layer Networks
Rangriz, Parsa
Machine Learning
Probability
Statistics Theory
This paper studies the high-dimensional scaling limits of online stochastic gradient descent (SGD). Building on the recent work of Ben Arous, Gheissari, and Jagannath on the effective dynamics of SGD, we study the critical scaling regime of the step size for single-layer networks. Below this critical regime, the effective dynamics are governed by deterministic (ballistic) limits, whereas at the critical scale, a new correction term emerges that changes the phase diagram. In this regime, near fixed points, the corresponding diffusive (SDE) limits of the effective dynamics reduce to an Ornstein-Uhlenbeck process under certain conditions. These results highlight how the information exponent controls sample complexity and illustrate the limitations of deterministic scaling limits in capturing stochastic fluctuations in high-dimensional learning dynamics.
title Limit Theorems for Stochastic Gradient Descent in High-Dimensional Single-Layer Networks
topic Machine Learning
Probability
Statistics Theory
url https://arxiv.org/abs/2511.02258