/BSautoconvnet

Primary LanguageJupyter Notebook

BSautoconvnet

Boolean Structured Autoencoder Convolutional Deep Learning Network (BSautoconvnet)

Main Takeaways

  • Our model has only 44000+ parameters instead of millions parameters in U-net
  • Use monotone Boolean algebra convolutional layers, so it does not overfit
  • Achieved near perfect result of masked image reconstruction on CIFAR10 dataset
  • Able to be trained on online using Google Colab with GPU
  • No data augmentations, no regularization such as weight decay and dropout

How to Run

Download the jupyter notebook

Open using Google Colab Colab It can also be run using a jupyter notebook

Follow the steps in the notebook, run each block of codes starting from the top to the bottom Model can be trained in 50 epoches.

If you do not want to train the model, you can load the pretrained model (1 megabytes only).

Model

Network design

Experiment Results

Input Input

Ground Truth Groundtruth

Reconstructured Output Reconstructured Output

Links

BSnet paper link

BSautonet paper link

BSconvnet paper link

BSautoconvnet paper link

BSnet GitHub

Discrete Markov Random Field Relaxation

Slideshare

That's it. Have a Nice Day!!!