GRASP: Geometry-aware Residual Alignment for Scalable Pretraining Data Attribution

Yue Min, Ruining Chen, Yujun Li


Abstract
Pretraining-data attribution typically assigns scalar scores to individual training examples. However, practical data curation involves retaining or removing coherent subsets, making additive pointwise assumptions mathematically restrictive. To address this, we formalize attribution as subset-level counterfactual utility prediction and propose GRASP (Geometry-aware Residual Alignment for Subset Prediction). GRASP combines first-order residual alignment with a low-dimensional train-train geometry term, derived from a smoothness lower bound. By explicitly modeling data interactions via compact 65-dimensional feature sketches, GRASP reduces upfront artifact construction time by nearly an order of magnitude compared to standard baselines. It efficiently scores 100,000 candidate subsets in just 5 seconds on a single CPU, completely bypassing the prohibitive cost of combinatorial retraining. Empirically, GRASP strictly dominates scalable baselines, more than doubling the average task-level subset-utility rank correlation on Linear Datamodeling Score (LDS) benchmarks, while fixed-compute diagnostics on Common Crawl further qualify its practical utility in large-scale curation.
Anthology ID:
2026.emnlp-main.1430
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
32663–32683
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1430/
DOI:
Bibkey:
Cite (ACL):
Yue Min, Ruining Chen, and Yujun Li. 2026. GRASP: Geometry-aware Residual Alignment for Scalable Pretraining Data Attribution. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 32663–32683, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
GRASP: Geometry-aware Residual Alignment for Scalable Pretraining Data Attribution (Min et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1430.pdf