bohu615 / structured-generative-modeling

This repository contains the implementation for the paper Information Theoretic StructuredGenerative Modeling,

Specially thanks for the open-source codes shared by sagelywizard/pytorch-mdn and PyTorch-GAN

Main Requirements

• Pytorch
• A GPU Machine

Usage

The main experiments in the paper are put in the notebook format.

Each file can be run independently

• GAN Example
• MINE Example
• Density Estimation
• Conditonal Estimation

To run ood_visualization.ipynb, please download the pretrained model in the ./model/ folder. (google drive link later)

The other baselines can be run by calling

python MDN.py
python CGAN.py

GMM_VBGMM_CE.py provides codes for producing conditional CE for any mixture models obtained from scipy.

Simply calling

compute_conditionalCE(joint, gm_joint)

in python to obtain the value, where joint should be bs*K and gm_joint is the class obtained from scipy.