Cargando…

An Adaptive Moment estimation method for online AUC maximization

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 l...

Descripción completa

Detalles Bibliográficos
Autores principales: Liu, Xin, Pan, Zhisong, Yang, Haimin, Zhou, Xingyu, Bai, Wei, Niu, Xianghua
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Public Library of Science 2019
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC6478373/
https://www.ncbi.nlm.nih.gov/pubmed/31013283
http://dx.doi.org/10.1371/journal.pone.0215426
_version_ 1783413158815203328
author Liu, Xin
Pan, Zhisong
Yang, Haimin
Zhou, Xingyu
Bai, Wei
Niu, Xianghua
author_facet Liu, Xin
Pan, Zhisong
Yang, Haimin
Zhou, Xingyu
Bai, Wei
Niu, Xianghua
author_sort Liu, Xin
collection PubMed
description 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.
format Online
Article
Text
id pubmed-6478373
institution National Center for Biotechnology Information
language English
publishDate 2019
publisher Public Library of Science
record_format MEDLINE/PubMed
spelling pubmed-64783732019-05-07 An Adaptive Moment estimation method for online AUC maximization Liu, Xin Pan, Zhisong Yang, Haimin Zhou, Xingyu Bai, Wei Niu, Xianghua PLoS One Research Article 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. Public Library of Science 2019-04-23 /pmc/articles/PMC6478373/ /pubmed/31013283 http://dx.doi.org/10.1371/journal.pone.0215426 Text en © 2019 Liu et al http://creativecommons.org/licenses/by/4.0/ This is an open access article distributed under the terms of the Creative Commons Attribution License (http://creativecommons.org/licenses/by/4.0/) , which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
spellingShingle Research Article
Liu, Xin
Pan, Zhisong
Yang, Haimin
Zhou, Xingyu
Bai, Wei
Niu, Xianghua
An Adaptive Moment estimation method for online AUC maximization
title An Adaptive Moment estimation method for online AUC maximization
title_full An Adaptive Moment estimation method for online AUC maximization
title_fullStr An Adaptive Moment estimation method for online AUC maximization
title_full_unstemmed An Adaptive Moment estimation method for online AUC maximization
title_short An Adaptive Moment estimation method for online AUC maximization
title_sort adaptive moment estimation method for online auc maximization
topic Research Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC6478373/
https://www.ncbi.nlm.nih.gov/pubmed/31013283
http://dx.doi.org/10.1371/journal.pone.0215426
work_keys_str_mv AT liuxin anadaptivemomentestimationmethodforonlineaucmaximization
AT panzhisong anadaptivemomentestimationmethodforonlineaucmaximization
AT yanghaimin anadaptivemomentestimationmethodforonlineaucmaximization
AT zhouxingyu anadaptivemomentestimationmethodforonlineaucmaximization
AT baiwei anadaptivemomentestimationmethodforonlineaucmaximization
AT niuxianghua anadaptivemomentestimationmethodforonlineaucmaximization
AT liuxin adaptivemomentestimationmethodforonlineaucmaximization
AT panzhisong adaptivemomentestimationmethodforonlineaucmaximization
AT yanghaimin adaptivemomentestimationmethodforonlineaucmaximization
AT zhouxingyu adaptivemomentestimationmethodforonlineaucmaximization
AT baiwei adaptivemomentestimationmethodforonlineaucmaximization
AT niuxianghua adaptivemomentestimationmethodforonlineaucmaximization