Abstract:The algorithms used to train neural networks, like stochastic gradient descent (SGD), have close parallels to natural processes that navigate a high-dimensional parameter space -- for example protein folding or evolution. Our study uses a Fokker-Planck approach, adapted from statistical physics, to explore these parallels in a single, unified framework. We focus in particular on the stationary state of the system in the long-time limit, which in conventional SGD is out of equilibrium, exhibiting persistent currents in the space of network parameters. As in its physical analogues, the current is associated with an entropy production rate for any given training trajectory. The stationary distribution of these rates obeys the integral and detailed fluctuation theorems -- nonequilibrium generalizations of the second law of thermodynamics. We validate these relations in two numerical examples, a nonlinear regression network and MNIST digit classification. While the fluctuation theorems are universal, there are other aspects of the stationary state that are highly sensitive to the training details. Surprisingly, the effective loss landscape and diffusion matrix that determine the shape of the stationary distribution vary depending on the simple choice of minibatching done with or without replacement. We can take advantage of this nonequilibrium sensitivity to engineer an equilibrium stationary state for a particular application: sampling from a posterior distribution of network weights in Bayesian machine learning. We propose a new variation of stochastic gradient Langevin dynamics (SGLD) that harnesses without replacement minibatching. In an example system where the posterior is exactly known, this SGWORLD algorithm outperforms SGLD, converging to the posterior orders of magnitude faster as a function of the learning rate.

A Gibbs distribution that learns from GA dynamics

A Hybrid Dna Based Genetic Algorithm for Parameter Estimation of Dynamic Systems

Generalized Glauber dynamics for inference in biology

Machine learning algorithms based on generalized Gibbs ensembles

Information-theoretic Analysis of the Gibbs Algorithm: An Individual Sample Approach

Learning Graphical Models From the Glauber Dynamics

Machine learning in and out of equilibrium

Kinetic description and convergence analysis of genetic algorithms for global optimization

Structure Learning in Gaussian Graphical Models from Glauber Dynamics

On the Validation of Gibbs Algorithms: Training Datasets, Test Datasets and their Aggregation

Gibbs state sampling via cluster expansions

The Novel Approach of Adaptive Twin Probability for Genetic Algorithm

Fitness Approximation through Machine Learning

Modeling And Dynamical Behavior Of Genetic Algorithms

Optimality of Glauber dynamics for general-purpose Ising model sampling and free energy approximation

A New Approach to Dynamics Analysis of Genetic Algorithms Without Selection

An efficient approach to learning inhomogeneous gibbs model

A generalization of regularized dual averaging and its dynamics

DynGMA: a robust approach for learning stochastic differential equations from data