Published October 1, 2020 | Version v1
Journal article

A dynamical mean-field theory for learning in restricted Boltzmann machines

  • 1. Artificial Intelligence Group, Technische Universität Berlin (Germany)

Description

We define a message-passing algorithm for computing magnetizations in restricted Boltzmann machines, which are Ising models on bipartite graphs introduced as neural network models for probability distributions over spin configurations. To model nontrivial statistical dependencies between the spins' couplings, we assume that the rectangular coupling matrix is drawn from an arbitrary bi-rotation invariant random matrix ensemble. Using the dynamical functional method of statistical mechanics we exactly analyze the dynamics of the algorithm in the large system limit. We prove the global convergence of the algorithm under a stability criterion and compute asymptotic convergence rates showing excellent agreement with numerical simulations. (paper)

Availability note (English)

Available from http://dx.doi.org/10.1088/1742-5468/abb8c9

Additional details

Identifiers

Publishing Information

Journal Title
Journal of Statistical Mechanics
Journal Volume
2020
Journal Issue
10
Journal Page Range
[32 p.]
ISSN
1742-5468