Open Vocabulary Extreme Classification Using Generative Models

7Citations
Citations of this article
43Readers
Mendeley users who have this article in their library.

Abstract

The extreme multi-label classification (XMC) task aims at tagging content with a subset of labels from an extremely large label set. The label vocabulary is typically defined in advance by domain experts and assumed to capture all necessary tags. However in real world scenarios this label set, although large, is often incomplete and experts frequently need to refine it. To develop systems that simplify this process, we introduce the task of open vocabulary XMC (OXMC): given a piece of content, predict a set of labels, some of which may be outside of the known tag set. Hence, in addition to not having training data for some labels-as is the case in zero-shot classification-models need to invent some labels on-the-fly. We propose GROOV, a fine-tuned seq2seq model for OXMC that generates the set of labels as a flat sequence and is trained using a novel loss independent of predicted label order. We show the efficacy of the approach, experimenting with popular XMC datasets for which GROOV is able to predict meaningful labels outside the given vocabulary while performing on par with state-of-the-art solutions for known labels.

Cite

CITATION STYLE

APA

Simig, D., Petroni, F., Yanki, P., Popat, K., Du, C., Riedel, S., & Yazdani, M. (2022). Open Vocabulary Extreme Classification Using Generative Models. In Proceedings of the Annual Meeting of the Association for Computational Linguistics (pp. 1561–1583). Association for Computational Linguistics (ACL). https://doi.org/10.18653/v1/2022.findings-acl.123

Register to see more suggestions

Mendeley helps you to discover research relevant for your work.

Already have an account?

Save time finding and organizing research with Mendeley

Sign up for free