Code Monkey home page Code Monkey logo

m-adda's Introduction

M-ADDA: Metric-based Adversarial Discriminative Domain Adaptation [Paper]

Accepted in the ICML 2018 Workshop of Domain Adaptation for Visual Understanding (DAVU).

Description

The idea is to cluster the source dataset using the triplet loss and then cluster the target dataset using adversarial learning and the center-magnet loss. The Figure below shows the resultant clusters after using this method. The different colored stars represent the cluster centers, each corresponding to a different digit category.

Source (MNIST) Target (USPS)

Requirements

  • pytorch 0.4.0.
  • torchvision 0.2.0

Download the datasets

cd datasets
bash download.sh

Test results

To obtain the test results, do the following two steps:

1. Download the checkpoints

cd checkpoints
bash download.sh

2. Run the following command,

python main.py -e usps2mnist mnist2usps uspsBig2mnistBig mnistBig2uspsBig -m test_model

The output should be,

mnist2usps          0.955676
mnistBig2uspsBig    0.980541
usps2mnist          0.951500
uspsBig2mnistBig    0.983100

which represent the accuracies obtained on the target test set.

  • mnistBig, and uspsBig use the full training set.
  • mnist, and usps use 2000 images from MNIST and 1800 images from USPS for training, respectively.

Training the models

0. Download the pretrained models and the usps dataset

cd checkpoints
bash download.sh
cd ..
cd datasets
bash download.sh
cd ..

1. Experiment USPS => MNIST

python main.py -e usps2mnist  -m train -rt 1

2. Experiment MNIST => USPS

python main.py -e mnist2usps  -m train -rt 1

3. Experiment MNIST ALL => USPS ALL

python main.py -e mnistBig2uspsBig  -m train -rt 1

4. Experiment USPS ALL => MNIST ALL

python main.py -e uspsBig2mnistBig  -m train -rt 1

Citation

If you find this useful for your research, please cite:

@Article{laradji2018m,
    title={M-ADDA: Unsupervised Domain Adaptation with Deep Metric Learning},
    author={Laradji, Issam and Babanezhad, Reza},
    journal={arXiv preprint arXiv:1807.02552},
    year = {2018}
}

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.