Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

STRIDE: Training Data Attribution via Sparse Recovery from Subset Perturbations

About

Training Data Attribution (TDA) seeks to trace a model's predictions back to its training data. The gold standard for TDA relies on causal interventions, observing how a model changes when data is added or removed, but repeated retraining is computationally challenging for Large Language Models (LLMs). Consequently, most approaches approximate this effect in the parameter space using gradients. However, tracking gradients across billions of parameters is not only prohibitively expensive but relies on local approximations. In this work, we propose a shift: rather than estimating parameter changes, we model the functional effect of training data in the activation space. We introduce STRIDE (Steering-based Training Data Influence Decomposition), a framework that formulates TDA as a sparse recovery problem in the spirit of compressive sensing. STRIDE learns lightweight "steering operators" that mimic the behavioral shift caused by training on data subsets. By measuring how these operators perturb test predictions, we recover individual training example influences via sparse linear decomposition. STRIDE achieves state-of-the-art for LLM pre-training attribution while being an order of magnitude ($13\times$) faster than previous art. We further validate its practical utility through downstream applications including data selection, data contamination, and qualitative analysis.

Rishit Dagli, Abir Harrasse, Luke Zhang, Florent Draye, Amirali Abdullah, Bernhard Sch\"olkopf, Zhijing Jin• 2026

Related benchmarks

TaskDatasetResultRank
Training Data AttributionNanochat Unseen (test)
LDS Correlation0.1792
26
Data AttributionAlpaca
LDS Correlation0.2426
10
Data AttributionTulu
LDS Correlation0.1611
10
Data AttributionSafeRLHF
LDS Correlation0.3995
10
Data AttributionFlan
LDS Correlation0.1932
10
Data SelectionFLAN 100K (test)
Unigram F149.65
6
Dataset-contamination attributionMATH replicated controlled leaks in OpenWebText
Recall74.2
3
Showing 7 of 7 rows

Other info

Follow for update