K-文化的一切——从回归到 K-美妆,发送到您的邮箱订阅邮件

METAL MEDIA

SuTRA : Structurally-Unified Tokenization with Root Awareness

arXiv:2608.180872026-08-20

让分词器不再乱切印度语系单词词根,而是尊重词素边界的新方法

像BPE这样的子词分词器只追求统计压缩效率,导致在印地语、马拉地语、古吉拉特语等形态丰富的印度语言中,词根和词缀经常被随意切开,作者将这种现象称为形态破碎。他们提出了SuTRA,在保持文字系统音节单位完整的同时,在类BPE的合并训练过程中对跨越词素边界的合并施加惩罚。基于团队新建的大规模金标准数据集测试,SuTRA在边界对齐度、语义可还原性和机器翻译质量上都优于标准BPE。

METAL MEDIA 解读图

让分词器不再乱切印度语系单词词根,而是尊重词素边界的新方法

  1. 01BPE、WordPiece等标准分词器仅按频率切分单词,导致例如印地语单词asādhāraṇ中的否定前缀a被错误地与词根sādhāraṇ融合在一起,而不是被正确分开
  2. 02SuTRA分两个阶段运作:第一阶段应用书写规则,确保天城文等文字中辅音与依附元音符号组成的音节单位不被拆散,并利用词典或针对未知词的序列到序列模型标记禁止穿越的词素边界;第二阶段在BPE式合并打分中加入惩罚项,对违反这些边界的合并进行降权,且该约束强度随训练进程逐渐放松
  3. 03团队构建了一个约56万词规模的印地语、马拉地语、古吉拉特语金标准词素切分数据集,采用规则提取与大语言模型验证相结合的方式
  4. 04在边界F1指标上,SuTRA相对BPE最高提升14.7个百分点,在印地语语义可还原性上提升34%,机器翻译平均提升8.08个chrF2分数
这是 METAL MEDIA 制作的解读图,并非论文作者提供的原图。

他们做了什么

  1. BPE、WordPiece等标准分词器仅按频率切分单词,导致例如印地语单词asādhāraṇ中的否定前缀a被错误地与词根sādhāraṇ融合在一起,而不是被正确分开
  2. SuTRA分两个阶段运作:第一阶段应用书写规则,确保天城文等文字中辅音与依附元音符号组成的音节单位不被拆散,并利用词典或针对未知词的序列到序列模型标记禁止穿越的词素边界;第二阶段在BPE式合并打分中加入惩罚项,对违反这些边界的合并进行降权,且该约束强度随训练进程逐渐放松
  3. 团队构建了一个约56万词规模的印地语、马拉地语、古吉拉特语金标准词素切分数据集,采用规则提取与大语言模型验证相结合的方式
  4. 在边界F1指标上,SuTRA相对BPE最高提升14.7个百分点,在印地语语义可还原性上提升34%,机器翻译平均提升8.08个chrF2分数
Figure 1: Morphological Shattering vs. Root Preservation. For the Hindi word asādhāraṇ, standard Bpe fuses the negation prefix into the root ([asā]+[dhāraṇ]), while SuTRA cleanly separates prefix and root ([a]+[sādhāraṇ]). We term such frequency-driven prefix–root fusion Morphological Shattering; SuTRA's root-preserving segmentations yield more stable subword units and reduce semantic blindness. Figure generated using PaperBanana [paperbanana].
Figure 1: Morphological Shattering vs. Root Preservation. For the Hindi word asādhāraṇ, standard Bpe fuses the negation prefix into the root ([asā]+[dhāraṇ]), while SuTRA cleanly separates prefix and root ([a]+[sādhāraṇ]). We term such frequency-driven prefix–root fusion Morphological Shattering; SuTRA's root-preserving segmentations yield more stable subword units and reduce semantic blindness. Figure generated using PaperBanana [paperbanana].
Table 1: Morphological Datasets. We provide the first high-scale, LLM-verified morphological coverage for Indic scripts.
DatasetLangs (Indic)SizeMethodVerification
UniMorph 4.0 [batsuren2022unimorph40universalmorphology]Multi (Partial)∼10MSchema-basedAlgorithmic
MorphyNet [batsuren-etal-2021-morphynet]15 (None)10.6MRule+EnrichmentManual (Expert)
GujMorph [baxi-bhatt-2022-gujmorph]1 (Gujarati)∼80kUnsup.None
Ours (Gold)3(Hi, Mr, Gu)560kHybridLLM-Verif.
Figure 2: Overview of SuTRA. Phase 1 (Pre-tokenization) applies orthographic rules Φ to map each word into akshara-like units and uses a gold morphological lexicon or a seq2seq model to mark forbidden boundaries (morpheme boundaries that merges should not cross). Phase 2 (Morphology-Aware Merging) runs a BPE-style algorithm with scores S​(a,b)=f​(a,b)​Ψ​(a,b)γt, where Ψ downweights merges that violate forbidden boundaries and γt controls rigidity over training, biasing vocabulary toward merges that respect Indic script structure and morpheme boundaries. Figure generated using PaperBanana [paperbanana].
Figure 2: Overview of SuTRA. Phase 1 (Pre-tokenization) applies orthographic rules Φ to map each word into akshara-like units and uses a gold morphological lexicon or a seq2seq model to mark forbidden boundaries (morpheme boundaries that merges should not cross). Phase 2 (Morphology-Aware Merging) runs a BPE-style algorithm with scores S​(a,b)=f​(a,b)​Ψ​(a,b)γt, where Ψ downweights merges that violate forbidden boundaries and γt controls rigidity over training, biasing vocabulary toward merges that respect Indic script structure and morpheme boundaries. Figure generated using PaperBanana [paperbanana].
Table 2: Gold Standard Statistics. Hybrid of rule-based extraction and LLM-verified morphological segmentation.
LanguageSourceUnique Words
HindiIndicCorp [ai4bharat_corpus]≃ 160,000
MarathiIndicCorp [ai4bharat_corpus]≃ 200,000
GujaratiIndicCorp [ai4bharat_corpus]≃ 200,000
Total≃ 560,000
Figure 4: Sample examples from the dataset used for Hindi to Marathi Machine Translation Task
Figure 4: Sample examples from the dataset used for Hindi to Marathi Machine Translation Task
Table 3: Morphological Alignment Evaluation. Evaluation of Boundary F1 and Fertility Ratio across Hindi, Marathi, and Gujarati. Higher F1 indicates superior structural integrity, while controlled fertility prevents arbitrary character-level fragmentation. Best F1 scores are in bold.
TokenizerHindiMarathiGujarati
F1 ↑Fert. ↓F1 ↑Fert. ↓F1 ↑Fert. ↓
BPE (ACL'16)0.4821.2850.4701.2250.5911.126
WordPiece (arXiv'12)0.4111.2140.5271.3000.5961.173
SentencePiece (EMNLP'18)0.4381.3150.0831.1830.5911.156
Unigram (ACL'18)0.4391.3100.5071.2560.6691.137
SuperBPE (COLM'25)0.0892.5170.0843.0840.0963.113
MorphTok (ICML-W'25)0.1902.6010.2163.4820.2473.494
\rowcolor[gray]0.9 SuTRA (Ours)0.5861.4120.6171.7550.5841.454
Figure 5: Loss curves for BPE and SuTRA
Figure 5: Loss curves for BPE and SuTRA
Table 4: Semantic Recoverability (R2). Linear models evaluate immediate compositionality, while deeper MLP models test the preservation of recoverable structural signals.
LanguageLinear R2 (Layer 0)MLP R2 (Layer 2+)
SuTRABpeSuTRABpe
Hindi0.44640.33290.50480.3358
Marathi0.46340.46190.53310.4649
Gujarati0.46240.46400.50550.4510
Table 5: Machine Translation. SuTRA achieves the highest scores in Marathi → Hindi and remains comparable to the strongest baseline in the reverse task, demonstrating that root contiguity improves cross-lingual alignment.
TokenizerHindi → MarathiMarathi → Hindi
chrF2 ↑COMET ↑chrF2 ↑COMET ↑
\rowcolor[gray]0.95 Statistical Baselines
Bpe (ACL'16)24.620.505436.550.6253
WordPiece (arXiv'12)30.370.609327.180.5223
SentencePiece (EMNLP'18)25.010.496428.100.5147
Unigram (ACL'18)26.550.515729.190.5349
\rowcolor[gray]0.95 Morphological Baselines
SuperBpe (COLM'25)15.840.304716.270.3422
MorphTok (ICML-W'25)26.750.575029.420.6007
\rowcolor[gray]0.9 SuTRA (Ours)29.960.575238.840.6554
Table 6: Morphological Robustness Comparison. Results show that SuTRA effectively mitigates root fragmentation.
TokenizerHindi (HI)Marathi (MR)Gujarati (GU)
Jac.(↑)R.Aff.(↓)Jac.(↑)R.Aff.(↓)Jac.(↑)R.Aff.(↓)
Bpe0.3860.2280.3050.3240.3190.296
Unigram0.3630.2320.2940.3400.3200.308
WordPiece0.4130.2490.2930.3550.3140.316
SentencePiece0.3710.2350.2980.3250.3210.287
SuperBpe0.6240.1460.6550.1270.6310.126
MorphTok0.8010.0670.8230.0550.7920.051
\rowcolor[gray]0.9 SuTRA (Ours)0.8750.0420.8850.0380.7880.040
Table 7: Effect of the Annealed Rigidity Constraint. At the start of training (t=0,γ=4), the exponential penalty prioritizes structurally safe merges (Ψ≈1.0), forcing the tokenizer to build valid semantic roots despite lower raw frequencies. By the end of training (t=T,γ=0), the penalty disappears (Ψ0=1), reducing the score to standard Bpe frequency to optimize compression.
Candidate Merge (a,b)Freq. fValidity ΨEffective Score S​(a,b)
t=0 (γ=4)t=T (γ=0)
Safe Merge (e.g., intra-root)10001.00𝟏𝟎𝟎𝟎1000
Boundary Violation (e.g., prefix+root)12000.80491𝟏𝟐𝟎𝟎
Severe Violation (e.g., shattered prefix)15000.5093𝟏𝟓𝟎𝟎
Table 8: Time Complexity Summary. Comparison of training and inference time complexity. Variables: N (corpus size), V (vocab size), M (unique candidate pairs), Vi​n (unique corpus words), and |w| (word length).
TokenizerTrainingInference (per word)
Standard Bpe𝒪​(N+V​log⁡M)𝒪​(|w|)
MorphTok𝒪​(Vi​n⋅|w|2+N+V​log⁡M)𝒪​(|w|2)
SuTRA (Ours)𝒪​(Vi​n⋅|w|2+N+V​log⁡M)𝒪​(|w|)
Table 9: Comparison of Tokenizers (Normalized) across Hindi, Marathi, and Gujarati. EM: Exact Match Acc, Prec: Boundary Precision, Rec: Boundary Recall, F1: Boundary F1, Fert: Fertility Ratio. Best F1 scores are in bold.
TokenizerHindiMarathiGujarati
EMPrec.Rec.F1Fert.EMPrec.Rec.F1Fert.EMPrec.Rec.F1Fert.
BPE0.2710.3980.6090.4821.2850.2680.3960.5780.4701.2250.3960.5340.6620.5911.126
SentencePiece0.1910.3570.5660.4381.3150.0000.0431.0000.0831.1830.3780.5230.6790.5911.156
SuperBPE0.0000.0560.2150.0892.5170.0000.0500.2640.0843.0840.0000.0580.2890.0963.113
Unigram0.1850.3590.5660.4391.3100.3940.5110.7780.5071.2560.4600.6000.7560.6691.137
WordPiece0.1660.3530.4930.4111.2140.2740.4270.6880.5271.3000.3780.5220.6940.5961.173
MorphTok0.0040.1220.5310.1902.6010.0080.1260.7630.2163.4820.0060.1450.8350.2473.494
SuTRA0.2120.4590.8100.5861.4120.1320.3540.8980.6171.7550.2130.4480.8360.5841.454
Table 10: Hyperparameters for the downstream Causal Language Modeling task.
HyperparameterValue
ArchitectureGPT-2 (124M parameters)
Number of Layers12
Number of Heads12
Embedding Dimension (dm​o​d​e​l)768
Context Window (nc​t​x)1024
Vocabulary SizeLanguage-dependent
OptimizerAdamW
Learning Rate5×10−4
Weight Decay0.01
Warmup Steps1,000
Target Training Tokens1.5 Billion
Batch Size (per device)16
PrecisionMixed (bf16 or fp16)
Table 11: Downstream performance on Causal Language Modeling (CLM). We report average Negative Log-Likelihood (Avg NLL) and Perplexity (PPL) on out-of-distribution (OOD) corpora. Lower values indicate better compression and predictive efficiency.
LanguageTokenizerVocab SizeAvg NLL ↓Perplexity ↓
Hindi (HI)BPE320003.11618.87
SuTRA320003.01616.08
Table 12: Morphological Shattering Comparison. Results show that SuTRA effectively mitigates root fragmentation.
TokenizerHindi (HI)Marathi (MR)Gujarati (GU)
Jac.(↑)R.Aff.(↓)Jac.(↑)R.Aff.(↓)Jac.(↑)R.Aff.(↓)
Bpe0.3860.2280.3050.3240.3190.296
Unigram0.3630.2320.2940.3400.3200.308
WordPiece0.4130.2490.2930.3550.3140.316
SentencePiece0.3710.2350.2980.3250.3210.287
SuperBpe0.6240.1460.6550.1270.6310.126
MorphTok0.8010.0670.8230.0550.7920.051
\rowcolor[gray]0.9 SuTRA (Ours)0.8750.0420.8850.0380.7880.040

为什么重要

分词器是文本进入语言模型的第一道关口,如果这道关口把有意义的词汇单位切碎,模型即便规模再大也要浪费能力去重新拼凑基本语义。这项工作提供的数据集和方法,为构建面向形态复杂语言的多语言模型提供了具体参考。

本文术语

  • 子词分词器 · 将单词切分成更小的统计常见片段以构建词表的工具,BPE是常见代表
  • 音节单位(akshara) · 天城文等文字中由辅音与依附元音符号组合而成的书写音节单位
  • 词素边界 · 词根与词缀等有意义单位之间的分界点
  • 连音(Sandhi) · 印度雅利安语系中单词或词素边界处发音融合或变化的现象
  • 边界F1 · 衡量分词器预测的切分点与金标准词素边界匹配程度的精确率和召回率综合指标

无法转载的图表

  • Figure 3: Qualitative Comparison of Morphological Segmentation Across Tokenizers. SuTRA consistently matches the Gold Standard by respecting phonetic and morphological boundaries.
在原文中查看图表 →

论文原文摘要(英文)

Existing subword tokenizers optimize statistical compression but ignore morphological structure, particularly the relationship between roots and affixes. This is harmful for morphologically rich Indic languages, where basic units are complex orthographic syllables (aksharas) rather than letters. Frequency-based methods over-fragment words, arbitrarily splitting roots and affixes - a phenomenon we term Morphological Shattering. We propose SuTRA (Structurally-Unified Tokenization with Root Awareness), a morphology-aware algorithm that preserves akshara indivisibility and penalizes merges crossing morphological boundaries. We also release a new morphological segmentation dataset for Hindi, Marathi, and Gujarati. SuTRA reduces shattering, achieving peak gains of +14.7% in morphological alignment (Boundary F1) and +34% in semantic recoverability (Hindi) over BPE. These structural gains yield an average improvement of +8.08 chrF2 in machine translation.

作者 · Vaibhav Rathore, Siddhant Gole, Dadhichi Telwadkar, Rooshil Bhatia, Maulik Ruparel, Siddharth Surekha, Neha Bhargava

在 arXiv 阅读

最新论文

全部论文 →

METAL MEDIA 最新报道

图片来源: Vaibhav Rathore et al., arXiv:2608.18087, CC BY 4.0