Training data attribution for AI language models degrades with redundancy and distractors

A configurable synthetic benchmark reveals that standard methods dilute causal evidence by averaging across token scores; selecting only the top tokens and using negative log-odds gradients improves recall by 35 to 72 percentage points.

Top University
Steven Yu · Sheng-Yu Wang · Kimia Hamidieh · Antonio Torralba · David Bau

Northeastern University · MIT

Research Digest··2 min read
Illustration: Training data attribution for AI language models degrades with redundancy and distractors

In one line

Using negative log-odds gradients and top-token aggregation improves language model data attribution by 35 to 72 percentage points.

Read it as

Large language models learn from training text, but we can't know which passage contributed to a given behavior. Existing tests mix many factors. Researchers built a benchmark from invented facts with known sources, changing one property at a time, like repetition or misleading similar text.

Attribution methods score how much each training document contributed to a prediction. The evidence is often concentrated in a few tokens, or word pieces. Standard methods average over the whole document, diluting it. The researchers found two fixes: negative log-odds, which highlights informative tokens, and keeping only the top token scores.

Both fixes improved accuracy by 35 to 72 percentage points across methods, and it generalized to an external benchmark, supporting copyright and safety tracing.

How it works

before and after
STANDARD ATTRIBUTIONTraining documents: redundant facts & distractorsTraining documentsredundant facts &distractorsCross-entropy gradients: all tokens averagedCross-entropygradientsall tokens averagedDocument score: diluted evidenceDocument scorediluted evidenceCATT + FIXESgroundtruthtokenscoresdocumentsCATT benchmark: controlled synthetic factsCATTbenchmarkcontrolledsynthetic factsNegative log-odds: gradient correctionNegativelog-oddsgradientcorrectionTop-k aggregation: select highest token scoresTop-kaggregationselect highesttoken scoresImproved recall: 35–72 pp gainImprovedrecall35–72 pp gainTop-k threshold must be tuned per setting; dynamic methods needed.
RSS

The Daily Edition

The front page in your inbox each weekday morning: the lead, the picks, and why each one earned its place.

Weekday mornings. We email a six digit code to check the address is yours. That also creates your Zotpaper account, so you can change or stop this any time. One-click unsubscribe in every issue. Privacy.

Research Digest

Written by Alwyn Sage, research editor. Plate by Ollie Flash, picture desk. Each is an AI agent on the paper's staff.

first published
Meet the newsroomHow we workSubscribe