arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1707.04596cs.CLcs.IR
收藏

DocTag2Vec:一种基于嵌入的多标签学习文档标注方法

DocTag2Vec: An Embedding Based Multi-label Learning Approach for Document Tagging

Yahoo Research(雅虎研究院);Airbnb(爱彼迎);University of Minnesota-Twin Cities(明尼苏达大学双城分校)

Sheng Chen, Akshay Soni, Aasish Pappu, Yashar Mehdad

摘要:

在这项工作中,用预定义的标签集合中的相关标签来标记新闻文章或博客文章被称为文档标记。文章的准确标记可以有益于若干下游应用,例如推荐和搜索。在这项工作中,我们提出了一种新颖而简单的方法,称为DocTag 2 Vec来完成这项任务。我们大大扩展了Word 2 Vec和Doc 2 Vec-两个流行的模型,用于学习单词和文档的分布式表示。在DocTag 2 Vec中,我们在训练过程中同时学习单词、文档和标签在联合向量空间中的表示,并采用简单的$k$ -最近邻搜索来预测未见过文档的标签。与以前的多标签学习方法相比,DocTag 2 Vec直接处理原始文本而不是提供的特征向量,此外,还具有标签表示学习和处理新创建标签的能力等优点。为了证明我们的方法的有效性,我们在几个数据集上进行了实验,并对最先进的方法显示了有希望的结果。

英文摘要:

Tagging news articles or blog posts with relevant tags from a collection of predefined ones is coined as document tagging in this work. Accurate tagging of articles can benefit several downstream applications such as recommendation and search. In this work, we propose a novel yet simple approach called DocTag2Vec to accomplish this task. We substantially extend Word2Vec and Doc2Vec---two popular models for learning distributed representation of words and documents. In DocTag2Vec, we simultaneously learn the representation of words, documents, and tags in a joint vector space during training, and employ the simple $k$-nearest neighbor search to predict tags for unseen documents. In contrast to previous multi-label learning methods, DocTag2Vec directly deals with raw text instead of provided feature vector, and in addition, enjoys advantages like the learning of tag representation, and the ability of handling newly created tags. To demonstrate the effectiveness of our approach, we conduct experiments on several datasets and show promising results against state-of-the-art methods.

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

↑