TensorX
返回文献探索

Paper · arXiv 2501.16975

Over-Tokenized Transformer: Vocabulary is Generally Worth Scaling

Hongzhi Huang, Defa Zhu, Banggu Wu, Yutao Zeng, Ya Wang, Qiyang Min, Xun Zhou

35 upvotesJanuary 28, 2025arXiv 预印本
AI 摘要

Over-Tokenized Transformers improve language modeling by decoupling input and output vocabularies, leading to better performance with larger input vocabularies.

Over-Tokenized Transformersmulti-gram tokensinput vocabulary sizetraining losslanguage modelingtokenizationscaling laws

Abstract

Tokenization is a fundamental component of large language models (LLMs), yet its influence on model scaling and performance is not fully explored. In this paper, we introduce Over-Tokenized Transformers, a novel framework that decouples input and output vocabularies to improve language modeling performance. Specifically, our approach scales up input vocabularies to leverage multi-gram tokens. Through extensive experiments, we uncover a log-linear relationship between input vocabulary size and training loss, demonstrating that larger input vocabularies consistently enhance model performance, regardless of model size. Using a large input vocabulary, we achieve performance comparable to double-sized baselines with no additional cost. Our findings highlight the importance of tokenization in scaling laws and provide practical insight for tokenizer design, paving the way for more efficient and powerful LLMs.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号