The Real Price of Bandit Information in Multiclass Classification

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Erez, Liad, Cohen, Alon, Koren, Tomer, Mansour, Yishay, Moran, Shay
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910495013863424
author Erez, Liad
Cohen, Alon
Koren, Tomer
Mansour, Yishay
Moran, Shay
author_facet Erez, Liad
Cohen, Alon
Koren, Tomer
Mansour, Yishay
Moran, Shay
contents We revisit the classical problem of multiclass classification with bandit feedback (Kakade, Shalev-Shwartz and Tewari, 2008), where each input classifies to one of $K$ possible labels and feedback is restricted to whether the predicted label is correct or not. Our primary inquiry is with regard to the dependency on the number of labels $K$, and whether $T$-step regret bounds in this setting can be improved beyond the $\smash{\sqrt{KT}}$ dependence exhibited by existing algorithms. Our main contribution is in showing that the minimax regret of bandit multiclass is in fact more nuanced, and is of the form $\smash{\widetildeΘ\left(\min \left\{|H| + \sqrt{T}, \sqrt{KT \log |H|} \right\} \right) }$, where $H$ is the underlying (finite) hypothesis class. In particular, we present a new bandit classification algorithm that guarantees regret $\smash{\widetilde{O}(|H|+\sqrt{T})}$, improving over classical algorithms for moderately-sized hypothesis classes, and give a matching lower bound establishing tightness of the upper bounds (up to log-factors) in all parameter regimes.
format Preprint
id arxiv_https___arxiv_org_abs_2405_10027
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle The Real Price of Bandit Information in Multiclass Classification
Erez, Liad
Cohen, Alon
Koren, Tomer
Mansour, Yishay
Moran, Shay
Machine Learning
Artificial Intelligence
We revisit the classical problem of multiclass classification with bandit feedback (Kakade, Shalev-Shwartz and Tewari, 2008), where each input classifies to one of $K$ possible labels and feedback is restricted to whether the predicted label is correct or not. Our primary inquiry is with regard to the dependency on the number of labels $K$, and whether $T$-step regret bounds in this setting can be improved beyond the $\smash{\sqrt{KT}}$ dependence exhibited by existing algorithms. Our main contribution is in showing that the minimax regret of bandit multiclass is in fact more nuanced, and is of the form $\smash{\widetildeΘ\left(\min \left\{|H| + \sqrt{T}, \sqrt{KT \log |H|} \right\} \right) }$, where $H$ is the underlying (finite) hypothesis class. In particular, we present a new bandit classification algorithm that guarantees regret $\smash{\widetilde{O}(|H|+\sqrt{T})}$, improving over classical algorithms for moderately-sized hypothesis classes, and give a matching lower bound establishing tightness of the upper bounds (up to log-factors) in all parameter regimes.
title The Real Price of Bandit Information in Multiclass Classification
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2405.10027