CoRect: Context-Aware Logit Contrast for Hidden State Rectification to Resolve Knowledge Conflicts

Xuhua Ma, Richong Zhang, Zhijie Nie


Abstract
Retrieval-Augmented Generation (RAG) can fail under knowledge conflicts, where a model’s parametric prior competes with retrieved evidence and leads to unfaithful outputs. Existing mitigation methods typically operate on the final decoding distribution or rely on weight editing with access to ground-truth targets. Through layer-wise analysis, we identify a **parametric suppression** pattern: in many conflict cases, evidence-supported answers become salient in intermediate representations but are weakened by later layer outputs. To address this, we propose CoRect (**Co**ntext-Aware Logit Contrast for Hidden State **Rect**ification), a training-free inference-time method. By contrasting contextualized and context-free forward passes, CoRect first selects an evidence-supported token direction without using gold labels. It then rectifies hidden states in layers whose outputs suppress this direction. Across Question Answering (QA) and summarization benchmarks, CoRect improves average faithfulness and reduces hallucinations compared with strong decoding baselines.
Anthology ID:
2026.emnlp-main.539
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:
12414–12432
Language:
URL:
https://aclanthology.org/2026.emnlp-main.539/
DOI:
Bibkey:
Cite (ACL):
Xuhua Ma, Richong Zhang, and Zhijie Nie. 2026. CoRect: Context-Aware Logit Contrast for Hidden State Rectification to Resolve Knowledge Conflicts. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 12414–12432, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
CoRect: Context-Aware Logit Contrast for Hidden State Rectification to Resolve Knowledge Conflicts (Ma et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.539.pdf