arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1609.03426cs.LG
收藏

具有可证明保证的多标记学习

Multi-Label Learning with Provable Guarantee

Sayantan Dasgupta

摘要:

在这里,我们研究了大型文本语料库的学习标签的问题,其中每个文本可以被分配一个可变数量的标签。当标签维度很小时,这个问题可能看起来微不足道,并且可以使用一系列one-vs-all分类器轻松解决。然而,随着标签维数增加到几千,参数空间变得非常大,不再可能使用one-vs-all技术。在这里,我们提出了一个模型的基础上的高阶矩的语料库中,以及标签和词之间的交叉矩的多标签预测。我们的模型提供了保证的估计参数的收敛界。此外,我们的模型只需要三次通过训练数据集来提取参数,从而产生高度可扩展的算法,该算法可以使用具有16 GB RAM的单个处理器的标称资源来训练由数百万文档和数十万标签组成的GB数据。我们的模型在大规模数据集上实现了10倍-15倍的速度提升,同时与现有的基准算法相比具有竞争力的性能。

英文摘要:

Here we study the problem of learning labels for large text corpora where each text can be assigned a variable number of labels. The problem might seem trivial when the label dimensionality is small and can be easily solved using a series of one-vs-all classifiers. However, as the label dimensionality increases to several thousand, the parameter space becomes extremely large, and it is no longer possible to use the one-vs-all technique. Here we propose a model based on the factorization of higher order moments of the words in the corpora, as well as the cross moment between the labels and the words for multi-label prediction. Our model provides guaranteed convergence bounds on the estimated parameters. Further, our model takes only three passes through the training dataset to extract the parameters, resulting in a highly scalable algorithm that can train on GB's of data consisting of millions of documents with hundreds of thousands of labels using a nominal resource of a single processor with 16GB RAM. Our model achieves 10x-15x order of speed-up on large-scale datasets while producing competitive performance in comparison with existing benchmark algorithms.

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

↑