模型改造与架构搭建
构建适合字节化的潜分词语言模型架构
在字节嵌入中加入最长的子词后缀嵌入,局部编码器用mLSTM层,非因果边界预测器使用1字节前瞻,池化为patch表示,经冻结的全局Transformer后解池并由局部解码器处理,语言建模头预测下一字节及边界
Retrofitting language models to operate over bytes
含完整架构与两阶段训练流程、多规模模型改造、多基准评测及压缩效率和任务算术后训练验证
Bolmo 7B和Bolmo 1B字节级模型 Bwen 8B和Blama 8B字节级模型 源子词级模型(Olmo 3 7B、OLMo 2 1B、Qwen3 8B Base、Llama 3 8B)
两阶段训练:阶段1使用9.8B tokens,阶段2使用39.3B tokens,总计49.1B tokens 阶段1损失权重λB=4、λE=1、λD,Distill=1、λD,CE=1,蒸馏温度τ=5 边界预测器使用1字节前瞻的余弦距离,局部编码器、解码器采用mLSTM层 评测采样设置温度=0.6、top_p=0.6,统计检验采用10,000次bootstrap重采样及Holm–Bonferroni校正 数据混合约172B tokens,另加约75M tokens的CUTE式字符级数据(词表150,000词)
提炼研究问题、关键发现与证据,快速把握文章的核心贡献。
按研究推进顺序梳理实验设计、验证步骤与关键观察。
构建适合字节化的潜分词语言模型架构
在字节嵌入中加入最长的子词后缀嵌入,局部编码器用mLSTM层,非因果边界预测器使用1字节前瞻,池化为patch表示,经冻结的全局Transformer后解池并由局部解码器处理,语言建模头预测下一字节及边界
以最小成本使字节级模型复现源子词模型行为
冻结全局模型,训练局部编码器、解码器、边界预测器和语言建模头,使用边界损失、编码器L2损失与蒸馏损失及交叉熵损失的组合
使模型适应端到端设定并学习利用字节级信息
全部参数(含全局模型)联合训练,仅保留边界损失与交叉熵损失,可选提高压缩比
比较字节化模型与源模型及早期字节级模型,并评估压缩效率与后训练迁移
在7B和1B评测套件上评测字符理解、代码、数学、STEM等任务,测试SuperBPE与不同合并策略下的压缩效率,并用任务算术从RL-Zero检查点向Bolmo迁移指令遵循能力
按研究目的归类文中使用的方法,便于定位所需技术。
| 实验环节 | 名称 | 品牌 | 货号 |
|---|---|---|---|
| 模型训练与推理 | PyTorch 2.8 | -- | -- |
| 文本生成 | vLLM 0.11.0 | -- | -- |
| 分词训练 | FOCUS | -- | -- |
| 字符级数据生成 | CUTE代码库 | -- | -- |
| 性能测量 | H100 GPU | -- | -- |
汇总复现实验时建议重点确认的条件及原文阅读提示。
| 环节 | 核对要点 |
|---|---|
| 两阶段字节化训练 | 阶段1使用9.8B tokens、阶段2使用39.3B tokens,总计49.1B tokens;数据混合约172B tokens加约75M tokens CUTE式数据;损失权重λB=4、λE=1、λD,Distill=1、λD,CE=1,蒸馏温度τ=5 阅读提示:查阅Methods中Experimental set-up的Data与Training小节及Extended Data Table 4 |
| 边界预测器设计 | 非因果边界预测使用1字节前瞻的余弦距离,解码时由局部解码器输出边界符号;字节词表由256扩展为512 阅读提示:查阅Byteified language model architecture小节及Fig. 1 |
| 评测与统计检验 | 7B与1B评测套件组成、采样设置温度=0.6与top_p=0.6、bootstrap次数N=10,000及Holm–Bonferroni校正 阅读提示:查阅Methods中Evaluation与Statistical tests小节及Extended Data Tables 1和2 |
| 效率与压缩实验 | SuperBPE词表大小200k和400k、10-GB文本样本、每点150,000步训练、效率测量batch size=1于H100 GPU 阅读提示:查阅Increasing model efficiency小节、Fig. 2与Extended Data Fig. 3 |
| 任务算术后训练 | 使用Olmo 3 RL-Zero检查点,将后训练检查点与基础Olmo 3的Transformer层权重差加到Bolmo对应层,评测IFEval(n=834,10,000次bootstrap) 阅读提示:查阅Post-training byteified models through task arithmetic小节与Fig. 3 |