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

面向长上下文推理的提示压缩方法特性研究

Characterizing Prompt Compression Methods for Long Context Inference

UC Berkeley(加州大学伯克利分校)

Siddharth Jha, Lutfi Eren Erdogan, Sehoon Kim, Kurt Keutzer, Amir Gholami

AI总结:

针对长上下文推理的提示压缩方法缺乏统一对比、结论矛盾的问题,本文通过标准化分析评估三类压缩方法,发现抽取式压缩性能最优,可实现10倍压缩且精度损失极小,token剪枝方法表现不及抽取式压缩,仅在摘要任务有边际提升。

摘要:

长上下文推理在系统层面面临计算与内存需求增加的挑战,在准确性层面也存在长上下文推理能力的难题。近期,学界提出了多种压缩提示以缩短上下文长度的方法。然而,目前鲜有研究通过标准化分析,在不同任务上对比这些已提出的方法,进而导致了相互矛盾的结论。为解决这一问题,本文对不同的提示压缩方法开展了全面的特性分析与评估,具体研究了抽取式压缩、基于摘要的生成式压缩以及token剪枝方法。令人意外的是,我们发现抽取式压缩的表现通常优于所有其他方法,可实现最高10倍的压缩比且准确率下降极小。有趣的是,我们还发现尽管近期有不少相关论断,但token剪枝方法的表现往往落后于抽取式压缩,我们仅在摘要任务上观测到其带来的边际提升。

英文摘要:

Long context inference presents challenges at the system level with increased compute and memory requirements, as well as from an accuracy perspective in being able to reason over long contexts. Recently, several methods have been proposed to compress the prompt to reduce the context length. However, there has been little work on comparing the different proposed methods across different tasks through a standardized analysis. This has led to conflicting results. To address this, here we perform a comprehensive characterization and evaluation of different prompt compression methods. In particular, we analyze extractive compression, summarization-based abstractive compression, and token pruning methods. Surprisingly, we find that extractive compression often outperforms all the other approaches, and enables up to 10x compression with minimal accuracy degradation. Interestingly, we also find that despite several recent claims, token pruning methods often lag behind extractive compression. We only found marginal improvements on summarization tasks.

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

↑