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

利用先验知识引导BERT的注意力进行语义文本匹配

Using Prior Knowledge to Guide BERT's Attention in Semantic Textual Matching Tasks

Jilin University(吉林大学);University of North Carolina at Chapel Hill(北卡罗来纳大学教堂山分校)

Tingyu Xia, Yue Wang, Yuan Tian, Yi Chang

摘要:

我们研究了将先验知识纳入基于transformer的深度模型的问题,即,双向编码器表示从Transformers(BERT),以提高其性能的语义文本匹配任务。通过探索和分析BERT在解决这个任务时已经知道的知识,我们可以更好地理解BERT最需要什么特定的知识,以及在哪里最需要。分析进一步激励我们采取不同于大多数现有作品的方法。我们没有使用先验知识来创建新的训练任务来微调BERT,而是直接将知识注入BERT的多头注意力机制。这使我们找到了一种简单而有效的方法,该方法具有快速的训练阶段,因为它使模型免于在主任务以外的其他数据或任务上进行训练。大量的实验表明,所提出的知识增强的BERT是能够不断提高语义文本匹配性能比原来的BERT模型,性能优势是最显着的训练数据是稀缺的。

英文摘要:

We study the problem of incorporating prior knowledge into a deep Transformer-based model,i.e.,Bidirectional Encoder Representations from Transformers (BERT), to enhance its performance on semantic textual matching tasks. By probing and analyzing what BERT has already known when solving this task, we obtain better understanding of what task-specific knowledge BERT needs the most and where it is most needed. The analysis further motivates us to take a different approach than most existing works. Instead of using prior knowledge to create a new training task for fine-tuning BERT, we directly inject knowledge into BERT's multi-head attention mechanism. This leads us to a simple yet effective approach that enjoys fast training stage as it saves the model from training on additional data or tasks other than the main task. Extensive experiments demonstrate that the proposed knowledge-enhanced BERT is able to consistently improve semantic textual matching performance over the original BERT model, and the performance benefit is most salient when training data is scarce.

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

↑