English

Regime-Aware Peer Specialization for Robust RAG under Heterogeneous Knowledge Conflicts

Computation and Language 2026-06-29 v1

Abstract

Retrieval-augmented generation (RAG) improves language models by grounding generation in external context. However, it can be fragile when the retrieved context conflicts with the model's parametric knowledge. Such conflicts span a reliability spectrum, ranging from reliable and partially reliable evidence to adversarial context. Existing remedies often handle such heterogeneous conflicts with regime-agnostic supervision, which can conflate incompatible learning signals across reliability regimes. To disentangle these signals, we propose RAPS-DA, a regime-aware peer specialization framework that addresses conflict at two complementary granularities. At the sample level, conflicts are divided into three regimes, including Grounding, Arbitration, and Resistance, with one same-scale peer specialist trained per regime from a shared base model. Each sample is then hard-routed to its regime-matched peer for on-policy reverse-KL supervision. At the token level, a dual-layer selector uses inter-teacher disagreement, student-teacher divergence, and student entropy to filter uninformative or unstable tokens, upweight confidently misaligned ones, and gradually focus supervision on high-conflict tokens as the student matures. Gains stem from specialization at a fixed model scale, not from a stronger teacher, and the peer specialists exist only during training, so the deployed student requires no regime labels or peer access. Experiments on five conflict scenarios and two out-of-distribution benchmarks show RAPS-DA surpasses all prompting, decoding, fine-tuning, RL, and single-teacher baselines.

Cite

@article{arxiv.2606.30518,
  title  = {Regime-Aware Peer Specialization for Robust RAG under Heterogeneous Knowledge Conflicts},
  author = {Bo Wang and Heyan Huang and Yaolin Li and Yanghao Zhou and Jiahao Teng and Ziyi Yang and Ge Shi and Chong Feng},
  journal= {arXiv preprint arXiv:2606.30518},
  year   = {2026}
}

Comments

Working in Progress