search

Optimization-AI / LibAUC

LibAUC: A Deep Learning Library for X-Risk Optimization
https://libauc.org/
MIT License
273 stars 37 forks source link

Question on implementation of AUC Margin Loss #35

Closed xyupeng closed 9 months ago

xyupeng commented 11 months ago

image

In LibAUC==1.2.0, why the margin is not used in this implementation of AUC Margin Loss (class AUCM_MultiLabel_V1), as indicated by the red line?

optmai commented 9 months ago

Thanks! it was not a bug. The loss here is only used for computing the gradient of model parameters, whether to use the margin here does not matter. In the optimizer when we update the dual, we do have the margin parameter. To avoid confusion, we will update it soon. Thank you!