当前位置：首页 > article >正文

Re62：读论文 GPT-2 Language Models are Unsupervised Multitask Learners

article 2025/1/6 17:47:57

诸神缄默不语-个人CSDN博文目录
诸神缄默不语的论文阅读笔记和分类

论文全名：Language Models are Unsupervised Multitask Learners
论文下载地址：https://cdn.openai.com/better-language-models/language_models_are_unsupervised_multitask_learners.pdf

官方博文：
Better language models and their implications
GPT-2: 6-month follow-up
GPT-2: 1.5B release

官方GitHub项目：openai/gpt-2: Code for the paper “Language Models are Unsupervised Multitask Learners”
对代码的整理和用代码实现GPT-2的解决方案请参考本文第4章

本文是OpenAI于2019年推出的GPT-2论文，核心思想是用无监督思路构建的语言模型来直接解决NLP问题，做一个通用大模型而非领域专家。
GPT-2本身也是个纯英文的模型。

Transformer基础可参考我撰写的博文：Transformer原理纯享版

GPT-1：Re45：读论文 GPT-1 Improving Language Understanding by Generative Pre-Training

GPT-2是因果/单向LM，预训练目标是通过前文预测下一个单词。

huggingface提供的在线试用GPT-2-large的平台：https://transformer.huggingface.co/doc/gpt2-large
打字后按Tab获得提示。
在这里插入图片描述

文章目录

1. 研究背景
2. 模型
3. 实验
- 3.1 数据集
- 3.2 实验结果
- - LM实验结果
  - 下游任务实验结果
  - Generalization vs Memorization
  - OOD生成样例
4. 复现
- 4.1 官方代码
- 4.2 基于transformers实现GPT-2
- - 4.2.1 基于GPT2LMHeadModel + PyTorch原生框架实现S2S任务微调和推理
  - 4.2.2 基于GPT2LMHeadModel + transformers.Trainer微调GPT-2
5. 本文撰写过程中参考的其他网络资料

1. 研究背景

现在的有监督方法的缺点：有监督模型不鲁棒，只能在IID测试集上表现良好。当前已经开始研究多任务通用模型，但多任务学习表现不好，而且要求高标注量。

现在的预训练方法还需要有监督微调。

2. 模型

语言模型：
在这里插入图片描述

在这里插入图片描述

可以用文本生成任务来建模所有NLP任务，这种思路统一了无监督和有监督任务的形式。

不同大小的GPT-2：
在这里插入图片描述

tokenization
Byte Pair Encoding (BPE) 改造版：原始BPE是贪心，这导致重复相似词出现，导致非最优和占空间。本文改为不让它在字节序列之间合并。
Layer normalization：加到每个sub-block输入，和最后一个self-attention block后
词表、上下文长度、batchsize增大
学习率是手动调整的，目标使测试集（5% WebText）perplexity最低
模型仍未收敛
位置向量
绝对位置编码，默认pad到右边。

3. 实验

3.1 数据集

语料：WebText
来源于爬虫

在预训练时删掉了非英语网页（所以翻译效果不好也很正常）

语料中蕴含的翻译知识：
在这里插入图片描述

下游多任务：
阅读理解：CoQA
翻译：WMT-14 Fr-En
摘要：CNN and Daily Mail
QA：Natural Questions

3.2 实验结果

LM实验结果

基本上都是模型尺寸越大，效果越好。

在这里插入图片描述

CBT：
在这里插入图片描述

LAMBADA

Winograd Schema Challenge：commonsense reasoning
在这里插入图片描述

在这里插入图片描述

下游任务实验结果

零样本多任务，模型尺寸越大，效果越好：
在这里插入图片描述

生成摘要的prompt：TL;DR:
在这里插入图片描述

翻译给出少样本上下文（english sentence = french sentence）和prompt（english sentence =）

QA：
在这里插入图片描述
可以看出GPT-2置信度高的样本准确率确实高。

Generalization vs Memorization

训练集与测试集重复度较低，所以可以认为模型是真的实现了泛化，而不是单纯的记忆。

语料训练集与测试集的重复程度：（第一行是原始数据的训练集，第二行是WebText的训练集）
在这里插入图片描述
文字描述的下游数据上的重复率略。

这个是通过8-gram布隆过滤器计算的，具体略。

8-gram覆盖率

在这里插入图片描述

OOD生成样例

在这里插入图片描述

更多示例见论文附录和官方博客，我就不列出来了。

4. 复现

4.1 官方代码

官方代码是5年前的TensorFlow 1，想想就知道我不可能会。TensorFlow是不可能学的，这辈子都不可能学TensorFlow的，PyTorch又好用，社区丰富，超喜欢这里的。

BPE tokenize：https://github.com/openai/gpt-2/blob/master/src/encoder.py

建模：https://github.com/openai/gpt-2/blob/master/src/model.py

生成：https://github.com/openai/gpt-2/blob/master/src/sample.py

无条件生成文本：https://github.com/openai/gpt-2/blob/master/src/generate_unconditional_samples.py

有条件（基于prompt）生成文本：https://github.com/openai/gpt-2/blob/master/src/interactive_conditional_samples.py

4.2 基于transformers实现GPT-2

GPT-2是AR语言模型，所以要用语言模型（AutoModelWithLMHead）来实现各项任务。不是seq2seq模型，所以应该不能用seq2seq模型的框架……
不过我也没试过那么干就是了……

官方提供的GPT-2权重（都是纯英文的）：
openai-community/gpt2 · Hugging Face：OpenAI官方提供的最小的GPT-2权重

https://huggingface.co/openai-community/gpt2-medium
https://huggingface.co/openai-community/gpt2-large
https://huggingface.co/openai-community/gpt2-xl

https://huggingface.co/distilbert/distilgpt2

4.2.1 基于GPT2LMHeadModel + PyTorch原生框架实现S2S任务微调和推理

数据集构建：https://github.com/PolarisRisingWar/Math_Word_Problem_Collection/blob/master/codes/finetune/gpt2/GPT2DataSet.py

微调：https://github.com/PolarisRisingWar/Math_Word_Problem_Collection/blob/master/codes/finetune/gpt2/finetune.py

推理：https://github.com/PolarisRisingWar/Math_Word_Problem_Collection/blob/master/codes/finetune/gpt2/test.py
关于generate()函数的使用可以参考这篇博文：文本生成解码策略及其在transformers中的代码实现及配套的代码，展示了不同解码方案输出的效果：https://github.com/PolarisRisingWar/all-notes-in-one/blob/main/decode_examples_in_GPT2.ipynb

4.2.2 基于GPT2LMHeadModel + transformers.Trainer微调GPT-2

参考代码：https://github.com/PolarisRisingWar/Math_Word_Problem_Collection/blob/master/codes/finetune/gpt2/finetune_w_Trainer.py

5. 本文撰写过程中参考的其他网络资料

OpenAI GPT2
https://github.com/huggingface/transformers/blob/main/examples/pytorch/text-generation/run_generation.py：这个是transformers的AR语言模型的示例
Fine-tune a non-English GPT-2 Model with Huggingface：用德语语料训练GPT-2的示例，本文撰写基于Trainer微调时参考了这篇博文的代码
python - How to build a dataset for language modeling with the datasets library as with the old TextDataset from the transformers library - Stack Overflow：这篇的意义就是跳转到下面这个讨论↓
Help understanding how to build a dataset for language as with the old TextDataset - 🤗Datasets - Hugging Face Forums
图解GPT-2 | The Illustrated GPT-2 (Visualizing Transformer Language Models)-CSDN博客：这篇具体介绍了GPT-2原理
我之前写的Transformer和GPT-1的博文其实也有涵盖，但是没有这篇这么有针对性。而且这篇对attention的介绍与我写的理解角度不同，可资参考。补充的点在于 ① 最后一个block输出与embedding相乘得到logits，对这个logits进行采样得到预测token ② GPT-2一般用top-k=40
图解自注意力机制_masked self-attention-CSDN博客：上文中的自注意力部分独立成章
这个masked attention图画得挺到位的，带了数字而且颜色更好看：

这篇补充多头注意力机制peojection过程是不改变维度的
https://github.com/huggingface/transformers/blob/870ff9e1dab249e4ffd8363ce132aa5145c94604/src/transformers/models/gpt2/modeling_gpt2.py：transformers官方实现
代码集成性太高了，一下子看不懂，以后有时间慢慢研究吧
GPT-2 论文+代码笔记 | Yam：这篇对论文做了解读，同时对GPT-2原代码做了解读，内容很全面。
用Conv1d做线性转换这事真奇怪啊……
GPT-2代码解读[2]：Attention_gpt attention qpv-CSDN博客：这篇主要的特色是对GPT-2原代码中attention部分进行解读
↑虽然但是，我也不用TF 1……总之是值得在需要时进一步了解的。但是，有没有PyTorch实现更简洁的解读啊
Training CodeParrot 🦜 from Scratch：这篇博文涵盖了更多大模型训练技巧，包括对多卡的处理。由于我暂时用不到所以没细看，如果有需要可以看看。