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

不要根据最后一层来判断语言模型:具有分层注意力池的对比学习

Don't Judge a Language Model by Its Last Layer: Contrastive Learning with Layer-Wise Attention Pooling

Korea University(高丽大学);George Washington University(乔治华盛顿大学)

Dongsuk Oh, Yejin Kim, Hodong Lee, H. Howie Huang, Heuiseok Lim

摘要:

最近的预训练语言模型(PLM)通过学习语言特征和上下文化的句子表示,在许多自然语言处理任务上取得了巨大的成功。由于在PLM的堆叠层中捕获的属性没有被清楚地识别,因此通常首选直接的方法,例如嵌入最后一层,以从PLM中导出句子表示。本文介绍了基于注意力的池化策略,该策略使模型能够保留在每一层中捕获的逐层信号,并为下游任务学习消化的语言特征。对比学习目标可以使逐层注意力池适应无监督和有监督的方式。它导致预训练嵌入的各向异性空间正则化,并且更加均匀。我们评估我们的模型上的标准语义文本相似性(STS)和语义搜索任务。因此,我们的方法提高了基本对比学习BERT_base和变体的性能。

英文摘要:

Recent pre-trained language models (PLMs) achieved great success on many natural language processing tasks through learning linguistic features and contextualized sentence representation. Since attributes captured in stacked layers of PLMs are not clearly identified, straightforward approaches such as embedding the last layer are commonly preferred to derive sentence representations from PLMs. This paper introduces the attention-based pooling strategy, which enables the model to preserve layer-wise signals captured in each layer and learn digested linguistic features for downstream tasks. The contrastive learning objective can adapt the layer-wise attention pooling to both unsupervised and supervised manners. It results in regularizing the anisotropic space of pre-trained embeddings and being more uniform. We evaluate our model on standard semantic textual similarity (STS) and semantic search tasks. As a result, our method improved the performance of the base contrastive learned BERT_base and variants.

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

↑