arXiv:2502.13063cs.CLcs.LG2025-02ACL被引 27

用优化方法将1568个词元压缩进单一向量,突破传统压缩瓶颈。

Cramming 1568 Tokens into a Single Vector and Back Again: Exploring the Limits of Embedding Space Capacity

  • 用样本级优化替代编码器,实现高压缩比
  • 最高压缩比达1500倍,远超现有方法的10倍上限
  • 压缩极限由信息不确定性决定,非输入长度限制

近期研究致力于将词元序列压缩为更短的实值向量,以替代词嵌入或键值缓存,主要目标是降低计算量而非存储字节数。尽管使用强大模型作为编码器,最大无损压缩比通常不超过10倍。这一现象令人费解,因为理论上大尺寸实值向量的信息容量远超当前表现,即使在16位精度下也如此。本文通过用每样本优化过程替代编码器,探索压缩极限,发现压缩比可达1500倍,揭示了现有方案与理论潜力之间两个数量级的差距。进一步实验证明,压缩极限并非由输入长度决定,而是由该序列在无条件情况下的交叉熵损失所反映的不确定性决定。这表明输入嵌入的理论容量与实际利用间存在巨大鸿沟,为模型设计优化提供了广阔空间。

原文摘要 · Abstract (English)

A range of recent works addresses the problem of compression of sequence of tokens into a shorter sequence of real-valued vectors to be used as inputs instead of token embeddings or key-value cache. These approaches are focused on reduction of the amount of compute in existing language models rather than minimization of number of bits needed to store text. Despite relying on powerful models as encoders, the maximum attainable lossless compression ratio is typically not higher than x10. This fact is highly intriguing because, in theory, the maximum information capacity of large real-valued vectors is far beyond the presented rates even for 16-bit precision and a modest vector size. In this work, we explore the limits of compression by replacing the encoder with a per-sample optimization procedure. We show that vectors with compression ratios up to x1500 exist, which highlights two orders of magnitude gap between existing and practically attainable solutions. Furthermore, we empirically show that the compression limits are determined not by the length of the input but by the amount of uncertainty to be reduced, namely, the cross-entropy loss on this sequence without any conditioning. The obtained limits highlight the substantial gap between the theoretical capacity of input embeddings and their practical utilization, suggesting significant room for optimization in model design.

压缩嵌入空间优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。