Skip to content

HierMUD: a Hierarchical Multi-task Unsupervised Domain adaptation framework

License

Notifications You must be signed in to change notification settings

jingxiaoliu/HierMUD

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

multi-task-UDA

This is the repository for the paper:

  • Jingxiao Liu

[slides][paper][video]

Description

We introduce a novel approach for multi-task unsupervised domain adaptation. This approach is developed for bridge health monitoring using drive-by vehicle vibrations, but it can be applied to other problems, such as digit recognition, image classification, etc.

The architecture of our hierarchical multi-task and domain-adversarial learning algorithm. The red and black arrows between blocks represent source and target domain data stream, respectively. Orange blocks are feature extractors, blue blocks are task predictors, and red blocks are domain classifiers.

In this repository, we demonstrate our approach through two examples:

  • A digit recognition example, which transfers model learned using MNIST data to MNIST-M data and conducts two tasks: odd-even classification and digits comparison.
  • A drive-by bridge health monitoring example, which transfers model learned using vehicle vibration data collected from one bridge to detect, localize and quantify damage on another bridge.

Note: the drive-by bridge health monitoring experiment involves data that is not publicly available. We will work towards making the experiment replicable without violating data usage policy.

Code Usage

git clone https://github.com/jingxiaoliu/multi-task-UDA.git
cd multi-task-UDA

Run the digit recognition example with 'demo_mnist.ipynb'. Run the drive-by bridge health monitoring example with 'demo_dbbhm.ipynb'.

Contact

Feel free to send any questions to:

  • Jingxiao Liu, Ph.D. Candidate at Stanford University, Department of Civil and Environmental Engineering.

Citation

If you use this implementation, please cite our paper as follows:

About

HierMUD: a Hierarchical Multi-task Unsupervised Domain adaptation framework

Topics

Resources

License

Stars

Watchers

Forks