AUC Maximization in Bayesian hierarchical models

Mehmet Gonen

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

The area under the curve (AUC) measures such as the area under the receiver operating characteristics curve (AUROC) and the area under the precision-recall curve (AUPR) are known to be more appropriate than the error rate, especially, for imbalanced data sets. There are several algorithms to optimize AUC measures instead of minimizing the error rate. However, this idea has not been fully exploited in Bayesian hierarchical models owing to the difficulties in inference. Here, we formulate a general Bayesian inference framework, called Bayesian AUC Maximization (BAM), to integrate AUC maximization into Bayesian hierarchical models by borrowing the pairwise and listwise ranking ideas from the information retrieval literature. To showcase our BAM framework, we develop two Bayesian linear classifier variants for two ranking approaches and derive their variational inference procedures. We perform validation experiments on four biomedical data sets to demonstrate the better predictive performance of our framework over its error-minimizing counterpart in terms of average AUROC and AUPR values.

Original languageEnglish (US)
Title of host publicationFrontiers in Artificial Intelligence and Applications
PublisherIOS Press
Pages21-27
Number of pages7
Volume285
ISBN (Electronic)9781614996712
DOIs
StatePublished - 2016
Externally publishedYes
Event22nd European Conference on Artificial Intelligence, ECAI 2016 - The Hague, Netherlands
Duration: Aug 29 2016Sep 2 2016

Publication series

NameFrontiers in Artificial Intelligence and Applications
Volume285
ISSN (Print)09226389

Other

Other22nd European Conference on Artificial Intelligence, ECAI 2016
CountryNetherlands
CityThe Hague
Period8/29/169/2/16

ASJC Scopus subject areas

  • Artificial Intelligence

Fingerprint Dive into the research topics of 'AUC Maximization in Bayesian hierarchical models'. Together they form a unique fingerprint.

  • Cite this

    Gonen, M. (2016). AUC Maximization in Bayesian hierarchical models. In Frontiers in Artificial Intelligence and Applications (Vol. 285, pp. 21-27). (Frontiers in Artificial Intelligence and Applications; Vol. 285). IOS Press. https://doi.org/10.3233/978-1-61499-672-9-21