amirrezarajabi / tensorflow-forward-forward

Implementation of Hinton's forward-forward (FF) algorithm in tensorflow - an alternative to back-propagation

Tensorflow-Forward-Forward

Hinton's Forward-Forward algorithm pdf implementation in Tensorflow

Datasets

• MNIST dataset

  • Notebook
  • Train Acc : 98.24%
  • Test acc : 96.83%

• Fashion-MNIST dataset

  • Notebook
  • Train Acc : 90.15%
  • Test acc : 87.23%

Acknowledgement

The code is mainly inspired by pytorch implementation in this repo

tensorflow-forward-forward's Issues

Can the labels be seperated from the images?

Thanks for this repo! I have a quick question - can the labels be seperated from the images, so I can train the network on a bunch of RGB images with classified labels?