https://github.com/pumpikano/tf-dann
Domain-Adversarial Neural Network in Tensorflow
https://github.com/pumpikano/tf-dann
adversarial-learning domain-adaptation tensorflow-models
Last synced: 3 months ago
JSON representation
Domain-Adversarial Neural Network in Tensorflow
- Host: GitHub
- URL: https://github.com/pumpikano/tf-dann
- Owner: pumpikano
- License: mit
- Created: 2016-06-14T06:19:44.000Z (about 9 years ago)
- Default Branch: master
- Last Pushed: 2021-12-05T21:27:30.000Z (over 3 years ago)
- Last Synced: 2025-03-28T13:08:00.623Z (3 months ago)
- Topics: adversarial-learning, domain-adaptation, tensorflow-models
- Language: Jupyter Notebook
- Homepage:
- Size: 1.25 MB
- Stars: 629
- Watchers: 15
- Forks: 222
- Open Issues: 10
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Domain-Adversarial Training of Neural Networks in Tensorflow
"[Unsupervised Domain Adaptation by Backpropagation](http://sites.skoltech.ru/compvision/projects/grl/files/paper.pdf)" introduced a simple and effective method for accomplishing domain adaptation with SGD with a gradient reversal layer. This work was elaborated and extended in "[Domain-Adversarial Training of Neural Networks](http://jmlr.org/papers/volume17/15-239/15-239.pdf)". For more information as well as a link to an equivalent implementation in Caffe, see http://sites.skoltech.ru/compvision/projects/grl/.
The `Blobs-DANN.ipynb` shows some basic experiments on a very simple dataset. The `MNIST-DANN.ipynb` recreates the MNIST experiment from the papers on a synthetic dataset. Instructions to generate the synthetic dataset are below.
Requires TensorFlow>=1.0 and tested with Python 2.7 and Python 3.4.
## Gradient Reversal Layer
The `flip_gradient` operation is implemented in Python by using `tf.gradient_override_map` to override the gradient of `tf.identity`. Refer to `flip_gradient.py` to see how this is implemented.
```python
from flip_gradient import flip_gradient
# Flip the gradient of y w.r.t. x and scale by l (defaults to 1.0)
y = flip_gradient(x, l)
```
## MNIST Experiments
The `MNIST-DANN.ipynb` notebook implements the MNIST experiments for the paper with the same model and optimization parameters, including the learning rate and adaptation parameter schedules. Rough results are below (more training would likely improve results - # epochs are not reported in the paper).
| Method | Target acc (paper) | Target acc (this repo w/ 10 epochs) |
| ------ | ------------------ | ----------------------------------- |
| Source Only | 0.5225 | 0.4801 |
| DANN | 0.7666 | 0.7189 |
### Build MNIST-M dataset
The MNIST-M dataset consists of MNIST digits blended with random color patches from the [BSDS500](http://www.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/resources.html#bsds500) dataset. To generate a MNIST-M dataset, first download the BSDS500 dataset and run the `create_mnistm.py` script:
```bash
curl -L -O http://www.eecs.berkeley.edu/Research/Projects/CS/vision/grouping/BSR/BSR_bsds500.tgz
python create_mnistm.py
```
This may take a couple minutes and should result in a `mnistm_data.pkl` file containing generated images.
## Contribution
It would be great to add other experiments to this repository. Feel free to make a PR if you decide to recreate other results from the papers or new experiments entirely.