InfoMax Variational Autoencoder

Learning representaion by maximizaing mutual information in variational autoencoders in PyTorch

This is a PyTorch implementation of Learning Representations by Maximizing Mutual Information in Variational Autoencoder.

Requirements

• Python 3
• PyTorch 0.4
• matplotlib
• scikit-learn
• install requirements via pip install -r requirements.txt

How to run

python MNIST/Save_InfoMaxVAE_CNN.py

Learning Representations by Maximizing Mutual Information in Variational Autoencoders https://arxiv.org/abs/1912.13361