过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?

符尧(yao.fu@ed.ac.uk),爱丁堡大学 (University of Edinburgh) 博士生,本科毕业于北京大学。

ChatGPT 大火之后,在 2023 年 2 月 24 日,LLaMA 的出现让 instruction tuning 这个方向变得火热;3 月 18 日,Alpaca 让大家看到从成熟的模型 distill 小模型成为还不错的 ChatBot 的可能性,从而引发羊驼系模型寒武纪大爆发。但仅仅过去三个月,大家开始发现意识到用 ChatGPT 的数据训练 LLaMA 的各种问题。本文回顾在过去三个月内的 LLaMA 系模型的发展,讨论 Instruction Tuning 的下一步挑战。

Disclaimer: 这篇文章算是一个 quick research memo,是从我近期的一个分享大纲里 edit 出来的,做了一些删减和补充;现阶段开源社区对于 LLM 训练清楚 / 不清楚的地方同时存在,我尽量做到引用 / 讨论的内容都是有切实证据,而不是基于流言。很多的内容是我跟对应论文的原作者直接讨论过的。但即便这样,我的 take 也可能有误,很多也讨论不出来,所以请大家直接在评论区 comment,积极参与讨论,真理越辩越明。

目录

过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?图片

1 – 起源

最开始三篇

  • InstructGPT: Training language models to follow instructions with human feedback
  • FLANv1: Finetuned Language Models Are Zero-Shot Learners
  • T0: Multitask Prompted Training Enables Zero-Shot Task Generalization

对比

  • InstructGPT 的目标是对齐,zero-shot /cross lingual 是副产物
  • 这篇文章用的 7B 的 Reward model 来对应 175B 的 Policy model,然后被 DeepSpeed Chat 以及之后一系列 RL 的开源工作 follow,这种做法应该是错的。
  • 正确的做法应该是用 Reward model scale up 换取 policy model 减小,见 [Scaling Laws for Reward Model Overoptimization](https://arxiv.org/abs/2210.10760) — 也就是说把两个模型大小换过来,用 175B 的 reward 去 PPO 7B 的 policy
  • 模型上线现阶段 10-50B 是一个比较跑得起的量级,再大太贵了
  • FLANv1 和 T0 的目标是 zero-shot,所以不对齐

然后是 Self-instruct

  • Self-Instruct: Aligning Language Models with Self-Generated Instructions

注意 self-instruct 的重点

  • Base model 可以是任意,不需要是经过了 alignment 之后的模型 (ChatGPT)
  • 复现了从初代 davinci 到 text-davinci-001 的过程 — 非常 insightful!!

过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?图片

然后是 FLANv2 — 很重要,我可能读了十遍以上,建议背诵全文

  • Scaling Instruction-Finetuned Language Models
  • 效果除了不加 human preference 之外其他都加,等下专门讨论
  • Human preference 确实是喜欢能说的,但是能说的模型不一定能干活。Flan 能干活,但是不能说,跟程序员一样

2 – LLaMA 出现之后

  • Alpaca:起始文章,但是模型本身强度并不多高
  • Vicuna
  • 在开源中只做对话强度不错,格式符合人类喜好,生成内容多,unique token 多
  • Automatic eval 中,可能 in-context learning /reasoning/knowledge suboptimal (体现在 MMLU,BBH 分数),不是说它不行,而是说它可以更好
  • GPT-4 eval 到底行不行还不好说,LMSys 团队自己说行,前提是 prompt engineering 做得足够到位:Judging LLM-as-a-judge with MT-Bench and Chatbot Arena
  • 另外 LMSys 的团队在 efficiency 方面非常强,模型的 serve 看 [vllm](https://github.com/vllm-project/vllm) 这个 project,或许是开源最快的
  • 然后一系列以 GPT-4 做 judge 然后号称自己达到了 GPT3.5 x% 水准的模型,全部不推荐,因为 Eval 不可靠
  • 但是存在几篇工作在 alignment 的时候没有依赖 ChatGPT,这些工作推荐,它们包括
  • LIMA: Less Is More for Alignment — 关注他们选数据的方法,推荐花一个小时的时间把他们的 [数据](https://huggingface.co/datasets/GAIR/lima) 有感情地朗读一遍,这样就知道什么样的 SFT 的数据是好数据了
  • Dromedary: Principle-Driven Self-Alignment of Language Models from Scratch with Minimal Human Supervision — 关注他们 prompt engineering 的方法,这个基本上是一个 LLaMA 版的 Constitutional AI – SFT
  • 然后是一些 paper (终于) 开始分析 instruction tuning 的 data mixture
  • Tulu: How Far Can Camels Go? Exploring the State of Instruction Tuning on Open Resources
  • 结果非常 mix,没办法下结论哪种 mixture 好
  • 但是知道哪种不好:NLP benchmark

3 – Eval 怎么做

首先,不要在一堆 benchmark 上算分数看平均,特别是不要在 GPT-3 的测试任务上看平均,因为平均下来大家都一样;推荐只看核心的有区分度的 benchmark

没有区分度的例子:

  • LM-Eval-Harness(https://github.com/EleutherAI/lm-evaluation-harness) ,benchmark 太多,平均下来掩盖优质 benchmark
  • 这里面其实涵盖了 MMLU 和 MATH,但是被其他数据集平均了
  • Summarization + Rouge / Translation + BLEU:
  • Rouge 和 BLEU 模型强弱只有四五分的差别,数字太小 v.s. accuracy 下模型强弱是 90 分和 10 分的差别,数字足够大
  • Rouge 和 BLEU 和人类偏好不 align — 注意 BLEU 也不完全 align

那么 Pretrain 建议看哪些呢?

  • 区分度,模型强弱需要能一眼看出
  • 分方向,现阶段可以暂时分成
  • 英文知识 — MMLU
  • 中文知识 — C-Eval
  • 推理 — GSM8k / BBH
  • 代码 — HumanEval / MBPP
  • 解决上面四项平衡之后,可以接着做
  • MATH:高难度 reasoning
  • Dialog:这个可能只有 human eval 才行,automatic eval 搞不定

接下来讲 Automatic Eval 

Automatic Eval – 适合 pretrained checkpoint – 基本上可以看  https://github.com/FranxYao/chain-of-thought-hub 的做法

  • Knowledge: MMLU
  •  这个数据集很稳定,基本上没有 sensitivity issue
  • Reasoning:
  • GSM8k: 也比较稳定,但要注意答案提取函数的提出率,低于九十的话得多加 regular expression
  • BBH – Algorithmic:
  • 不是很稳定,需要注意答案提出率
  • BBH – Language:
  • 不是很稳定,需要注意答案提出率 — Chain-of-thought Hub 马上会出一个答案提出率对于结果的 sensitivity 的分析,结论是 BBH 比较 sensitive
  • 现在除了增大模型之外,还不清楚哪些操作可以增加 BBH 数据集上的分数
  • Coding:
  • Human Eval / MBPP: 似乎比较稳定但需要注意做 unbiased estimation
  • 先看上面的几个数据集,分数能够 match llama 之后,就看 MATH
  • MATH
  • 超级难,GPT-4 的分数
  • naive prompting: 42
  • → complexity based prompting: 50 https://openreview.net/forum?id=yf1icZHC-l9
  • → progressive hint prompting: 53 https://arxiv.org/abs/2304.09797
  • → majority voting over 18k: 69.6
  • → best of n with outcome based reward modeling: 72.4
  • → best of n with [process-based reward modeling](https://arxiv.org/abs/2305.20050): 78.2
  • → PPO + process-based reward modeling = ? 推测会上 90
  • 泛化?— 应该是比较强的,泛化一般而言跟基础模型大小正相关,跟 SFT 数据总量负相关,跟 SFT 数据丰富度正相关
  • 如果不是 GPT-4
  • Minerva / PaLM-2: 34.3
  • Galactica: 33.6 — 这篇文章操作很好,因为 Hallucination 被喷下架导致重要性被严重低估
  • 88B paper + 7B code + 7B encyclopedias, textbooks and educational material + 2B KB + 1B CC + 0.4B prompt /instruction * 4 epochs
  • LLaMA 65B: 10.6
  • 其他:低于 10 分

对于一个已经 finetune 成了 chatbot 的模型

  • 首先把上述 benchmark 用 few-shot 的方式过一遍,确保不要掉点
  • 如果只是 dialog finetuning 的话可能会伤已有的能力 (MMLU / BBH)
  • 如果掉点,则考虑 LM mixing / FLANv2 mixing
  • 注意 Chatbot 的 few-shot prompting 要用 dialog 的版本因为 single round 里塞很多 in-context example 模型可能不 instruction-following 不够强,见 CoT Hub 的 standard prompt library(https://github.com/FranxYao/chain-of-thought-hub/blob/main/spl/gsm8k/chat/few_shot_cot.chatml)
  • 然后就是去 eval 用户偏好了,这个时候只能人做
  • 如果有很大的,已经训练好了的 reward model,可以用它 eval 上线的小型 / 中等模型,这个其实跟人做 eval 区别不大
  • 对于一个很大的 Policy Model
  • Online iterative RLHF 前期怎样都需要需要 expert eval
  • 后期需要 expert eval [with AI assistance](https://arxiv.org/abs/2206.05802)

那么能不能用稍微弱一点的模型做 eval 呢?— 可以用,但是注意 query 的难度和分布,注意 prompt engineering

  • 如果不经过 prompt engineering ,肯定不行,因为各种 bias
  • 如果 query 难度不够,diversity 不够,也不一定行
  • 如果 query 难度足够 + 经过了疯狂 prompt engineering,则对于 information seeking 类型的 eval ,或许可以,see https://arxiv.org/abs/2306.05685
  • 但是对于 reasoning 相关,non-information seeking 相关(比如 TLDR),又不一定行
  • 对于 information seeking 相关的 query 会 biased 到长的回复

回复越长,GPT-4 越喜欢,分越高

4 – FLANv2 的效果,Long-Context

FLANv2 是一个很神奇的数据集,它除了不加 user preference 之外什么都加

过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?

  • 注意 CoT prompting
  • 只在 62B 之后才会比 Direct 更好
  • 不加 knowledge (MMLU) 只加 reasoning (BBH)
  • FLANv2 增加的效果有
  •  knowledge (MMLU)
  • reasoning (BBH)
  • Multilingual (TyDiQA / MGSM)
  •  注意 FLAN 的作者们验证过,没有数据泄露
  • 注意以上内容对 in-context learning 和 zero-shot 均成立
  • 但是 FLAN 的回复短,所以不加 user preference — Flan 的性格就像直男,能干活儿,话太少

注意区分数据泄漏和分布内泛化

  • 如果一个数据集的测试集被用来训练模型,叫做数据泄漏,此时模型的分数会特别高,不可信
  • 如果一个数据集的训练集被用来训练模型,叫做分布内泛化,此时模型的分数是可信的
  • 有些数据集分布内泛化的难度不高,比如 MMLU / C-Eval,基本上做 data scaling 就可以加分
  • 有些数据集,如果模型不强,即使看过了训练集,模型在测试集上也做不好,比如 GSM8K — 这种类型的数据集是优质 eval 数据集
  • 代码的难度可能介于 MMLU 和 GSM8k 之间,分布内泛化不像 GSM8K 那么难,但也不简单

然后根据最近Zero-Scrolls 上的数据(https://www.zero.scrolls-benchmark.com/leaderboard),FLAN 还可以加 Long-context Reasoning ,不知道为什么

过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?图片

注意这里 FlanT5 和 T0pp 只有 instruction 的数据集有区别,但是 FlanT5 仅靠 T5 的 relative positional encoding 来 naively scale 到 8k 的 context length 会显著高于 T0

Long Context 或许 data engineering 跟 neural architecture engineering 同样重要

5 – 代码

两篇文章的 data engineering 非常出色

  • WizardCoder: Empowering Code Large Language Models with Evol-Instruct
  • 通过不断 prompt AlpacaCoder 构造 instruction tuning 数据集,基于wizardlm的方法,使用了 chatgpt 来生成复杂指令和对应的回答
  • HumanEval,DS-1000 仅次于 GPT-4,超过 Claude / Bard
  • base model 用的是 StarCoder,这意味着 The Stack V3 的质量再次得到验证,同时注意 pretrain code data 可以过多个 epoch 但网页只过一个 epoch
  • Phi-1: Textbooks Are All You Need
  • Pretrain 数据集来源于 filtered code + prompt ChatGPT
  • Instruction tuning 的数据集来自于 prompt ChatGPT
  • base model 只有 1B

怎么评价

  • 一定要好好研究他们是如何 prompt base model 的 — 要对 base model 有信心,只要 MMLU / BBH / HumanEval 分高,它的潜力就超过你的想象
  • prompt 出来的数据集相当于给 HumanEval / MBPP 这种比较短的算法题搞了一个超大训练集
  • 但是不可以认为它对着测试集优化,因为它泛化的空间应该大于 HumanEval / MBPP — 这个泛化空间跟 model scale 显著正相关
  • 在此基础上,比较难的点是
  • Repo-level code understanding /completion — HumanEval / MBPP 还是有点短
  • Ability balance — 如果照着 Phi-1 的做法,除了代码之外的其他能力都会被冲掉

另外关于代码和文本的 data mixture: https://arxiv.org/abs/2305.16264

过去三个月,LLaMA系模型发展如何?指令微调的核心问题又是什么?图片

Continue training 时使用 50% 的代码作为 data mixture 不会伤模型 language 的能力,反而会提升 coding 和 reasoning 

6 – Putting them together: 能力平衡

目标:

  • 构造一个 instruction tuning data mixture,使得 dialog /coding 增加
  • 同时 MMLU (English knowledge) / C-Eval (Chinese knowledge) / BBH and GSM8K (reasoning) 不掉点
  • In-context learning 不掉点

思路

  • 可以用 FLAN 打底 — 它非常大几乎相当于 continue training
    考虑做一个中文版的 FLAN — 最近智源发的COIG-PC(https://huggingface.co/datasets/BAAI/COIG-PC) 似乎有点像
  • code 的部分参照 WizardCoder 和 Phi-1 的做法
  • 以上数据做好之后,搜 instruction tuning 的 data mixture and data curriculum 的超参数
  • 用上面提到的方法做 Eval

7 – 总结

  • 现阶段 instruction tuning 核心问题是能力平衡
  • 基础能力的 Eval 可以参照 Chain-of-thought Hub,但 dialog 还是得人来,且人也不一定 eval 得足够好
  • FLAN 非常神奇,可以考虑做一个中文版
  • 抓紧把 instruction tuning 收尾,快点进到 reward modeling 阶段
  • 注意要先把 reward modeling 本身做好,确保 reward model 有判断力,再去做 PPO
  • 不要 reward model 还没搞清楚就上 PPO ,步子迈太大容易扯到
© 版权声明

相关文章