Conference Paper
Machine Learning

UniversalRAG

Woongyeong Yeo, Kangsan Kim, Soyeong Jeong, Jinheon Baek, Sung Ju Hwang
April 29, 2025Annual Meeting of the Association for Computational Linguistics15 citations

15

Citations

1

Influential Citations

Annual Meeting of the Association for Computational Linguistics

Venue

2025

Year

Abstract

Retrieval-Augmented Generation (RAG) has shown substantial promise in improving factual accuracy by grounding model responses with external knowledge relevant to queries. However, most existing approaches are limited to a text-only corpus, and while recent efforts have extended RAG to other modalities such as images and videos, they typically operate over a single modality-specific corpus. In contrast, real-world queries vary widely in the type of knowledge they require, which a single type of knowledge source cannot address. To address this, we introduce UniversalRAG, an any-to-any RAG framework designed to retrieve and integrate knowledge from heterogeneous sources with diverse modalities and granularities. Specifically, motivated by the observation that forcing all modalities into a unified representation space derived from a single aggregated corpus causes a modality gap, where the retrieval tends to favor items from the same modality as the query, we propose modality-aware routing, which dynamically identifies the most appropriate modality-specific corpus and performs targeted retrieval within it, and further justify its effectiveness with a theoretical analysis. Moreover, beyond modality, we organize each modality into multiple granularity levels, enabling fine-tuned retrieval tailored to the complexity and scope of the query. We validate UniversalRAG on 10 benchmarks of multiple modalities, showing its superiority over various modality-specific and unified baselines.

Analysis

Why This Paper Matters

Retrieval-Augmented Generation (RAG) has become a cornerstone for improving factual accuracy in language models, but existing systems are largely confined to text-only corpora or single-modality extensions. Real-world queries often require knowledge from multiple modalities—text, images, videos, etc.—and a single type of source cannot address this diversity. UniversalRAG tackles this critical gap by introducing an any-to-any RAG framework that retrieves from heterogeneous sources with varying modalities and granularities. This is significant because it moves beyond the limitations of unified representation spaces, which suffer from a modality gap where retrieval biases toward same-modality items. By dynamically routing queries to the most appropriate modality-specific corpus, UniversalRAG enables more accurate and context-aware knowledge integration, directly addressing a practical need in multimodal AI applications.

Technical Contributions

  • Modality-aware routing: Dynamically identifies the best modality-specific corpus for a query, avoiding the modality gap problem inherent in unified spaces. The authors provide theoretical justification for this approach.
  • Multi-granularity organization: Each modality is structured into multiple granularity levels (e.g., coarse to fine), allowing retrieval to match the complexity and scope of the query.
  • Any-to-any retrieval: Supports queries and knowledge sources across different modalities, enabling flexible integration of heterogeneous information.
  • Theoretical analysis: Formalizes the modality gap and demonstrates why modality-aware routing outperforms unified representations.

Results

UniversalRAG was evaluated on 10 benchmarks spanning multiple modalities, including text, image, and video. It consistently outperformed both modality-specific baselines (e.g., text-only RAG, image-only RAG) and unified multimodal baselines. While specific numerical metrics are not detailed in the abstract, the paper claims superiority across all benchmarks, indicating robust generalization. The framework's ability to handle diverse query types and knowledge sources is a key empirical strength.

Significance

UniversalRAG has broad implications for AI systems that need to ground responses in real-world, multimodal knowledge. By enabling retrieval from heterogeneous sources, it enhances factual accuracy and relevance for applications like question answering, content generation, and decision support. The modality-aware routing concept could inspire new architectures for multimodal learning, moving beyond simple fusion to dynamic, context-dependent retrieval. This work also highlights the importance of addressing modality gaps in representation learning, potentially influencing future research in multimodal RAG and beyond.