资料摘要:Llama 2

Meta 推出的开源大模型系列(7B/13B/70B),首次对 RLHF 对齐方法论进行全面透明披露,对话模型在人类评估中与 ChatGPT 相近。 Llama 2 - Open Foundation and Fine-Tuned Chat Models

核心要点

  • 完整开源:预训练 + 对话微调(RLHF)权重全部开放,附带商业可用许可
  • RLHF 透明披露:详尽记录数据收集、奖励模型训练、多轮迭代 RLHF、安全性工作的全流程
  • 2T tokens 训练,上下文从 2K 翻倍到 4K;34B/70B 采用 GQA(分组查询注意力)
  • 70B 可媲美 ChatGPT:人类评估中 Llama 2-Chat 70B 与 GPT-3.5-turbo 互有胜负

详细笔记

模型规格

规模GQAtokensLR
7B2.0T3.0e-4
13B2.0T3.0e-4
34B2.0T1.5e-4
70B2.0T1.5e-4

:34B 已训练并出现在论文各表中,但 Meta 因红队测试未达发布标准未公开发布 34B 权重——故引言「开源系列」只列 7B/13B/70B。

架构基于 Llama 1:RMSNorm + SwiGLU + RoPE,新增 4K 上下文、GQA(34B/70B)。

RLHF 方法(详尽记录)

数据收集:

  • SFT:27,540 条高质量人工标注数据
  • RM 训练:1,418,091 条人工偏好对比数据(二元比较)
  • 每条对话平均 3.9 轮,每条样本平均 798.5 token

分离式奖励模型:

  • 分别训练 Helpfulness RM 和 Safety RM
  • RM 从预训练对话模型初始化,分类头替换为标量回归头
  • Helpfulness RM Meta 测试集准确率 63.2%,Safety RM 64.5%

迭代式 RLHF(V1→V5):

  1. Rejection Sampling:每条 prompt 采样 K 个输出,RM 选最优作为新金标准,70B 模型执行 RS、小模型蒸馏。最优温度 T=1.2-1.3
  2. PPO:KL 奖励 = whiten(logit(R_c)) - β·D_KL;R_c 综合 safety/helpfulness RM;β=0.01(7B/13B)、0.005(34B/70B);200-400 轮,70B 每轮约 330s

Ghost Attention (GAtt):将系统指令在所有轮次中合成拼接至用户消息;训练时除最后轮次外 token 损失置零;实现 20+ 轮多轮一致性

安全性上下文蒸馏:在对抗性 prompt 前添加安全预提示生成安全回答,去除预提示后在该输出上微调;仅当 safety RM 评分提升时应用

关键基准

基准Llama 2 70BGPT-3.5GPT-4
MMLU68.970.086.4
GSM8K56.857.192.0
HumanEval29.948.167.0

训练细节

  • 优化器 AdamW(β1=0.9, β2=0.95),cosine LR、2000 步 warmup、衰减至峰值 10%
  • 权重衰减 0.1、梯度裁剪 1.0、全局 batch 4M tokens
  • Meta RSC 集群 A100-80GB,总计 330 万 GPU 小时
  • 碳排放 539 tCO2eq(100% 抵消)
  • 验证 RoCE 在 2000 GPU 规模下几乎与 InfiniBand 同等扩展

影响

Llama 2 为整个开源社区提供了 RLHF 对齐的完整蓝图(数据→RM→迭代→安全),商业许可加速了生态采纳。GAtt、分离式 RM、迭代 RLHF 等方法被后续模型广泛借鉴。

相关