OC

Knowledge OS
分词也能成为瓶颈:Gigatoken 把大模型最不起眼的一步跑到 GB/s
科技 · 2026-07-23 · AI 工程 · 阅读 1

分词也能成为瓶颈:Gigatoken 把大模型最不起眼的一步跑到 GB/s

作者:林岚|OC 开发者生态编辑

作者:林岚|OC 开发者生态编辑

Gigatoken GitHub 项目 介绍,这个 Rust 实现的分词工具在多核 CPU 上针对多种 BPE tokenizer 做了深度优化。项目基准显示,在 AMD EPYC 9565 上,部分 tokenizer 可达到 24.5GB/s 级别;在 Apple M4 Max 上,示例命令显示可处理约 8.3GB/s 文本。

一句话结论:分词不是 AI 里最性感的环节,但当上下文、RAG 和训练数据规模变大,它会从“小工具”变成真实成本。

普通用户很少关心 tokenizer。开发者也常把它当成模型前面的一个函数:文本进去,token 出来。但在大规模场景里,这一步会反复发生。训练前要分词,RAG 入库要分词,长文档处理要分词,日志和网页清洗也要分词。数据越大,原来“不值得优化”的环节就越显眼。

Gigatoken 的作者把优化点放在 SIMD、缓存层级、减少 Python 交互、并行处理和 pretokenization 上。项目 README 里有一个很夸张但有启发的例子:在 EPYC CPU 的速度下,理论上可以在数小时级别处理 Common Crawl 规模的 tokenization。

分词流水线性能瓶颈示意

当然,OC 不建议把 README 基准直接当成所有场景的性能承诺。不同 tokenizer、不同文本分布、不同硬件、是否需要验证一致性,都会影响结果。项目自己也说明 SentencePiece 类 tokenizer 优化不足,Windows 测试也不充分。

但这个项目提醒开发者一件很实际的事:AI 工程不是只有模型推理。很多时候,你的钱和时间花在“模型周边”:分词、切块、去重、向量化、索引、重排、缓存、日志清洗。模型越强,这些管道越不能随便糊。

林岚认为,Gigatoken 的报道价值不在于“比 Hugging Face 快多少倍”这个数字本身,而在于它把一个被忽视的底层环节重新拿出来看。AI 应用进入工业化之后,任何高频步骤都会变成工程优化对象。

这里也有一个容易被忽略的问题:分词速度提升不一定直接让聊天机器人回答更快。在线推理的瓶颈常常在模型计算和网络延迟,不在 tokenizer。但在离线处理、大规模索引、训练数据准备、批量评测和日志回放里,tokenizer 可能会被调用数十亿次。这个时候,几倍甚至几十倍差距就会变成机器数量和等待时间。

所以 Gigatoken 更像是给 AI 工程团队看的工具,而不是普通 App 开发者马上要替换的依赖。你只有在数据量足够大、分词确实出现在 profile 里时,才应该认真考虑这种优化。

关键事实

  • 来源:Gigatoken GitHub
  • 涉及技术:Tokenizer、BPE、Rust、SIMD、缓存优化
  • 适用场景:训练数据处理、RAG 入库、长上下文预处理、大规模文本清洗
  • 关键数字:README 示例中部分硬件和 tokenizer 可达到 GB/s 级吞吐

OC 判断

AI 应用的成本优化会越来越向底层移动。开发者不一定都要换 tokenizer,但应该知道 tokenization、chunking、embedding 和缓存并不是免费的。

为什么重要

  • 对开发者:大规模 RAG 和数据处理项目要把分词吞吐纳入性能预算。
  • 对企业:AI 成本不只有模型 API,还包括数据预处理管道。
  • 对用户:更快、更便宜的 AI 产品,往往来自这些看不见的工程优化。

参考来源

相关阅读

基于标题、摘要和正文内容自动匹配。

更多科技

评论

围绕这篇文章补充信息、提出问题或分享观察。

0
暂无评论。

发表评论