arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.01007cs.CL
收藏

将令牌训练模型蒸馏为字节级模型

Distilling Token-Trained Models into Byte-Level Models

Fuzhou University(福州大学);Fudan University(复旦大学);Together AI;Nous Research

Zishuo Bao, Jiaqi Leng, Junxiong Wang, Bowen Peng, Yucheng Lu

AI总结:

本文提出了一种将令牌训练模型高效蒸馏为字节级模型的方法,通过两阶段课程实现端到端生成,验证了在有限数据下保持模型性能的可行性。

摘要:

字节语言模型(BLMs)已成为在令牌化之外扩展语言模型的一种有前途的方向。然而,现有的BLMs通常需要从零开始在万亿字节上进行训练,使其成本高昂。在本文中,我们提出了一种高效的蒸馏配方,将现有的令牌训练语言模型转换为BLMs,同时保持相当的能力。我们的配方遵循两个阶段的课程:(1)渐进知识蒸馏,将字节级表示与令牌训练教师模型的嵌入对齐;以及(2)字节级监督微调,使端到端生成完全在字节空间中进行。我们验证了我们的方法在多个模型家族中的有效性,包括Llama、Qwen和OLMo,并证明使用仅约125B字节,蒸馏后的BLMs保留了大多数教师模型的性能。

英文摘要:

Byte Language Models (BLMs) have emerged as a promising direction for scaling language models beyond tokenization. However, existing BLMs typically require training from scratch on trillions of bytes, making them prohibitively expensive. In this paper, we propose an efficient distillation recipe that converts existing token-trained LLMs into BLMs while retaining comparable capabilities. Our recipe follows a two-stage curriculum: (1) Progressive Knowledge Distillation, which aligns byte-level representations with the embeddings of the token-trained teacher model; and (2) Byte-Level Supervised Fine-Tuning, which enables end-to-end generation entirely in the byte space. We validate our approach across multiple model families, including Llama, Qwen, and OLMo, and demonstrate that the distilled BLMs retain most of the teacher models' performance using only approximately 125B bytes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑