arrow
Return

ReflectRAG: Enhancing retrieval-augmented generation with GRPO-optimized iterative reflection

delete2026-05-21
delete0
PRE
AI
X
Xuanhe Chen
Y
Yuchen Li *
Y
Youyi Bi *
S
Shuaiqiang Wang
L
Linghe Kong
D
Dawei Yin
DOI:10.1016/j.neucom.2026.134047delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Retrieval-Augmented Generation (RAG) enhances question answering (QA) by grounding responses in external documents, yet struggles with irrelevant retrieval and single-pass generation errors. To address these issues, we propose ReflectRAG, a framework that advances RAG through structured planning, iterative reflection, and reinforcement learning (RL). Specifically, ReflectRAG first retrieves and re-ranks documents and generates answers guided by fact-based plans. Then, ReflectRAG leverages a reflection module to iteratively critique and refine plans and answers, and optimizes the reflection module via RL. By integrating planning, reflection, and RL, ReflectRAG ensures precise, contextually aligned answers, offering a robust solution for open-domain QA. To demonstrate the effectiveness of our proposed model, we carry out extensive experiments on MS MARCO v2.1, TriviaQA (TQA), and Natural Questions (NQ). The comparative results on three open-domain datasets show that ReflectRAG achieves up to +0.31 absolute F1 points over the BASE RAG baseline across MS MARCO v2.1, TriviaQA, and NQ, establishing ReflectRAG as an effective solution for enhancing factual accuracy and coherence in QA systems. The source code is available at https://github.com/uncxlirh/ReflectRAG .
Keywords:
Retrieval-Augmented Generation
Iterative Reflection
Reinforcement Learning
Question Answering
Structured Planning

Journal

Neurocomputing cover
Neurocomputing
IF:
6.5
Papers:
2.5W
Citations:
6.5W

Organization

B
baidu inc.
Scholars:
62
Papers: 26
Citations: 0
S
shanghai jiao tong university
Scholars:
15.6W
Papers: 11.6W
Citations: 159