arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.05932cs.CLcs.LG

用于零样本多标签文本分类的元数据诱导对比学习

Metadata-Induced Contrastive Learning for Zero-Shot Multi-Label Text Classification

University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Microsoft(微软);University of California, Los Angeles(加利福尼亚大学洛杉矶分校)

Yu Zhang, Zhihong Shen, Chieh-Han Wu, Boya Xie, Junheng Hao, Ye-Yi Wang, Kuansan Wang, Jiawei Han

AI总结:

本文研究零样本多标签文本分类,提出利用文档元数据诱导对比学习(MICoL)方法,通过文档元数据构建相似文档对训练分类器,在无需标注数据下性能媲美有监督方法并缓解长尾标签问题。

摘要:

大规模多标签文本分类(LMTC)旨在将文档与来自大型候选集的相关标签进行关联。现有的多数LMTC方法依赖大量人工标注的训练数据,这些数据获取成本高昂,且存在长尾标签分布问题(即许多标签在训练集中仅出现几次)。本文研究零样本设置下的LMTC,该设置不需要任何带有标签的标注文档,仅依赖标签的表面名称和描述。为训练一个计算文档与标签之间相似度得分的分类器,我们提出了一种新颖的元数据诱导对比学习(MICoL)方法。与以往基于文本的对比学习技术不同,MICoL利用网络上广泛可用的文档元数据(例如研究论文的作者、发表场所和参考文献),来推导相似的文档-文档对。在两个大规模数据集上的实验结果表明:(1)MICoL显著优于强大的零样本文本分类和对比学习基线方法;(2)MICoL与在10K-200K标注文档上训练的最先进的有监督元数据感知LMTC方法表现相当;(3)MICoL倾向于预测比有监督方法更多的不频繁标签,从而缓解了长尾标签上的性能下降问题。

英文摘要:

Large-scale multi-label text classification (LMTC) aims to associate a document with its relevant labels from a large candidate set. Most existing LMTC approaches rely on massive human-annotated training data, which are often costly to obtain and suffer from a long-tailed label distribution (i.e., many labels occur only a few times in the training set). In this paper, we study LMTC under the zero-shot setting, which does not require any annotated documents with labels and only relies on label surface names and descriptions. To train a classifier that calculates the similarity score between a document and a label, we propose a novel metadata-induced contrastive learning (MICoL) method. Different from previous text-based contrastive learning techniques, MICoL exploits document metadata (e.g., authors, venues, and references of research papers), which are widely available on the Web, to derive similar document-document pairs. Experimental results on two large-scale datasets show that: (1) MICoL significantly outperforms strong zero-shot text classification and contrastive learning baselines; (2) MICoL is on par with the state-of-the-art supervised metadata-aware LMTC method trained on 10K-200K labeled documents; and (3) MICoL tends to predict more infrequent labels than supervised methods, thus alleviates the deteriorated performance on long-tailed labels.

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

↑