Align Documents to Questions: Question-Oriented Document Rewriting for Retrieval-Augmented Generation

Jiaang Li, Zhendong Mao, Quan Wang, Yuning Wan, Yongdong Zhang


Abstract
Retrieval-Augmented Generation (RAG) enhances the factuality of Large Language Models (LLMs) by incorporating retrieved documents and/or generated context. However, LLMs often exhibit a stylistic bias when presented with mixed contexts, favoring fluent but hallucinated generated content over factually grounded yet disorganized retrieved evidence. This phenomenon reveals that the utility of retrieved information is bottlenecked by its presentation. To bridge this gap, we propose QREAM, a style-controlled rewriter that aligns retrieved documents with a question-oriented style while preserving facts, better for LLM readers to utilize. Our framework consists of two stages: (1) QREAM-ICL, which uses stylistic seeds to guide iterative rewriting exploration; and (2) QREAM-FT, a lightweight student model distilled from denoised ICL outputs. QREAM-FT employs dual-criteria rejection sampling, filtering based on answer correctness and factual consistency to ensure high-quality supervision. QREAM seamlessly integrates into existing RAG pipelines as a plug-and-play module. Experiments demonstrate that QREAM consistently enhances advanced RAG pipelines, yielding up to 8% relative improvement with negligible latency overhead, effectively balancing question relevance with factual grounding.
Anthology ID:
2026.findings-acl.884
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
17838–17849
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.884/
DOI:
Bibkey:
Cite (ACL):
Jiaang Li, Zhendong Mao, Quan Wang, Yuning Wan, and Yongdong Zhang. 2026. Align Documents to Questions: Question-Oriented Document Rewriting for Retrieval-Augmented Generation. In Findings of the Association for Computational Linguistics: ACL 2026, pages 17838–17849, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Align Documents to Questions: Question-Oriented Document Rewriting for Retrieval-Augmented Generation (Li et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.884.pdf
Checklist:
 2026.findings-acl.884.checklist.pdf