Convex Surrogates for Unbiased Loss Functions in Extreme Classification With Missing Labels

Extreme Classification (XC) refers to supervised learning where each training/test instance is labeled with small subset of relevant labels that are chosen from a large set of possible target labels. The framework of XC has been widely employed in web applications such as automatic labeling of web-encyclopedia, prediction of related searches, and recommendation systems. While most state-of-the-art models in XC achieve high overall accuracy by performing well on the frequently occurring labels, they perform poorly on a large number of infrequent (tail) labels. This arises from two statistical challenges, (i) missing labels, as it is virtually impossible to manually assign every relevant label to an instance, and (ii) highly imbalanced data distribution where a large fraction of labels are tail labels. In this work, we consider common loss functions that decompose over labels, and calculate unbiased estimates that compensate missing labels according to Natarajan et al. [26]. This turns out to be disadvantageous from an optimization perspective, as important properties such as convexity and lower-boundedness are lost. To circumvent this problem, we use the fact that typical loss functions in XC are convex surrogates of the 0-1 loss, and thus propose to switch to convex surrogates of its unbiased version. These surrogates are further adapted to the label imbalance by combining with label-frequency-based rebalancing. We show that the proposed loss functions can be easily incorporated into various different frameworks for extreme classification. This includes (i) linear classifiers, such as DiSMEC, on sparse input data representation, (ii) attention-based deep architecture, AttentionXML, learnt on dense Glove embeddings, and (iii) XLNet-based transformer model for extreme classification, APLC-XLNet. Our results demonstrate consistent improvements over the respective vanilla baseline models, on the propensity-scored metrics for precision and nDCG.

DOI

10.1145/3442381.3450139

Citation

Mohammadnia Qaraei, M, Schultheis, E, Gupta, P & Babbar, R 2021, Convex Surrogates for Unbiased Loss Functions in Extreme Classification With Missing Labels . in The Web Conference 2021 - Proceedings of the World Wide Web Conference, WWW 2021 . ACM, pp. 3711-3720, The Web Conference, Ljubljana, Slovenia, 19/04/2021 . https://doi.org/10.1145/3442381.3450139

Permanent link to this item

https://urn.fi/URN:NBN:fi:aalto-202108098273

Collections

[article-cris] Perustieteiden korkeakoulu / SCI

Show all metadata

Convex Surrogates for Unbiased Loss Functions in Extreme Classification With Missing Labels

Access rights

URL

Journal Title

Journal ISSN

Volume Title

Authors

Date

Department

Major/Subject

Mcode

Degree programme

Language

Pages

Series

Abstract

Description

Keywords

Other note

DOI

Citation

Permanent link to this item

Collections