Abstract:Restricted Boltzmann Machines (RBMs) are effective tools for modeling complex systems and deriving insights from data. However, training these models with highly structured data presents significant challenges due to the slow mixing characteristics of Markov Chain Monte Carlo processes. In this study, we build upon recent theoretical advancements in RBM training, to significantly reduce the computational cost of training (in very clustered datasets), evaluating and sampling in RBMs in general. The learning process is analogous to thermodynamic continuous phase transitions observed in ferromagnetic models, where new modes in the probability measure emerge in a continuous manner. Such continuous transitions are associated with the critical slowdown effect, which adversely affects the accuracy of gradient estimates, particularly during the initial stages of training with clustered data. To mitigate this issue, we propose a pre-training phase that encodes the principal components into a low-rank RBM through a convex optimization process. This approach enables efficient static Monte Carlo sampling and accurate computation of the partition function. We exploit the continuous and smooth nature of the parameter annealing trajectory to achieve reliable and computationally efficient log-likelihood estimations, enabling online assessment during the training, and propose a novel sampling strategy named parallel trajectory tempering (PTT) which outperforms previously optimized MCMC methods. Our results show that this training strategy enables RBMs to effectively address highly structured datasets that conventional methods struggle with. We also provide evidence that our log-likelihood estimation is more accurate than traditional, more computationally intensive approaches in controlled scenarios. The PTT algorithm significantly accelerates MCMC processes compared to existing and conventional methods.

Parallel tempering with equi-energy moves for training of restricted boltzmann machines

Training Restricted Boltzmann Machines with Binary Synapses Using the Bayesian Learning Rule

Hyperparameters Adaptation for Restricted Boltzmann Machines Based on Free Energy

Fast training and sampling of Restricted Boltzmann Machines

Policy Gradients for Optimal Parallel Tempering MCMC

Non-Reversible Parallel Tempering: a Scalable Highly Parallel MCMC Scheme

Uniform Ergodicity of Parallel Tempering With Efficient Local Exploration

Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering

Bayesian Parameter Estimation for Biochemical Reaction Networks Using Region-Based Adaptive Parallel Tempering

Parallel Tempering for Dynamic Generalized Linear Models

Restricted Boltzmann machine based algorithm for multi-objective optimization

Barriers and dynamical paths in alternating Gibbs sampling of restricted Boltzmann machines

Adaptive hyperparameter updating for training restricted Boltzmann machines on quantum annealers

Mean-Field Assisted Deep Boltzmann Learning with Probabilistic Computers

Efficient Training of Energy-Based Models Using Jarzynski Equality

Generative and discriminative training of Boltzmann machine through Quantum annealing

Training Restricted Boltzmann Machines on Word Observations

"Parallel-Tempering"-Assisted Hybrid Monte Carlo Algorithm For Bayesian Inference In Dynamical Systems

Efficient training of energy-based models using Jarzynski equality *

Temperature based Restricted Boltzmann Machines

Online calibration scheme for training restricted Boltzmann machines with quantum annealing