AdvRBMs Julia package
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.
Installation
This package is registered. Install with:
using Pkg
Pkg.add("AdvRBMs")This package does not export any symbols.
Related packages
Implementation of standard Restricted Boltzmann Machines in Julia: https://github.com/cossio/RestrictedBoltzmannMachines.jl.
Ising model simulations were carried out using IsingModels.jl.
Citation
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.