URaG: Unified Retrieval and Generation in Multimodal LLMs for Efficient Long Document Understanding

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Shi, Yongxin, Wang, Jiapeng, Shan, Zeyu, Peng, Dezhi, Lin, Zening, Jin, Lianwen
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866908650682974208
author Shi, Yongxin
Wang, Jiapeng
Shan, Zeyu
Peng, Dezhi
Lin, Zening
Jin, Lianwen
author_facet Shi, Yongxin
Wang, Jiapeng
Shan, Zeyu
Peng, Dezhi
Lin, Zening
Jin, Lianwen
contents Recent multimodal large language models (MLLMs) still struggle with long document understanding due to two fundamental challenges: information interference from abundant irrelevant content, and the quadratic computational cost of Transformer-based architectures. Existing approaches primarily fall into two categories: token compression, which sacrifices fine-grained details; and introducing external retrievers, which increase system complexity and prevent end-to-end optimization. To address these issues, we conduct an in-depth analysis and observe that MLLMs exhibit a human-like coarse-to-fine reasoning pattern: early Transformer layers attend broadly across the document, while deeper layers focus on relevant evidence pages. Motivated by this insight, we posit that the inherent evidence localization capabilities of MLLMs can be explicitly leveraged to perform retrieval during the reasoning process, facilitating efficient long document understanding. To this end, we propose URaG, a simple-yet-effective framework that Unifies Retrieval and Generation within a single MLLM. URaG introduces a lightweight cross-modal retrieval module that converts the early Transformer layers into an efficient evidence selector, identifying and preserving the most relevant pages while discarding irrelevant content. This design enables the deeper layers to concentrate computational resources on pertinent information, improving both accuracy and efficiency. Extensive experiments demonstrate that URaG achieves state-of-the-art performance while reducing computational overhead by 44-56%. The code is available at https://github.com/shi-yx/URaG.
format Preprint
id arxiv_https___arxiv_org_abs_2511_10552
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle URaG: Unified Retrieval and Generation in Multimodal LLMs for Efficient Long Document Understanding
Shi, Yongxin
Wang, Jiapeng
Shan, Zeyu
Peng, Dezhi
Lin, Zening
Jin, Lianwen
Computation and Language
Recent multimodal large language models (MLLMs) still struggle with long document understanding due to two fundamental challenges: information interference from abundant irrelevant content, and the quadratic computational cost of Transformer-based architectures. Existing approaches primarily fall into two categories: token compression, which sacrifices fine-grained details; and introducing external retrievers, which increase system complexity and prevent end-to-end optimization. To address these issues, we conduct an in-depth analysis and observe that MLLMs exhibit a human-like coarse-to-fine reasoning pattern: early Transformer layers attend broadly across the document, while deeper layers focus on relevant evidence pages. Motivated by this insight, we posit that the inherent evidence localization capabilities of MLLMs can be explicitly leveraged to perform retrieval during the reasoning process, facilitating efficient long document understanding. To this end, we propose URaG, a simple-yet-effective framework that Unifies Retrieval and Generation within a single MLLM. URaG introduces a lightweight cross-modal retrieval module that converts the early Transformer layers into an efficient evidence selector, identifying and preserving the most relevant pages while discarding irrelevant content. This design enables the deeper layers to concentrate computational resources on pertinent information, improving both accuracy and efficiency. Extensive experiments demonstrate that URaG achieves state-of-the-art performance while reducing computational overhead by 44-56%. The code is available at https://github.com/shi-yx/URaG.
title URaG: Unified Retrieval and Generation in Multimodal LLMs for Efficient Long Document Understanding
topic Computation and Language
url https://arxiv.org/abs/2511.10552