mlsanityData to 14 Oct 2026

Directional Curvature from Armijo Backtracking: A Low-Cost Sharpness Probe and a Calibration-Free Learning-Rate Safeguard for Adam

arXiv 2607.03998 ↗First posted 4 Jul 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Local sharpness, defined by the largest Hessian eigenvalue λ1, sets the maximum stable gradient update size, but its computation would usually require running Lanczos or Hessian-vector products. However, we notice that even a single Armijo backtracking line search already contains this information with just a few forward passes, as the accepted step α determines the directional curvature along the search direction up to the multiplicative band set by the backtracking factor. The correlation between logα and logλ1 on CIFAR-10, Fashion-MNIST and Imagenette reaches -0.91 to -0.95 in Pearson correlation, and even after removing the trend per run the correlation remains at -0.60 to -0.70. This allows for a cheap online Edge-of-Stability estimate of the slow sharpness component. The employed probing mechanism searches along Adam's first-step update direction, at initialisation and nine times over the course of the first 50 optimiser steps. The learning-rate cap is set as twice the smallest observed step size, and in the studied learning-rate ranges (10-3 to 3.0) and GPT-2 pretraining experiments all capped runs avoid divergence; in the general architecture analysis, one MLP architecture is still sensitive to the initial batch order. This probing protocol incurs an approximately one percent overhead, and using a non-binding cap means that the optimiser's state and first update are bit-identical. There is no fine-tuning of any of the protocol parameters to any specific architecture; this is the sense in which this is a calibration-free safeguard. It is meant as a way of avoiding divergence, not achieving accuracy. At GPT-2 scale, multiple measurements also illustrate why an initialisation-only cap is insufficient: directional curvature rises substantially in the first five optimiser steps, which motivates the short probationary window.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 36 references that are in the map. First posted on arXiv 4 Jul 2026.

Builds on

25 of 36 references.

References · 25

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  4. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
  5. 2016Pointer Sentinel Mixture Modelscited by 3,096
  6. 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
  7. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  8. 2019On the Convergence of Adam and Beyondcited by 841
  9. 2015Cyclical Learning Rates for Training Neural Networkscited by 629
  10. 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
  11. 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 292
  12. 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
13 more
  1. 2024The Road Less Scheduledcited by 163
  2. 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 155
  3. 2020Stochastic Polyak Step-size for SGD: An Adaptive Learning Rate for Fast Convergencecited by 137
  4. 2023Prodigy: An Expeditiously Adaptive Parameter-Free Learnercited by 133
  5. 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 128
  6. 2019Painless Stochastic Gradient: Interpolation, Line-Search, and Convergence Ratescited by 119
  7. 2023Learning-Rate-Free Learning by D-Adaptationcited by 113
  8. 2024Why Warmup the Learning Rate? Underlying Mechanisms and Improvementscited by 46
  9. 2024Edge of Stochastic Stability: Revisiting the Edge of Stability for SGDcited by 30
  10. 2023Mechanic: A Learning Rate Tunercited by 20
  11. 2019Parabolic Approximation Line Search for DNNscited by 14
  12. 2024Stepping on the Edge: Curvature Aware Learning Rate Tunerscited by 8
  13. 2026A Scalable Measure of Loss Landscape Curvature for Analyzing the Training Dynamics of LLMscited by 8

Built on by

Not cited yet.