A Structured Prediction Approach for Label Ranking

Anna Korba (1, 2) , Alexandre Garcia (2, 1) , Florence d'Alché-Buc (3, 1)

1 S2A - Signal, Statistique et Apprentissage
2 LTCI - Laboratoire Traitement et Communication de l'Information
3 LTCI - Laboratoire Traitement et Communication de l'Information

We propose to solve a label ranking problem as a structured output regression task. In this view, we adopt a least square surrogate loss approach that solves a supervised learning problem in two steps: a regression step in a well-chosen feature space and a pre-image (or decoding) step. We use specific feature maps/embeddings for ranking data, which convert any ranking/permutation into a vector representation. These embeddings are all well-tailored for our approach, either by resulting in consistent estimators, or by solving trivially the pre-image problem which is often the bottleneck in structured prediction. Their extension to the case of incomplete or partial rankings is also discussed. Finally, we provide empirical results on synthetic and real-world datasets showing the relevance of our method.

Domaines: Informatique [cs]

Anna Korba, Alexandre Garcia, Florence d'Alché-Buc. A Structured Prediction Approach for Label Ranking. Nips 2018, Dec 2018, Montreal, Canada. ⟨hal-02371075⟩