← Search

Rohit Babbar

8 accepted papers

2026

Hardware-Aware Dynamic Sparse Training for Large Output Spaces

ICML 2026poster

Extreme multi-label classification (XMC) involves learning deep learning models over large output spaces with millions of labels, making the output layer of the network a major bottleneck in memory and compute. While sparsity-based methods reduce arithmetic complexity, they often fail to yield propo…

Cited by 0SourceScholar
2025

ELMO : Efficiency via Low-precision and Peak Memory Optimization in Large Output Spaces

ICML 2025poster

Large output spaces, also referred to as Extreme multilabel classification (XMC), is a setting that arises, e.g., in large-scale tagging and product-to-product recommendation, and is characterized by the number of labels ranging from hundreds of thousands to millions. This means that the linear clas…

Cited by 0SourcePDFScholar
2025

Large Language Model as a Teacher for Zero-shot Tagging at Extreme Scales

COLING 2025main

Extreme Multi-label Text Classification (XMC) entails selecting the most relevant labels for an instance from a vast label set. Extreme Zero-shot XMC (EZ-XMC) extends this challenge by operating without annotated data, relying only on raw text instances and a predefined label set, making it particul…

2024

A General Online Algorithm for Optimizing Complex Performance Metrics

ICML 2024poster

We consider sequential maximization of performance metrics that are general functions of a confusion matrix of a classifier (such as precision, F-measure, or G-mean). Such metrics are, in general, non-decomposable over individual instances, making their optimization very challenging. While they have…

Cited by 0SourcePDFScholar
2024

Consistent algorithms for multi-label classification with macro-at-$k$ metrics

ICLR 2024poster

We consider the optimization of complex performance metrics in multi-label classification under the population utility framework. We mainly focus on metrics linearly decomposable into a sum of binary classification utilities applied separately to each label with an additional requirement of exactly…

2024

Navigating Extremes: Dynamic Sparsity in Large Output Spaces

NeurIPS 2024poster

In recent years, Dynamic Sparse Training (DST) has emerged as an alternative to post-training pruning for generating efficient models. In principle, DST allows for a much more memory efficient training process, as it maintains sparsity throughout the entire training run. However, current DST implem…

Cited by 2SourcePDFScholar
2023

Generalized test utilities for long-tail performance in extreme multi-label classification

NeurIPS 2023poster

Extreme multi-label classification (XMLC) is the task of selecting a small subset of relevant labels from a very large set of possible labels. As such, it is characterized by long-tail labels, i.e., most labels have very few positive instances. With standard performance measures such as precision@k…

2022

CascadeXML: Rethinking Transformers for End-to-end Multi-resolution Training in Extreme Multi-label Classification

NeurIPS 2022accept

Extreme Multi-label Text Classification (XMC) involves learning a classifier that can assign an input with a subset of most relevant labels from millions of label choices. Recent approaches, such as XR-Transformer and LightXML, leverage a transformer instance to achieve state-of-the-art performance.…