Hugging Face 优化技巧让 BERT 分词器提速 50 倍

art_zucker · x · 2026-10-15

Hugging Face 工程师 Simon Brandeis 撰文详解如何把 BERT 类模型的分词速度再提升 5.7 倍——相比 tokenizers v0.23 总计快约 50 倍(目前处于 PR 阶段)。

背景

为什么 BERT 值得优化

BERT 类编码器是 embeddings、搜索、RAG、重排和分类器的底座。Hub 上下载量前三的模型(all-MiniLM-L6-v2 月下载 2.18 亿、ms-marco-MiniLM-L6-v2 7900 万、bge-small-en-v1.5 5900 万)全部走 BERT 的 normalizer;万大下载榜中 WordPiece 分词器相关下载达每月 6.27 亿次。

核心思路

文章标题点出关键:跳过(skipping)ASCII 相关处理路径是让 BERT normalizer 大幅提速的手段,配合其他工程优化实现额外的 5.7 倍加速。

所属事件:HF Tokenizers 冲刺 v1:BERT 分词器再提速 5 倍(5 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →