mlsanityData to 7 Oct 2026

Efficient line search for optimizing Area Under the ROC Curve in gradient descent

arXiv 2410.08635 ↗First posted 11 Oct 2024cs.LG cs.AI stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Receiver Operating Characteristic (ROC) curves are useful for evaluation in binary classification and changepoint detection, but difficult to use for learning since the Area Under the Curve (AUC) is piecewise constant (gradient zero almost everywhere). Recently the Area Under Min (AUM) of false positive and false negative rates has been proposed as a differentiable surrogate for AUC. In this paper we study the piecewise linear/constant nature of the AUM/AUC, and propose new efficient path-following algorithms for choosing the learning rate which is optimal for each step of gradient descent (line search), when optimizing a linear model. Remarkably, our proposed line search algorithm has the same log-linear asymptotic time complexity as gradient descent with constant step size, but it computes a complete representation of the AUM/AUC as a function of step size. In our empirical study of binary classification problems, we verify that our proposed algorithm is fast and exact; in changepoint detection problems we show that the proposed algorithm is just as accurate as grid search, but faster.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 21 references that are in the map. First posted on arXiv 11 Oct 2024.

Builds on

6 of 21 references.

References · 6

  1. 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
  2. 2020Survival regression with accelerated failure time model in XGBoostcited by 37
  3. 2014On Optimal Multiple Changepoint Algorithms for Large Datacited by 15
  4. 2023LibAUC: A Deep Learning Library for X-Risk Optimizationcited by 15
  5. 2017Maximum Margin Interval Treescited by 7
  6. 2023A Log-linear Gradient Descent Algorithm for Unbalanced Binary Classification using the All Pairs Squared Hinge Losscited by 1

Built on by

Not cited yet.