中文

MBA:小批量 AUC 优化

机器学习 2018-06-01 v2 机器学习

摘要

受试者工作特征曲线下面积(AUC)是广泛应用于信号处理和机器学习问题的重要指标,近期已有可扩展的 AUC 优化方法被提出。然而,处理极大规模数据集对该问题而言仍是一个开放性挑战。本文提出了一种新的 AUC 最大化方法,该方法基于对正负样本对的小批量进行采样,并计算 U 统计量以逼近全局风险最小化问题。所得算法简单、快速且无需学习率。我们证明,获得良好性能所需的样本数与可用的样本对数量无关,而后者是正负实例数量的二次函数。大量实验表明了所提方法的实用价值。

关键词

引用

@article{arxiv.1805.11221,
  title  = {MBA: Mini-Batch AUC Optimization},
  author = {San Gultekin and Avishek Saha and Adwait Ratnaparkhi and John Paisley},
  journal= {arXiv preprint arXiv:1805.11221},
  year   = {2018}
}