必收藏！程序员必学：大语言模型(LLM)核心原理从Token到Next Token Prediction深度解析

关注

原创最新推荐文章于 2026-07-12 10:30:00 发布 1k 阅读 AI 写同款· GEO 优化 ›

VLM 系列——Object Recognition as Next Token Prediction——论文解读本文主要介绍《Object Recognition as Next Token Prediction》：结合了CLIP 的视觉编码器+语言模型Llama 的部分参数，将常见的图片描述任务转变为只输出属性，换言之将图片分类转变为预测下一个文本输出token。这样就能够生成图片的top K属性（英文），用于开放域的图片Tag 场景。阅读详情

本文详细解释了大语言模型(LLM)的核心工作原理，重点介绍了Token的概念和Next Token Prediction机制。首先说明了Token是文本的基本单位，不同语言对Token的定义各异。然后介绍了模型如何通过Token Embedding和位置编码处理文本序列，以及如何使用Transformer结构进行自回归生成。最后详细讲解了训练阶段和推理阶段的区别，包括温度参数(Temperature)对生成结果随机性的影响，并提供了代码实现示例，帮助读者理解LLM的实际工作方式。

什么是 token

token是指文本中的一个词或者子词，给定一句文本，送入语言模型前首先要做的是对原始文本进行tokenize，也就是把一个文本序列拆分为离散的token序洌

其中，tokenize是在无标签的语料上训练得到的一个token数量固定且唯一的分词器，这里的token数量就是大家常说的词表

英文中的 Token

在英文中，Token 通常是单词、子词或标点符号。一个单词可能对应一个 Token，也可能被拆分为多个 Token。例如，“unhappiness” 可能被拆分为 “un”、“happi” 和 “ness”。
一般来说，1 个 Token 对应 3 至 4 个字母，或者约 0.75 个单词。

中文中的 Token

在中文中，Token 通常是单个汉字或经过分词后的词语。例如，“人工智能” 可能被拆分为 “人工” 和 “智能”。
不同平台对 Token 的定义有所不同。例如，通义千问和千帆大模型中 1 Token 等于 1 个汉字，而腾讯混元大模型中 1 Token 约等于 1.8 个汉字

当我们对文本进行分词后，每个token可以对应一个embedding，这也就是语言模型中的embedding层，获得某个token的embedding就类似一个查表的过程

我们知道文本序列是有顺序的，而常见的语言模型都是基于注意力机制的transformer结构，无法自动考虑文本的前后顺序，因此需要自动加上位置编码，也就是每个位置有一个位置embedding 然后和对应位置的token embedding进行相加

在模型训练或推理阶段大家经常会听到上下文长度这个词，它指的是模型训练时接收的token训练的最大长度，如果在训练阶段只学习了一个较短长度的位置embedding，那模型在推理阶段就不能够适用于较长文本(因为它没见过长文本的位置编码)

语言模型的预训练

当我们有了token的embedding和位置embedding后，将它们送入一个decoder-only的transofrmer模型，它会在每个token的位置输出一个对应的embedding(可以理解为就像是做了个特征加工)

有了每个token的一个输出embedding后，我们就可以拿它来做 next token prediction了，其实就是当作一个分类问题来看待:

首先我们把输出embedding送入一个线性层，输出的维度是词表的大小，就是让预测这个token的下一个token属于词表的"哪一个"
为了将输出概率归一化，需要再进行一个softmax变换
训练时就是最大化这个概率使得它能够预测真实的下一个token
推理时就是从这个概率分布中采样下一个token

训练阶段: 因为有因果自注意力（Causal Self-Attention）的存在，我们可以一次性对一整个句子每个token进行下一个token的预测，并计算所有位置token的loss

因果自注意力通过引入一个“掩码”（mask）来实现这一机制。具体来说：

在计算注意力权重时，模型会将当前时刻之后的所有位置的注意力权重设置为零。
这样，模型在预测下一个词时，只能基于已经生成的词（即前面的词）来进行预测。

推理阶段:以自回归的方式进行预测

其中，在预测下一个token时，每次我们都有一个概率分布用于采样，根据不同场景选择采样策略会略有不同，比如有贪婪策略、核采样、Top-k采样等，另外经常会看到Temperature这个概念，它是用来控制生成的随机性的，温度系数越小越稳定。

代码实现

https://github.com/karpathy/nanoGPT/tree/master

对于各种基于Transformer的模型，它们都是由很多个Block堆起来的，每个Block主要有两个部分组成：

Multi-headed Causal Self-Attention
Feed-forward Neural Network

结构的示意图如下：

看图搭一下单个Block

class Block(nn.Module):
def __init__(self， config):
super().__init__()
self.ln_1 = LayerNorm(config.n_embd， bias=config.bias)
self.attn = CausalSelfAttention(config)
self.ln_2 = LayerNorm(config.n_embd， bias=config.bias)
self.mlp = MLP(config)
def forward(self， x):
x_ = x + self.attn(self.ln_1(x))
x = x + self.mlp(self.ln_2(x_))
return x

整个nano-GPT的结构

class GPT(nn.Module):
def __init__(self， config):
super().__init__()
assert config.vocab_size is not None
assert config.block_size is not None
self.config = config
self.transformer = nn.ModuleDict(dict(
wte = nn.Embedding(config.vocab_size， config.n_embd)，
wpe = nn.Embedding(config.block_size， config.n_embd)，
drop = nn.Dropout(config.dropout)，
h = nn.ModuleList([Block(config) for _ in range(config.n_layer)])，
ln_f = LayerNorm(config.n_embd， bias=config.bias)，
))
self.lm_head = nn.Linear(config.n_embd， config.vocab_size， bias=False)
# with weight tying when using torch.compile() some warnings get generated:
# "UserWarning: functional_call was passed multiple values for tied weights.
# This behavior is deprecated and will be an error in future versions"
# not 100% sure what this is， so far seems to be harmless. TODO investigate
self.transformer.wte.weight = self.lm_head.weight # https://paperswithcode.com/method/weight-tying
# init all weights
self.apply(self._init_weights)
# apply special scaled init to the residual projections， per GPT-2 paper
for pn， p in self.named_parameters():
if pn.endswith('c_proj.weight'):
torch.nn.init.normal_(p， mean=0.0， std=0.02/math.sqrt(2 * config.n_layer))

最后一层用来分类的线性层的权重和token embedding层的权重共享。

训练和推理的forward

首先需要构建token embedding和位置embedding，把它们叠加起来后过一个dropout，然后就可以送入transformer的block中了。

pos = torch.arange(0， t， dtype=torch.long， device=device) # shape (t)
# forward the GPT model itself
tok_emb = self.transformer.wte(idx) # token embeddings of shape (b， t， n_embd)
pos_emb = self.transformer.wpe(pos) # position embeddings of shape (t， n_embd)
x = self.transformer.drop(tok_emb + pos_emb)
for block in self.transformer.h:
x = block(x)
x = self.transformer.ln_f(x)

接下来看推理阶段

根据当前输入序列进行一次前向传播
利用温度系数对输出概率分布进行调整
通过softmax进行归一化
从概率分布进行采样下一个token
拼接到当前句子并再进入下一轮循环

@torch.no_grad()
def generate(self， idx， max_new_tokens， temperature=1.0， top_k=None):
"""
Take a conditioning sequence of indices idx (LongTensor of shape (b，t)) and complete
the sequence max_new_tokens times， feeding the predictions back into the model each time.
Most likely you'll want to make sure to be in model.eval() mode of operation for this.
"""
for _ in range(max_new_tokens):
# if the sequence context is growing too long we must crop it at block_size
idx_cond = idx if idx.size(1) <= self.config.block_size else idx[:， -self.config.block_size:]
# forward the model to get the logits for the index in the sequence
logits， _ = self(idx_cond)
# pluck the logits at the final step and scale by desired temperature
logits = logits[:， -1， :] / temperature
# optionally crop the logits to only the top k options
if top_k is not None:
v， _ = torch.topk(logits， min(top_k， logits.size(-1)))
logits[logits < v[:， [-1]]] = -float('Inf')
# apply softmax to convert logits to (normalized) probabilities
probs = F.softmax(logits， dim=-1)
# sample from the distribution
idx_next = torch.multinomial(probs， num_samples=1)
# append sampled index to the running sequence and continue
idx = torch.cat((idx， idx_next)， dim=1)
return idx

温度参数（Temperature）的作用

温度参数 temperature 是一个超参数，用于控制生成文本的随机性。它的作用是调整 logits 的分布，从而影响最终的概率分布。

具体来说：

temperature > 1：增加随机性
当温度参数大于 1 时，会放大 logits 的值，使得 logits 的分布更加“平坦”。
这意味着在 softmax 转换为概率分布后，各个 token 的概率会更加接近，从而增加生成结果的随机性。
例如，假设原始 logits 是 [10， 20， 30]，除以温度参数 2 后变为 [5， 10， 15]，经过 softmax 后，概率分布会更加均匀。
temperature < 1：减少随机性
当温度参数小于 1 时，会缩小 logits 的值，使得 logits 的分布更加“尖锐”。
这意味着在 softmax 转换为概率分布后，高概率的 token 会更加突出，而低概率的 token 的概率会进一步降低，从而减少生成结果的随机性。
例如，假设原始 logits 是 [10， 20， 30]，除以温度参数 0.5 后变为 [20， 40， 60]，经过 softmax 后，概率分布会更加集中在高概率的 token 上。
temperature = 1：保持原始分布
当温度参数等于 1 时，logits 不变，模型的输出概率分布保持原始的预测结果。

为什么要调整温度参数？

增加随机性（temperature > 1）：
有助于生成更多样化的文本，避免模型总是生成相同的、高概率的 token。
适用于需要创造性或多样性的场景，例如诗歌生成、故事创作等。
减少随机性（temperature < 1）：
有助于生成更稳定、更符合预期的文本，减少生成的噪声。
适用于需要高准确性的场景，例如机器翻译、问答系统等。

如何学习大模型 AI ？

由于新岗位的生产效率，要优于被取代岗位的生产效率，所以实际上整个社会的生产效率是提升的。

但是具体到个人，只能说是：

“最先掌握AI的人，将会比较晚掌握AI的人有竞争优势”。

这句话，放在计算机、互联网、移动互联网的开局时期，都是一样的道理。

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

这份完整版的大模型 AI 学习资料已经上传优快云，朋友们如果需要可以微信扫描下方优快云官方认证二维码免费领取【保证100%免费】