Saved in:
Bibliographic Details
Main Authors: Zhang, Jianxin, Scott, Clayton
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2305.19470
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918132216496128
author Zhang, Jianxin
Scott, Clayton
author_facet Zhang, Jianxin
Scott, Clayton
contents Label embedding is a framework for multiclass classification problems where each label is represented by a distinct vector of some fixed dimension, and training involves matching model output to the vector representing the correct label. While label embedding has been successfully applied in extreme classification and zero-shot learning, and offers both computational and statistical advantages, its theoretical foundations remain poorly understood. This work presents an analysis of label embedding in the context of extreme multiclass classification, where the number of classes $C$ is very large. We present an excess risk bound that reveals a trade-off between computational and statistical efficiency, quantified via the coherence of the embedding matrix. We further show that under the Massart noise condition, the statistical penalty for label embedding vanishes with sufficiently low coherence. Our analysis supports an algorithm that is simple, scalable, and easily parallelizable, and experimental results demonstrate its effectiveness in large-scale applications.
format Preprint
id arxiv_https___arxiv_org_abs_2305_19470
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Label Embedding via Low-Coherence Matrices
Zhang, Jianxin
Scott, Clayton
Machine Learning
Label embedding is a framework for multiclass classification problems where each label is represented by a distinct vector of some fixed dimension, and training involves matching model output to the vector representing the correct label. While label embedding has been successfully applied in extreme classification and zero-shot learning, and offers both computational and statistical advantages, its theoretical foundations remain poorly understood. This work presents an analysis of label embedding in the context of extreme multiclass classification, where the number of classes $C$ is very large. We present an excess risk bound that reveals a trade-off between computational and statistical efficiency, quantified via the coherence of the embedding matrix. We further show that under the Massart noise condition, the statistical penalty for label embedding vanishes with sufficiently low coherence. Our analysis supports an algorithm that is simple, scalable, and easily parallelizable, and experimental results demonstrate its effectiveness in large-scale applications.
title Label Embedding via Low-Coherence Matrices
topic Machine Learning
url https://arxiv.org/abs/2305.19470