Label Filters for Large Scale Multilabel Classification

被引:0
作者
Niculescu-Mizil, Alexandru [1 ]
Abbasnejad, Ehsan [2 ]
机构
[1] NEC Labs Amer, Princeton, NJ 08540 USA
[2] Australian Natl Univ, Canberra, ACT 0200, Australia
来源
ARTIFICIAL INTELLIGENCE AND STATISTICS, VOL 54 | 2017年 / 54卷
关键词
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页码:1448 / 1457
页数:10
相关论文
共 29 条
  • [1] [Anonymous], 2009, Proc. of Neural Information Processing Systems
  • [2] [Anonymous], 2009, CVPR09
  • [3] [Anonymous], 2011, Advances in Neural Information Processing Systems
  • [4] [Anonymous], 2012, P 18 ACM SIGKDD INT
  • [5] [Anonymous], 2013, P 30 INT C INT C MAC
  • [6] [Anonymous], 2010, Advances in Neural Information Processing Systems
  • [7] [Anonymous], 2013, International Conference on Machine Learning
  • [8] [Anonymous], 2012, ADV NEURAL INFORM PR, DOI DOI 10.1055/S-0031-1291042
  • [9] [Anonymous], ABS14127479 CORR
  • [10] DiSMEC - Distributed Sparse Machines for Extreme Multi-label Classification
    Babbar, Rohit
    Schoelkopf, Bernhard
    [J]. WSDM'17: PROCEEDINGS OF THE TENTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2017, : 721 - 729