K. Balasubramanian

activity

20122025

most citedAn Analysis of Constant Step Size SGD in the Non-convex Regime: Asymptotic Normality and Bias

14 citations · 79 across the 38 of their papers we have counts for

collaborators

Showing 2023 · math.PRShow all

2 papers · 2 filters

math.PR2023

Gaussian random field approximation via Stein's method with applications to wide random neural networks

Krishnakumar Balasubramanian, Larry Goldstein, Nathan Ross +1

We derive upper bounds on the Wasserstein distance ( $W_{1}$ ), with respect to $sup$ -norm, between any continuous $R^{d}$ valued random field indexed by the $n$ -sphere and the…

math.PR2023★ 2 cited

High-dimensional scaling limits and fluctuations of online least-squares SGD with smooth covariance

Krishnakumar Balasubramanian, Promit Ghosal, Ye He

We derive high-dimensional scaling limits and fluctuations for the online least-squares Stochastic Gradient Descent (SGD) algorithm by taking the properties of the data generating…