The Fair Language Model Paradox

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Pinto, Andrea, Galanti, Tomer, Balestriero, Randall
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914974159339520
author Pinto, Andrea
Galanti, Tomer
Balestriero, Randall
author_facet Pinto, Andrea
Galanti, Tomer
Balestriero, Randall
contents Large Language Models (LLMs) are widely deployed in real-world applications, yet little is known about their training dynamics at the token level. Evaluation typically relies on aggregated training loss, measured at the batch level, which overlooks subtle per-token biases arising from (i) varying token-level dynamics and (ii) structural biases introduced by hyperparameters. While weight decay is commonly used to stabilize training, we reveal that it silently introduces performance biases detectable only at the token level. In fact, we empirically show across different dataset sizes, model architectures and sizes ranging from 270M to 3B parameters that as weight decay increases, low-frequency tokens are disproportionately depreciated. This is particularly concerning, as these neglected low-frequency tokens represent the vast majority of the token distribution in most languages, calling for novel regularization techniques that ensure fairness across all available tokens.
format Preprint
id arxiv_https___arxiv_org_abs_2410_11985
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle The Fair Language Model Paradox
Pinto, Andrea
Galanti, Tomer
Balestriero, Randall
Computation and Language
Artificial Intelligence
Machine Learning
Large Language Models (LLMs) are widely deployed in real-world applications, yet little is known about their training dynamics at the token level. Evaluation typically relies on aggregated training loss, measured at the batch level, which overlooks subtle per-token biases arising from (i) varying token-level dynamics and (ii) structural biases introduced by hyperparameters. While weight decay is commonly used to stabilize training, we reveal that it silently introduces performance biases detectable only at the token level. In fact, we empirically show across different dataset sizes, model architectures and sizes ranging from 270M to 3B parameters that as weight decay increases, low-frequency tokens are disproportionately depreciated. This is particularly concerning, as these neglected low-frequency tokens represent the vast majority of the token distribution in most languages, calling for novel regularization techniques that ensure fairness across all available tokens.
title The Fair Language Model Paradox
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2410.11985