用句级主成分分析优化静态词向量,提升句子语义表示效果
Static Word Embeddings for Sentence Semantic Representation
- 从句向量模型提取词向量,再通过句级主成分分析改进
- 平均词向量即可表征句子,在多个任务上超越现有静态模型
- 能剔除无关语义分量,按词重要性调整向量长度
我们提出一种针对句子语义表征优化的静态词向量。首先从预训练的 Sentence Transformer 中提取词向量,再通过句级主成分分析进行改进,并结合知识蒸馏或对比学习。推理时仅需对词向量求平均,计算开销极低。在单语和跨语言任务上评估,结果表明该模型在句子语义任务中显著优于现有静态模型,甚至在文本嵌入基准上超过基础版 Sentence Transformer(SimCSE)。最后的分析显示,该方法成功移除了与句子语义关联不高的词向量分量,并根据词对句子语义的影响调整向量范数。
原文摘要 · Abstract (English)
We propose new static word embeddings optimised for sentence semantic representation. We first extract word embeddings from a pre-trained Sentence Transformer, and improve them with sentence-level principal component analysis, followed by either knowledge distillation or contrastive learning. During inference, we represent sentences by simply averaging word embeddings, which requires little computational cost. We evaluate models on both monolingual and cross-lingual tasks and show that our model substantially outperforms existing static models on sentence semantic tasks, and even surpasses a basic Sentence Transformer model (SimCSE) on a text embedding benchmark. Lastly, we perform a variety of analyses and show that our method successfully removes word embedding components that are not highly relevant to sentence semantics, and adjusts the vector norms based on the influence of words on sentence semantics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。