Git Product home page Git Product logo

cd3a's Introduction

Curriculum based Dropout Discriminator for Domain Adaptation(CD3A)

Torch code for Domain Adaptation model(CD3A) . For more information, please refer the paper

Accepted at [BMVC 2019]

Abstract

Domain adaptation is essential to enable wide usage of deep learning based networks trained using large labeled datasets. Adversarial learning based techniques have shown their utility towards solving this problem using a discriminator that ensures source and target distributions are close. However, here we suggest that rather than using a point estimate, it would be useful if a distribution based discriminator could be used to bridge this gap. This could be achieved using multiple classifiers or using traditional ensemble methods. In contrast, we suggest that a Monte Carlo dropout based ensemble discriminator could suffice to obtain the distribution based discriminator. Specifically, we propose a curriculum based dropout discriminator that gradually increases the variance of the sample based distribution and the corresponding reverse gradients are used to align the source and target feature representations. The detailed results and thorough ablation analysis show that our model outperforms state-of-the-art results.

Result

Requirements

This code is written in Lua and requires Torch.

You also need to install the following package in order to sucessfully run the code.

Download Dataset

Prepare Datasets
  • Download the dataset

Training Steps

We have prepared everything for you ;)

####Clone the repositotry

git clone https://github.com/DelTA-Lab-IITK/CD3A

Dataset prepare

  • Downalod dataset

  • put all source images inside mydataset/train/ such that folder name is class name

  mkdir -p /path_to_wherever_you_want/mydataset/train/ 
  • put all target images inside mydataset/val/ such that folder name is class name
mkdir -p /path_to_wherever_you_want/mydataset/val/ 
  • creare softlink of dataset
 cd CD3A/
 ln -sf /path_to_wherever_you_want/mydataset dataset

Pretrained Alexnet model

  • Download Alexnet pretraine caffe model Link
cd CD3A/  
ln -sf /path_to_where_model_is_downloaded/ pretrained_network 

Train model

cd CD3A/  
./train.sh 

Reference

If you use this code as part of any published research, please acknowledge the following paper

@article{kurmi2019curriculum,
  title={Curriculum based Dropout Discriminator for Domain Adaptation},
  author={Kurmi, Vinod Kumar and Bajaj, Vipul and Subramanian, Venkatesh K and Namboodiri, Vinay P},
  journal={arXiv preprint arXiv:1907.10628},
  year={2019}
}

Contributors

cd3a's People

Contributors

vinodkkurmi avatar vipulbjj avatar

Watchers

 avatar

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.