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...
Autores principales: | , , , , , |
---|---|
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 |