arXiv:2605.22705cs.CL2026-05被引 1

新分词方法ToaST通过树形结构压缩文本,显著减少令牌数量。

Tokenization with Split Trees

  • 基于字节n元组统计构建分裂树,递归选择词汇表中首个匹配节点
  • 在40,960以上词表规模下,令牌数比BPE等少11%以上
  • 适合追求长上下文、高效推理的语言模型训练

我们提出一种名为分拆树分词(ToaST)的子词分词方法,其通过新的递归推理过程直接优化压缩效果。ToaST利用预计算的字节n元组统计,将每个预分词贪婪地分割为完整的二叉树,独立于任何词汇表。给定词汇表后,推理过程递归遍历每棵分裂树,沿路径首次命中词汇表中的节点即输出。词汇表选择被建模为整数规划(IP),目标是最小化所有分裂树下的总令牌数。实际中线性规划(LP)松弛近似整数解,可获得近最优词汇表,训练时间随分裂树数量呈二次增长。在英文文本上,当词汇表大小超过40,960时,ToaST比BPE、WordPiece和UnigramLM减少超过11%的令牌数,降低模型推理所需令牌数,从而延长有效上下文长度。ToaST也更少使用常见单字节令牌,显著提升Renyi效率。在训练15亿参数语言模型的实验中,ToaST取得最高CORE分数,相比基线提升2.6%至7.6%,其中两项差异显著,在22项任务中有13项排名第一。

原文摘要 · Abstract (English)

We introduce Tokenization with Split Trees (ToaST), a subword tokenization method that directly optimizes compression under a new recursive inference procedure. ToaST greedily splits each pretoken into a full binary tree using precomputed byte n-gram counts, independent of any vocabulary. Given a vocabulary, inference recursively descends each split tree and emits the first in-vocabulary node reached on each path. Vocabulary selection is formulated as an Integer Program (IP) that minimizes the total token count over all split trees under this inference procedure. The Linear Programming (LP) relaxation is near-integral in practice, yielding provably near-optimal vocabularies, with training time empirically scaling quadratically in the number of split trees. On English text, ToaST reduces token counts by more than 11% compared to BPE, WordPiece, and UnigramLM at vocabulary sizes of 40,960 and above, reducing the number of inference tokens for models using this tokenizer, thus extending the effective context length. ToaST also uses common single-byte tokens less frequently than these baselines, leading to a substantial improvement in Renyi efficiency. In experiments training 1.5B parameter language models, ToaST achieves the highest CORE score, outperforming baselines by 2.6%--7.6%, with significance for two of three, and scoring best on 13 of 22 individual tasks.

分词方法压缩优化语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。