通过词元级最优传输实现跨语言表示对齐,提升多语言模型性能。
Cross-Lingual Representation Alignment by Token-Level Optimal Transport in a Language-Agnostic Space

- 在语言无关空间中识别并保留语言特异性表示,词元级对齐
- 推理时对齐提升准确率最高11.2点,保持输入输出语言一致
- 适用于需要高精度多语言能力的场景,如跨语言任务迁移
跨语言对齐(CLA)旨在对齐大语言模型在不同语言下的表示,以提升多语言能力。现有方法常忽略表示中的语言特异性信息,仅进行句级对齐,可能导致性能不佳及输入输出语言不匹配。本文提出CAROT(基于最优传输的跨语言表示对齐),包含两步:识别大语言模型内部状态中的语言特异性表示,并在语言无关空间中通过最优传输进行词元级对齐,同时显式保留语言特异性信息。推理阶段的引导实验表明,CAROT生成的表示是有效的对齐目标,可使多语言性能最高提升11.2个准确率点,且维持输入输出语言一致性。进一步将这些表示作为训练目标,内化对齐表示。训练后的模型在18个评估设置中的11个上优于现有方法(3模型×3任务×同域/异域语言)。本工作为大语言模型中有效对齐目标提供了新见解。代码已公开于https://github.com/ynklab/CAROT。
原文摘要 · Abstract (English)
Cross-lingual alignment (CLA) aims to align the representations of large language models (LLMs) across languages, enabling cross-lingual transfer to improve multilingual capabilities. Previous CLA methods often ignore language-specific information encoded in representations and only consider sentence-level alignment, which may lead to suboptimal performance and input-output language mismatch. We propose CAROT (Cross-Lingual Alignment of Representations in a Language-Agnostic Space via Optimal Transport), which consists of two steps: identifying language-specific representations in LLMs' internal states and aligning language-agnostic representations across languages at the token level by optimal transport, while explicitly preserving language-specific representations. Inference-time steering experiments show that the representations computed by CAROT are effective alignment targets, improving multilingual performance by up to 11.2 points in accuracy while maintaining input-output language consistency. We further use the representations obtained by CAROT as training targets, internalizing the aligned representations. The trained models outperform existing CLA methods in 11 of 18 evaluation settings (3 models $\times$ 3 tasks $\times$ ID/OOD languages). Our work provides insights into what constitutes effective alignment targets for CLA in LLMs. Code is available at https://github.com/ynklab/CAROT
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。