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

被引:9
作者
Cakmak, Burak [1 ]
Opper, Manfred [1 ]
机构
[1] Tech Univ Berlin, Artificial Intelligence Grp, Berlin, Germany
关键词
cavity and replica method; machine learning; message-passing algorithms; spin glasses; STATISTICAL-MECHANICS;
D O I
10.1088/1742-5468/abb8c9
中图分类号
O3 [力学];
学科分类号
08 ; 0801 ;
摘要
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.
引用
收藏
页数:32
相关论文
共 46 条
[1]  
[Anonymous], 2008, P 25 INT C MACH LEAR
[2]  
[Anonymous], 1986, PARALLEL DISTRIBUTED
[3]  
[Anonymous], 2011, Random Matrix Methods for Wireless Communications
[4]  
[Anonymous], 2008, J PHYS C SERIES
[5]  
Baker A., 2020, ARXIV200401571
[6]   The Dynamics of Message Passing on Dense Graphs, with Applications to Compressed Sensing [J].
Bayati, Mohsen ;
Montanari, Andrea .
IEEE TRANSACTIONS ON INFORMATION THEORY, 2011, 57 (02) :764-785
[7]   Rectangular R-Transform as the Limit of Rectangular Spherical Integrals [J].
Benaych-Georges, Florent .
JOURNAL OF THEORETICAL PROBABILITY, 2011, 24 (04) :969-987
[8]   An Iterative Construction of Solutions of the TAP Equations for the Sherrington-Kirkpatrick Model [J].
Bolthausen, Erwin .
COMMUNICATIONS IN MATHEMATICAL PHYSICS, 2014, 325 (01) :333-366
[9]  
Cakmak B, 2016, THESIS
[10]   Analysis of Bayesian inference algorithms by the dynamical functional approach [J].
Cakmak, Burak ;
Opper, Manfred .
JOURNAL OF PHYSICS A-MATHEMATICAL AND THEORETICAL, 2020, 53 (27)