Hallucination-Aware Multi-Agent RAG via Internal Attention Rewards

Runsong Jia, Qian Chang, Mengjia Wu, Ciprian Doru Giurcaneanu, Jie Lu, Yi Zhang


Abstract
Retrieval-augmented generation (RAG) grounds language models in retrieved evidence, but its outputs often contain claims unsupported by that evidence. Recent multi-agent RAG systems decompose the pipeline into cooperative decisions, yet they typically optimize all agents under a shared task reward such as EM or F1, which measures answer overlap rather than faithfulness to evidence. We propose a hallucination-aware multi-agent RAG framework that combines the task reward with an internal hallucination-risk signal computed from the generator’s attention-head norms. Three role-specific LoRA adapters, a query planner, an evidence selector, and an answer generator, share a frozen backbone. A lightweight critic is trained once on TruthfulQA truthful/false pairs and then kept frozen, providing a complementary risk score that is added to the shared reward during cooperative PPO training. Across three backbones (Qwen3-8B, Llama3-8B, Llama2-13B) and three benchmarks (RAGTruth, HotpotQA, 2WikiMultiHopQA), our method improves task performance and reduces hallucination rates over the strongest multi-agent baseline. Ablations and critic diagnostics show that the TruthfulQA-calibrated signal provides useful reward shaping that transfers across RAG tasks.
Anthology ID:
2026.findings-emnlp.1809
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:
38648–38664
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1809/
DOI:
Bibkey:
Cite (ACL):
Runsong Jia, Qian Chang, Mengjia Wu, Ciprian Doru Giurcaneanu, Jie Lu, and Yi Zhang. 2026. Hallucination-Aware Multi-Agent RAG via Internal Attention Rewards. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 38648–38664, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Hallucination-Aware Multi-Agent RAG via Internal Attention Rewards (Jia et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1809.pdf