用关键点分段池化提升长文本表示,兼顾局部与全局信息
LMK > CLS: Landmark Pooling for Dense Embeddings
- 将序列分块插入关键点令牌,再对这些令牌均值池化
- 长上下文任务性能显著提升,短任务表现不降反升
- 适合需要长文本理解的检索、分类等应用
表示学习在搜索、聚类、分类和重排序等下游任务中至关重要。当前主流序列编码器通常使用池化操作将变长词元序列压缩为单一向量,最常见的是特殊[CLS]标记或对词元嵌入进行均值池化。本文指出这两种策略存在系统性缺陷:[CLS]倾向于集中初始位置信息,弱化分布证据;均值池化则可能稀释显著局部信号,导致短上下文表现下降。为此,我们提出地标(LMK)池化:将序列分块,在块间插入地标令牌,最终通过均值池化地标令牌嵌入获得表示。该简单机制在不牺牲局部显著特征的前提下,显著提升长上下文外推能力,仅引入少量特殊令牌。实验表明,LMK池化在短上下文检索任务中表现相当,长上下文任务上取得显著提升,是一种实用且可扩展的替代方案。
原文摘要 · Abstract (English)
Representation learning is central to many downstream tasks such as search, clustering, classification, and reranking. State-of-the-art sequence encoders typically collapse a variable-length token sequence to a single vector using a pooling operator, most commonly a special [CLS] token or mean pooling over token embeddings. In this paper, we identify systematic weaknesses of these pooling strategies: [CLS] tends to concentrate information toward the initial positions of the sequence and can under-represent distributed evidence, while mean pooling can dilute salient local signals, sometimes leading to worse short-context performance. To address these issues, we introduce Landmark (LMK) pooling, which partitions a sequence into chunks, inserts landmark tokens between chunks, and forms the final representation by mean-pooling the landmark token embeddings. This simple mechanism improves long-context extrapolation without sacrificing local salient features, at the cost of introducing a small number of special tokens. We empirically demonstrate that LMK pooling matches existing methods on short-context retrieval tasks and yields substantial improvements on long-context tasks, making it a practical and scalable alternative to existing pooling methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。