Skip to content

Code for "A Spectral Approach to Gradient Estimation for Implicit Distributions" (ICML'18)

License

Notifications You must be signed in to change notification settings

thjashin/spectral-stein-grad

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

spectral-stein-grad

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

A Spectral Approach to Gradient Estimation for Implicit Distributions (ICML 18)

https://arxiv.org/abs/1806.02925

Dependencies

Get Started

python -m toy.guassian

Implicit VAEs

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

Citation

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},
}

About

Code for "A Spectral Approach to Gradient Estimation for Implicit Distributions" (ICML'18)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published