Multiclass classification, information, divergence and surrogate risk

John Duchi, Khashayar Khosravi, Feng Ruan

Research output: Contribution to journalArticlepeer-review

26 Scopus citations

Abstract

We provide a unifying view of statistical information measures, multiway Bayesian hypothesis testing, loss functions for multiclass classification problems and multidistribution f-divergences, elaborating equivalence results between all of these objects, and extending existing results for binary outcome spaces to more general ones. We consider a generalization of f-divergences to multiple distributions, and we provide a constructive equivalence between divergences, statistical information (in the sense of DeGroot) and losses for multiclass classification. A major application of our results is in multiclass classification problems in which we must both infer a discriminant function γ —for making predictions on a label Y from datum X—and a data representation (or, in the setting of a hypothesis testing problem, an experimental design), represented as a quantizer q from a family of possible quantizers Q. In this setting, we characterize the equivalence between loss functions, meaning that optimizing either of two losses yields an optimal discriminant and quantizer q, complementing and extending earlier results of Nguyen et al. [Ann. Statist. 37 (2009) 876–904] to the multiclass case. Our results provide a more substantial basis than standard classification calibration results for comparing different losses: we describe the convex losses that are consistent for jointly choosing a data representation and minimizing the (weighted) probability of error in multiclass classification problems.

Original languageEnglish (US)
Pages (from-to)3246-3275
Number of pages30
JournalAnnals of Statistics
Volume46
Issue number6B
DOIs
StatePublished - 2018

Funding

Received September 2016; revised October 2017. 1Supported in part by NSF-CAREER Award 1553086 and the SAIL-Toyota Center for AI Research. Feng Ruan additionally supported by the Stanford Graduate Fellowship. MSC2010 subject classifications. 62G10, 62K05, 62C05, 94A17, 68Q32. Key words and phrases. f -divergence, risk, surrogate loss function, hypothesis test. Supported in part by NSF-CAREER Award 1553086 and the SAIL-Toyota Center for AI Research. Feng Ruan additionally supported by the Stanford Graduate Fellowship.

Keywords

  • F-divergence
  • Hypothesis test
  • Risk
  • Surrogate loss function

ASJC Scopus subject areas

  • Statistics and Probability
  • Statistics, Probability and Uncertainty

Fingerprint

Dive into the research topics of 'Multiclass classification, information, divergence and surrogate risk'. Together they form a unique fingerprint.

Cite this