← 返回 2026-10-09

大语言模型基础 Foundations of Large Language Models

Tong Xiao, Jingbo Zhu 📅 2026-10-08 👍 26 2026-11-11 18:30
RLHF对齐 推理加速 推理模型 提示工程 教材与综述 预训练

系统梳理大语言模型六大基石:预训练、生成建模、提示、对齐、推理与推理增强的教材级综述

前置知识

Transformer 与自注意力机制

Transformer 是当前几乎所有大语言模型的骨干网络。自注意力让序列中每个位置通过 Query、Key、Value 三个矩阵计算与其他位置的关联权重:$\mathrm{Att}(Q,K,V)=\mathrm{Softmax}(QK^{\top}/\sqrt{d}+\mathrm{Mask})V$,实现对全序列的并行编码;堆叠 $L$ 层并配合多头机制与前馈网络即构成完整模型。

全书第1-2章的预训练任务、第5章的 KV 缓存与推理优化全部建立在 Transformer 的注意力计算之上,不懂自注意力就无法理解后续的架构改造与加速技术。

语言建模与极大似然训练

语言模型对下一个 token 的条件概率 $\Pr(y_i\mid x,y_{<i})$ 建模,训练目标是在语料上最大化对数似然 $\hat{\theta}=\arg\max_\theta\sum_{x}\sum_i\log\Pr_\theta(y_i\mid x,y_{<i})$,等价于最小化交叉熵损失。BERT 式掩码建模与之同源,只是把预测目标换成被遮盖的词。

预训练(第1章)、生成模型训练(第2章)、SFT(第4章)与推理学习(第6章)的所有损失函数都是这一目标的变形,是贯穿全书的数学主线。

强化学习基础(策略、奖励与 PPO)

强化学习把决策建模为智能体在状态 $s_t$ 下依策略 $\pi_\theta(a_t\mid s_t)$ 采取动作并获得奖励的过程。策略梯度以 $\nabla_\theta\log\pi_\theta(a_t\mid s_t)A(s_t,a_t)$ 方向更新参数,A2C 引入价值函数估计优势 $A(s_t,a_t)$,PPO 再用重要性采样构造替代目标并以 KL 惩罚限制策略偏移。

第4章的 RLHF 与第6章的过程监督推理训练都用这套语言描述 LLM 的对齐与优化,缺少 RL 词汇会完全读不懂相关章节。

KL 散度

KL 散度 $D_{\mathrm{KL}}(p\,\|\,q)=\sum_i p_i\log\frac{p_i}{q_i}$ 度量两个概率分布的差异,非负且不对称。在 LLM 中它既用于蒸馏(学生逼近教师的输出分布),也用于约束微调后的模型不偏离参考模型太远;DPO 更是从带 KL 约束的奖励最大化目标严格推导而来。

对齐章节的 KL 惩罚、DPO 的完整推导以及第6章在线策略蒸馏的损失函数全部以 KL 散度为核心构件。

KV 缓存与推理显存瓶颈

自回归解码时每生成一个 token 都要 attend 到之前所有位置的 Key/Value 向量,推理系统因此把它们缓存在显存中避免重复计算。缓存大小随序列长度与批大小线性增长,使解码阶段成为访存受限(memory-bound)而非计算受限的过程,这是推理加速必须解决的核心矛盾。

第2章长序列建模中的 MQA/GQA 与缓存压缩、第5章的 PagedAttention、continuous batching 等技术全部围绕 KV 缓存展开。

缩放定律与幂律拟合

缩放定律用幂律函数刻画测试损失随模型参数量 $N$、数据量 $D$ 等因素的下降规律,如 Kaplan 的 $L(N)=(N_c/N)^{0.076}$、$L(D)=(D_c/D)^{0.095}$,以及加入不可约误差项的 Chinchilla 公式 $L(N,D)=\frac{406.4}{N^{0.34}}+\frac{410.7}{D^{0.28}}+1.69$。

它解释了社区为何持续扩大模型与数据规模,也是理解第2章训练策略与第5-6章推理时扩展(test-time scaling)的背景知识。

研究动机

大语言模型在 2020 年后爆发式发展,但相关知识高度碎片化:预训练、提示、对齐、推理加速、推理模型分散在成千上万篇论文中,术语与记号互不兼容,初学者很难建立完整图景。同时该领域迭代极快——从 BERT 的掩码语言建模,到 GPT-3 的 175B 参数、Llama3.1-405B 的 15.6T token 预训练,再到 o1 与 DeepSeek-R1 的长思维链推理——教材与课程严重滞后,工程人员往往只会调库而不理解每个环节的原理与代价(例如为何解码是访存受限、为何 RLHF 需要 KL 惩罚、为何 Chinchilla 主张少堆参数多补数据)。对非英语母语、缺乏系统科研训练的读者而言,缺少一份既讲清数学推导、又覆盖最新推理模型的系统入门资料,是进入该领域最现实的障碍。

本文的目标是本书的目标是以『基础』为准绳,用统一记号系统讲解大语言模型从构建到应用的完整技术栈:第1章讲预训练范式与 BERT;第2章讲生成式 LLM、大规模训练(数据准备、架构改造、分布式训练、缩放定律)与长序列建模;第3章讲提示工程、上下文学习、思维链与自动/软提示设计;第4章讲指令微调与 RLHF、DPO 对齐;第5章讲解码算法、推理加速系统与推理时扩展;第6章讲推理模型的训练与评测。作者希望读者读完后能理解每个模块的输入输出、关键公式与设计动机,具备跟踪前沿和动手实践的能力;各章尽量自成体系,即使没有 Transformer 基础也不至于无法阅读。

与已有工作不同的是,本书的独特之处在于教科书式的统一视角:它不是罗列最新论文的综述目录,而是把现代 LLM 技术与经典 NLP/机器学习思想贯通起来——例如把 RLHF 的奖励模型与判别式打分模型对照,把 self-consistency 解释为最小贝叶斯风险搜索,把弱到强泛化归结为知识蒸馏问题,把过程监督的奖励整形 $r_t=v_t-v_{t-1}$ 与 RL 中的 reward shaping 相衔接。全书选自 NiuTrans 实验室多年教学积累的 NLPBook 项目(github.com/NiuTrans/NLPBook),每章自成体系、公式推导完整(如从 KL 约束优化逐步消去配分函数推出 DPO 损失),并专节讨论『LLM 能否真正推理』这类批判性问题,这在只有论文堆砌式综述的现有文献中是明显的差异化贡献。

核心方法

直觉上,本书把大语言模型的发展组织成一条『先预训练、后适配』的主线:先用自监督任务在海量无标注文本上训练基础模型,再通过微调、对齐或提示把通用能力转化为具体任务能力。技术路线上,第1章从无监督/有监督/自监督三种预训练范式的对比出发,给出语言建模、掩码语言建模、置换语言建模、去噪自编码等任务的统一形式,并以 BERT 为完整示例;第2章转向 decoder-only 生成模型,讨论数据准备(质量、多样性、偏见、隐私)、模型改造(Pre-Norm、SwiGLU、RoPE 等)、数据/张量/流水线并行、Kaplan 与 Chinchilla 缩放定律,以及长上下文的 KV 缓存压缩、MQA/GQA 与位置外推/插值;第3章覆盖从 few-shot 提示到思维链、问题分解、自我修正、集成、RAG 与工具使用,再到自动提示优化与软提示;第4章给出 SFT → RLHF → DPO 的对齐流水线;第5章把推理拆成 prefilling 与 decoding 两阶段并讨论解码算法与系统级加速;第6章从提示推理、推理时扩展与过程监督训练三条路径解释 o1/R1 类推理模型。

核心思想是用一套数学记号贯通所有 LLM 技术。书中把各种方法还原为概率建模框架下的本质:预训练任务都是对 $\log\Pr(\cdot\mid\cdot)$ 不同条件形式的构造;解码是搜索问题 $\hat{y}=\arg\max_y\log\Pr(y\mid x)$;对齐是在 KL 约束下的奖励最大化,其闭式解 $\pi^*(y\mid x)\propto\pi_{\mathrm{ref}}(y\mid x)\exp(r(x,y)/\beta)$ 直接导出 DPO;推理时扩展则是搜索空间扩展加验证器。与已有工作的本质区别在于:市面综述多为文献分类与结论罗列,本书提供可推导、可复算的数学主线,并把『训练时扩展』与『推理时扩展』这对概念作为组织新范式的钥匙(见 Table 6.1 的五维对比)——更大模型与更长思考之间存在成本权衡,小模型配合推理时计算扩展可以胜过大模型的即时作答。

方法步骤详情

全书按六步推进。第一步(第1章)定义统一序列模型 $o=g_\theta(x_0,\dots,x_m)$,剖析 decoder-only、encoder-only、encoder-decoder 三类预训练任务(Table 1.1),并以 BERT 为例走通 15% 随机遮盖、80/10/10 替换、损失 $\mathrm{Loss}_{\mathrm{BERT}}=\mathrm{Loss}_{\mathrm{mlm}}+\mathrm{Loss}_{\mathrm{nsp}}$ 的训练流程。第二步(第2章)给出极大似然目标与模型规模数据,讲数据清洗、Pre-Norm/SwiGLU 架构改造、DP/TP/PP 并行、Kaplan 与 Chinchilla 缩放定律,再处理长序列:IO 优化、稀疏/线性注意力、MQA/GQA、RoPE 外推与位置插值。第三步(第3章)从基础提示设计讲到上下文学习、思维链(含 zero-shot 版)、问题分解、自我修正、self-consistency 集成、RAG 与工具调用,再到自动提示优化与软提示。第四步(第4章)推导 SFT 损失、self-instruct 数据合成、Bradley-Terry 奖励模型 $r(x,y)=h_{\mathrm{last}}W_r$、PPO 训练与 KL 惩罚,再到 DPO 损失 $\mathcal{L}_{\mathrm{dpo}}(\theta)=-\mathbb{E}\log\Pr_\theta(y_a\succ y_b\mid x)$。第五步(第5章)定义 prefilling/decoding 两阶段框架(Table 5.1)、贪心/束搜索/top-$k$/top-$p$/投机解码等算法、TTFT/ITL 指标与 PagedAttention 等系统优化。第六步(第6章)讨论推理模型:多路径采样、树搜索、生成式与形式化验证器、 监督数据、过程奖励整形、在线策略蒸馏,以及 RL 的 S 形缩放曲线与 Pass@k、Cons@k 评测。

技术新颖性

作为教材型综述,其新颖性体现在三点。其一,覆盖范围的完整性:它较早把推理模型(o1/DeepSeek-R1 式长思维链、过程监督、在线策略蒸馏、生成式验证器)纳入基础教材体系,并收录了 RL 计算量与推理性能呈 S 形曲线等 2025-2026 年的最新实证结论。其二,推导的严格性:书中不止给出公式,还保留完整推导链——例如从 KL 约束的奖励最大化出发,经配分函数 $Z(x)$ 的消去得到 DPO 损失;或证明过程奖励的伸缩求和 $R(x,y)=\sum_t(v_t-v_{t-1})=v_{n_p}-v_0$ 与最终步骤评分一致;这类推导在一般综述中往往被省略。其三,概念的甄别与批判:作者设专节讨论外部推理 vs 内部推理(CoT 文本未必反映模型的真实内部计算,故引入暂停 token、COCONUT、Quiet-STaR 等潜空间推理)、记忆 vs 推理(grokking 现象表明泛化在记忆饱和后突现)以及 LLM 能否真正推理(形式语言能力强、功能语言能力弱,规划任务中把实体名换成无意义词后性能骤降),体现了超越技术罗列的学术判断。

A running example of BERT-style masked language modeling. First, 15% of the tokens are randomly selected.
Fig. 1.5: A running example of BERT-style masked language modeling. First, 15% of the tokens are randomly selected.
The model architecture of BERT (Transformer encoder).
Fig. 1.6: The model architecture of BERT (Transformer encoder).
The Transformer-decoder architecture for language modeling.
Fig. 2.1: The Transformer-decoder architecture for language modeling.
An overview of RLHF. There are 4 key steps involved.
Fig. 2.2: An overview of RLHF. There are 4 key steps involved.
Illustrations of self-attention, sparse attention, linear attention and recurrent models.
Fig. 2.5: Illustrations of self-attention, sparse attention, linear attention and recurrent models.
Illustration of QKV attention based on different multi-head and sharing mechanisms.
Fig. 2.8: Illustration of QKV attention based on different multi-head and sharing mechanisms.
Illustrations of different positional embedding behaviors across a sequence length continuum.
Fig. 2.9: Illustrations of different positional embedding behaviors across a sequence length continuum.
CoT in four different reasoning tasks, including CSQA, StrategyQA, Dyck languages, and Last Letter Concatenation.
Fig. 3.1: CoT in four different reasoning tasks, including CSQA, StrategyQA, Dyck languages, and Last Letter Concatenation.
Ensembling methods for LLMs.
Fig. 3.2: Ensembling methods for LLMs.
Illustration of hard and soft prompts.
Fig. 3.3: Illustration of hard and soft prompts.
Schematic illustration of the pre-training-then-alignment pipeline for developing LLMs.
Fig. 4.1: Schematic illustration of the pre-training-then-alignment pipeline for developing LLMs.
Illustration of self-instruct.
Fig. 4.4: Illustration of self-instruct.
Architecture of a Transformer-based reward model.
Fig. 4.8: Architecture of a Transformer-based reward model.
Standard RLHF (PPO) vs. DPO.
Fig. 4.10: Standard RLHF (PPO) vs. DPO.
Illustration of the KV cache.
Fig. 5.2: Illustration of the KV cache.
Illustration of the prefilling and decoding processes.
Fig. 5.3: Illustration of the prefilling and decoding processes.
Illustrations of greedy decoding, beam decoding, top-k decoding and top-p decoding methods.
Fig. 5.5: Illustrations of greedy decoding, beam decoding, top-k decoding and top-p decoding methods.
Illustration of one step of speculative decoding.
Fig. 5.7: Illustration of one step of speculative decoding.
Illustration of memory allocation in PagedAttention.
Fig. 5.12: Illustration of memory allocation in PagedAttention.
Illustrations of parallel scaling and sequential scaling.
Fig. 5.16: Illustrations of parallel scaling and sequential scaling.
Illustration of step-level search with verifiers.
Fig. 5.17: Illustration of step-level search with verifiers.
Comparison between discriminative and generative verification approaches.
Fig. 6.1: Comparison between discriminative and generative verification approaches.
Comparison of supervised fine-tuning, process supervision, and on-policy distillation methods.
Fig. 6.2: Comparison of supervised fine-tuning, process supervision, and on-policy distillation methods.

实验结果

本书不报告原创实验,而是系统汇编领域关键实证结论。规模与数据:GPT-3 有 175B 参数、96 层、宽度 12288;Llama3.1-405B 用 15.6T token 预训练;DeepSeek-V3 达 671B 参数、14.8T token;Qwen3 语料高达 36T token。缩放定律:Kaplan 拟合出 $L(N)=(N/8.8\times10^{13})^{-0.076}$、$L(D)=(D/5.4\times10^{13})^{-0.095}$ 的幂律,Chinchilla 给出 $L(N,D)=\frac{406.4}{N^{0.34}}+\frac{410.7}{D^{0.28}}+1.69$,并导出参数与数据应等比扩展的结论。数据:Penedo 等证明清洗后丢弃 90% 网页数据反而提升效果。对齐:LIMA 仅用 1,000 条精选样本微调 Llama-65B 即具竞争力,而 FLAN 用了 1,836 个任务、1,500 万样本,对比支持『表层对齐假说』;Burns 等实验中 GPT-2 级监督使 GPT-4 级模型在 22 个分类任务上恢复约 80%(PGR≈0.8)的差距。推理:CoT 与 self-consistency 显著提升 GSM8K 等基准成绩;过程监督(PRM800K)优于纯结果监督;大规模 RL 催生 DeepSeek-R1 的『aha moment』,RL 计算量与推理性能呈 S 形曲线。

Comparison of pre-training tasks, including language modeling, masked language modeling, permuted language modeling, discriminative training, and denoising autoencoding.
Table 1.1: Comparison of pre-training tasks, including language modeling, masked language modeling, permuted language modeling, discriminative training, and denoising autoencoding.
Comparison of some LLMs in terms of model size, model depth, model width, and number of heads.
Table 2.2: Comparison of some LLMs in terms of model size, model depth, model width, and number of heads.
Reported pre-training token counts and major data categories for selected LLMs.
Table 2.3: Reported pre-training token counts and major data categories for selected LLMs.
Prefilling vs Decoding.
Table 5.1: Prefilling vs Decoding.
Comparison of training-time and inference-time scaling.
Table 6.1: Comparison of training-time and inference-time scaling.
A scaling law of test error against a variable of interest (e.g., training dataset size).
Fig. 2.3: A scaling law of test error against a variable of interest (e.g., training dataset size).
Test loss against model size (N) and training dataset size (D).
Fig. 2.4: Test loss against model size (N) and training dataset size (D).
查看结构化数据
任务指标本文基线提升
语言建模测试损失随规模的变化 测试损失的幂律拟合指数 汇编 Kaplan 结果:$L(N)\propto N^{-0.076}$,$L(D)\propto D^{-0.095}$,万亿 token 后仍服从幂律 传统观点认为性能增益会在某个规模后彻底消失 为继续扩大模型与数据规模提供了定量依据,催生了后续大规模预训练实践
预训练算力最优分配(Chinchilla) 每 token 测试损失 $L(N,D)=\frac{406.4}{N^{0.34}}+\frac{410.7}{D^{0.28}}+1.69$,参数与数据应等比扩展 Kaplan 缩放定律只单独拟合 $N$ 或 $D$,倾向堆大模型 同等算力下应优先补数据而非堆参数,直接影响了 70B 级模型配万亿 token 的训练配方
指令微调数据效率 指令跟随能力(人工/自动评测) LIMA:1,000 条精选样本微调 Llama-65B 即具竞争力 FLAN:1,836 个任务、约 1,500 万样本的大规模指令微调 证明数据质量远胜数量,支持表层对齐假说,大幅降低对齐门槛
弱到强泛化(22 个 NLP 分类任务) 性能差距恢复率 PGR 约 0.8(GPT-2 级弱监督 → GPT-4 级学生模型) 无有效监督时 PGR=0,纯模仿弱模型则无法超越弱教师 弱监督可恢复约 80% 的差距,为监督超人类 AI 系统提供了初步可行性证据
网页数据清洗(RefinedWeb/Falcon) 清洗后保留的数据比例 仅保留约 10% 的高质量网页数据用于训练 直接使用未过滤的原始网页抓取语料 清洗后训练效果更优,确立了质量优先于数量的数据工程准则

局限与改进

作者坦承本书定位为『基础』而非前沿全景:为控制篇幅省略了多模态、智能体、安全对齐/红队测试、RAG 系统实现细节等主题,多处以『此处从略』收尾并指向 NLPBook 其余章节。作为教材它不做原创实验,所有数字均引自第三方论文,个别结论(如弱到强泛化 PGR≈0.8)来自特定任务设置,普适性有限。以我的观察另有几点局限:其一,领域迭代极快,v4 版于 2026 年 10 月更新,推理模型一章引用的 S 形缩放曲线、生成式验证器等文献非常新,一两年内可能部分过时;其二,全书以纯文本 LLM 为主线,对视觉、语音与具身推理仅一笔带过,而『推理跨越数据类型』恰是作者自己列出的关键未来方向;其三,缺少可运行的代码、超参数表与算力估算工具,动手复现门槛仍然不低;其四,部分前沿话题(如多语言推理、低资源场景) coverage 较薄,低资源语言的表现问题仅在数据多样性一段被提及。

独立分析的弱点

独立分析本书的弱点:第一,缺乏端到端代码与算力估算——书中讲清了数据/张量/流水线并行的原理,但没有给出『训练一个 7B 模型需要多少 GPU 时』之类的量化工具,初学者仍难落地,改进方向是配套开源 notebook 与成本计算器。第二,评测讨论偏浅:第6章正确指出了 Pass@k、Cons@k 与数据污染问题,但没有提供污染语料去重的可操作流程或抗污染新基准清单。第三,系统与算法割裂:continuous batching、PagedAttention 属系统层,self-consistency、树搜索属算法层,两者在第5章并列陈述却缺少统一的成本模型(如每 token 延迟与搜索宽度的权衡公式),可补充排队论或 roofline 模型视角。第四,安全与伦理着墨不足:偏见、隐私、奖励作弊(reward hacking)各只有一小节,缺少对越狱攻击、幻觉缓解、内容水印等专题的系统处理。第五,公式密集但没有习题与参考实现,教学场景下需要教师自行设计练习;若每章增加一个『最小可运行示例』(例如用百兆参数模型演示缩放定律拟合或 DPO 训练),教学价值会显著提升。

未来方向

作者在第6章明确提出若干方向:让模型学会动态分配思考时间——简单问题不浪费算力、难题才放慢;把推理从数学、代码等有唯一答案的领域扩展到开放式现实任务;实现跨模态(图像、音频、物理环境)推理;以及把推理能力用于自主智能体的『思考-行动-观察』循环并解决循环中迷失与失败恢复的问题。基于本书成果还可延伸:其一,把『记忆 vs 推理』的成本权衡形式化,指导 RAG 外部知识与参数化知识的自动分工——作者已指出 RAG 的本质是把记忆负担外包给数据库;其二,将过程监督推广到缺少唯一正确步骤的开放领域,生成式验证器(自然语言批评)是有希望的起点;其三,研究思维链的忠实性——Turpin 等已证明 CoT 解释可能具有欺骗性,需要结合神经激活分析与机制可解释性工具;其四,弱到强泛化目前只在分类任务上验证,扩展到生成与推理任务是对齐超人类模型的关键一步;其五,构建持续更新、防污染的评测基准与私有保留集,缓解基准饱和问题。

复现评估

本书本身是开放获取资料,源自 NiuTrans 实验室的 NLPBook 项目(github.com/NiuTrans/NLPBook,niutrans.github.io/NLPBook 在线阅读),采用 CC BY-NC 4.0 许可,arXiv 版(2501.09223v4)可自由下载,获取成本近乎为零——有机器学习与 Transformer 基础即可通读。但它不是软件工件:没有统一代码仓库,涉及的实验(BERT、GPT-3、Chinchilla、RLHF、DPO、o1/R1)来自数十个团队,开放程度差异大——BERT、Llama、DeepSeek-V3 已开源,GPT-3/4 与 o1 只有论文。动手验证方面:缩放定律拟合、CoT 提示、self-consistency、DPO 可在单卡到数张 GPU 上复现(开源 7B 模型加 LoRA/DPO 库即可);完整 RLHF 流水线涉及策略、参考、奖励、价值四个大模型,大规模 RL 训练还需高频 rollout 与验证器调用,属机构级工程;弱到强泛化实验依赖闭源 GPT-4 级模型,已无法严格复现。总体而言,读懂与复现小规模实验难度为『低』,完整复现章节级系统难度为『极高』。

相似论文推荐