πŸ“–
Did the Model Use That? Counterfactual Training Boosts Self-Attribution in Language Models
Zikai Zhou, Jiacheng Sang, Yilin Xu, Yufei Jin, Monica S. Lam

Did the Model Use That? Counterfactual Training Boosts Self-Attribution in Language Models

Computer Science & Mathematics Published 2026 Verification Pending

Overview & Synopsis

A fundamental challenge for explainable and accountable language models is revealing how external information influences their answers. This paper examines whether models are self-attributive: that is, whether they indicate when -- and only when -- their responses to questions are influenced by information provided alongside the question. Furthermore, we explore how self-attribution can be improved through counterfactual training, which explicitly teaches models to distinguish between answers that do and do not depend on supplied information. We found that most of the eleven frontier models rarely attribute influential information in their answers, and the rest attribute non-influential information. Training acknowledgment into reasoning does not fix this: the acknowledgment does not reach the final answer. We introduce CREDIT, which directly supervises final-answer self-attribution with counterfactual labels, without the need for human annotation. CREDIT substantially improves attribution in four open models, including two distilled reasoning models, and outperforms reasoning-level fine-tuning and prompting baselines. Among answers that agree with a user's suggestion, CREDIT distinguishes those changed by the suggestion from those the model would have given anyway. Gains transfer partly to unseen information types and prompts containing sources that support conflicting answers.

Advertisement
SPONSOR CONTENT
Indexed from repository: arXiv (arxiv.org)
Loading document from verified source...
Keyboard Shortcuts: ← / → for pages • + / - for zoom • 0 to fit width Powered by Mozilla PDF.js v3.11

Document Information

Document Title Did the Model Use That? Counterfactual Training Boosts Self-Attribution in Language Models
Author / Organization Zikai Zhou, Jiacheng Sang, Yilin Xu, Yufei Jin, Monica S. Lam
Subject & Discipline cs.CL
Categories Computer Science & Mathematics
Publication Year 2026
License / Rights arXiv Open Access / Creative Commons
Resource Verification Listed PDF (Verification Pending)
Original Host Website arXiv (arxiv.org)
File Format application/pdf (Portable Document Format)
Language EN
Catalog ID DOC-3477
Original Source Link Visit Source Page on arxiv.org »