arXiv:2502.18915cs.CLcs.AI2025-02被引 1

通过早期丢弃噪声上下文,提升大模型生成效率与质量

END: Early Noise Dropping for Efficient and Effective Context Denoising

  • 在模型早期层用线性探测器识别有用与噪声内容
  • 在多个数据集上显著提升性能并降低计算开销
  • 无需微调,适用于长/短上下文任务场景

大型语言模型在自然语言处理任务中表现优异,但输入序列中的无关或噪声信息会降低输出质量,影响检索增强生成、表格问答和上下文学习等场景。我们发现,大模型在生成标记前的早期层即可隐式判断输入是否包含有用信息。基于此,提出无需微调的早期噪声丢弃(END)方法:将输入分块,利用模型早期层的线性探测器区分信息性与噪声块,并提前丢弃后者,从而保留关键信息、减少干扰、降低计算开销。大量实验表明,END 在多个大模型和数据集上均显著提升性能与效率。此外,通过探测器分析模型对输入的隐式理解,进一步揭示了大模型内部上下文推理机制。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated remarkable performance across a wide range of natural language processing tasks. However, they are often distracted by irrelevant or noisy context in input sequences that degrades output quality. This problem affects both long- and short-context scenarios, such as retrieval-augmented generation, table question-answering, and in-context learning. We reveal that LLMs can implicitly identify whether input sequences contain useful information at early layers, prior to token generation. Leveraging this insight, we introduce Early Noise Dropping (\textsc{END}), a novel approach to mitigate this issue without requiring fine-tuning the LLMs. \textsc{END} segments input sequences into chunks and employs a linear prober on the early layers of LLMs to differentiate between informative and noisy chunks. By discarding noisy chunks early in the process, \textsc{END} preserves critical information, reduces distraction, and lowers computational overhead. Extensive experiments demonstrate that \textsc{END} significantly improves both performance and efficiency across different LLMs on multiple evaluation datasets. Furthermore, by investigating LLMs' implicit understanding to the input with the prober, this work also deepens understanding of how LLMs do reasoning with contexts internally.

大模型上下文去噪高效推理

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