Localizing Prompt Ambiguity in Large Language Models with Probe-Targeted Attribution

Govind Ramesh, Yao Dou, Wei Xu


Abstract
Prompt ambiguity is a common source of failure in large language models, but is difficult to localize because it is a latent property of the prompt, while existing attribution methods are designed to explain observable outputs such as logits or generated tokens. We introduce PRIG, a gradient attribution method that uses a probe logit to attribute latent properties to token positions. Specifically, PRIG trains a linear probe to distinguish clear prompts from ambiguous prompts and attributes the probe score to earlier token representations in the residual stream. To enable token-level evaluation, we construct synthetic ambiguity datasets across coding, math, and writing by rewriting one task-critical sentence per prompt, and complement them with a human-written gold benchmark. In this setting, PRIG localizes ambiguous spans substantially better than gradient attribution baselines, achieving 0.840 AUROC on the combined synthetic benchmark. It also outperforms GPT-5.4 on sentence-level ambiguity identification and retains useful signal out-of-domain. More broadly, these results show that latent prompt properties can be localized through intermediate representations, rather than through output-level attribution.
Anthology ID:
2026.findings-emnlp.2103
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:
44774–44795
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.2103/
DOI:
Bibkey:
Cite (ACL):
Govind Ramesh, Yao Dou, and Wei Xu. 2026. Localizing Prompt Ambiguity in Large Language Models with Probe-Targeted Attribution. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 44774–44795, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Localizing Prompt Ambiguity in Large Language Models with Probe-Targeted Attribution (Ramesh et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.2103.pdf