Research

Publications

• 
  LibAUC: A Deep Learning Library for X-risk Optimization.

  Zhuoning Yuan, Dixian Zhu, Zi-Hao Qiu, Gang Li, Xuanhui Wang, Tianbao Yang
  ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD 2023)
  Read more

• 
  Algorithmic Foundation of Deep X-risk Optimization.

  Tianbao Yang
  preprint
  Read more

• 
  Not All Semantics are Created Equal: Contrastive Self-supervised Learning with Automatic Temperature Individualization.

  Zi-Hao Qiu, Quanqi Hu, Zhuoning Yuan, Denny Zhou, Lijun Zhang, Tianbao Yang
  International Conference on Machine Learning (ICML 2023)
  Read more

• 
  Provable Multi-instance Deep AUC Maximization with Stochastic Pooling.

  Dixian Zhu, Bokun Wang, Zhi Chen, Yaxing Wang, Milan Sonka, Xiaodong Wu, Tianbao Yang
  International Conference on Machine Learning (ICML 2023)
  Read more

• 
  Finite-Sum Coupled Compositional Stochastic Optimization: Theory and Applications.

  Bokun Wang, Tianbao Yang
  International Conference on Machine Learning (ICML 2022)
  Read more

• 
  Provable Stochastic Optimization for Global Contrastive Learning: Small Batch Does Not Harm Performance.

  Zhuoning Yuan, Yuexin Wu, Zihao Qiu, Xianzhi Du, Lijun Zhang, Denny Zhou, Tianbao Yang
  International Conference on Machine Learning (ICML 2022)
  Read more

• 
  Large-scale Stochastic Optimization of NDCG Surrogates for Deep Learning with Provable Convergence.

  Zi-hao Qiu*, Quanqi Hu*, Yongjian Zhong, Lijun Zhang, Tianbao Yang
  International Conference on Machine Learning (ICML 2022)
  Read more

• 
  When AUC meets DRO: Optimizing Partial AUC for Deep Learning with Non-Convex Convergence Guarantee.

  Dixian Zhu*, Gang Li*, Bokun Wang, Xiaodong Wu, Tianbao Yang
  International Conference on Machine Learning (ICML 2022)
  Read more

• 
  AUC Maximization in the Era of Big Data and AI: A Survey.

  Tianbao Yang, Yiming Ying
  preprint
  Read more

• 
  Compositional Training for End-to-End Deep AUC Maximization.

  Zhuoning Yuan, Zhishuai Guo, Nitesh Chawla, Tianbao Yang
  International Conference on Learning Representations (ICLR 2022)
  Read more

• 
  Advanced Graph and Sequence Neural Networks for Molecular Property Prediction and Drug Discovery.

  Zhengyang Wang, Meng Liu, Youzhi Luo, Zhao Xu, Yaochen Xie, Limei Wang, Lei Cai, Qi Qi, Zhuoning Yuan, Tianbao Yang, Shuiwang Ji (Bioinformatics)
  Read more

• 
  Momentum Accelerates the Convergence of Stochastic AUPRC Maximization.

  Guanghui Wang, Ming Yang, Lijun Zhang, Tianbao Yang
  International Conference on Artificial Intelligence and Statistics (AISTATS 2022)
  Read more

• 
  Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image Classification.

  Zhuoning Yuan, Yan Yan, Milan Sonka, Tianbao Yang
  International Conference on Computer Vision (ICCV 2021)
  Read more

• 
  Federated Deep AUC Maximization for Heterogeneous Data with a Constant Communication Complexity.

  Zhuoning Yuan*, Zhishuai Guo*, Yi Xu, Yiming Ying, Tianbao Yang (*equal contribution)
  International Conference on Machine Learning (ICML 2021)
  Read more

• 
  Stochastic Optimization of Area Under Precision-Recall Curve for Deep Learning with Provable Convergence.

  Qi Qi*, Youzhi Luo*, Zhao Xu, Shuiwang Ji, Tianbao Yang (*equal contribution)
  Neural Information Processing Systems (NeurIPS2021)
  Read more

• 
  Non-Convex Min-Max Optimization: Provable Algorithms and Applications in Machine Learning.

  Hassan Rafique, Mingrui Liu, Qihang Lin, Tianbao Yang
  Optimization Methods and Software (2021)
  Read more

• 
  Communication-Efficient Distributed Stochastic AUC Maximization with Deep Neural Networks.

  Zhishuai Guo, Mingrui Liu, Zhuoning Yuan, Li Shen, Wei Liu, Tianbao Yang
  International Conference on Machine Learning (ICML 2020)
  Read more

• 
  Optimal Epoch Stochastic Gradient Descent Ascent Methods for Min-Max Optimization.

  Yan Yan, Yi Xu, Qihang Lin, Wei Liu, Tianbao Yang
  Neural Information Processing Systems (NeurIPS2020)
  Read more

• 
  Fast Objective and Duality Gap Convergence for Non-convex Strongly-concave Min-max Problems.

  Zhishuai Guo, Yan Yan, Zhuoning Yuan, Tianbao Yang
  Preprints (October 2020)
  Read more

• 
  Stochastic AUC Maximization with Deep Neural Networks.

  Mingrui Liu, Zhuoning Yuan, Yiming Ying, Tianbao Yang
  International Conference on Learning Representations (ICLR 2020)
  Read more

• 
  Fast Stochastic AUC Maximization with O(1/n)-Convergence Rate.

  Mingrui Liu, Xiaoxuan Zhang, Zaiyi Chen, Xiaoyu Wang, Tianbao Yang
  International Conference on Machine Learning (ICML 2018)
  Read more

• 
  Online AUC Maximization.

  Peilin Zhao, Steven Hoi, Rong Jin, Tianbao Yang
  International Conference on Machine Learning (ICML 2011)
  Read more