LogoCookLLM文档
LogoCookLLM文档
首页CookLLM

原理精讲

词元化
Tokenization 基础BPE 算法详解GPT 系列 TokenizerBPE 训练工程化
模型架构
Transformer LM
从 token ids 到 logitsEmbedding 与 LM Head
Attention 机制
Self-Attention 到 GQAAttention Sink
位置编码
位置编码基础RoPE 数学推导RoPE 代码实现长度外推
GPU 编程基础
GPU 架构基础张量布局Triton 入门:向量加法
FlashAttention
Flash Attention 原理详解从朴素实现到 Auto-TuningBlock Pointer 与多维支持Causal Masking 优化Grouped Query Attention反向传播实现
分布式训练
数据并行ZeRO 优化器全分片数据并行张量并行流水线并行多维混合并行
推理优化
KV CacheContinuous BatchingPagedAttention

动手训练

概述
预训练
预训练数据Tokenizer 训练模型架构数据流水线训练循环监控与验证
X (Twitter)
基础知识模型架构Transformer LM

Embedding 与 LM Head

会员专享

理解 token ids 如何进入连续向量空间,以及 hidden states 如何投影回词表 logits

在权益中心获取代码

为什么输入输出层值得单独看?

Decoder-only Transformer 最重的计算发生在中间的 blocks 里,但语言模型的输入输出关系其实由两端决定:

登录以继续阅读

这是一篇付费内容,请登录您的账户以访问完整内容。

token ids→embeddinghidden states→LM Headnext-token logits\text{token ids} \xrightarrow{\text{embedding}} \text{hidden states} \xrightarrow{\text{LM Head}} \text{next-token logits}token idsembedding​hidden statesLM Head

如果这两端没有看清楚,后面很多问题都会混在一起:为什么输入是整数、为什么输出不是 token id、为什么 loss 需要 shift targets、为什么词表大小会显著影响参数量,甚至为什么有些模型会共享输入 embedding 和输出 LM Head 的权重。

本章主线只讲输入输出两端:Embedding 和 LM Head。

从 token ids 到 logits

理解 Decoder-only Transformer 如何把 token ids 变成 next-token logits

Attention 机制

从 MHA / Causal / GQA,到 Attention Sink 与 Gated Attention,理解注意力机制的设计、缺陷与演进

目录

为什么输入输出层值得单独看?
Embedding
按 token id 查表
训练后的 embedding 学到了什么?
加新的 special token:padded 槽位的妙用
SFT 把 embedding 改了什么
从 special token 读 SFT 重点
从普通 BPE token 读 SFT 内容
拼出来的 SFT 配方
LM Head
几何视角:为什么形状一样不是巧合
代码里就是一行赋值
代价:tying 不是免费的
LoRA包装一层之后,tying 还成立吗
总结
​
next-token logits