mlsanityData to 14 Oct 2026

Scaling Laws for Dynamic Mini-Batch SGD in Sketched Linear Regression

arXiv 2605.24316 ↗First posted 23 May 2026cs.LGcited by 2

Not on the map. Why?

Abstract (arXiv)

Mini-batching is central to large-scale optimization, yet its role in statistical scaling laws remains limited. We study one-pass and multi-pass batch SGD for sketched linear regression under power-law spectral and source conditions. Our analysis reveals a two-horizon phenomenon induced by warmup--stable--decay schedules: deterministic learning is governed by the full optimization trajectory, while stochastic error retains only a shorter terminal memory. For dynamic batch schedules, the individual batch sizes enter through influence-weighted summaries that measure how strongly each update affects the final risk. Consequently, batching leaves the approximation and optimization-bias laws unchanged at a fixed update horizon, but controls the one-pass variance and the multi-pass fluctuation around full-batch gradient descent. We obtain matching one-pass variance bounds and nearly matching multi-pass fluctuation bounds, recover static-batch and full-batch behavior as special cases, and derive an oracle square-root rule for allocating a fixed iteration budget. These results identify WSD horizon separation and final-risk influence as the mechanisms governing dynamic mini-batch scaling.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 12 of its 42 references that are in the map. First posted on arXiv 23 May 2026.

Builds on

12 of 42 references.

References · 12

  1. 2020Scaling Laws for Neural Language Modelscited by 6,586
  2. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  3. 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
  4. 2017Large Batch Training of Convolutional Networkscited by 535
  5. 2020Scaling Laws for Autoregressive Generative Modelingcited by 522
  6. 2018An Empirical Model of Large-Batch Trainingcited by 281
  7. 2022A Solvable Model of Neural Scaling Lawscited by 116
  8. 2021Learning Curve Theorycited by 85
  9. 2024Chinchilla Scaling: A replication attemptcited by 72
  10. 2020A Neural Scaling Law from the Dimension of the Data Manifoldcited by 52
  11. 2024Scaling and renormalization in high-dimensional regressioncited by 50
  12. 2024Scaling Law for Language Models Training Considering Batch Sizecited by 22

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.