Skip to content

Official implementation of the paper "Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence" ICML2022.

Notifications You must be signed in to change notification settings

Optimization-AI/NDCG-Optimization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

NDCG-Optimization

Code to reproduce the results in paper "Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence". The proposed methods in this paper, SONG and K-SONG, are implemented in LibAUC.

Citation

@inproceedings{qiu2022large,
  title={Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence},
  author={Qiu, Zi-Hao and Hu, Quanqi and Zhong, Yongjian and Zhang, Lijun and Yang, Tianbao},
  booktitle={International Conference on Machine Learning},
  pages={18122--18152},
  year={2022},
  publisher={PMLR},
}

About

Official implementation of the paper "Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence" ICML2022.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published