← Search

Kunal Dahiya

4 accepted papers

2026

Large Language Models Meet Extreme Multi-label Classification: Scaling and Multi-modal Framework

AAAI 2026technical

Foundation models have revolutionized artificial intelligence across numerous domains, yet their transformative potential remains largely untapped in Extreme Multi-label Classification (XMC). Queries in XMC are associated with relevant labels from extremely large label spaces, where it is critical t

Cited by 0SourcePDFScholar
2025

Prototypical Extreme Multi-label Classification with a Dynamic Margin Loss

NAACL 2025long

Extreme Multi-label Classification (XMC) methods predict relevant labels for a given query in an extremely large label space. Recent works in XMC address this problem using deep encoders that project text descriptions to an embedding space suitable for recovering the closest labels. However, learnin…

2022

Multi-Modal Extreme Classification

CVPR 2022poster

This paper develops the MUFIN technique for extreme classification (XC) tasks with millions of labels where datapoints and labels are endowed with visual and textual descriptors. Applications of MUFIN to product-to-product recommendation and bid query prediction over several millions of products are…

Cited by 16PDFcodeScholar
2021

SiameseXML: Siamese Networks meet Extreme Classifiers with 100M Labels

ICML 2021spotlight

Deep extreme multi-label learning (XML) requires training deep architectures that can tag a data point with its most relevant subset of labels from an extremely large label set. XML applications such as ad and product recommendation involve labels rarely seen during training but which nevertheless h…