English

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

Computation and Language 2026-02-10 v1 Artificial Intelligence

Abstract

Retrieval-Augmented Generation (RAG) often struggles with knowledge conflicts, where model-internal parametric knowledge overrides retrieved evidence, leading to unfaithful outputs. Existing approaches are often limited, relying either on superficial decoding adjustments or weight editing that necessitates ground-truth targets. Through layer-wise analysis, we attribute this failure to a parametric suppression phenomenon: specifically, in deep layers, certain FFN layers overwrite context-sensitive representations with memorized priors. To address this, we propose CoRect (Context-Aware Logit Contrast for Hidden State Rectification). By contrasting logits from contextualized and non-contextualized forward passes, CoRect identifies layers that exhibit high parametric bias without requiring ground-truth labels. It then rectifies the hidden states to preserve evidence-grounded information. Across question answering (QA) and summarization benchmarks, CoRect consistently improves faithfulness and reduces hallucinations compared to strong baselines.

Keywords

Cite

@article{arxiv.2602.08221,
  title  = {CoRect: Context-Aware Logit Contrast for Hidden State Rectification to Resolve Knowledge Conflicts},
  author = {Xuhua Ma and Richong Zhang and Zhijie Nie},
  journal= {arXiv preprint arXiv:2602.08221},
  year   = {2026}
}
R2 v1 2026-07-01T10:27:12.148Z