Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 

README.md

Wasserstein Introspective Neural Networks

This repository contains the code for the paper:
Wasserstein Introspective Neural Networks
Kwonjoon Lee, Weijian Xu, Fan Fan, Zhuowen Tu
CVPR 2018 (Oral)

Introduction

We present Wasserstein introspective neural networks (WINN) that are both a generator and a discriminator within a single model. WINN provides a significant improvement over the recent introspective neural networks (INN) method by enhancing INN's generative modeling capability. WINN has three interesting properties: (1) A mathematical connection between the formulation of the INN algorithm and that of Wasserstein generative adversarial networks (WGAN) is made. (2) The explicit adoption of the Wasserstein distance into INN results in a large enhancement to INN, achieving compelling results even with a single classifier -- e.g., providing nearly a 20 times reduction in model size over INN for unsupervised generative modeling. (3) When applied to supervised classification, WINN also gives rise to improved robustness against adversarial examples in terms of the error reduction. In the experiments, we report encouraging results on unsupervised learning problems including texture, face, and object modeling, as well as a supervised classification task against adversarial attacks.

Citation

If you use this code for your research, please cite our paper:

@inproceedings{lee2018winn,
  title={Wasserstein Introspective Neural Networks},
  author={Lee, Kwonjoon and Xu, Weijian and Fan, Fan and Tu, Zhouwen},
  booktitle={CVPR},
  year={2018}
}

Acknowledgments

This code is based on the implementations of INNg and WGAN-GP.

About

Wasserstein Introspective Neural Networks (CVPR 2018 Oral)

Topics

Resources

Releases

No releases published

Packages

No packages published
You can’t perform that action at this time.