长文本评测指引¶
介绍¶
虽然大语言模型(LLM)如GPT-4在处理自然语言任务已经展现出明显的优势,但目前的开源模型大多只能处理数千个token长度以内的文本,这限制了模型阅读书籍、撰写文本摘要等需要处理长文本的能力。为了探究模型在应对长文本能力时的表现,我们采用L-Eval和LongBench两个长文本数据集来测试模型长文本能力。
现有算法及模型¶
在处理长文本输入时,推理时间开销和灾难性遗忘是大模型面临的两大主要挑战。最近,大量研究致力于扩展模型长度,这些研究集中于以下三个改进方向。
注意力机制。这些方法的最终目的多为减少query-key对的计算开销,但可能对下游任务的效果产生影响。
输入方法。部分研究将长文本输入分块或将部分已有文本段重复输入模型以增强模型处理长文本能力,但这些方法只对部分任务有效,难以适应多种下游任务。
位置编码。这部分研究包括RoPE, ALiBi,位置插值等,在长度外推方面展现出了良好的效果。这些方法已经被用于训练如ChatGLM2-6b-32k和LongChat-32k等长文本模型。
首先,我们介绍一些流行的位置编码算法。
RoPE¶
RoPE是一种在Transformer中注入位置信息的位置嵌入方法。它使用旋转矩阵对绝对位置进行编码,并同时在自注意力公式中融入显式的相对位置依赖关系。下图是RoPE机制的一个示例。
RoPE具有一些有价值的特性,例如可以扩展到任意序列长度、随着相对距离增加而减弱的token间依赖关系以及为线性自注意力提供相对位置编码的能力。
RoPE被应用于许多LLM模型,包括LLaMA、LLaMA 2和Vicuna-7b-v1.5-16k。
ALiBi¶
尽管RoPE和其他替代原始位置编码的方法(如T5 bias)改善了外推能力,但它们的速度比原始方法慢得多,并且使用了额外的内存和参数。因此,作者引入了具有线性偏置的注意力(ALiBi)来促进高效的外推。
对于长度为L的输入子序列,注意力子层在每个head中计算第i个query
的注意力分数,给定前i个键
其中d是head维度。
ALiBi通过与相关key和query之间的距离成比例的线性递减惩罚来负向偏置注意力分数。它唯一的修改是在query-key点积之后,在其中添加了一个静态的、非学习的偏置。
其中m是在训练之前固定的head特定的斜率。
ALiBi去除了位置嵌入部分,它与原始位置编码方法一样快。它被用于包括mpt-7b-storywriter在内的大语言模型,该模型能够处理非常长的输入。
位置插值(PI)¶
许多现有的预训练LLM模型包括LLaMA,使用具有弱外推性质(例如RoPE)的位置编码。作者提出了一种位置插值方法,它可以轻松地实现非常长的上下文窗口,同时相对保持模型在其原始上下文窗口大小内的处理质量。
位置插值的关键思想是直接缩小位置索引,使得最大位置索引与预训练阶段的先前上下文窗口限制相匹配。换句话说,为了容纳更多的输入token,该算法在相邻的整数位置插值位置编码,利用位置编码可以应用于非整数位置的优势,它不需要在训练位置之外进行外推从而导致灾难性值的出现。该算法只需要很少的微调时间,模型就能完全适应大大扩展的上下文窗口。
下图展现了位置插值方法的机制。图中左下方说明了位置插值方法,它将位置索引(蓝色和绿色点)本身从[0, 4096]缩小到[0, 2048],从而使它们位于预训练范围内。
位置插值使得基于ChatGLM2-6B的ChatGLM2-6B-32k模型能够处理32k的上下文窗口大小。
接下来,我们将介绍一些我们纳入评测范围的模型。
XGen-7B-8k¶
XGen-7B-8k是使用标准的注意力机制训练的,训练文本最长为8k,总计1.5T个token。为了减少训练时间开销, XGen-7B-8k在不同阶段逐步增加输入文本长度。首先, 模型在序列长度为2k的文本上训练总计800B的token, 随后在长度为4k的文本上训练总计400B的token, 最后, 在长度为8k的文本上训练总计300B的token。
Vicuna-7b-v1.5-16k¶
Vicuna-7b-v1.5-16k是从LLaMA 2微调而来的,它使用了有监督指导微调和线性RoPE扩展方法。训练数据量约为125K个对话,这些对话是从ShareGPT收集而来的。ShareGPT是一个用户可以分享他们与ChatGPT对话的网站。这些对话被打包成每个包含16k个token的序列。
LongChat-7b-v1.5-32k¶
LongChat-7b-v1.5-32k也是从LLaMA 2模型微调得到, LLaMA 2模型最初使用4k的上下文长度进行预训练。LongChat-7b-v1.5-32k的第一步是压缩RoPE。由于LLaMA 2模型在预训练阶段没有训练输入位置大于4096的token,LongChat将位置大于4096的token压缩到0到4096之间。第二步是在对话数据上微调LongChat模型。在这一步中,LongChat使用FastChat中的步骤对数据进行清洗,并将对话文本截断到模型的最大长度。
ChatGLM2-6B-32k¶
ChatGLM2-6B-32k进一步增强了ChatGLM2-6B的长文本能力。它采用位置插值方法,在对话对齐过程中使用32k上下文长度进行训练,因此ChatGLM2-6B-32k能够更好地处理长达32K的上下文长度。
L-Eval¶
L-Eval是由OpenLMLab构建的一个长文本数据集,由18个子任务组成,其中包含法律、经济、科技等各个领域的文本。数据集总计411篇文档,超过2000条测例,文档平均长度为7217词。该数据集将子任务划分为close-ended和open-ended两类,5个close-ended任务使用完全匹配(Exact Match)作为评测标准,而13个open-ended任务则使用Rouge分数评测。
LongBench¶
LongBench是由THUDM构建的长文本数据集,由21个子任务构成,总计4750条测例。该数据集是第一个包含中英双语的长文本数据集,其中英语文本长度平均为6711词,中文文本平均长度为13386字。21个子任务分为以下6种类型,对模型各方面能力提供了较为全面的评测。
评测方法¶
由于不同模型能够接受的最大输入长度不同,为了更加公平地比较这些大模型,在输入长度超过模型最大输入限制时,我们将裁剪输入文本的中间部分,从而避免提示词缺失的情况。
长文本能力榜单¶
在LongBench和L-Eval能力榜单中,我们选取各模型在子任务上排名的平均值 (排名数值越低越好) 作为标准。可以看到GPT-4和GPT-3.5-turbo-16k在长文本任务中仍然占据领先地位,而例如ChatGLM2-6B-32k在基于ChatGLM2-6B使用位置插值后在长文本能力方面也有明显提升。
原始分数如下所示。
L-Eval |
GPT-4 |
GPT-3.5-turbo-16k |
chatglm2-6b-32k |
vicuna-7b-v1.5-16k |
xgen-7b-8k |
internlm-chat-7b-8k |
longchat-7b-v1.5-32k |
chatglm2-6b |
---|---|---|---|---|---|---|---|---|
coursera |
61.05 |
50 |
45.35 |
26.74 |
33.72 |
40.12 |
27.91 |
38.95 |
gsm100 |
92 |
78 |
27 |
11 |
8 |
19 |
5 |
8 |
quality |
81.19 |
62.87 |
44.55 |
11.39 |
33.66 |
45.54 |
29.7 |
41.09 |
tpo |
72.93 |
74.72 |
56.51 |
17.47 |
44.61 |
60.59 |
17.1 |
56.51 |
topic_retrieval |
100 |
79.33 |
44.67 |
24.67 |
1.33 |
0 |
25.33 |
1.33 |
financialqa |
53.49 |
50.32 |
35.41 |
44.59 |
39.28 |
25.09 |
34.07 |
17.82 |
gov_report |
50.84 |
50.48 |
42.97 |
48.17 |
38.52 |
31.29 |
36.52 |
41.88 |
legal_contract_qa |
31.23 |
27.97 |
34.21 |
24.25 |
21.36 |
19.28 |
13.32 |
17.59 |
meeting_summ |
31.44 |
33.54 |
29.13 |
28.52 |
27.96 |
17.56 |
22.32 |
15.98 |
multidocqa |
37.81 |
35.84 |
28.6 |
26.88 |
24.41 |
22.43 |
21.85 |
19.66 |
narrativeqa |
25.87 |
25.73 |
18.24 |
20.58 |
16.87 |
13.81 |
16.87 |
1.16 |
nq |
67.36 |
66.91 |
41.06 |
36.44 |
29.43 |
16.42 |
35.02 |
0.92 |
news_summ |
34.52 |
40.41 |
32.72 |
33.98 |
26.87 |
22.48 |
30.33 |
29.51 |
paper_assistant |
42.26 |
41.76 |
34.59 |
35.83 |
25.39 |
28.25 |
30.42 |
30.43 |
patent_summ |
48.61 |
50.62 |
46.04 |
48.87 |
46.53 |
30.3 |
41.6 |
41.25 |
review_summ |
31.98 |
33.37 |
21.88 |
29.21 |
26.85 |
16.61 |
20.02 |
19.68 |
scientificqa |
49.76 |
48.32 |
31.27 |
31 |
27.43 |
33.01 |
20.98 |
13.61 |
tvshow_summ |
34.84 |
31.36 |
23.97 |
27.88 |
26.6 |
14.55 |
25.09 |
19.45 |
LongBench |
GPT-4 |
GPT-3.5-turbo-16k |
chatglm2-6b-32k |
longchat-7b-v1.5-32k |
vicuna-7b-v1.5-16k |
internlm-chat-7b-8k |
chatglm2-6b |
xgen-7b-8k |
---|---|---|---|---|---|---|---|---|
NarrativeQA |
31.2 |
25.79 |
19.27 |
19.19 |
23.65 |
12.24 |
13.09 |
18.85 |
Qasper |
42.77 |
43.4 |
33.93 |
30.36 |
31.45 |
24.81 |
22.52 |
20.18 |
MultiFieldQA-en |
55.1 |
54.35 |
45.58 |
44.6 |
43.38 |
25.41 |
38.09 |
37 |
MultiFieldQA-zh |
64.4 |
61.92 |
52.94 |
32.35 |
44.65 |
36.13 |
37.67 |
14.7 |
HotpotQA |
59.85 |
52.49 |
46.41 |
34.43 |
34.17 |
27.42 |
27.35 |
28.78 |
2WikiMQA |
67.52 |
41.7 |
33.63 |
23.06 |
20.45 |
26.24 |
22.83 |
20.13 |
Musique |
37.53 |
27.5 |
21.57 |
12.42 |
13.92 |
9.75 |
7.26 |
11.34 |
DuReader (zh) |
38.65 |
29.37 |
38.53 |
20.25 |
20.42 |
11.11 |
17.18 |
8.57 |
GovReport |
32.09 |
29.92 |
32.47 |
29.83 |
29.27 |
18.38 |
22.86 |
23.37 |
QMSum |
24.37 |
23.67 |
23.19 |
22.71 |
23.37 |
18.45 |
21.23 |
21.12 |
Multi_news |
28.52 |
27.05 |
25.12 |
26.1 |
27.83 |
24.52 |
24.7 |
23.69 |
VCSUM (zh) |
15.54 |
16.88 |
15.95 |
13.46 |
15.76 |
12.91 |
14.07 |
0.98 |
TREC |
78.5 |
73.5 |
30.96 |
29.23 |
32.06 |
39 |
24.46 |
29.31 |
TriviaQA |
92.19 |
92.75 |
80.64 |
64.19 |
46.53 |
79.55 |
64.19 |
69.58 |
SAMSum |
46.32 |
43.16 |
29.49 |
25.23 |
25.23 |
43.05 |
20.22 |
16.05 |
LSHT (zh) |
41.5 |
34.5 |
22.75 |
20 |
24.75 |
20.5 |
16 |
18.67 |
Passage Count |
8.5 |
3 |
3 |
1 |
3 |
1.76 |
3 |
1 |
PassageRetrieval-en |
75 |
73 |
57.5 |
20.5 |
16.5 |
7 |
5.5 |
12 |
PassageRetrieval-zh |
96 |
82.5 |
58 |
15 |
21 |
2.29 |
5 |
3.75 |
LCC |
59.25 |
53.49 |
53.3 |
51.46 |
49.3 |
49.32 |
46.59 |
44.1 |
RepoBench-P |
55.42 |
55.95 |
46.66 |
52.18 |
41.49 |
35.86 |
41.97 |
41.83 |