Do Association-Reweighted BPE Merge Rules Help Data-Efficient Language Modeling?

Zy Li, Phu Han, Trinh Tran, Lam Duong, Andrew Fang


Abstract
Subword tokenization determines a language model’s symbol inventory and sequence length, yet standard BPE selects merges by frequency alone. We test whether association-reweighted merges help on BabyLM Strict-Small. Holding the merge budget and architecture fixed and approximately matching word exposure, we train 97M-parameter causal decoders with three seeds using standard BPE, the S-BPE merge score with fixed-budget stopping, and exponential positive-PMI reweighting (PMI-BPE). Neither alternative reliably improves held-out bits-per-byte (BPB) or BLiMP. PMI-BPE’s mean BPB is slightly higher than standard BPE’s, but no pairwise difference is statistically reliable at three seeds. Both alternatives alter the merge tables and measurably change compression, ruling out a trivial no-intervention explanation. Yet intrinsic association rankings depend on scoring measure and aggregation. Type-uniform rankings reverse between post-merge PMI and thresholded held-out PPMI, while application-weighted PPMI is nearly identical across tokenizers. These results do not rule out association-aware tokenization; they cover one merge budget, one operating point per score, and two evaluation axes. Within that scope, the tested scores do not reliably help, and intrinsic merge association does not consistently align with the BPB and BLiMP point-estimate orderings.
Anthology ID:
2026.babylm-main.8
Volume:
Proceedings of the Second BabyLM Workshop
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Leshem Choshen, Ryan Cotterell, Mustafa Omer Gul, Jaap Jumelet, Tal Linzen, Aaron Mueller, Suchir Salhan, Raj Sanjay Shah, Alex Warstadt, Ethan Gotlieb Wilcox
Venues:
BabyLM | WS
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
93–104
Language:
URL:
https://aclanthology.org/2026.babylm-main.8/
DOI:
Bibkey:
Cite (ACL):
Zy Li, Phu Han, Trinh Tran, Lam Duong, and Andrew Fang. 2026. Do Association-Reweighted BPE Merge Rules Help Data-Efficient Language Modeling?. In Proceedings of the Second BabyLM Workshop, pages 93–104, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Do Association-Reweighted BPE Merge Rules Help Data-Efficient Language Modeling? (Li et al., BabyLM 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.babylm-main.8.pdf