Update: Please find our latest python library for nonparametric score estimation (including the spectral estimator) at https://github.com/miskcoo/kscore
This repository contains the official code for
https://arxiv.org/abs/1806.02925
python -m toy.guassian
Train plain VAE on CelebA:
python -m vae.vae_celeba
Train implicit VAE on CelebA using entropy gradients estimated by SSGE:
python -m vae.vae_celeba_implicit
To cite this work, please use
@InProceedings{shi2018spectral,
title = {A Spectral Approach to Gradient Estimation for Implicit Distributions},
author = {Shi, Jiaxin and Sun, Shengyang and Zhu, Jun},
booktitle = {Proceedings of the 35th International Conference on Machine Learning},
pages = {4651--4660},
year = {2018},
}