中文

无均值特征:上下文压缩的简单而强大的基线

计算与语言 2026-05-12 v2 人工智能 机器学习

摘要

上下文压缩通过用较短的预计算表示取代长度较长的输入来降低 Transformer 推理成本。这一技术为检索增强生成 (RAG) 带来了显著益处,并吸引了日益增长的研究关注。然而,由于评估不一致且基线方法不统一,进展难以衡量。我们设计了一个标准且易于复现的评估套件 BenchPress,以及用于英文阅读理解的简单高性能基线。BenchPress 支持跨模型规模、数据集、压缩比率以及从短 (<1K<1K token) 到中等 (<8K<8K token) 上下文的基准测试。虽然该套件适用于任何压缩范式,但我们的基线针对软上下文压缩。我们建立了两个简单且强大的基线,显著优于广泛使用的因果压缩-标记方法:均值池化和双向压缩-标记变体。我们的结果表明,在计算压缩表示时使用双向注意力的好处,简单池化也是一种具有表达力的压缩算子。

关键词

引用

@article{arxiv.2510.20797,
  title  = {No Mean Feat: Simple, Strong Baselines for Context Compression},
  author = {Yair Feldman and Yoav Artzi},
  journal= {arXiv preprint arXiv:2510.20797},
  year   = {2026}
}

备注

Code available at https://github.com/lil-lab/benchpress