Skip to main navigation Skip to search Skip to main content

Label filters for large scale multilabel classification

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

Abstract

When assigning labels to a test instance, most multilabel and multiclass classifiers systematically evaluate every single label to decide whether it is relevant or not. This linear scan over labels becomes prohibitive when the number of labels is very large. To alleviate this problem we propose a two step approach where computationally efficient label filters pre-select a small set of candidate labels before the base multiclass or multilabel classifier is applied. The label filters select candidate labels by projecting a test instance on a filtering line, and retaining only the labels that have training instances in the vicinity of this projection. The filter parameters are learned directly from data by solving a constraint optimization problem, and are independent of the base multilabel classifier. The proposed label filters can be used in conjunction with any multiclass or multilabel classifier that requires a linear scan over the labels, and speed up prediction by orders of magnitude without significant impact on performance.

Original languageEnglish
Title of host publicationProceedings of the 20th International Conference on Artificial Intelligence and Statistics, AISTATS 2017
PublisherProceedings of Machine Learning Research (PMLR)
Publication statusPublished - 2017
Externally publishedYes
EventInternational Conference on Artificial Intelligence and Statistics 2017 - Fort Lauderdale, United States of America
Duration: 20 Apr 201722 Apr 2017
Conference number: 20th
https://aistats.org/aistats2017// (Website)
https://proceedings.mlr.press/v54/ (Proceedings)

Conference

ConferenceInternational Conference on Artificial Intelligence and Statistics 2017
Abbreviated titleAISTATS 2017
Country/TerritoryUnited States of America
CityFort Lauderdale
Period20/04/1722/04/17
Internet address

Cite this