mlsanityData to 7 Oct 2026

Leap+Verify: Regime-Adaptive Speculative Weight Prediction for Accelerating Neural Network Training

arXiv 2602.19580 ↗First posted 23 Feb 2026cs.LG econ.GN q-fin.ECcited by 0

Not on the map. Why?

Abstract (arXiv)

We introduce Leap+Verify, a framework that applies speculative execution -- predicting future model weights and validating predictions before acceptance -- to accelerate neural network training. Inspired by speculative decoding in language model inference and by the Automatically Scalable Computation (ASC) architecture for program execution, Leap+Verify decomposes training into three dynamically detected regimes (chaotic, transition, stable) using activation-space cosine similarity as a real-time Lyapunov proxy signal. Within each regime, analytic weight predictors (momentum, linear, quadratic extrapolation) attempt to forecast model parameters K training steps ahead; predictions are accepted only when validated against a held-out loss criterion. We evaluate Leap+Verify on GPT-2 124M and Qwen 2.5-1.5B trained on WikiText-103 across five random seeds, sweeping prediction depth K in {5, 10, 25, 50, 75, 100}. Momentum-based prediction (Adam moment extrapolation) fails catastrophically at both scales, with predicted losses exceeding actuals by 100-10,000x -- a universal norm explosion in optimizer-state extrapolation. Finite-difference predictors (linear, quadratic) succeed where momentum fails: at 124M, they achieve 24% strict acceptance at K=5 in stable regimes; at 1.5B, they achieve 37% strict acceptance in transition regimes. The scale-dependent finding is in regime distribution: GPT-2 124M spends 34% of training in stable regime, while Qwen 1.5B spends 64% in chaotic regime and reaches stable in only 0-2 of 40 checkpoints. Larger models are more predictable when predictable, but less often predictable -- the practical bottleneck shifts from predictor accuracy to regime availability. Cross-seed results are highly consistent (less than 1% validation loss variance), and the three-regime framework produces identical phase boundaries (plus or minus 50 steps) across seeds.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 32 references that are in the map. First posted on arXiv 23 Feb 2026.

Builds on

22 of 32 references.

References · 22

  1. 2020Scaling Laws for Neural Language Modelscited by 6,411
  2. 2016Pointer Sentinel Mixture Modelscited by 2,992
  3. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
  4. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
  5. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
  6. 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
  7. 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 975
  8. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  9. 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 763
  10. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
  11. 2015Cyclical Learning Rates for Training Neural Networkscited by 627
  12. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
10 more
  1. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
  2. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
  3. 2019Lookahead Optimizer: k steps forward, 1 step backcited by 333
  4. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
  5. 2018Blockwise Parallel Decoding for Deep Autoregressive Modelscited by 318
  6. 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
  7. 2024Accelerating Training with Neuron Interaction and Nowcasting Networkscited by 12
  8. 2023XGrad: Boosting Gradient-Based Optimizers With Weight Predictioncited by 5
  9. 2017Introspection: Accelerating Neural Network Training By Learning Weight Evolutioncited by 4
  10. 2023Enhancing Deep Neural Network Training Efficiency and Performance through Linear Predictioncited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Verification-Aware Training for Speculative Decoding
  2. 2025Draft, Verify, and Improve: Toward Training-Aware Speculative Decoding
  3. 2026Learning to Draft: Adaptive Speculative Decoding with Reinforcement Learning
  4. 2026TAPS: Task Aware Proposal Distributions for Speculative Sampling
  5. 2026From Tokens to Steps: Verification-Aware Speculative Decoding for Efficient Multi-Step Reasoning
  6. 2026When Drafts Evolve: Speculative Decoding Meets Online Learning
  7. 2023Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding
  8. 2025Scaling Laws for Speculative Decoding