The Gaussian-Multinoulli Restricted Boltzmann Machine: A Potts Model Extension of the GRBM
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Many real-world tasks, from associative memory to symbolic reasoning, benefit from discrete, structured representations that standard continuous latent models can struggle to express. We introduce the Gaussian-Multinoulli Restricted Boltzmann Machine (GM-RBM), a generative energy-based model that extends the Gaussian-Bernoulli RBM (GB-RBM) by replacing binary hidden units with q-state categorical (Potts) units, yielding a richer latent state space for multivalued concepts. We provide a self-contained derivation of the energy, conditional distributions, and learning rules, and detail practical training choices (contrastive divergence with temperature annealing and intra-slot diversity constraints) that avoid state collapse. To separate architectural effects from sheer latent capacity, we evaluate under both capacity-matched and parameter-matched setups, comparing GM-RBM with GB-RBM configured to have the same number of possible latent assignments. On analogical recall and structured memory benchmarks, GM-RBM achieves competitive, and in several regimes improved, recall at equal capacity with comparable training cost, despite using only Gibbs updates. The discrete q-ary formulation is also amenable to efficient implementation. These results clarify when categorical hidden units provide a simple, scalable alternative to binary latents for discrete inference within tractable RBMs.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 11 of its 31 references that are in the map. First posted on arXiv 16 May 2025.
Builds on
11 of 31 references.
References · 11
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2013Efficient Estimation of Word Representations in Vector Spacecited by 4,375
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2016Conditional Image Generation with PixelCNN Decoderscited by 862
- 2016Pixel Recurrent Neural Networkscited by 816
- 2023Energy Transformercited by 71
- 2020Learning Discrete Energy-based Models via Auxiliary-variable Local Explorationcited by 30
- 2022CMOS-compatible Ising and Potts Annealing Using Single Photon Avalanche Diodescited by 5
- 2024Energy-Based Modelling for Discrete and Mixed Data via Heat Equations on Structured Spacescited by 5
- 2014Mixed-Variate Restricted Boltzmann Machinescited by 3
- 2022ReCAB-VAE: Gumbel-Softmax Variational Inference Based on Analytic Divergencecited by 1
Built on by
Citing papers per year: 2025: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024The effect of priors on Learning with Restricted Boltzmann Machines
- 2024Modeling Structured Data Learning with Restricted Boltzmann Machines in the Teacher-Student Setting
- 2026Spherical Boltzmann machines: a solvable theory of learning and generation in energy-based models
- 2023Hopfield model with planted patterns: a teacher-student self-supervised learning model
- 2025Memorization to Generalization: Emergence of Diffusion Models from Associative Memory
- 2025Maximum Likelihood Learning of Latent Dynamics Without Reconstruction
- 2026Stochastic Attention via Langevin Dynamics on the Modern Hopfield Energy
- 2026Hierarchical Prototype Emergence in Modern Hopfield Models