mlsanityData to 14 Oct 2026

A Unified Noise-Curvature View of Loss of Trainability

arXiv 2509.19698 ↗First posted 24 Sep 2025cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

Loss of trainability refers to a phenomenon in continual learning where parameter updates no longer make progress on the optimization objective, so accuracy stalls or degrades as the learning problem changes over time. In this paper, we analyze loss of trainability through an optimization lens and find that the phenomenon is not reliably predicted by existing individual indicators such as Hessian rank, sharpness level, weight or gradient norms, gradient-to-parameter ratios, and unit-sign entropy. Motivated by our analysis, we introduce two complementary indicators: a batch-size-aware gradient-noise bound and a curvature volatility-controlled bound. We then combine these two indicators into a per-layer adaptive noise threshold on the effective step-size that anticipates trainability behavior. Using this insight, we propose a step-size scheduler that keeps each layer's effective parameter update below this bound, thereby avoiding loss of trainability. We demonstrate that our scheduler can improve the accuracy maintained by previously proposed approaches, such as concatenated ReLU (CReLU), Wasserstein regularizer, and L2 weight decay. Surprisingly, our scheduler produces adaptive step-size trajectories that, without tuning, mirror the manually engineered step-size decay schedules.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 28 of its 34 references that are in the map. First posted on arXiv 24 Sep 2025.

Builds on

28 of 34 references.

References · 28

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  4. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  5. 2017Rethinking Atrous Convolution for Semantic Image Segmentationcited by 2,161
  6. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  7. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
  8. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
  9. 2018How Does Batch Normalization Help Optimization?cited by 357
  10. 2017Snapshot Ensembles: Train 1, get M for freecited by 350
  11. 2017Don't Decay the Learning Rate, Increase the Batch Sizecited by 320
  12. 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 300
16 more
  1. 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 292
  2. 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 290
  3. 2018An Empirical Model of Large-Batch Trainingcited by 281
  4. 2017Spectral Norm Regularization for Improving the Generalizability of Deep Learningcited by 182
  5. 2023Understanding plasticity in neural networkscited by 174
  6. 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 128
  7. 2023Loss of Plasticity in Continual Deep Reinforcement Learningcited by 121
  8. 2016Understanding and Improving Convolutional Neural Networks via Concatenated Rectified Linear Unitscited by 86
  9. 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 86
  10. 2021A study on the plasticity of neural networkscited by 44
  11. 2024Does SGD really happen in tiny subspaces?cited by 36
  12. 2023Directions of Curvature as an Explanation for Loss of Plasticitycited by 26
  13. 2023Symmetry Induces Structure and Constraint of Learningcited by 25
  14. 2024Plastic Learning with Deep Fourier Featurescited by 14
  15. 2024Position: Lifetime tuning is incompatible with continual reinforcement learningcited by 11
  16. 2025Preserving Plasticity in Continual Learning with Adaptive Linearity Injectioncited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.