Relative Entropic Optimal Transport: a (Prior-aware) Matching Perspective to (Unbalanced) Classification
Liangliang Shi, Haoyu Zhen, Gu Zhang, Junchi Yan,
conda create -n reot python=3.7
conda activate reot
pip install -r requirements.txt
$ python main.py --cfg config/{config_file}.yaml
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}
}
- The code is based on BalancedMetaSoftmax and classifier-balancing.