ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
15
Citations
1
Influential Citations
Annual Meeting of the Association for Computational Linguistics
Venue
2025
Year
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.
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.
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.
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.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba