An Adaptive Moment estimation method for online AUC maximization

PLoS One. 2019 Apr 23;14(4):e0215426. doi: 10.1371/journal.pone.0215426. eCollection 2019.

Abstract

Area Under the ROC Curve (AUC) is a widely used metric for measuring classification performance. It has important theoretical and academic values to develop AUC maximization algorithms. Traditional methods often apply batch learning algorithm to maximize AUC which is inefficient and unscalable for large-scale applications. Recently some online learning algorithms have been introduced to maximize AUC by going through the data only once. However, these methods sometimes fail to converge to an optimal solution due to the fixed or rapid decay of learning rates. To tackle this problem, we propose an algorithm AdmOAM, Adaptive Moment estimation method for Online AUC Maximization. It applies the estimation of moments of gradients to accelerate the convergence and mitigates the rapid decay of the learning rates. We establish the regret bound of the proposed algorithm and implement extensive experiments to demonstrate its effectiveness and efficiency.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Area Under Curve
  • Machine Learning*
  • ROC Curve

Grants and funding

Our work is supported by the National Key Research Development Program of China (No. 2017YFB0802800). The funders participated in the design of algorithms and commented on the manuscript.