Original Paper: https://arxiv.org/abs/2407.11963

By: Mo LiSongyang ZhangYunxin LiuKai Chen

Abstract:

In evaluating the long-context capabilities of large language models (LLMs), identifying content relevant to a user's query from original long documents is a crucial prerequisite for any LLM to answer questions based on long text.

We present NeedleBench, a framework consisting of a series of progressively more challenging tasks for assessing bilingual long-context capabilities, spanning multiple length intervals (4k, 8k, 32k, 128k, 200k, 1000k, and beyond) and different depth ranges, allowing the strategic insertion of critical data points in different text depth zones to rigorously test the retrieval and reasoning capabilities of models in diverse contexts.

We use the NeedleBench framework to assess how well the leading open-source models can identify key information relevant to the question and apply that information to reasoning in bilingual long texts.

Furthermore, we propose the Ancestral Trace Challenge (ATC) to mimic the complexity of logical reasoning challenges that are likely to be present in real-world long-context tasks, providing a simple method for evaluating LLMs in dealing with complex long-context situations.

Our results suggest that current LLMs have significant room for improvement in practical long-context applications, as they struggle with the complexity of logical reasoning challenges that are likely to be present in real-world long-context tasks.


Summary Notes

https://arxiv.org/html/2407.11963v1/x2.png

                                                     Figure: *NeedleBench* Framework

Introduction

In the ever-evolving landscape of artificial intelligence, Large Language Models (LLMs) have emerged as powerful tools capable of processing vast amounts of text.

These capabilities are particularly crucial for applications such as legal document retrieval, academic research, and business intelligence.