MAGNET: Improving the Multilingual Fairness of Language Models with Adaptive Gradient-Based Tokenization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ahia, Orevaoghene, Kumar, Sachin, Gonen, Hila, Hofmann, Valentin, Limisiewicz, Tomasz, Tsvetkov, Yulia, Smith, Noah A.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929594582433792
author Ahia, Orevaoghene
Kumar, Sachin
Gonen, Hila
Hofmann, Valentin
Limisiewicz, Tomasz
Tsvetkov, Yulia
Smith, Noah A.
author_facet Ahia, Orevaoghene
Kumar, Sachin
Gonen, Hila
Hofmann, Valentin
Limisiewicz, Tomasz
Tsvetkov, Yulia
Smith, Noah A.
contents In multilingual settings, non-Latin scripts and low-resource languages are usually disadvantaged in terms of language models' utility, efficiency, and cost. Specifically, previous studies have reported multiple modeling biases that the current tokenization algorithms introduce to non-Latin script languages, the main one being over-segmentation. In this work, we propose MAGNET; multilingual adaptive gradient-based tokenization to reduce over-segmentation via adaptive gradient-based subword tokenization. MAGNET learns to predict segment boundaries between byte tokens in a sequence via sub-modules within the model, which act as internal boundary predictors (tokenizers). Previous gradient-based tokenization methods aimed for uniform compression across sequences by integrating a single boundary predictor during training and optimizing it end-to-end through stochastic reparameterization alongside the next token prediction objective. However, this approach still results in over-segmentation for non-Latin script languages in multilingual settings. In contrast, MAGNET offers a customizable architecture where byte-level sequences are routed through language-script-specific predictors, each optimized for its respective language script. This modularity enforces equitable segmentation granularity across different language scripts compared to previous methods. Through extensive experiments, we demonstrate that in addition to reducing segmentation disparities, MAGNET also enables faster language modelling and improves downstream utility.
format Preprint
id arxiv_https___arxiv_org_abs_2407_08818
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MAGNET: Improving the Multilingual Fairness of Language Models with Adaptive Gradient-Based Tokenization
Ahia, Orevaoghene
Kumar, Sachin
Gonen, Hila
Hofmann, Valentin
Limisiewicz, Tomasz
Tsvetkov, Yulia
Smith, Noah A.
Computation and Language
In multilingual settings, non-Latin scripts and low-resource languages are usually disadvantaged in terms of language models' utility, efficiency, and cost. Specifically, previous studies have reported multiple modeling biases that the current tokenization algorithms introduce to non-Latin script languages, the main one being over-segmentation. In this work, we propose MAGNET; multilingual adaptive gradient-based tokenization to reduce over-segmentation via adaptive gradient-based subword tokenization. MAGNET learns to predict segment boundaries between byte tokens in a sequence via sub-modules within the model, which act as internal boundary predictors (tokenizers). Previous gradient-based tokenization methods aimed for uniform compression across sequences by integrating a single boundary predictor during training and optimizing it end-to-end through stochastic reparameterization alongside the next token prediction objective. However, this approach still results in over-segmentation for non-Latin script languages in multilingual settings. In contrast, MAGNET offers a customizable architecture where byte-level sequences are routed through language-script-specific predictors, each optimized for its respective language script. This modularity enforces equitable segmentation granularity across different language scripts compared to previous methods. Through extensive experiments, we demonstrate that in addition to reducing segmentation disparities, MAGNET also enables faster language modelling and improves downstream utility.
title MAGNET: Improving the Multilingual Fairness of Language Models with Adaptive Gradient-Based Tokenization
topic Computation and Language
url https://arxiv.org/abs/2407.08818