/Graph-Optimal-Transport

Code for ICML 2020 "Graph Optimal Transport for Cross-Domain Alignment"

Primary LanguagePythonMIT LicenseMIT

Graph Optimal Transport framework for cross-domain alignment

Summary

In this work, both Gromov-Wasserstein and Wasserstein distance are applied to improve the performance of different cross-domain alignment tasks, such as VQA, Machine Translation, etc

Implemntation

Note that the VQA model is developed upon BAN model.

And the machine translation model is developed upon texar.

Citation

Please cite our ICML 2020 paper if you found the code useful.

@article{chen2020graph,
  title={Graph Optimal Transport for Cross-Domain Alignment},
  author={Chen, Liqun and Gan, Zhe and Cheng, Yu and Li, Linjie and Carin, Lawrence and Liu, Jingjing},
  journal={ICML},
  year={2020}
}