为什么说 The Pile 是好的训练集和评测基准?

更新时间

The Pile 的价值可以从两个角度看:作为训练集,它把 22 个高质量数据集组合成 825 GiB 的多样文本,有助于模型获得跨领域知识和下游泛化能力;作为评测基准,它的 Pile BPB 指标要求模型同时理解书籍、代码仓库、网页、聊天记录以及医学、物理、数学、计算机科学、哲学论文等差异很大的领域,因此能较全面地衡量通用跨领域文本建模能力。是否适合你,取决于你要做的是训练、微调还是评测,以及你能否处理 jsonlines + zstandard 格式的数据。

作为训练集:多样性带来跨领域能力

The Pile 是一个 825 GiB 的开源语言建模数据集,由 22 个较小的、高质量的数据集合并而成。它的训练价值主要来自“多样性”这一设计:

  • 数据来源覆盖多个差异明显的领域,而不是单一网页抓取或单一书籍语料。
  • 官方说明指出,近期工作表明,尤其对大型模型而言,数据来源的多样性可以改善模型的通用跨领域知识,以及下游泛化能力。
  • 在官方评测中,用 The Pile 训练的模型不仅在传统语言建模基准上有适度提升,在 Pile BPB 上也有显著改善。

适用条件:如果你希望模型不只擅长某一类文本,而是对代码、学术论文、网页、对话等多种输入都有基本建模能力,The Pile 的组成方式与这一目标匹配。反过来,如果你的任务高度垂直、只需要某一领域语料,全量使用 The Pile 未必是最经济的选择。

作为评测基准:Pile BPB 考的是跨领域理解

Pile BPB(bits per byte)是 The Pile 提供的评测指标。官方对它的定位很明确:要在 Pile BPB 上取得好成绩,模型必须能够理解许多互不相同的领域,包括书籍、GitHub 仓库、网页、聊天记录,以及医学、物理、数学、计算机科学和哲学论文。

因此,Pile BPB 衡量的是模型在这些领域中的世界知识与推理能力,被官方描述为衡量大型语言模型通用、跨领域文本建模能力的稳健基准。

使用时有两点需要注意:

  • 官方排行榜区分了 Zero-shot 设置,表示并非 Pile 的所有组成部分都出现在训练数据中;排行榜中带 * 的条目表示可能存在测试集重叠。解读分数时应结合这些标注,而不是只看排名。
  • 排行榜展示的是特定时间点的结果,例如 GPT-3(Zero-Shot)为 0.7177、GPT-2(Zero-Shot)为 1.2253,并附有 EleutherAI 2021 的评测代码。这些数字适合作为参照,不宜直接当作当前模型的横向结论。

怎么获取和使用

The Pile 由 Eye 托管,可从官网的 Download 入口获取。数据格式为 jsonlines,并使用 zstandard 压缩。这意味着使用前需要先准备能解压 zstandard 并逐行读取 jsonlines 的处理流程,再按你的训练或评测框架切分和加载。

如果你有使用或评测 The Pile 的模型,官网提供了反馈入口。引用方面,官方要求使用 The Pile 或其任何组成部分时进行引用,给出的条目为 Gao 等人 2020 年的 arXiv 论文《The Pile: An 800GB Dataset of Diverse Text for Language Modeling》(arXiv:2101.00027)。

选择判断

你的目标 The Pile 是否合适 理由
训练通用跨领域语言模型 合适 22 个高质量数据源组合,官方称多样性改善跨领域知识与下游泛化
评测模型的通用文本建模能力 合适 Pile BPB 覆盖书籍、代码、网页、对话及多学科论文
只做单一垂直领域任务 需斟酌 全量数据规模大,垂直任务未必需要全部 22 个组成部分
需要开箱即用的标准格式 需准备 数据为 jsonlines + zstandard 压缩,需自行搭建读取流程

一句话总结:The Pile 的“好”来自多样性与统一评测口径的结合——训练时用多领域数据拓宽模型能力,评测时用 Pile BPB 检验这种能力是否真的跨得过去。