Skip to content

LibAUC 2.0.0

Latest

Choose a tag to compare

@PenGuln PenGuln released this 29 May 20:16

New Features

  • Added the LibAUC Trainer, enabling users to easily set up and run training pipelines.
  • Added distributed training support for all optimizers used with AUCMLoss, AveragePrecisionLoss, pAUC_CVaR_Loss, pAUC_KL_Loss, tpAUC_CVaR_Loss, tpAUC_KL_Loss, ListwiseCELoss, and NDCGLoss.
  • Integrated Hugging Face models into the LibAUC framework.
  • Updated AUCMLoss and PESG to support resume training. All optimizers now support checkpoint-based training resumption.

New Optimizers

New Tutorials

Bug Fixes

  • Fixed a tensor shape mismatch issue when training models with iSogCLR.
  • Fixed a sampler bug triggered under extremely small imbalance ratios.

Key contributors

  • Siqi Guo, Gang Li, Xiyuan Wei, Linli Zhou, Xingyu Chen, Tianbao Yang