[3G1-GS-2g] 機械学習：分類

[3G1-GS-2g-03] Extreme Multi-Label Classification of Images via Multiscale k-Nearest Neighbour

〇Takuma Tanaka^1,3, Akifumi Okuno^2,3, Hidetoshi Shimodaira^1,3 (1. Graduate School of Informatics, Kyoto University, 2. The Institute of Statistical Mathematics, 3. RIKEN Center for Advanced Intelligence Project)

Keywords:multiscale k-nearest neighbour, extreme multi-label classification, image processing, recommender system

We consider the extreme multi-label classification (XMC) problem, which aims at finding positive labels of a query from an extreme variety of labels, e.g., diverse text-tags of images posted on social networking services. k-nearest neighbour (k-NN) can be applied to XMC problem: k-NN predicts the positive label probabilities by averaging the labels of k objects nearest to the query. However, the predicted probability with small k can be unintentionally stick to 0 in many cases, as many labels are often sparse in XMC setting. Conversely, k-NN estimator with large k has large bias, as it leverages the labels of objects distant from the query. For solving these issues, we employ multiscale k-NN, which reduces the bias of the k-NN asymptotically. Through NUS-WIDE dataset experiments, we examine the multiscale k-NN and its modification using a sigmoid function, as a first work of the practical XMC application of the MS-k-NN.