What Kinds of Tokens Benefit from Distant Text? An Analysis on Long Context Language Modeling

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hu, Yutong, Huang, Quzhe, Luo, Kangcheng, Feng, Yansong
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913393371250688
author Hu, Yutong
Huang, Quzhe
Luo, Kangcheng
Feng, Yansong
author_facet Hu, Yutong
Huang, Quzhe
Luo, Kangcheng
Feng, Yansong
contents As the context length that large language models can handle continues to increase, these models demonstrate an enhanced ability to utilize distant information for tasks such as language modeling. This capability contrasts with human reading and writing habits, where it is uncommon to remember and use particularly distant information, except in cases of foreshadowing. In this paper, we aim to explore which kinds of words benefit more from long contexts in language models. By analyzing the changes in token probabilities with increasing context length, we find that content words (e.g., nouns, adjectives) and the initial tokens of words benefit the most. Frequent patterns in the context (N-grams) also significantly impact predictions. Additionally, the model's prior knowledge plays a crucial role in influencing predictions, especially for rare tokens. We also observe that language models become more confident with longer contexts, resulting in sharper probability distributions. This overconfidence may contribute to the increasing probabilities of tokens with distant contextual information. We hope that our analysis will help the community better understand long-text language modeling and contribute to the design of more reliable long-context models.
format Preprint
id arxiv_https___arxiv_org_abs_2406_11238
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle What Kinds of Tokens Benefit from Distant Text? An Analysis on Long Context Language Modeling
Hu, Yutong
Huang, Quzhe
Luo, Kangcheng
Feng, Yansong
Computation and Language
As the context length that large language models can handle continues to increase, these models demonstrate an enhanced ability to utilize distant information for tasks such as language modeling. This capability contrasts with human reading and writing habits, where it is uncommon to remember and use particularly distant information, except in cases of foreshadowing. In this paper, we aim to explore which kinds of words benefit more from long contexts in language models. By analyzing the changes in token probabilities with increasing context length, we find that content words (e.g., nouns, adjectives) and the initial tokens of words benefit the most. Frequent patterns in the context (N-grams) also significantly impact predictions. Additionally, the model's prior knowledge plays a crucial role in influencing predictions, especially for rare tokens. We also observe that language models become more confident with longer contexts, resulting in sharper probability distributions. This overconfidence may contribute to the increasing probabilities of tokens with distant contextual information. We hope that our analysis will help the community better understand long-text language modeling and contribute to the design of more reliable long-context models.
title What Kinds of Tokens Benefit from Distant Text? An Analysis on Long Context Language Modeling
topic Computation and Language
url https://arxiv.org/abs/2406.11238