atlas

Research map / Large language models / Language, safety and interpretability

Arrows of Time for Large Language Models

ICML 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that autoregressive LLMs trained forward on natural language reach lower log-perplexity than when trained backward on time-reversed text (an "Arrow of Time"), and explains it with computational hardness and sparsity arguments on synthetic data.

Problem

Whether next-token (forward) and previous-token (backward) prediction are equally learnable by autoregressive LLMs; information theory says the joint likelihood is identical, so any consistent difference in log-perplexity would reflect an asymmetry in the data distribution.

Approach

Train matched forward and time-reversed GPT/GRU/LSTM models on the same data and measure the cross-entropy gap; for large enough models and context lengths a consistent forward advantage appears across languages. Explains it via computational hardness of inverting maps (factoring example) and sparsity of linear maps versus their inverses.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Predict the next token instead of the previous tokenGPT2-Medium · CC-100 Englishvalidation loss: 2.902 -> 2.88Helped
Predict the next token instead of the previous tokenGPT2-Medium · CC-100 Frenchvalidation loss: 2.862 -> 2.788Helped
Increase GPT model sizeForward and backward GPT models · CC-100 FrenchBW relative loss difference: +0.15% -> +2.65%Helped
Increase GPT model sizeForward and backward GPT models · CC-100 EnglishBW relative loss difference: -0.33% -> +0.76%Mixed
Increase context window lengthGPT2-Medium · CC-100 English and FrenchBW/FW loss percentage differenceHelped
Predict the next token instead of the previous tokenGRU and LSTM models · CC-100 English and Frenchfinal loss: -0.07% to +0.66%Mixed
Predict the next token instead of the previous tokenGPT2-Medium and GPT2-XL · CC-100 multilingualfinal loss: +0.61% to +3.17%Helped
Predict the next token instead of the previous tokenGPT2-Medium · Prime multiplication datasettotal log-perplexity: 22.2 nats -> 30.2 natsHelped
Predict the next token instead of the previous tokenGPT · Linear language sparse updates, 2-bit flipsloss: $0.353\pm 0.008$ -> $0.347\pm 0.009$Helped
Predict the next token instead of the previous tokenGPT · Linear language sparse updates, 4-bit flipsloss: $0.371\pm 0.011$ -> $0.354\pm 0.007$Helped
Predict the next token instead of the previous tokenGPT · Linear language sparse updates, 6-bit flipsloss: $0.387\pm 0.013$ -> $0.367\pm 0.014$Helped
Train the BPE tokenizer on character-reversed text (ablation)GPT2-Medium · CC-100 Greek and FrenchlossNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: CC-100 synthetic prime multiplication dataset synthetic linear language dataset

Benchmarks: CC-100 validation loss (English, French, German, Turkish, Finnish, Vietnamese, Greek, Indonesian, Tagalog, Arabic, Hebre

Metrics: cross-entropy / log-perplexity relative BW-FW loss difference

Baselines: GRU LSTM

Builds on: GPT-2 minGPT BPE AdamW Shannon's forward/backward prediction experiments

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy