English

Interactive Learning of Single-Index Models via Stochastic Gradient Descent

Machine Learning 2026-02-23 v1 Machine Learning Statistics Theory Statistics Theory

Abstract

Stochastic gradient descent (SGD) is a cornerstone algorithm for high-dimensional optimization, renowned for its empirical successes. Recent theoretical advances have provided a deep understanding of how SGD enables feature learning in high-dimensional nonlinear models, most notably the \textit{single-index model} with i.i.d. data. In this work, we study the sequential learning problem for single-index models, also known as generalized linear bandits or ridge bandits, where SGD is a simple and natural solution, yet its learning dynamics remain largely unexplored. We show that, similar to the optimal interactive learner, SGD undergoes a distinct ``burn-in'' phase before entering the ``learning'' phase in this setting. Moreover, with an appropriately chosen learning rate schedule, a single SGD procedure simultaneously achieves near-optimal (or best-known) sample complexity and regret guarantees across both phases, for a broad class of link functions. Our results demonstrate that SGD remains highly competitive for learning single-index models under adaptive data.

Keywords

Cite

@article{arxiv.2602.17876,
  title  = {Interactive Learning of Single-Index Models via Stochastic Gradient Descent},
  author = {Nived Rajaraman and Yanjun Han},
  journal= {arXiv preprint arXiv:2602.17876},
  year   = {2026}
}

Comments

26 pages, 2 figures

R2 v1 2026-07-01T10:43:41.179Z