dwavesystems / dwavesystems/dwave-pytorch-plugin

Use free energy in the loss function of RBM

Open
#20 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
39
Forks
25
Avg merge
28m
Merged PRs (30d)
1

Description

Allow users to use free energy in the objective of the RBM i.e., in
https://github.com/dwavesystems/dwave-pytorch-plugin/blob/main/dwave/plugins/torch/boltzmann_machine.py#L323
instead of using energy in the objective of RBM allow user to alternatively use free energy. Free energy is obtained after marginalizing out the hidden variables. This will eliminate the need for sampling hidden variables, will reduce noise or variance that is introduced by sampling hidden units.
The current loss function is ` - ` the new loss function using free energy will remove `h` from this equation.
Please see https://proceedings.neurips.cc/paper_files/paper/2013/file/7bb060764a818184ebb1cc0d43d382aa-Paper.pdf for free energy calculation.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.