Leap+Verify: Regime-Adaptive Speculative Weight Prediction for Accelerating Neural Network Training
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
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 975
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 763
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2015Cyclical Learning Rates for Training Neural Networkscited by 627
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
10 more
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2019Lookahead Optimizer: k steps forward, 1 step backcited by 333
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
- 2018Blockwise Parallel Decoding for Deep Autoregressive Modelscited by 318
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
- 2024Accelerating Training with Neuron Interaction and Nowcasting Networkscited by 12
- 2023XGrad: Boosting Gradient-Based Optimizers With Weight Predictioncited by 5
- 2017Introspection: Accelerating Neural Network Training By Learning Weight Evolutioncited by 4
- 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
- 2026Verification-Aware Training for Speculative Decoding
- 2025Draft, Verify, and Improve: Toward Training-Aware Speculative Decoding
- 2026Learning to Draft: Adaptive Speculative Decoding with Reinforcement Learning
- 2026TAPS: Task Aware Proposal Distributions for Speculative Sampling
- 2026From Tokens to Steps: Verification-Aware Speculative Decoding for Efficient Multi-Step Reasoning
- 2026When Drafts Evolve: Speculative Decoding Meets Online Learning
- 2023Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding
- 2025Scaling Laws for Speculative Decoding