arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.08387cs.CL

LEALLA:利用知识蒸馏学习轻量级语言无关句子嵌入

LEALLA: Learning Lightweight Language-agnostic Sentence Embeddings with Knowledge Distillation

Google Research(谷歌研究院);Kyoto University(京都大学)

Zhuoyuan Mao, Tetsuji Nakagawa

AI总结:

本研究提出LEALLA,通过知识蒸馏和薄深编码器,在109种语言上构建轻量级语言无关句子嵌入,在多个基准上验证了其有效性。

摘要:

大规模语言无关句子嵌入模型(如LaBSE(Feng等人,2022))在平行句对齐任务上取得了最先进的性能。然而,这些大规模模型可能面临推理速度和计算开销方面的问题。本研究系统性地探索了使用轻量级模型学习语言无关句子嵌入的方法。我们证明,一个薄而深的编码器可以为109种语言构建稳健的低维句子嵌入。通过我们提出的蒸馏方法,我们通过融入教师模型的知识进一步提升了性能。在Tatoeba、United Nations和BUCC上的实证结果显示了我们的轻量级模型的有效性。我们在TensorFlow Hub上发布了我们的轻量级语言无关句子嵌入模型LEALLA。

英文摘要:

Large-scale language-agnostic sentence embedding models such as LaBSE (Feng et al., 2022) obtain state-of-the-art performance for parallel sentence alignment. However, these large-scale models can suffer from inference speed and computation overhead. This study systematically explores learning language-agnostic sentence embeddings with lightweight models. We demonstrate that a thin-deep encoder can construct robust low-dimensional sentence embeddings for 109 languages. With our proposed distillation methods, we achieve further improvements by incorporating knowledge from a teacher model. Empirical results on Tatoeba, United Nations, and BUCC show the effectiveness of our lightweight models. We release our lightweight language-agnostic sentence embedding models LEALLA on TensorFlow Hub.

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

↑