通用开源大模型在特定任务微调后,往往仍沿用庞杂的通用词表,造成严重的显存与计算浪费。AI研发团队Dharma-AI提出VocabX方法,通过以成本为导向定制领域专有词表,在完全不改动底座模型与训练数据的前提下,将生产环境推理成本直降77%。
利用 Qwen 或 Gemma 等通用开源大模型做领域微调时,开发者通常直接沿用原装分词器(Tokenizer)。然而,通用词表本是为覆盖多语言、多场景而设计的妥协方案。直接在专有业务中使用,往往会引入隐蔽却高昂的推理成本。
在葡萄牙语作文评分系统的生产部署中,Dharma-AI 团队提出了一种名为 VocabX 的专有词表重构方案。在模型权重、训练数据与超参数完全不变的前提下,仅通过定制词表,就将生产环境的端到端推理成本削减了 77%。

主流大语言模型以 Token 为基本读写单元。通用词表由多语言代码、法律文书、网页等海量语料共同训练而成。对于面向特定垂类(例如固定格式的评分、合同提取或代码执行)的模型,通用词表会带来两个致命浪费:
第一是碎片化(Fragmentation)。模型频繁输出的高频模版(如固定的 JSON 结构、标准评估短语),在通用词表中会被生生切碎成十几个细小的 Token。每调用一次,模型就得逐一生成并支付相应的计算开销。

第二是词表冗余(Excess)。通用词表往往包含 15 万以上的词条。一个专门用于英文生物阅卷的模型,哪怕永远不会输出中文,也必须在每次解码生成时,由最后的 LM Head 输出层为那数万个从不用到的汉字 Token 计算打分(Logits)。对于 4B 级别的轻量级模型,LM Head 矩阵参数甚至能占到整机权重的近 10%。这些无用参数在显存受限的解码阶段反复搬运,白白拖累速度。

优化分词不能仅看 Token 数量,必须看真实的硬件账单。
大模型推理分为两个阶段:
实测表明,在评分系统的典型并发环境下,生成一个 Token 的实际耗时成本是读取一个 Token 的 7.4 倍。如果只按全局出现频率去合并词表,而忽略了输入与输出端近乎 7 倍的成本差异,压缩效果就会大打折扣。

现有方案如快速词表迁移(FVT)或 Liquid AI 的原位词表扩展(in-place BPE),大都仅按频次筛选,且局限于单一子词(Sub-word)级别。VocabX 从三个维度打破了局限:

在搭载 RTX 5090、运行 vLLM 的评测环境中,团队针对 Qwen3-4B 基座训练了 7 个不同词表版本:
在优化过程中,团队发现了一个关键经验:如果盲目追求极端压缩,将评分结果与外层 JSON 强行融合成单个宏词,由于模型需要单步决定“结构+分数”,评分质量反而会恶化。为关键决策预留独立的 Token,不仅能保住模型推理能力,整体响应依然比激进压缩版本更便宜。

免费获取企业 AI 成熟度诊断报告,发现转型机会
在合同、医疗、科研论文、新闻与 Python 代码五大通用语料库,以及涵盖 Qwen3、Llama-3.1 与 Gemma-3 的跨模型验证中,VocabX 均全面击败现有主流方案。
专用模型的优化往往聚焦于数据清洗和权重微调。但这项实践证明,早在一个训练步骤都还没开始的分词阶段,词表本身的轻量化与专有化,同样是一处极具潜力的成本降维打击点。
关注公众号

扫码关注,获取最新 AI 资讯
3 步完成企业诊断,获取专属转型建议
已有 200+ 企业完成诊断