arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2206.09591cs.CL
收藏

基于结构化知识的领域自适应文本分类

Domain-Adaptive Text Classification with Structured Knowledge from Unlabeled Data

Peking University(北京大学);University of California, Berkeley(加州大学伯克利分校);Sun Yat-sen University(中山大学)

Tian Li, Xiang Chen, Zhen Dong, Weijiang Yu, Yijun Yan, Kurt Keutzer, Shanghang Zhang

摘要:

领域自适应文本分类对于大规模预训练语言模型来说是一个具有挑战性的问题,因为它们通常需要昂贵的额外标记数据来适应新的领域。现有的作品通常无法利用跨域词之间的隐含关系。在本文中,我们提出了一种新的方法,称为领域自适应与结构化知识(DASK),提高领域自适应,利用词级语义关系。DASK首先构建一个知识图谱,以捕获目标领域中的枢轴术语(领域独立词)和非枢轴术语之间的关系。然后在训练过程中,DASK将相关的知识图谱信息注入到源领域文本中。对于下游任务,这些知识注入的文本被馈送到能够处理知识注入的文本数据的BERT变体中。由于知识注入,我们的模型根据非枢轴与枢轴的关系学习非枢轴的域不变特征。DASK通过在使用伪标签进行训练期间通过候选枢轴的极性得分动态推断来确保枢轴具有域不变行为。我们在广泛的跨域情感分类任务上验证了DASK,并观察到20个不同域对的绝对性能比基线提高了高达2.9%。代码将在https://github.com/hikaru-nara/DASK上提供。

英文摘要:

Domain adaptive text classification is a challenging problem for the large-scale pretrained language models because they often require expensive additional labeled data to adapt to new domains. Existing works usually fails to leverage the implicit relationships among words across domains. In this paper, we propose a novel method, called Domain Adaptation with Structured Knowledge (DASK), to enhance domain adaptation by exploiting word-level semantic relationships. DASK first builds a knowledge graph to capture the relationship between pivot terms (domain-independent words) and non-pivot terms in the target domain. Then during training, DASK injects pivot-related knowledge graph information into source domain texts. For the downstream task, these knowledge-injected texts are fed into a BERT variant capable of processing knowledge-injected textual data. Thanks to the knowledge injection, our model learns domain-invariant features for non-pivots according to their relationships with pivots. DASK ensures the pivots to have domain-invariant behaviors by dynamically inferring via the polarity scores of candidate pivots during training with pseudo-labels. We validate DASK on a wide range of cross-domain sentiment classification tasks and observe up to 2.9% absolute performance improvement over baselines for 20 different domain pairs. Code will be made available at https://github.com/hikaru-nara/DASK.

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

↑