Tiny Web LLM

Tiny Web LLM 实验

在浏览器中训练小型 next-token 模型并生成文本。

展示语言模型核心机制:分词、上下文窗口、logits、softmax 与基于梯度的更新。

这是一个很小的教学示例。你可以查看训练损失并比较 Greedy、Sampling、Top-k。

Token 由空格分词。模型对当前上下文的 token embedding 求平均并投影为词表 logits。

$$ h = \frac{1}{K}\sum E(t_i),\quad z = W^Th + b,\quad p = softmax(z) $$

训练使用逐 token 交叉熵与简单 SGD 更新。

1) 训练语料

加载示例:

2) 超参数

3) 训练运行

0%

                

4) 文本生成