/RE-OT

Primary LanguagePython


Relative Entropic Optimal Transport: a (Prior-aware) Matching Perspective to (Unbalanced) Classification

Liangliang Shi, Haoyu Zhen, Gu Zhang, Junchi Yan,

NeurIPS 2023

Logo

Environment

conda create -n reot python=3.7
conda activate reot
pip install -r requirements.txt

Training

$ python main.py --cfg config/{config_file}.yaml

Citation

If you find our work useful in your research, please consider citing:

@inproceedings{shi2023relative,
  title={Relative Entropic Optimal Transport: a (Prior-aware) Matching Perspective to (Unbalanced) Classification},
  author={Shi, Liangliang and Zhen, Haoyu and Zhang, Gu and Yan, Junchi},
  booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
  year={2023}
}

References