The Restricted Boltzmann Machine (RBM) is a type of neural network that was popularized in the early 2000s by Geoffrey Hinton and his colleagues.
RBM is a type of energy-based model that learns to represent the underlying probability distribution of input data by minimizing the energy function of the system. It is a two-layer neural network that consists of a visible layer and a hidden layer, where each neuron in the visible layer is connected to every neuron in the hidden layer but there are no connections between neurons within each layer.
The RBM is used in various applications, such as dimensionality reduction, feature learning, collaborative filtering, data augmentation and recommendation systems. It can also be used as a building block in more complex deep learning architectures, such as Deep Belief Networks (DBNs) and Convolutional Neural Networks (CNNs).
- Our work was done using PyTorch and is accessible in the
rbm.pyfile. - More details about RMBs are available in the notebook
mnist.ipynb.
Pretrained models are available in the /save/model folder and can be loaded as follows :
# --- RBM
rbm = RBM(k=1)
rbm.load_state_dict(torch.load("/save/model/rbm.pth"))
# --- Classifiers with RBM features
clf = pickle.load(open("save/model/model_name.pkl", "rb"))- Geoffrey Hinton (2010) A Practical Guide to Training Restricted Boltzmann Machines, UTML TR 2010–003
- https://github.com/odie2630463/Restricted-Boltzmann-Machines-in-pytorch
Copyright © 2023 Ahmed OSMAN & Data Saiyentist.
This project is MIT License licensed.