K. Balasubramanian

activity

20122025

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

14 citations · 59 across the 20 of their papers we have counts for

collaborators

Showing math.PRShow all

2 papers · 1 filter

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

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…