FOCUS: Effective Embedding Initialization for Monolingual Specialization of Multilingual Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dobler, Konstantin, de Melo, Gerard
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914741055651840
author Dobler, Konstantin
de Melo, Gerard
author_facet Dobler, Konstantin
de Melo, Gerard
contents Using model weights pretrained on a high-resource language as a warm start can reduce the need for data and compute to obtain high-quality language models for other, especially low-resource, languages. However, if we want to use a new tokenizer specialized for the target language, we cannot transfer the source model's embedding matrix. In this paper, we propose FOCUS - Fast Overlapping Token Combinations Using Sparsemax, a novel embedding initialization method that initializes the embedding matrix effectively for a new tokenizer based on information in the source model's embedding matrix. FOCUS represents newly added tokens as combinations of tokens in the overlap of the source and target vocabularies. The overlapping tokens are selected based on semantic similarity in an auxiliary static token embedding space. We focus our study on using the multilingual XLM-R as a source model and empirically show that FOCUS outperforms random initialization and previous work in language modeling and on a range of downstream tasks (NLI, QA, and NER).
format Preprint
id arxiv_https___arxiv_org_abs_2305_14481
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle FOCUS: Effective Embedding Initialization for Monolingual Specialization of Multilingual Models
Dobler, Konstantin
de Melo, Gerard
Computation and Language
Using model weights pretrained on a high-resource language as a warm start can reduce the need for data and compute to obtain high-quality language models for other, especially low-resource, languages. However, if we want to use a new tokenizer specialized for the target language, we cannot transfer the source model's embedding matrix. In this paper, we propose FOCUS - Fast Overlapping Token Combinations Using Sparsemax, a novel embedding initialization method that initializes the embedding matrix effectively for a new tokenizer based on information in the source model's embedding matrix. FOCUS represents newly added tokens as combinations of tokens in the overlap of the source and target vocabularies. The overlapping tokens are selected based on semantic similarity in an auxiliary static token embedding space. We focus our study on using the multilingual XLM-R as a source model and empirically show that FOCUS outperforms random initialization and previous work in language modeling and on a range of downstream tasks (NLI, QA, and NER).
title FOCUS: Effective Embedding Initialization for Monolingual Specialization of Multilingual Models
topic Computation and Language
url https://arxiv.org/abs/2305.14481