Train and sample Restricted Boltzmann Machines (RBMs) in Julia, with 1st and 2nd order adversarial constraints on the weights, that promote concentratation of information about labeled features into selected hidden units.

This package is registered. Install with:

using Pkg
Pkg.add("AdvRBMs")

This package does not export any symbols.

Implementation of standard Restricted Boltzmann Machines in Julia:

• https://github.com/cossio/RestrictedBoltzmannMachines.jl

Ising model simulations were carried out using IsingModels.jl.

If you use this package in a publication, please cite:

• Jorge Fernandez-de-Cossio-Diaz, Simona Cocco, and Remi Monasson. "Disentangling representations in Restricted Boltzmann Machines without adversaries." Physical Review X 13, 021003 (2023).

Or you can use the included CITATION.bib.