测试大模型从超长文本中提取序列信息的能力,发现顶尖模型准确率仅63.5%。
Sequential-NIAH: A Needle-In-A-Haystack Benchmark for Extracting Sequential Needles from Long Contexts
- 设计三种生成方式构建含序列信息的长文本数据集
- 最长上下文达128K,测试集准确率最高仅63.5%
- 可作为评估模型长文本理解能力的重要基准
评估大语言模型处理长上下文的能力至关重要,尤其在从其中检索与查询相关的信息时。本文提出Sequential-NIAH基准,专门用于评测模型从长文本中提取序列信息(称为“针”)的能力。该基准包含三种针生成流程:合成时间、真实时间与真实逻辑顺序,上下文长度覆盖8K至128K,共14,000个样本(其中2,000用于测试)。为更可靠地评估模型表现,我们训练了一个评估模型,通过对比生成结果与真实答案的完整性和序列一致性来判断正确性,优于GPT-4或Claude的评估方式。在六个主流LLM上进行实验,结果显示即使最优模型在测试集上的准确率也仅达63.50%。进一步分析表明,随着上下文长度或针数量增加,挑战显著上升,凸显当前模型仍有巨大提升空间。噪声分析验证了基准的可靠性和难度,使Sequential-NIAH成为推动长文本信息抽取研究的重要参考。
原文摘要 · Abstract (English)
Evaluating the ability of large language models (LLMs) to process lengthy contexts is critical, especially for retrieving query-relevant information embedded within them. We introduce Sequential-NIAH, a benchmark specifically designed to evaluate the capability of LLMs to extract sequential information items (known as \emph{needles}) from long contexts. The benchmark includes three needle generation pipelines: synthetic-temporal, real-temporal, and real-logical orders, with context lengths ranging from 8K to 128K, which comprises 14,000 samples (2,000 for testing). To facilitate the evaluation of this benchmark, we trained an evaluation model that assesses the correctness of LLM responses by comparing their completeness and sequential consistency against the ground truth, which provides a more reliable evaluation metric than GPT-4 or Claude. We conducted experiments on six well-known LLMs, revealing that even the best-performing model achieved a maximum accuracy of only 63.50% on test set of this benchmark. Further analysis highlights the growing challenges posed by increasing the context length or the number of needles, underscoring substantial room for improvement of LLMs. Additionally, noise analysis validates the reliability and challenge of the benchmark, making Sequential-NIAH an important reference for advancing research on long text information extraction capabilities of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。