← 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 0Source ↗Scholar ↗
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 0Source ↗PDF ↗Scholar ↗
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 0Source ↗PDF ↗Scholar ↗
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 2Source ↗PDF ↗Scholar ↗
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.…