大模型词元化算法详解:BPE、WordPiece与Unigram原理对比与实战选型
1. 项目概述:为什么词元化是大模型的第一道“工序”?
如果你最近在折腾大语言模型,无论是想自己微调一个,还是想深入理解ChatGPT、Claude这些“智能体”的内部运作,有一个概念你绝对绕不过去,那就是“词元化”。听起来有点学术?别怕,你可以把它想象成厨师做菜前的“备菜”过程。模型没法直接读懂我们写的“今天天气真好”这句话,它需要先把这句话切成一小块一小块它能处理的“食材”,这个过程就是词元化。
我刚开始接触时也犯过迷糊,觉得这不就是分词嘛,中文按词分,英文按空格分。但大模型的世界要复杂和精妙得多。你试试把“ChatGPT is amazing!”扔给模型,它可能不会切成
[“ChatGPT”, “is”, “amazing", "!”]
,更可能切成像
[“Chat”, “G”, “PT”, “ is”, “ amazing”, “!”]
这样奇怪的片段。这些片段就是“词元”。为什么这么切?这背后就是BPE、WordPiece、Unigram这些算法在发挥作用。它们决定了模型如何看待和理解文本,直接影响了模型的词汇表大小、训练效率、对生僻词或新造词的处理能力,甚至是最终生成文本的流畅度和准确性。
所以,无论你是开发者想要优化自己的模型输入管道,还是研究者希望理解不同模型架构的差异,或者仅仅是AI爱好者想弄明白这些技术热词,吃透词元化都是至关重要的一步。这篇文章,我就结合自己趟过的坑和读过的源码,把这三种主流词元化算法掰开揉碎了讲清楚,让你不仅知道它们是什么,更明白它们为什么这么设计,以及在实际项目中该如何选择和调优。
2. 核心概念与算法原理深度拆解
在深入算法之前,我们必须统一几个核心概念,否则后续的讨论会像鸡同鸭讲。
词元 :文本被切分后的基本单位。它可能是一个完整的词(如“apple”),一个子词(如“ing”),甚至是一个字符(如“a”)。词元是模型输入输出的直接对象。
词汇表 :一个模型所有可能词元的集合,通常是一个从词元到唯一ID的映射字典。词汇表的大小是一个关键超参数,通常在几万到几十万之间。太大了训练慢且容易过拟合,太小了表达能力不足。
词元化器 :执行切分算法的组件。它的核心功能有两个:1. 编码 :将原始文本字符串转换为一系列词元ID。2. 解码 :将词元ID序列转换回(尽可能接近的)原始文本。
现在,让我们进入正题,看看三大算法是如何构建词汇表和进行切分的。
2.1 Byte Pair Encoding:从数据压缩到NLP的经典迁移
BPE最早是一种数据压缩算法,核心思想非常直观: 迭代地合并最频繁共现的字节对 。OpenAI的GPT系列、早期的BERT都采用了BPE或其变种。
它的训练(构建词汇表)过程可以概括为以下几步:
-
初始化
:将训练语料库中的所有文本拆分成单个字符(或字节),作为初始词汇表。例如,“low”被拆成
[‘l’, ‘o’, ‘w’]。 -
统计与合并
:在整个语料库中,统计所有相邻符号对的出现频率。找到频率最高的那个对(比如
(‘l’, ‘o’)出现最多),将它们合并成一个新的符号(比如‘lo’),并将这个新符号加入词汇表。 - 迭代 :重复步骤2,直到合并了预定的次数(即词汇表达到了目标大小),或者直到下一个最高频对的频率为1。
我举个例子,假设我们的语料只有三个单词:“low”, “lower”, “newest”。初始词汇表是所有字符:
{‘l’, ‘o’, ‘w’, ‘e’, ‘r’, ‘n’, ‘s’, ‘t’}
。
-
第一轮:统计相邻对。
(‘l’, ‘o’)出现2次(在”low”和”lower”中),(‘o’, ‘w’)出现2次,(‘e’, ‘s’)出现1次等。假设(‘l’, ‘o’)胜出,我们合并得到‘lo’。现在词汇表加入‘lo’,语料变为:[‘lo’, ‘w’],[‘lo’, ‘w’, ‘e’, ‘r’],[‘n’, ‘e’, ‘w’, ‘e’, ‘s’, ‘t’]。 -
第二轮:现在
(‘lo’, ‘w’)出现了2次,合并为‘low’。词汇表加入‘low’。 -
继续迭代,我们可能会得到
‘low’,‘er’,‘new’,‘est’等子词。
编码(应用)过程 则稍有不同,它采用一种 贪婪匹配 策略。对于一个新词,比如“lowest”,词元化器会:
-
从词汇表中找到最长的、能与该词开头匹配的子词。假设词汇表里有
‘low’和‘est’。 -
匹配到
‘low’, 然后对剩余部分‘est’重复此过程,匹配到‘est’。 -
最终切分为
[‘low’, ‘est’]。如果‘est’不在词汇表,则会继续向下拆分成更小的单位,比如[‘e’, ‘s’, ‘t’], 直到所有部分都在词汇表中。
注意 :BPE的贪婪匹配有时会导致非最优切分。例如,对于“unhappily”,如果词汇表里有
‘un’,‘happy’,‘ly’, 理想切分是[‘un’, ‘happy’, ‘ly’]。但如果词汇表里还有一个更长的‘unh’, 贪婪算法会先匹配‘unh’, 导致剩余部分‘appily’无法正确处理,最终可能退化成字符级切分。这是BPE的一个已知缺点。
BPE的优势在于能有效地在词级和字符级之间取得平衡,对常见词保留完整形式,对生僻词则分解为有意义的子词,大大缓解了未登录词问题。它的实现相对简单,是很多开源项目的首选。
2.2 WordPiece:BERT的“御用”分词器
WordPiece是Google为BERT模型量身定做的算法。如果你用过
transformers
库里的
BertTokenizer
, 那你用的就是WordPiece。它和BPE非常相似,核心区别在于
合并符号对的标准不同
。
BPE合并最高频的符号对,而WordPiece合并能 最大程度提升语言模型似然值 的符号对。具体来说:
在训练过程中,假设我们有一个当前的词汇表和基于该词汇表训练的一个简单的语言模型(比如一元模型)。对于每一个可能的符号对
(A, B)
, 我们计算如果将它们合并为新符号
AB
, 会对整个训练语料的似然值产生多大提升。选择提升最大的那个对进行合并。
这个计算过程可以近似为以下公式:
score = freq(A, B) / (freq(A) * freq(B))
。它衡量的是A和B的共现频率是否远高于它们独立出现的频率的乘积(即是否显著相关)。如果
score
很高,说明A和B经常紧挨着出现,合并它们对语言模型有利。
还是用“low”, “lower”, “newest”的例子。假设当前词汇表是字符级。
(‘l’, ‘o’)
的共现频率是2,
freq(‘l’)
和
freq(‘o’)
可能都是2或3。
(‘e’, ‘s’)
的共现频率是1。WordPiece会计算每个对的
score
, 而不仅仅是看频率。可能
(‘e’, ‘s’)
虽然频率低,但
‘e’
和
‘s’
单独出现也少,其
score
反而更高,从而被优先合并。
编码过程
, WordPiece采用与BPE类似的
最长匹配优先
策略,但有一个关键补充:
它会在词前添加特殊前缀
(如
##
)。例如,一个词中间的片段“##ing”表示“ing”是一个后缀。对于“playing”, 可能被切分为
[‘play’, ‘##ing’]
。这样做的好处是,解码时能明确知道哪些词元是词的开头,哪些是词的中间或结尾部分,有助于模型更好地理解词的边界信息。
WordPiece的这种基于似然提升的合并策略,理论上能产生对语言建模更有效的词汇表。BERT在英语任务上的卓越表现,部分也归功于其精心设计的WordPiece词元化。
2.3 Unigram Language Model:一种概率驱动的全新视角
与前两种基于合并的“自底向上”方法不同,Unigram算法是一种“自顶向下”的概率模型。它的核心思想是: 预先定义一个很大的种子词汇表(比如所有字符和常见子串),然后通过迭代淘汰概率贡献低的词元,来得到一个最优大小的词汇表。
它的训练过程更像是一个剪枝优化问题:
- 初始化 :用一个足够大的种子词汇表覆盖训练语料,例如,可以用BPE先跑出一个很大的词汇表(比如10万)。
-
训练语言模型
:基于当前词汇表,训练一个一元语言模型。这个模型给每个词元
x分配一个概率p(x), 并且所有词元的概率之和为1。同时,对于任何一个句子,它的概率可以表示为所有可能切分方式中,各切分词元概率乘积的最大值(或和,取决于目标)。 -
评估损失
:计算如果从词汇表中移除某个词元
x, 对整个训练语料库的似然值会造成多大损失(例如,使用期望似然损失)。 - 剪枝 :移除那些造成损失最小的词元(即对整体似然贡献最小的词元)。通常,会按损失排序,移除一定比例(比如10%)的词元。
- 迭代 :用新的、缩小了的词汇表,重复步骤2-4,直到词汇表达到目标大小。
编码过程 也独具特色。给定一个训练好的Unigram词元化器(包含词汇表和每个词元的概率),对一个新词进行切分时,它会 找出所有可能的切分方式,然后选择概率乘积最大的那种切分 。这通常通过 维特比算法 来实现,是一种全局最优搜索,而不是BPE/WordPiece的贪婪局部最优。
例如,对于“unhappily”, 假设词汇表里有
{‘un’, ‘happ’, ‘ily’, ‘happy’, ‘ly’, ‘unh’, ‘app’, ‘il’}
等。Unigram会计算:
-
切分1:
[‘un’, ‘happ’, ‘ily’]的概率 = p(‘un’) * p(‘happ’) * p(‘ily’) -
切分2:
[‘un’, ‘happy’, ‘ly’]的概率 = p(‘un’) * p(‘happy’) * p(‘ly’) -
切分3:
[‘unh’, ‘app’, ‘ily’]的概率 = … 然后选择概率最高的切分作为结果。
Unigram的优势在于其灵活性。由于它基于概率模型,可以轻松地处理同一输入有多种有效切分的情况(这是语言中常见的歧义),并且通过调整词元概率,可以引导模型偏好更短或更长的切分。SentencePiece工具包中的
--model_type=unigram
模式就是这种算法的实现。
3. 三大算法对比与实战选型指南
了解了原理,我们放到一起对比一下,这就像选工具,得知道每把扳手适合拧什么螺丝。
| 特性维度 | BPE (Byte Pair Encoding) | WordPiece | Unigram Language Model |
|---|---|---|---|
| 核心思想 | 迭代合并最高频字节对 | 迭代合并能最大提升语言模型似然的符号对 | 从大种子词汇表出发,迭代剪枝贡献低的词元 |
| 训练方向 | 自底向上 (字符 -> 子词) | 自底向上 (字符 -> 子词) | 自顶向下 (大词汇表 -> 小词汇表) |
| 合并/选择标准 | 频率最高 |
互信息最大 (
freq(A,B)/(freq(A)*freq(B))
)
| 移除对整体似然损失最小的词元 |
| 编码策略 | 贪婪最长匹配 |
贪婪最长匹配 (带
##
前缀标识)
| 全局最大概率切分 (维特比算法) |
| 主要优势 | 实现简单,效率高,能有效平衡词与字符 | 合并标准更贴合语言模型目标,BERT验证有效 | 切分结果全局最优,灵活支持概率采样和多切分 |
| 主要劣势 | 贪婪匹配可能导致非最优切分 | 与BPE类似,也存在贪婪匹配的局部最优问题 | 训练过程更复杂,计算量相对较大 |
| 代表模型/工具 | GPT-2, GPT-3, RoBERTa (使用BPE变种) | BERT, DistilBERT | SentencePiece (Unigram模式), XLNet |
实战选型建议:
-
如果你要复现或微调BERT系列模型 :无脑选择WordPiece。使用Hugging Face
transformers库中的BertTokenizer是最省事、最兼容的做法。自己从头训练WordPiece词汇表可以使用tokenizers库(也是Hugging Face出品)。 -
如果你要处理GPT系列或类似的自回归模型 :BPE是更常见的选择。OpenAI的
tiktoken(用于GPT)就是BPE的高效实现。你也可以通过tokenizers库训练自己的BPE词元化器。 -
如果你的文本非常多样化、包含多语言或者大量未知领域术语 :强烈推荐考虑 Unigram 算法,特别是通过 SentencePiece 工具来实现。SentencePiece的Unigram模式有几个杀手级优势:
- 无需预分词 :它直接把文本当作Unicode字符序列处理,省去了针对不同语言设计分词器的麻烦,对中文、日文、代码混合文本等尤其友好。
- 子词正则化 :这是一个宝藏功能。在训练时,它可以随机采样多种可能的切分(而不是只用概率最高的一种),相当于一种数据增强,能提升模型的鲁棒性。
- 全局最优切分 :编码时寻找概率最大的切分,理论上比贪婪算法更合理。
-
对于绝大多数新的、从零开始的实验项目 :我个人更倾向于使用 SentencePiece with Unigram 。它的通用性最强,能减少很多预处理上的坑,尤其是在处理用户生成内容、社交媒体文本、混合语言场景时。虽然训练慢一点,但一劳永逸。
实操心得 :别盲目追求“最好”的算法。先花点时间用你的目标领域数据(比如医疗报告、法律文书、科技论坛帖子),分别用BPE和Unigram(SentencePiece)训练两个小规模的词汇表(比如3万大小),然后手动检查一些典型句子的切分结果。看看哪个算法产生的词元对你来说更“顺眼”、更“有意义”。这个简单的测试往往比理论对比更有指导价值。
4. 使用SentencePiece进行实战训练与调优
理论说再多,不如动手跑一遍。这里我以目前最流行、功能最全面的 SentencePiece 工具为例,展示如何从原始文本训练一个词元化模型,并详细解释关键参数。
4.1 环境准备与数据预处理
首先,安装SentencePiece。它既是一个独立的C++库,也提供了Python封装。
pip install sentencepiece
准备你的训练数据。数据应该是一个纯文本文件(例如
corpus.txt
), 每一行是一个句子或文档。数据质量至关重要:
- 清洗 :移除过多的空格、乱码、HTML标签等。但不必过度清洗,标点符号、大小写通常应保留,除非你有特殊理由要规范化。
- 规模 :对于训练一个通用的词汇表,几十MB到几GB的文本是常见的。数据越多,覆盖的词汇和语言现象越全。
- 代表性 :确保你的数据分布与模型未来要处理的数据分布一致。如果你做医疗问答,就用医学文献和问答记录,而不是用小说。
4.2 训练模型与核心参数解析
训练一个SentencePiece模型非常简单,主要参数都体现在下面这个命令或Python脚本中。
Python脚本方式(更灵活):
import sentencepiece as spm
# 定义训练参数
spm.SentencePieceTrainer.train(
input=‘corpus.txt‘, # 输入文件
model_prefix=‘sp_model‘, # 输出模型名前缀(会生成.spm.model和.spm.vocab)
vocab_size=32000, # 目标词汇表大小,这是最重要的参数之一
model_type=‘unigram‘, # 可选 ‘unigram‘, ‘bpe‘, ‘char‘, ‘word‘
character_coverage=0.9995, # 字符覆盖度,对于多语言或生僻字多的文本可调低(如0.98)
max_sentence_length=16384, # 单句最大长度,过滤超长句
pad_id=0, # pad token id, 通常为0
unk_id=1, # unknown token id, 通常为1
bos_id=2, # beginning of sentence token id, 通常为2
eos_id=3, # end of sentence token id, 通常为3
user_defined_symbols=[‘[CLS]‘, ‘[SEP]‘, ‘[MASK]‘], # 添加自定义特殊符号
split_digits=True, # 将数字单独拆开,提升对数字的泛化能力
byte_fallback=True, # 对于未知字符,回退到UTF-8字节表示,确保100%覆盖
)
关键参数详解:
-
vocab_size: 词汇表大小。这是 最需要权衡 的参数。常见范围在1.6万到10万之间。- 太小(<1.6万) : 词元粒度太粗,未登录词多,模型表达能力弱。
- 适中(3.2万-5万) : 大多数单语言模型的甜点区,在效率和表达间取得良好平衡。BERT-base是3万。
- 太大(>10万) : 词元粒度细,接近字符级,能很好处理生僻词,但会导致输入序列变长(因为一个词可能被拆成很多词元),增加计算开销,且模型需要学习更多嵌入参数。
-
model_type: 如前所述,‘unigram‘(默认)或‘bpe‘。对于大多数情况,unigram是更好的选择。 -
character_coverage: 字符覆盖度。设置为1.0意味着要求模型覆盖输入中所有出现的字符。如果你的语料包含很多非常罕见的Unicode字符(如某些特殊数学符号、生僻汉字),将其设置为1.0会导致词汇表中充斥大量仅出现一次的字符词元,挤占更有用子词的空间。对于像日语、中文这种字符集大的语言,通常设置为0.9995或0.999。对于英文,1.0通常没问题。 -
split_digits: 强烈建议设置为True 。这会把所有数字(0-9)都当作独立的词元拆开。例如,“123”会被切分成[‘1‘, ‘2‘, ‘3‘]。这极大地提升了模型对任意数字组合的泛化能力,而无需在词汇表中为“123”、“4567”等每一个数字组合都保留一个位置。 -
byte_fallback: 另一个强烈建议开启的选项 。当遇到一个完全不在训练字符集中的字符时,SentencePiece会将其回退到UTF-8字节序列来表示。例如,一个罕见表情符号可能会被编码为几个字节词元。这保证了 任何文本都能被无损(或近乎无损)地编码和解码 ,彻底解决了未登录字符问题。这是SentencePiece相比传统BPE/WordPiece的一大优势。 -
user_defined_symbols: 如果你需要一些特殊控制符号,如[CLS],[SEP],[MASK](用于BERT),<|endoftext|>(用于GPT), 一定要在这里加入。这些符号会被强制加入词汇表,并且 永远不会被拆分 。
训练完成后,你会得到两个文件:
sp_model.model
(模型文件)和
sp_model.vocab
(可读的词汇表文件)。
4.3 加载模型与编码解码实践
训练好模型后,就可以加载并使用它了。
import sentencepiece as spm
# 加载模型
sp = spm.SentencePieceProcessor()
sp.load(‘sp_model.model‘)
# 编码:文本 -> 词元ID列表
text = “大语言模型的词元化处理详解!”
ids = sp.encode_as_ids(text)
print(“Token IDs:”, ids)
# 输出可能类似:[2345, 123, 4567, 7890, 12, 345]
# 编码:文本 -> 词元字符串列表
pieces = sp.encode_as_pieces(text)
print(“Tokens:”, pieces)
# 输出可能类似:[‘▁大‘, ‘语言‘, ‘模型‘, ‘的‘, ‘词元‘, ‘化‘, ‘处理‘, ‘详解‘, ‘!’]
# 注意:`▁` 是SentencePiece用来表示空格或词开始的特殊符号(可替换)。
# 解码:词元ID列表 -> 文本
decoded_text = sp.decode_ids(ids)
print(“Decoded:”, decoded_text) # 应尽可能接近原始文本
# 解码:词元字符串列表 -> 文本
decoded_text_from_pieces = sp.decode_pieces(pieces)
print(“Decoded from pieces:”, decoded_text_from_pieces)
# 查看词汇表信息
print(f“词汇表大小: {sp.get_piece_size()}”) # 输出 32000
print(f”Unknown token ID: {sp.unk_id()}”) # 输出 1
print(f”Pad token ID: {sp.pad_id()}”) # 输出 0
编码输出中的
▁
符号
:这是SentencePiece的默认行为,它用一个特殊符号(通常是
▁
, U+2581)来表示空格或词的开始。这样做的好处是,解码时能准确恢复空格。如果你觉得这个符号在调试时碍眼,可以在训练时通过
--add_dummy_prefix=false
和
--remove_extra_whitespaces=false
等参数调整,但可能会影响空格处理的准确性。通常建议保留默认设置。
5. 高级话题与性能优化陷阱
当你掌握了基础训练和用法后,在实际部署和优化中,还会遇到一些更深层次的问题。
5.1 词汇表大小的影响:一个具体的权衡实验
词汇表大小
vocab_size
不是随便设的。为了让你有直观感受,我分享一个在英文维基百科数据上的小实验片段:
我们训练了四个Unigram模型,词汇表大小分别为8k, 16k, 32k, 64k。在同一批测试句上:
- 平均词元长度 : 随着词汇表增大,每个词元平均代表的字符数增加。8k模型下,平均每个词元约4.2个字符;64k模型下,约5.8个字符。这意味着 大词汇表能产生更长的词元,从而缩短序列长度 。
- 序列长度(词元数) : 对于同一段文本,64k模型产生的词元数量比8k模型少约30%。 序列越短,模型训练和推理时注意力机制的计算量就越小,速度越快 。
-
未登录词率
: 在领域内测试集上,8k模型的未登录词(被拆成
<unk>或大量子词)比例明显高于64k模型。 -
模型参数
: 词表嵌入层参数 =
vocab_size * hidden_dim。假设hidden_dim=768, 从32k增加到64k,仅嵌入层就多了约2400万个参数。 模型总参数量会增大,占用更多显存 。
结论与建议 :对于通用英文模型, 32k是一个经过充分验证的甜点值 。对于中文,由于汉字本身是字符,且常用字有限,词汇表可以稍小(如22k),但考虑到中文词汇的丰富性,32k也是一个安全的选择。如果你的领域有大量专业术语(如生物医学、法律),可以适当增大到5万-6万。 不要盲目追求超大词表 ,除非你确信你的数据和应用能从中受益,并且有足够的算力支撑。
5.2 处理多语言与代码混合文本
这是当前大模型面临的一个现实挑战。你的语料里可能同时有中文、英文、数字、公式和Python代码。
策略一:统一词表
。使用SentencePiece,将所有语言的文本混合在一起训练一个大的词表。这是最主流、最简单的方法。SentencePiece的
byte_fallback
和
character_coverage
参数在这里至关重要。确保
character_coverage
设置能覆盖所有语言用到的字符(例如,中文字符集很大,可能需要调低到0.9995)。
策略二:语言标识符
。在训练时,可以在每条文本前加上语言标签,如
[EN]
,
[ZH]
。然后将这些标签也加入
user_defined_symbols
。这样模型在编码时,第一个词元就是语言ID,理论上可以帮助模型进行语言间的切换。但实践表明,对于强大的Transformer模型,统一词表通常足够。
对于代码
:关键参数是
split_digits=True
和保持标点符号。代码中有大量数字、变量名(如
user_id
)、操作符(
+=
,
->
)。开启
split_digits
能让模型更好地泛化到不同的数字。变量名中的下划线通常会被保留或作为词元的一部分。
5.3 词元化器的性能瓶颈与优化
在生产环境中,词元化可能成为API服务或数据预处理管道的瓶颈,尤其是处理海量流式文本时。
-
Python vs. C++ : SentencePiece的Python接口在循环中编码大量短文本时,由于Python解释器开销,可能较慢。解决方案是 批量编码 。不要在一个for循环里反复调用
sp.encode(),而是将文本收集到一个列表中,一次性调用sp.encode_batch()或sp.encode_as_ids_batch()。 -
使用Hugging Face
tokenizers库 : 这是Hugging Face用Rust重写的高性能词元化库,支持BPE、WordPiece等多种算法,并且与transformers库无缝集成。它的速度通常比纯Python实现快一个数量级,且支持多线程。如果你的流水线是Python的,并且需要与Hugging Face生态兼容,tokenizers是更好的选择。from tokenizers import Tokenizer, models, trainers, pre_tokenizers, decoders # 使用BPE算法 tokenizer = Tokenizer(models.BPE()) # 配置训练器并训练... # 使用起来非常高效 output = tokenizer.encode_batch([“Hello world!”, “Another text.”]) -
缓存 : 对于重复出现的固定文本(如系统提示词、模板),将其编码结果缓存起来,避免重复计算。
-
序列化与加载 : 训练好的SentencePiece模型(
.model文件)加载很快。确保你的服务在启动时只加载一次,而不是每次请求都加载。
6. 常见问题排查与调试技巧实录
在实际操作中,你肯定会遇到各种奇怪的问题。这里记录了几个我踩过的坑和解决方法。
问题1:编码后再解码,文本和原本文本不一致,多了或少了空格。
-
原因
: 这几乎总是空格处理问题。SentencePiece默认用
▁表示词首空格。解码时,它会将▁转换为普通空格。但如果你的原始文本有连续多个空格、制表符或其他空白字符,SentencePiece在训练和编码时可能对其进行了规范化(比如合并多个空格为一个)。 -
排查
: 打印出
encode_as_pieces的结果,仔细查看词元列表。注意第一个词元前是否有▁, 以及词元内部是否包含空格。 -
解决
:
- 如果必须完全无损还原,SentencePiece可能不是最佳选择,可以考虑字符级词元化或字节级BPE。
-
对于大多数NLP任务,文本中额外的空格通常不影响语义,可以接受这种规范化。你可以在训练时调整
--normalization_rule_name和--remove_extra_whitespaces参数来控制行为。
问题2:遇到罕见emoji或特殊符号,模型输出一堆奇怪的
<0xXX>
词元。
-
原因
: 这是
byte_fallback机制在起作用。当遇到训练时未见过的字符时,它被分解为UTF-8字节,每个字节被表示为一个形如<0xE7>的词元。 -
排查
: 确认训练语料的
character_coverage是否设置得太高,导致这些罕见字符没有被收入词汇表。 -
解决
:
- 这通常 不是问题,而是特性 。这种表示确保了任何文本都可处理。模型会学习这些字节词元的组合。
-
如果你希望这些符号成为独立词元,你需要确保包含这些符号的文本出现在训练数据中,并且
character_coverage设置得足够高以覆盖它们。
问题3:模型在处理数字时表现很差,总是把“123.45”切分成奇怪的样子。
-
原因
: 没有开启
split_digits, 并且数字组合“123.45”可能没有出现在训练数据中,因此被拆成了更随机的子词。 -
解决
:
训练时务必加上
--split_digits=True。这能保证所有数字字符都被单独分开,模型会学会数字、小数点之间的通用组合模式,从而完美处理任何未见过的数字。
问题4:我自己训练的词汇表,在加载到Hugging Face Transformers里时出错。
-
原因
: Hugging Face Transformers对词元化器有一些预期,比如特殊的
[PAD],[UNK],[CLS],[SEP],[MASK]等符号的ID必须正确映射。 -
解决
:
-
在训练SentencePiece时,通过
pad_id,unk_id,bos_id,eos_id明确指定这些ID。通常保持默认(0, 1, 2, 3)即可,并与Transformers的配置对齐。 -
通过
user_defined_symbols参数明确添加所有你需要的特殊符号。 -
使用Hugging Face的
tokenizers库来训练,它能生成与Transformers完全兼容的词元化器文件(tokenizer.json)。
-
在训练SentencePiece时,通过
问题5:训练时内存溢出(OOM),语料太大。
- 原因 : SentencePiece需要将语料全部读入内存进行处理。
-
解决
:
- 采样 : 如果语料极大(如TB级),不要用全部数据。随机采样1%-10%的数据通常就能训练出高质量的词汇表。
- 分片 : 如果必须用全部数据,且内存不足,可以先将语料分成多个小文件,分别训练词表,然后合并词频统计信息,但这比较麻烦。更推荐方法1。
-
调整
--input_sentence_size: 这个参数控制用于训练的最大句子数,默认是0(全部)。可以设置为一个较大的数(如1000万)来限制内存使用。
词元化作为大模型流水线的第一步,其重要性怎么强调都不为过。一个设计良好的词元化器,就像一把锋利的厨刀,能让后续的模型“烹饪”事半功倍。希望这篇近万字的详解,能帮你从原理到实践,彻底掌握BPE、WordPiece和Unigram,为你自己的大模型项目打下坚实的基础。记住,没有“放之四海而皆准”的最佳算法,最适合你数据的那个,才是最好的。动手去实验,去分析切分结果,你的直觉会在这个过程中变得越来越准。
更多推荐


所有评论(0)