Same Task, More Tokens: the Impact of Input Length on the Reasoning Performance of Large Language Models

📄 arXiv: 2402.14848v2 📥 PDF

作者: Mosh Levy, Alon Jacoby, Yoav Goldberg

分类: cs.CL, cs.AI

发布日期: 2024-02-19 (更新: 2024-07-10)

备注: Accepted to ACL 2024


💡 一句话要点

提出新框架评估输入长度对LLM推理性能的影响

🎯 匹配领域: 支柱九:具身大模型 (Embodied Foundation Models)

关键词: 大型语言模型 输入长度 推理性能 问答系统 性能评估

📋 核心要点

  1. 现有大型语言模型在不同输入长度下的性能一致性尚不明确,导致推理能力的评估存在挑战。
  2. 论文提出了一种新颖的问答推理框架,通过多版本样本的不同填充长度、类型和位置来隔离输入长度的影响。
  3. 研究发现,LLMs在较短输入长度下的推理性能显著下降,且传统的下一个词预测指标与推理性能呈负相关。

📝 摘要(中文)

本文探讨了输入长度扩展对大型语言模型(LLMs)能力的影响。尽管LLMs在近期取得了显著进展,但其在不同输入长度下的性能一致性尚不明确。我们通过引入一种新颖的问答推理框架,专门评估输入长度的影响。研究发现,在输入长度远低于技术最大值时,LLMs的推理性能显著下降,并且这种下降趋势在我们数据集的每个版本中均有体现,尽管强度不同。此外,传统的下一个词预测指标与LLMs在推理数据集上的表现呈负相关。我们分析了结果并识别出失败模式,为未来研究提供了有价值的指导,可能为解决LLMs观察到的局限性提供策略。

🔬 方法详解

问题定义:本文旨在解决大型语言模型在不同输入长度下推理性能不一致的问题。现有方法未能充分评估输入长度对模型性能的影响,导致推理能力的评估存在盲点。

核心思路:论文通过引入一种新颖的问答推理框架,设计多版本样本以不同方式扩展输入长度,从而系统性地评估输入长度对LLMs推理性能的影响。

技术框架:整体架构包括数据预处理、样本生成、模型训练和性能评估四个主要模块。数据预处理阶段负责生成不同长度的输入样本,样本生成模块则通过填充不同长度和类型的内容来创建多版本数据集。模型训练阶段使用这些样本进行训练,最后在性能评估阶段分析模型在不同输入长度下的推理表现。

关键创新:最重要的技术创新在于提出了针对输入长度影响的专门评估框架,通过多版本样本的设计,系统性地揭示了输入长度对LLMs推理性能的显著影响,这与现有方法的单一输入长度评估方式本质上不同。

关键设计:在实验中,采用了不同的填充策略和位置,以确保输入长度的多样性。同时,使用了多种性能评估指标,特别关注下一个词预测与推理性能之间的关系,揭示了其负相关性。实验中还考虑了不同模型架构的影响,以确保结果的全面性。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

实验结果显示,在输入长度显著低于技术最大值时,LLMs的推理性能出现明显下降,且这种下降在不同版本的数据集中表现出不同的强度。此外,传统的下一个词预测指标与推理性能呈负相关,为未来研究提供了新的视角。

🎯 应用场景

该研究的潜在应用领域包括自然语言处理、问答系统和对话生成等。通过深入理解输入长度对LLMs推理性能的影响,研究结果可以为模型设计和优化提供指导,帮助开发更高效的语言理解系统,提升实际应用中的性能表现。

📄 摘要(原文)

This paper explores the impact of extending input lengths on the capabilities of Large Language Models (LLMs). Despite LLMs advancements in recent times, their performance consistency across different input lengths is not well understood. We investigate this aspect by introducing a novel QA reasoning framework, specifically designed to assess the impact of input length. We isolate the effect of input length using multiple versions of the same sample, each being extended with padding of different lengths, types and locations. Our findings show a notable degradation in LLMs' reasoning performance at much shorter input lengths than their technical maximum. We show that the degradation trend appears in every version of our dataset, although at different intensities. Additionally, our study reveals that the traditional metric of next word prediction correlates negatively with performance of LLMs' on our reasoning dataset. We analyse our results and identify failure modes that can serve as useful guides for future research, potentially informing strategies to address the limitations observed in LLMs.