Multimodal Adaptive Retrieval Augmented Generation through Internal Representation Learning
Organizations: Shanghaitech University School of Information Science and Technology · Chinese Academy of Sciences Institute of Automation · Ant Group
Abstract
Visual Question Answering systems face reliability issues due to hallucinations, where models generate answers misaligned with visual input or factual knowledge. While Retrieval Augmented Generation frameworks mitigate this issue by incorporating external knowledge, static retrieval often introduces irrelevant or conflicting content, particularly in visual RAG settings where visually similar but semantically incorrect evidence may be retrieved. To address this, we propose Multimodal Adaptive RAG (MMA-RAG), which dynamically assesses the confidence in the internal knowledge of the model to decide whether to incorporate the retrieved external information into the generation process. Central to MMA-RAG is a decision classifier trained through a layer-wise analysis, which leverages joint internal visual and textual representations to guide the use of reverse image retrieval. Experiments demonstrated that the model achieves a significant improvement in response performance in three VQA datasets. Meanwhile, ablation studies highlighted the importance of internal representations in adaptive retrieval decisions. In general, the experimental results demonstrated that MMA-RAG effectively balances external knowledge utilization and inference robustness in diverse multimodal scenarios.
Figures & tables
| Dataset | Infoseek | OK-VQA | E-VQA |
|---|---|---|---|
| Training | 3740 | 1000 | 1000 |
| Evaluation | 1646 | 4989 | 3345 |
| Model | Method | InfoSeek | OK-VQA | E-VQA |
|---|---|---|---|---|
| Zero shot | 19.9 | 58.7 | 10.0 | |
| Few shot | 15.9 | 58.5 | 14.1 | |
| RIR | 23.3 | 62.2 | 19.8 | |
| Qwen2VL | CoT | 20.4 | 46.7 | 14.4 |
| P(true) | 22.6 | 53.3 | 19.1 | |
| CLIP | 20.9 | 56.0 | 18.5 |
| Method | Infoseek | OK-VQA | E-VQA |
|---|---|---|---|
| RIR | 17.2 | 56.7 | 14.1 |
| wo-text | 17.9 | 58.6 | 14.2 |
| wo-vision | 19.3 | 59.9 | 14.2 |
| MM Adaptive RAG | 20.3 | 60.1 | 14.6 |
Multimodal RAG
Multimodal retrieval-augmented generation grounds AI answers in retrieved text, images, video, audio, and structured visual evidence.
See all 157 papers