MBA: Mini-Batch AUC Optimization

January 27, 2020

Abstract

Area under the receiver operating characteristics curve (AUC) is an important metric for a wide range of signal processing and machine learning problems, and scalable methods for optimizing AUC have recently been proposed. However, handling very large datasets remains an open challenge for this problem. This paper proposes a novel approach to AUC maximization, based on sampling mini-batches of positive/negative instance pairs and computing U-statistics to approximate a global risk minimization problem. The resulting algorithm is simple, fast, and learning-rate free. We show that the number of samples required for good performance is independent of the number of pairs available, which is a quadratic function of the positive and negative instances. Extensive experiments show the practical utility of the proposed method.

Download

Publication Type

Paper

Conference / Journal Name

IEEE Transactions on Neural Networks and Learning Systems

Authors

BibTeX


@inproceedings{
    author = {},
    title = {‌MBA: Mini-Batch AUC Optimization‌},
    booktitle = {Proceedings of IEEE Transactions on Neural Networks and Learning Systems‌},
    year = {‌2020‌}
}