Fluctuation-dissipation relations for stochastic gradient descent
Not on the map. Why?
Abstract (arXiv)
The notion of the stationary equilibrium ensemble has played a central role in statistical mechanics. In machine learning as well, training serves as generalized equilibration that drives the probability distribution of model parameters toward stationarity. Here, we derive stationary fluctuation-dissipation relations that link measurable quantities and hyperparameters in the stochastic gradient descent algorithm. These relations hold exactly for any stationary state and can in particular be used to adaptively set training schedule. We can further use the relations to efficiently extract information pertaining to a loss-function landscape such as the magnitudes of its Hessian and anharmonicity. Our claims are empirically verified.
Key facts. Cited by 47 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 Sep 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 13; 2022: 6; 2023: 6; 2024: 6; 2025: 7; 2026: 9.
47 papers.
Sort:
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
- 2021Strength of Minibatch Noise in SGDcited by 39
- 2021SGD in the Large: Average-case Analysis, Asymptotics, and Stepsize Criticalitycited by 33
- 2022Implicit regularization of dropoutcited by 24
- 2021Fractal Structure and Generalization Properties of Stochastic Optimization Algorithmscited by 23
- 2022Implicit Regularization or Implicit Conditioning? Exact Risk Trajectories of SGD in High Dimensionscited by 23
- 2021The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusioncited by 19
- 2026Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioningcited by 15
- 2021SVRG Meets AdaGrad: Painless Variance Reductioncited by 13
- 2021How to decay your learning ratecited by 11
- 2023On a continuous time model of gradient descent dynamics and instability in deep learningcited by 11
- 2021SGD Implicitly Regularizes Generalization Errorcited by 10
35 more
- 2022Born-Infeld (BI) for AI: Energy-Conserving Descent (ECD) for Optimizationcited by 10
- 2022Taming Fat-Tailed ("Heavier-Tailed'' with Potentially Infinite Variance) Noise in Federated Learningcited by 10
- 2024Formation of Representations in Neural Networkscited by 10
- 2024How Neural Networks Learn the Support is an Implicit Regularization Effect of SGDcited by 8
- 2025Neural Thermodynamics: Entropic Forces in Deep and Universal Representation Learningcited by 8
- 2024Stochastic weight matrix dynamics during learning and Dyson Brownian motioncited by 7
- 2022Optimization Planning for 3D ConvNetscited by 5
- 2023Stochastic Gradient Descent-Induced Drift of Representation in a Two-Layer Neural Networkcited by 5
- 2023Machine learning in and out of equilibriumcited by 5
- 2025SGD as Free Energy Minimization: A Thermodynamic View on Neural Network Trainingcited by 4
- 2021Fluctuation-dissipation Type Theorem in Stochastic Linear Learningcited by 3
- 2021Revisiting the Effects of Stochasticity for Hamiltonian Samplerscited by 3
- 2023Stochastic Gradient Descent-like relaxation is equivalent to Metropolis dynamics in discrete optimization and inference problemscited by 3
- 2021Generative Adversarial Network: Some Analytical Perspectivescited by 2
- 2021Shift-Curvature, SGD, and Generalizationcited by 2
- 2021Multilayer Lookahead: a Nested Version of Lookaheadcited by 2
- 2023Anti-Correlated Noise in Epoch-Based Stochastic Gradient Descent: Implications for Weight Variances in Flat Directionscited by 2
- 2025AutoSGD: Automatic Learning Rate Selection for Stochastic Gradient Descentcited by 1
- 2025Phase diagram and eigenvalue dynamics of stochastic gradient descent in multilayer neural networkscited by 1
- 2025Closed-form ℓr norm scaling with data for overparameterized linear regression and diagonal linear networks under ℓp biascited by 1
- 2025Contribution of task-irrelevant stimuli to drift of neural representationscited by 1
- 2026Spectral phase transitions and trainability in neural network learning dynamicscited by 1
- 2022Studying Generalization Through Data Averagingcited by 0
- 2023Climate Intervention Analysis using AI Model Guided by Statistical Physics Principlescited by 0
- 2024Dynamics of Supervised and Reinforcement Learning in the Non-Linear Perceptroncited by 0
- 2024Coupling-based Convergence Diagnostic and Stepsize Scheme for Stochastic Gradient Descentcited by 0
- 2024Random Matrix Theory for Stochastic Gradient Descentcited by 0
- 2025The Universe Learning Itself: On the Evolution of Dynamics from the Big Bang to Machine Intelligencecited by 0
- 2026Thermodynamics of Reinforcement Learning Curriculacited by 0
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forcescited by 0
- 2026The Kinetics of Training: A Driven-Nucleation Rate Law for Emergence, Plasticity Loss, and Circuit Control in Language Modelscited by 0
- 2026Speed Limit for Information Acquisition in Stochastic Learning Dynamicscited by 0
- 2026An Analytical Theory of Auxiliary Learningcited by 0
- 2026Optimizer-dependent training dynamics converge to the same one-third optimal data scalingcited by 0
- 2026Learning the identity: a case study of how SGD selects among functional decompositionscited by 0
- 2026Learning the identity: a case study of how SGD selects among functional decompositions30 Sep 2026
- 2026Optimizer-dependent training dynamics converge to the same one-third optimal data scaling29 Sep 2026
- 2026An Analytical Theory of Auxiliary Learning24 Sep 2026
- 2026Speed Limit for Information Acquisition in Stochastic Learning Dynamics8 Sep 2026
- 2026The Kinetics of Training: A Driven-Nucleation Rate Law for Emergence, Plasticity Loss, and Circuit Control in Language Models29 Jul 2026
- 2026Spectral phase transitions and trainability in neural network learning dynamics26 Jun 2026
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forces4 Jun 2026
- 2026Thermodynamics of Reinforcement Learning Curricula12 Mar 2026
- 2026Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioning10 Mar 2026
- 2025The Universe Learning Itself: On the Evolution of Dynamics from the Big Bang to Machine Intelligence18 Dec 2025
- 2025Contribution of task-irrelevant stimuli to drift of neural representations24 Oct 2025
- 2025Closed-form ℓr norm scaling with data for overparameterized linear regression and diagonal linear networks under ℓp bias25 Sep 2025
35 more
- 2025Phase diagram and eigenvalue dynamics of stochastic gradient descent in multilayer neural networks1 Sep 2025
- 2025SGD as Free Energy Minimization: A Thermodynamic View on Neural Network Training29 May 2025
- 2025AutoSGD: Automatic Learning Rate Selection for Stochastic Gradient Descent27 May 2025
- 2025Neural Thermodynamics: Entropic Forces in Deep and Universal Representation Learning18 May 2025
- 2024Random Matrix Theory for Stochastic Gradient Descent29 Dec 2024
- 2024Coupling-based Convergence Diagnostic and Stepsize Scheme for Stochastic Gradient Descent15 Dec 2024
- 2024Formation of Representations in Neural Networks3 Oct 2024
- 2024Dynamics of Supervised and Reinforcement Learning in the Non-Linear Perceptron5 Sep 2024
- 2024Stochastic weight matrix dynamics during learning and Dyson Brownian motion23 Jul 2024
- 2024How Neural Networks Learn the Support is an Implicit Regularization Effect of SGD17 Jun 2024
- 2023Stochastic Gradient Descent-like relaxation is equivalent to Metropolis dynamics in discrete optimization and inference problems11 Sep 2023
- 2023Anti-Correlated Noise in Epoch-Based Stochastic Gradient Descent: Implications for Weight Variances in Flat Directions8 Jun 2023
- 2023Machine learning in and out of equilibrium6 Jun 2023
- 2023Climate Intervention Analysis using AI Model Guided by Statistical Physics Principles7 Feb 2023
- 2023Stochastic Gradient Descent-Induced Drift of Representation in a Two-Layer Neural Network6 Feb 2023
- 2023On a continuous time model of gradient descent dynamics and instability in deep learning3 Feb 2023
- 2022Taming Fat-Tailed ("Heavier-Tailed'' with Potentially Infinite Variance) Noise in Federated Learning3 Oct 2022
- 2022Implicit regularization of dropout13 Jul 2022
- 2022Studying Generalization Through Data Averaging28 Jun 2022
- 2022Implicit Regularization or Implicit Conditioning? Exact Risk Trajectories of SGD in High Dimensions15 Jun 2022
- 2022Born-Infeld (BI) for AI: Energy-Conserving Descent (ECD) for Optimization26 Jan 2022
- 2022Optimization Planning for 3D ConvNets11 Jan 2022
- 2021Multilayer Lookahead: a Nested Version of Lookahead27 Oct 2021
- 2021Shift-Curvature, SGD, and Generalization21 Aug 2021
- 2021The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusion19 Jul 2021
- 2021Revisiting the Effects of Stochasticity for Hamiltonian Samplers30 Jun 2021
- 2021Fractal Structure and Generalization Properties of Stochastic Optimization Algorithms9 Jun 2021
- 2021Fluctuation-dissipation Type Theorem in Stochastic Linear Learning4 Jun 2021
- 2021Generative Adversarial Network: Some Analytical Perspectives25 Apr 2021
- 2021SGD Implicitly Regularizes Generalization Error10 Apr 2021
- 2021How to decay your learning rate23 Mar 2021
- 2021SVRG Meets AdaGrad: Painless Variance Reduction18 Feb 2021
- 2021Strength of Minibatch Noise in SGD10 Feb 2021
- 2021SGD in the Large: Average-case Analysis, Asymptotics, and Stepsize Criticality8 Feb 2021
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descent28 Jan 2021
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
- 2021Strength of Minibatch Noise in SGDcited by 39
- 2021SGD in the Large: Average-case Analysis, Asymptotics, and Stepsize Criticalitycited by 33
- 2022Implicit regularization of dropoutcited by 24
- 2021Fractal Structure and Generalization Properties of Stochastic Optimization Algorithmscited by 23
- 2022Implicit Regularization or Implicit Conditioning? Exact Risk Trajectories of SGD in High Dimensionscited by 23
- 2021The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusioncited by 19
- 2026Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioningcited by 15
- 2021SVRG Meets AdaGrad: Painless Variance Reductioncited by 13
- 2021How to decay your learning ratecited by 11
- 2023On a continuous time model of gradient descent dynamics and instability in deep learningcited by 11
- 2021SGD Implicitly Regularizes Generalization Errorcited by 10
12 most cited of 47 All 47 (JSON)
- 2026Learning the identity: a case study of how SGD selects among functional decompositions30 Sep 2026
- 2026Optimizer-dependent training dynamics converge to the same one-third optimal data scaling29 Sep 2026
- 2026An Analytical Theory of Auxiliary Learning24 Sep 2026
- 2026Speed Limit for Information Acquisition in Stochastic Learning Dynamics8 Sep 2026
- 2026The Kinetics of Training: A Driven-Nucleation Rate Law for Emergence, Plasticity Loss, and Circuit Control in Language Models29 Jul 2026
- 2026Spectral phase transitions and trainability in neural network learning dynamics26 Jun 2026
- 2026Nonreversible Gauge Fields in Fokker--Planck Dynamics: Supersymmetric Hamiltonians and Learned Finite Forces4 Jun 2026
- 2026Thermodynamics of Reinforcement Learning Curricula12 Mar 2026
- 2026Mousse: Rectifying the Geometry of Muon with Curvature-Aware Preconditioning10 Mar 2026
- 2025The Universe Learning Itself: On the Evolution of Dynamics from the Big Bang to Machine Intelligence18 Dec 2025
- 2025Contribution of task-irrelevant stimuli to drift of neural representations24 Oct 2025
- 2025Closed-form ℓr norm scaling with data for overparameterized linear regression and diagonal linear networks under ℓp bias25 Sep 2025
12 newest of 47 All 47 (JSON)