15亿参数的GPT-2被两个CS硕士复制出来了，没有语言建模经验，花了5万美元

torry

机器之心报道

参与：杜伟、张倩

前几天，OpenAI 发布了 7.74 亿参数量的新一版 GPT-2，并表示将在几个月之内发布 15.58 亿参数量的完整版本。但还没等到完整版官宣，有人就已经等不及了，索性自己动手复制了一个 15 亿参数量的 GPT-2，并将其命名为 OpenGPT-2。项目作者是来自布朗大学的两位硕士研究生，他们复制 GPT-2 的花费大约是 5 万美元。

. B( a* x4 k! ]( N( t0 E

; ]( i5 d0 R% O9 F: ~4 d项目的两位作者：Aaron Gokaslan 和 Vanya Cohen 。
- Z- x0 _% d- v3 s* D0 Z. h( J读者可以在 Google Colab 上访问该模型并生成文本。
5 g, g7 ]9 J9 ?/ jGoogle Colab 地址：https://colab.research.google.com/drive/1esbpDOorf7DQJV8GXWON24c-EQrSKOit
" n4 P6 f" p, b8 ^- m
; W& Q8 x( D5 C$ h8 E9 H7 o+ ~模型权重：https://drive.google.com/drive/u/0/folders/1KfK5MXtvgH8C615UUZoKPIUVJYIdJxX1
4 a7 `0 o" c  _. _) r, U5 T作者表示，复制 GPT-2 没有那么难，论文中的很多结果都是他们两人可以复制的，并且二人并没有语言建模的经验。所以，只要你感兴趣（壕），你也能自己复制一份。

$ J$ L+ x' ?+ g4 |! d  L! x7 O2 g
复制方法和成本
4 G6 T, k' k+ _! V8 V% X
1 u5 u+ x# ]( Q. V3 r4 }# q, G& lOpenGPT-2 的实现基于 Grover 模型，通过修改它们的代码库来达到 GPT-2 的语言建模训练目标。由于 Grover 模型是在类似的大型语料库上进行训练，所以很多代码和超参数很容易重复使用。他们也没有对 Grover 的超参数进行大量修改。
9 n0 T: b8 X+ n7 E8 j" S
+ a! e% g- U* f; M至于成本嘛，他们使用自己的代码从零开始训练 GPT-2 模型大约花费了 5 万美元。但需要注意的是，5 万美元只是云计算的估算成本，没有包含更细微的内在成本（在其他效率更低的易用计算资源上训练模型的效果会更差）。

数据集

OpenAI GPT-2 的原始论文中有对清理数据集的详解。在该论文中，Open AI 的研究人员用到了一个名为 WebText 的数据集，其中包含数百万个网页。

论文地址：https://d4mucfpksywv.cloudfront.net/better-language-models/language_models_are_unsupervised_multitask_learners.pdf
1 [8 d# V" Y( t0 q  V' G
+ ?: I% D6 b9 q6 s! Z4 p% O如 WebText 一样，本文中的两位研究者首先解析 Reddit 上 3 个 up-vote 以上的所有链接。同时，他们利用了 Pushshift Reddit scrape 数据集，该数据集集合了持续更新的 Reddit 帖子、评论和有关元数据。然后，作者对这些链接进行过滤，删除那些不太可能包含有用文本或 HTML 的文件类型（即视频文件、PDF 和 CSS 格式文件）的直接链接。
0 G0 L* J) m( K" e0 v0 e: y
' A+ y/ k2 }/ C1 t$ @此外，作者还过滤了网页，以删除被各种评估基准和数据集所使用的 Wikipedia 内容。他们并不能确定自己的过滤方法是否符合 OpenAI 的标准。因此，他们使用 Newspaper Python 库从 HTML 网页上提取文本，然后使用 fastText Python 库只过滤掉英文文本。
* g5 }8 s% w4 g* q9 t: }" j9 ?
, `0 J) Q. v! z& ^' f具体来说，作者使用了 WhatTheLang python Wrapper。他们利用局部敏感哈希（locally sensitive hashing，LSH）删除这些文档。最后，他们将这些文档散列到了 5-gram 的集合中，并删除了相似度阈值大于 0.5 的所有文档。
1 E9 {! a6 [) r$ B, M/ Y6 O( f
作者还从数据集中删除了 token 数少于 128 的文档。这些短文档往往质量较低。作者将这一数据集作为 OpenWebTextCorpus 发布。
0 s7 G  L! @1 L6 M# _
数据集链接：https://skylion007.github.io/OpenWebTextCorpus/

! ?: \4 \- p! C' z在编码数据集时，作者使用了 Radford 等人发布的适用于小模型的 Binary Pattern 编码器。他们还利用 OpenWebText 网页爬取代码库的修订版作为自身数据集集合的起始点。
- g5 U' a7 w, E
+ ?( t6 Z, {  u' X. I/ v从公开发布的 WebText 的 26 万篇文档的集合来看，作者发现所有文档的双字节编码（BPE）长度最小为 40，最大为 1024。而 OpenWebText 的不同之处在于作者将文档长度的最低值设为 128 个 token（替代 BPE），并且不限制文档最大长度。此外，原始 WebTextCorpus 是在这些样本可用之前发布的，因此作者没有使用这些信息来生成清理启发式。
  Z$ Y# o0 j" K
尽管在训练分布上存在差异，作者依然得出了与多数数据集接近的困惑度。

1 z8 J- e4 f1 t+ p- S  e2 Q( Y效果
9 a6 B0 u" J7 A. l
两位作者在 medium 博客中展示了 OpenGPT-2 的生成效果。他们提供给模型的提示为：「Recycling is good for the world. NO! YOU COULD NOT BE MORE WRONG!!」，输入模型之后得到了以下输出：
9 N6 c8 @1 |, q
% [& Q4 h1 R$ Y

6 v- {( _& j6 S" n+ k, R
reddit 网友也根据作者给出的连接进行了测试，并将测试结果与 OpenAI 前几天公布的 7.74 亿参数量版本进行了对比。有人表示：
7 x: _( ~0 J1 r, j: o& }

我使用相同的提示分别测试了 OpenGPT-2 和 OpenAI 的 GPT-2 7.74 亿参数版本，结果还是 OpenAI 的输出效果更好，所以作者的复制过程或许存在一些问题。

当然，也有人认为 OpenGPT-2 的效果更好：

- ^/ x0 X9 P: V% J/ l, f

我认为作者的 OpenGPT-2 效果优于 OpenAI 的 GPT-2 7.74 亿参数版本，但还不足以用来制造令人信服的假新闻。几次尝试输入「Shocking revelation! Vladimir Putin and Donald Trump are」，我得出了以下效果最佳的文本。该文本能够更长时间地保持语义连贯性，语句也更有意义，但还是比较容易被识破。但不可否认，OpenGPT-2 对研究确实很有帮助。
+ K0 A9 p# }" Q+ K  o+ P" e
& A5 F6 m6 d1 F0 @( x但我还有一个疑问，OpenGPT-2 到底只是把它读过的文本随机组合到一起，还是真正地创造出了新文本。

8 N, K* P" L+ y# O: D# ?, U! D5 d+ O/ X/ z
* }0 \: v% b+ H; _/ H

( o0 y" A: c5 V+ `7 C1 Z
+ p* E% _' Q0 `0 T! X7 H& s1 z至于 OpenGPT-2 的生成效果究竟如何，大家可以根据文中提供的链接测试一下。

( A' Z5 h- L3 {* u3 O( z5 D! |: @参考链接：

% R3 A) F7 [- d) `! U8 y
" o8 o' ~' F4 l7 \7 W
https://www.reddit.com/r/MachineLearning/comments/cu9xgi/p_opengpt2_we_replicated_gpt2_because_you_can_too/
* L- W2 _( t5 a4 fhttps://medium.com/@vanya_cohen/opengpt-2-we-replicated-gpt-2-because-you-can-too-45e34e6d36dc
( T6 A5 m  y7 ^; k" b- W- f

+ `1 ]3 E5 Q% I6 |

WAIC 2019 开发者日

将于 8 月 31 日在上海世博中心举办，包含 1 个主单元、4 个分单元、黑客马拉松比赛和开发者诊所互动区。

' f+ I9 ^* \, v9 ?

- l( U6 p& O9 U% j8 C- {; B  a

届时，全球顶尖 AI 专家、技术大牛、知名企业代表以及数千名开发者将齐聚上海，围绕人工智能前沿理论技术和开发实践进行分享与解读。

点击阅读原文，立即报名。

! H2 N5 w( K2 l$ D
来源：http://mp.weixin.qq.com/s?src=11&timestamp=1566630003&ver=1809&signature=lixpkgck3Jci-PJs9nuYQ2-e-AoYyRsK9T2JfY8UTHbrfqeLiSdjvkGmvntU-OVvGe8*llf7XPaBdBKSdmrL7CrALKS1LX-KQs*HOTH8twOu5W3TN*s6mdYxiqgYsckc&new=1
, y+ K9 Y, h, n$ i, F免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！