Black-Box Dataset-Level Source Ranking via Context Mixture Factorization

Milad Fotouhi, Mohammad Taha Bahadori, Payman Arabshahi, Oluwaseyi Feyisetan


Abstract
Training data attribution (TDA) for large language models is dominated by gradient- and retraining-based methods that require model internals, ruling out the most widely deployed closed-weight systems. We study TDA in the strictly black-box setting, observing only input output behavior accessed through retrieval- augmented prompting. We introduce Context Mixture Factorization (CMF), which models the LLM’s output distribution as a mixture of a base component and dataset-specific components and recovers per-dataset contribution weights by regularized matrix factorization via alternating projected least squares. CMF explicitly decomposes background (π0) and dataset- specific influence and requires only K+1 prompts per query for K candidate datasets. As a comparison method we also formulate the Shapley Context Method (SCM), a residual- similarity attribution score aggregated via Shapley values. We evaluate four instruction-tuned LLMs (GPT-4, Bloomz, Mistral-7B, Phi-3mini) on BoolQ, a synthetic FakeQ variant, and a post-cutoff Olympic2024 dataset against three baselines (TRAK, ContextCite, retrieval cosine similarity). CMF and SCM recover the ordering predicted by our exposure-proxy setup (BoolQ>FakeQ>Olympic2024); ContextCite shows the opposite ordering, attributing most to Olympic2024 because the model relies on context most when it has no parametric knowledge — supporting the claim that span-level context attribution and dataset-level exposure attribution answer different questions. A synthetic identifiability experiment, controlled fine- tuning sweeps, and a multi-checkpoint unlearning case study provide supporting evidence that the metrics behave as exposure-sensitive source-ranking proxies.
Anthology ID:
2026.findings-emnlp.1587
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
33781–33801
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1587/
DOI:
10.18653/2026.findings-emnlp.1587
Bibkey:
Cite (ACL):
Milad Fotouhi, Mohammad Taha Bahadori, Payman Arabshahi, and Oluwaseyi Feyisetan. 2026. Black-Box Dataset-Level Source Ranking via Context Mixture Factorization. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 33781–33801, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Black-Box Dataset-Level Source Ranking via Context Mixture Factorization (Fotouhi et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1587.pdf