用超网络提升大模型对低资源语言的词嵌入初始化效果
HYPEROFA: Expanding LLM Vocabulary to New Languages via Hypernetwork-Based Embedding Initialization
- 通过超网络将多语言词向量映射到模型嵌入空间,生成灵活的初始词向量
- 在持续预训练和下游任务中均优于随机初始化,媲美或超过OFA表现
- 适合需要扩展大模型多语言能力的研究者与工程实践
许多预训练语言模型在中低资源语言上表现不佳,主要因预训练阶段缺乏足够覆盖。常见策略是为目标语言引入新词元,初始化其嵌入,并在目标语言数据上进行持续预训练。其中,OFA(Liu等,2024a)提出基于相似性的子词嵌入初始化方法,高效且有效。但OFA限制目标语言词元嵌入仅为固定数量源语言嵌入的凸组合,表达能力受限。为此,我们提出HYPEROFA,一种基于超网络的更自适应的词元嵌入初始化方法。超网络通过源语言词元,从外部多语言词向量空间学习映射至模型词元嵌入空间。训练完成后,可为目标语言词元生成灵活嵌入,作为持续预训练的良好起点。实验表明,HYPEROFA在持续预训练收敛速度和下游任务性能上均持续优于随机初始化基线,且媲美或超越OFA表现。代码已公开。
原文摘要 · Abstract (English)
Many pre-trained language models (PLMs) exhibit suboptimal performance on mid- and low-resource languages, largely due to limited exposure to these languages during pre-training. A common strategy to address this is to introduce new tokens specific to the target languages, initialize their embeddings, and apply continual pre-training on target-language data. Among such methods, OFA (Liu et al., 2024a) proposes a similarity-based subword embedding initialization heuristic that is both effective and efficient. However, OFA restricts target-language token embeddings to be convex combinations of a fixed number of source-language embeddings, which may limit expressiveness. To overcome this limitation, we propose HYPEROFA, a hypernetwork-based approach for more adaptive token embedding initialization. The hypernetwork is trained to map from an external multilingual word vector space to the PLMs token embedding space using source-language tokens. Once trained, it can generate flexible embeddings for target-language tokens, serving as a good starting point for continual pretraining. Experiments demonstrate that HYPEROFA consistently outperforms random initialization baseline and matches or exceeds the performance of OFA in both continual pre-training convergence and downstream task performance. We make the code publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。