Extreme Multi-Label Classification

29 papers with code • 0 benchmarks • 2 datasets

Extreme Multi-Label Classification is a supervised learning problem where an instance may be associated with multiple labels. The two main problems are the unbalanced labels in the dataset and the amount of different labels.

Libraries

Use these libraries to find Extreme Multi-Label Classification models and implementations
3 papers
493

Propensity-scored Probabilistic Label Trees

mwydmuch/napkinXC 20 Oct 2021

Extreme multi-label classification (XMLC) refers to the task of tagging instances with small subsets of relevant labels coming from an extremely large set of all possible labels.

58
20 Oct 2021

Adaptive Elastic Training for Sparse Deep Learning on Heterogeneous Multi-GPU Servers

YMA33/HeteroGPU 13 Oct 2021

We address these challenges with Adaptive SGD, an adaptive elastic model averaging stochastic gradient descent algorithm for heterogeneous multi-GPUs that is characterized by dynamic scheduling, adaptive batch size scaling, and normalized model merging.

2
13 Oct 2021

DECAF: Deep Extreme Classification with Label Features

Extreme-classification/DECAF 1 Aug 2021

This paper develops the DECAF algorithm that addresses these challenges by learning models enriched by label metadata that jointly learn model parameters and feature representations using deep networks and offer accurate classification at the scale of millions of labels.

52
01 Aug 2021

ECLARE: Extreme Classification with Label Graph Correlations

Extreme-classification/ECLARE 31 Jul 2021

This paper presents ECLARE, a scalable deep learning architecture that incorporates not only label text, but also label correlations, to offer accurate real-time predictions within a few milliseconds.

41
31 Jul 2021

Extreme Multi-label Learning for Semantic Matching in Product Search

amzn/pecos 23 Jun 2021

In this paper, we aim to improve semantic product search by using tree-based XMC models where inference time complexity is logarithmic in the number of products.

493
23 Jun 2021

Priberam at MESINESP Multi-label Classification of Medical Texts Task

Priberam/mesinesp-svm 12 May 2021

Information retrieval tools are crucial in order to navigate and provide meaningful recommendations for articles and treatments.

4
12 May 2021

Stratified Sampling for Extreme Multi-Label Data

maxitron93/stratified_sampling_for_XML 5 Mar 2021

Extreme multi-label classification (XML) is becoming increasingly relevant in the era of big data.

7
05 Mar 2021

Top-$k$ eXtreme Contextual Bandits with Arm Hierarchy

rajatsen91/XtremeContextualBandits 15 Feb 2021

We show that our algorithm has a regret guarantee of $O(k\sqrt{(A-k+1)T \log (|\mathcal{F}|T)})$, where $A$ is the total number of arms and $\mathcal{F}$ is the class containing the regression function, while only requiring $\tilde{O}(A)$ computation per time step.

8
15 Feb 2021

Retrieving Skills from Job Descriptions: A Language Model Based Extreme Multi-label Classification Framework

wing-nus/jd2skills-bert-xmlc COLING 2020

We introduce a deep learning model to learn the set of enumerated job skills associated with a job description.

49
01 Dec 2020

Probabilistic Label Trees for Extreme Multi-label Classification

mwydmuch/napkinXC 23 Sep 2020

We first introduce the PLT model and discuss training and inference procedures and their computational costs.

58
23 Sep 2020