arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.07092cs.LGstat.ML
收藏

多标签分类的在线度量学习

Online Metric Learning for Multi-Label Classification

The University of Sydney(悉尼大学);The University of New South Wales(新南威尔士大学)

Xiuwen Gong, Jiahui Yang, Dong Yuan, Wei Bao

摘要:

现有的在线多标签分类研究,如在线顺序多标签极端学习机(OSML-ELM)和随机梯度下降(SGD),取得了可喜的成绩。然而,这些工作没有考虑标签依赖性,缺乏损失函数的理论分析。因此,我们提出了一种新的在线多标签分类度量学习范式,以填补目前的研究空白。本文首先提出了一种基于$k$ -最近邻($k$ NN)并结合大间隔原则的多标签分类度量。然后,我们将其应用到在线设置中,以获得我们的模型,该模型可以以更高的速度在线处理大量的流数据。具体来说,为了学习新的基于$k$ NN的度量,我们首先将训练数据集中的实例投影到标签空间中,这使得在相同维度上比较实例和标签成为可能。然后,我们将它们同时投影到一个新的低维空间中,这使得我们能够提取实例和标签之间的依赖结构。最后,我们利用大余量和$k$ NN原理,使用有效的优化算法来学习度量。此外,我们提供了理论分析的上界的累积损失,我们的方法。在许多基准多标签数据集上进行的全面实验验证了我们的理论方法,并表明我们提出的在线度量学习(OML)算法优于最先进的方法。

英文摘要:

Existing research into online multi-label classification, such as online sequential multi-label extreme learning machine (OSML-ELM) and stochastic gradient descent (SGD), has achieved promising performance. However, these works do not take label dependencies into consideration and lack a theoretical analysis of loss functions. Accordingly, we propose a novel online metric learning paradigm for multi-label classification to fill the current research gap. Generally, we first propose a new metric for multi-label classification which is based on $k$-Nearest Neighbour ($k$NN) and combined with large margin principle. Then, we adapt it to the online settting to derive our model which deals with massive volume ofstreaming data at a higher speed online. Specifically, in order to learn the new $k$NN-based metric, we first project instances in the training dataset into the label space, which make it possible for the comparisons of instances and labels in the same dimension. After that, we project both of them into a new lower dimension space simultaneously, which enables us to extract the structure of dependencies between instances and labels. Finally, we leverage the large margin and $k$NN principle to learn the metric with an efficient optimization algorithm. Moreover, we provide theoretical analysis on the upper bound of the cumulative loss for our method. Comprehensive experiments on a number of benchmark multi-label datasets validate our theoretical approach and illustrate that our proposed online metric learning (OML) algorithm outperforms state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑