Title :
Optimization of the Area under the ROC Curve
Author :
Castro, Cristiano Leite ; Braga, Antonio Padua
Author_Institution :
Depto. Eng. Eletron., Univ. Fed. de Minas Gerais, Belo Horizonte
Abstract :
In this paper, we propose a new binary classification algorithm (AUCtron), based on gradient descent learning, that directly optimizes AUC (area under the ROC curve). We compare it with a linear classifier and with AUCsplit proposed. The AUCtron algorithm implicitly considers class prior probabilities in the decision criteria. Our results demonstrated that AUC is a sensitive enough metric that when used in small and imbalanced data sets may lead to a better separation.
Keywords :
classification; learning (artificial intelligence); optimisation; sensitivity analysis; ROC curve; binary classification algorithm; gradient descent learning; linear classifier; optimization; Area measurement; Classification algorithms; Data mining; Decision making; Error analysis; Machine learning; Medical diagnostic imaging; Neural networks; Robustness; Signal detection; AUC; ROC Curve; binary classifier; imbalanced classes; learning; neural networks;
Conference_Titel :
Neural Networks, 2008. SBRN '08. 10th Brazilian Symposium on
Conference_Location :
Salvador
Print_ISBN :
978-1-4244-3219-6
Electronic_ISBN :
1522-4899
DOI :
10.1109/SBRN.2008.25