分词与 NLP 基础:BPE、SentencePiece 和 WordPiece
分词是将文本拆分为模型可处理的子单元(tokens)的过程。这一过程构成了现代 LLM 的基础,并直接影响模型性能。
什么是分词?
分词是将原始文本转换为数值表示的第一步:
"Hello world!" → ["Hello", "world", "!"] → [1234, 5678, 99]
分词级别
- 字符级: 每个字符是一个 token。
- 词级: 每个单词是一个 token。
- 子词级: 将单词拆分为更小的子单元(现代方法)。
词级分词
简单方法
1def word_tokenize(text): 2 return text.split() 3 4# Example 5text = "Artificial intelligence is shaping the future" 6tokens = word_tokenize(text) 7# ['Artificial', 'intelligence', 'is', 'shaping', 'the', 'future']
问题
- OOV(超出词表): 遇到训练时未出现的单词。
- 词表过大: 管理几十万单词非常低效。
- 形态变化丰富: 在土耳其语等语言中,因后缀变化会出现大量词形。
- 复合词问题: “Artificial intelligence” 应视为一个概念还是两个?
字符级分词
1def char_tokenize(text): 2 return list(text) 3 4# Example 5text = "Hello" 6tokens = char_tokenize(text) 7# ['H', 'e', 'l', 'l', 'o']
优点
- 没有 OOV 问题。
- 词表很小(约 100 个字符)。
缺点
- 序列非常长。
- token 层面语义缺失。
- 模型计算成本更高。
子词分词
现代 LLM 的选择:介于词级与字符级之间的平衡方案。
"tokenization" → ["token", "ization"] "unhappiness" → ["un", "happiness"] 或 ["un", "happy", "ness"]
BPE(Byte Pair Encoding)
最广泛使用的子词分词算法。
BPE 算法
- 将文本拆分为单独字符。
- 找到最频繁的相邻字符对。
- 将该字符对合并为一个新的 token。
- 重复此过程直到达到所需词表大小。
BPE 示例
1Starting vocabulary: ['l', 'o', 'w', 'e', 'r', 'n', 's', 't', 'i', 'd'] 2Corpus: "low lower newest lowest widest" 3 4Step 1: Most frequent pair 'e' + 's' → 'es' 5Step 2: Most frequent pair 'es' + 't' → 'est' 6Step 3: Most frequent pair 'l' + 'o' → 'lo' 7Step 4: Most frequent pair 'lo' + 'w' → 'low' 8... 9 10Final Result: ['low', 'est', 'er', 'new', 'wid', ...]
BPE 实现
1def get_stats(vocab): 2 pairs = {} 3 for word, freq in vocab.items(): 4 symbols = word.split() 5 for i in range(len(symbols) - 1): 6 pair = (symbols[i], symbols[i + 1]) 7 pairs[pair] = pairs.get(pair, 0) + freq 8 return pairs 9 10def merge_vocab(pair, vocab): 11 new_vocab = {} 12 bigram = ' '.join(pair) 13 replacement = ''.join(pair) 14 for word in vocab: 15 new_word = word.replace(bigram, replacement) 16 new_vocab[new_word] = vocab[word] 17 return new_vocab 18 19def train_bpe(corpus, num_merges): 20 vocab = get_initial_vocab(corpus) 21 22 for i in range(num_merges): 23 pairs = get_stats(vocab) 24 if not pairs: 25 break 26 best_pair = max(pairs, key=pairs.get) 27 vocab = merge_vocab(best_pair, vocab) 28 29 return vocab
WordPiece
由 Google 开发并用于 BERT 等模型的算法。
BPE vs WordPiece
| Feature | BPE | WordPiece |
|---|---|---|
| Merge Criterion | Frequency | Likelihood |
| Prefix | None | ## (for mid-word tokens) |
| Used In | GPT, LLaMA | BERT, DistilBERT |
WordPiece 示例
1"tokenization" → ["token", "##ization"] 2"playing" → ["play", "##ing"] 3## SentencePiece 4 5Google 开发的一种与语言无关的 tokenizer。 6 7### Features 8 9- **语言无关:** 不假设空格是词语分隔符。 10- **字节级:** 直接作用于原始文本。 11- **BPE + Unigram:** 支持多种算法。 12- **可逆:** 可以实现完美的逆向解码。 13 14### SentencePiece Usage 15 16```python 17import sentencepiece as spm 18 19# Training the model 20spm.SentencePieceTrainer.train( 21 input='corpus.txt', 22 model_prefix='tokenizer', 23 vocab_size=32000, 24 model_type='bpe' # or 'unigram' 25) 26 27# Loading and using the model 28sp = spm.SentencePieceProcessor() 29sp.load('tokenizer.model') 30 31# Encode 32tokens = sp.encode('Hello world', out_type=str) 33# ['▁Hello', '▁world'] 34 35ids = sp.encode('Hello world', out_type=int) 36# [1234, 5678, 9012] 37 38# Decode 39text = sp.decode(ids) 40# 'Hello world'
▁(下划线)符号
SentencePiece 使用 ▁ 标记词语开头:
"Hello world" → ["▁Hello", "▁world"] "New York" → ["▁New", "▁York"]
Tiktoken (OpenAI)
OpenAI 使用的专用 BPE 实现。
1import tiktoken 2 3# Loading the encoder 4enc = tiktoken.encoding_for_model("gpt-4") 5 6# Encode 7tokens = enc.encode("Hello world!") 8# [12345, 67890, 999] 9 10# Decode 11text = enc.decode(tokens) 12# "Hello world!" 13 14# Check token count 15print(len(tokens)) # 3
Model-Encoder Mappings
| Model | Encoder | Vocab Size |
|---|---|---|
| GPT-4 | cl100k_base | 100,277 |
| GPT-3.5 | cl100k_base | 100,277 |
| GPT-3 | p50k_base | 50,281 |
| Codex | p50k_edit | 50,281 |
Hugging Face Tokenizers
1from transformers import AutoTokenizer 2 3# Loading the tokenizer 4tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased") 5 6# Encode 7encoded = tokenizer("Hello, world!", return_tensors="pt") 8# { 9# 'input_ids': tensor([[101, 7592, 1010, 2088, 999, 102]]), 10# 'attention_mask': tensor([[1, 1, 1, 1, 1, 1]]) 11# } 12 13# Decode 14text = tokenizer.decode(encoded['input_ids'][0]) 15# "[CLS] hello, world! [SEP]" 16 17# Token List 18tokens = tokenizer.tokenize("Hello, world!") 19# ['hello', ',', 'world', '!']
Fast Tokenizers
1from tokenizers import Tokenizer, models, trainers, pre_tokenizers 2 3# Creating a new tokenizer 4tokenizer = Tokenizer(models.BPE()) 5tokenizer.pre_tokenizer = pre_tokenizers.Whitespace() 6 7trainer = trainers.BpeTrainer( 8 vocab_size=30000, 9 special_tokens=["[PAD]", "[UNK]", "[CLS]", "[SEP]", "[MASK]"] 10) 11 12tokenizer.train(files=["corpus.txt"], trainer=trainer) 13tokenizer.save("my_tokenizer.json")
Special Tokens
Common Special Tokens
| Token | Description | Use Case |
|---|---|---|
| [CLS] | 序列起始 | BERT 分类任务 |
| [SEP] | 片段分隔符 | 用于分隔句子对 |
| [PAD] | 填充 | 批处理对齐 |
| [UNK] | 未知词元 | 处理未登录词 |
| [MASK] | 掩码 | 掩码语言模型(MLM) |
| <|endoftext|> | 序列结束 | GPT 生成任务 |
Chat Tokens
1<|system|>You are a helpful assistant<|end|> 2<|user|>Hello!<|end|> 3<|assistant|>Hello! How can I help you today?<|end|>
Tokenization Challenges in Turkish
Morphological Richness
1"gelebileceklermiş" (they were said to be able to come) → A single word but complex structure 2gel (come) + ebil (can) + ecek (will) + ler (they) + miş (reportedly) 3 4Tokenization: 5- Poor: ["gelebileceklermiş"] (Single token, very rare) 6- Good: ["gel", "ebil", "ecek", "ler", "miş"]
Solutions
- 面向土耳其语优化的 tokenizer 训练。
- 结合形态学分析。
- 后缀感知的 BPE 应用。
Token 限制与管理
上下文窗口
| Model | 上下文长度 (Tokens) | 约当单词数量 |
|---|---|---|
| GPT-3.5 | 16K | ~12,000 |
| GPT-4 | 128K | ~96,000 |
| Claude 3 | 200K | ~150,000 |
Token 数量估算
1def estimate_tokens(text): 2 # Rough estimate: 1 token ≈ 4 characters (English) 3 # For Turkish: 1 token ≈ 3 characters 4 return len(text) // 3 5 6# More accurate calculation 7def count_tokens(text, model="gpt-4"): 8 enc = tiktoken.encoding_for_model(model) 9 return len(enc.encode(text))
结论
Tokenization 是 NLP 和 LLM 的基础构建模块。BPE、WordPiece 和 SentencePiece 等子词方法在现代语言模型的成功中发挥关键作用。选择并配置合适的 tokenizer 会直接影响模型的最终性能。
在 Veni AI,我们提供专为土耳其语 NLP 解决方案优化的 tokenization 策略。
