E2E-VisRAG: End-to-End Visual Retrieval-Augmented Generation with Differentiable Retrieval

Zehua Wang, Zhaojin Zhang, Xiaolong Weng, Yanzhi Chen, Ying Xiong, Buzhou Tang


Abstract
Retrieval-Augmented Generation (RAG) grounds large language models (LLMs) in external evidence, but existing RAG systems remain poorly matched to visually rich documents. OCR-based methods discard layout and non-textual visual cues, and although vision-language models (VLMs) can better understand native visual information, existing visual retrieval approaches often compress entire pages into global embedding vectors, obscuring fine-grained spatial information such as table cells and local annotations. Meanwhile, discrete Top-K selection in cascaded architectures blocks gradient flow between retrieval and generation, preventing joint optimization.In this work, we propose E2E-VisRAG, an end-to-end visual RAG framework that operates directly on native visual patches. Building upon patch-level visual encoding, we introduce a differentiable Top-K selection mechanism to enable joint training of retrieval and generation, and employ a gated residual fusion module to inject retrieval-enhanced visual features while suppressing noise. This design abandons the traditional paradigms of relying on OCR, global embeddings, and retrieval–generation separation, maximizing the preservation of original document visual semantics. Experiments on Vidore and MMLongBench-Doc show that, under matched 3B generator capacity, E2E-VisRAG outperforms non-end-to-end visual retrieval baselines, improving both Recall@1 and Accuracy@1 over OCR-based, global-embedding, and patch-level retrieval methods.
Anthology ID:
2026.emnlp-main.1030
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
23670–23683
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1030/
DOI:
Bibkey:
Cite (ACL):
Zehua Wang, Zhaojin Zhang, Xiaolong Weng, Yanzhi Chen, Ying Xiong, and Buzhou Tang. 2026. E2E-VisRAG: End-to-End Visual Retrieval-Augmented Generation with Differentiable Retrieval. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 23670–23683, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
E2E-VisRAG: End-to-End Visual Retrieval-Augmented Generation with Differentiable Retrieval (Wang et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1030.pdf