arXiv:2608.03494cs.CLcs.LG2026-08

优化大模型新词嵌入初始化,显著提升多语言扩展训练效率

Beyond Initialization Loss: A Systematic Study of Token Embedding Initialization Strategies for LLM Vocabulary Extension

论文配图:Beyond Initialization Loss: A Systematic Study of Token Embedding Initialization Strategies for LLM Vocabulary Extension
图 1 · 摘自论文原文
  • 采用子词组合策略,区分输入输出嵌入初始化方式
  • 新方法在500步内达基线3500步精度,节省超6倍训练步数
  • 轻量微调仅需50步即可可靠筛选最优初始化方案

词汇扩展是高效适配预训练大语言模型至新语言的手段,但新增词嵌入的初始化方式会显著影响继续预训练(CPT)效率。本文系统研究了在Nemotron-3-Nano-30B-A3B上对印地语词汇扩展的20余种初始化策略,涵盖词汇平均基准、外部与学习型初始化方法(如FOCUS、top-k语义检索、残差MLP映射)、子词组合、范数校准及输入输出不对称性。结果发现,子词组合方法优于词汇平均及外部/学习型初始化。在子词组合中,非对称变体表现最佳,输入与输出嵌入初始化偏好不同。最优配置为:输入嵌入矩阵采用均匀子词平均+印地语特定范数校准,输出语言建模头使用字符长度加权子词平均。相比标准均值基准(Mean-all),该完整初始化流程在超过6倍减少的CPT步数下达到相当验证损失,并在500步时超越基线3500步的MILU-Hindi准确率。最后,我们证明初始化损失和初始化比特每字节(Init BPB)无法可靠预测下游收敛,而仅需50步的轻量级微调即能提供经济且可靠的策略选择信号。

原文摘要 · Abstract (English)

Vocabulary extension is an efficient way to adapt pretrained large language models (LLMs) to new languages, but the initialization of newly added token embeddings can strongly affect continued pre-training (CPT) efficiency. We present a systematic study of more than 20 initialization strategies for Hindi vocabulary extension in Nemotron-3-Nano-30B-A3B. Our comparison spans vocabulary-averaging baselines; external and learned initialization methods, including FOCUS, top-k semantic retrieval, and residual MLP mappings; subword composition; norm calibration; and input-output asymmetry. We find that subword composition methods outperform both vocabulary averaging and external/learned initialization approaches. Within subword composition, asymmetric variants achieve the lowest observed early validation loss and reveal distinct preferences for input and output embedding initialization. The best observed configuration initializes the input embedding matrix with uniform subword averaging and Hindi-specific norm calibration, and the output language modeling head with character-length-weighted subword averaging. Relative to the standard Mean-all baseline, this full initialization pipeline reaches comparable validation loss with over a 6x reduction in CPT steps and exceeds the baseline's 3,500-step MILU-Hindi accuracy after only 500 steps. Finally, we show that initialization loss and initialization bits-per-byte (Init BPB) are unreliable predictors of downstream convergence, whereas lightweight CPT, as few as 50 steps, provides a cost-effective and reliable signal for selecting the best initialization strategy.

LLM扩展嵌入初始化多语言训练效率

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