A General Online Algorithm for Optimizing Complex Performance Metrics

Wojciech Kotłowski, Marek Wydmuch, Erik Schultheis, Rohit Babbar, Krzysztof Dembczyński

Research output: Contribution to journalConference articlepeer-review

Abstract

We consider sequential maximization of performance metrics that are general functions of a confusion matrix of a classifier (such as precision, F-measure, or G-mean). Such metrics are, in general, non-decomposable over individual instances, making their optimization very challenging. While they have been extensively studied under different frameworks in the batch setting, their analysis in the online learning regime is very limited, with only a few distinguished exceptions. In this paper, we introduce and analyze a general online algorithm that can be used in a straightforward way with a variety of complex performance metrics in binary, multi-class, and multi-label classification problems. The algorithm's update and prediction rules are appealingly simple and computationally efficient without the need to store any past data. We show the algorithm attains O(lnnn ) regret for concave and smooth metrics and verify the efficiency of the proposed algorithm in empirical studies.

Original languageEnglish
Pages (from-to)25396-25425
Number of pages30
JournalProceedings of Machine Learning Research
Volume235
Early online date27 Jul 2024
Publication statusE-pub ahead of print - 27 Jul 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 21 Jul 202427 Jul 2024

ASJC Scopus subject areas

  • Artificial Intelligence
  • Software
  • Control and Systems Engineering
  • Statistics and Probability

Fingerprint

Dive into the research topics of 'A General Online Algorithm for Optimizing Complex Performance Metrics'. Together they form a unique fingerprint.

Cite this