HypRAG: Hyperbolic Dense Retrieval for Retrieval Augmented Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Madhu, Hiren, Bui, Ngoc, Maatouk, Ali, Tassiulas, Leandros, Krishnaswamy, Smita, Yang, Menglin, Ganguly, Sukanta, Srinivasan, Kiran, Ying, Rex
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917257309847552
author Madhu, Hiren
Bui, Ngoc
Maatouk, Ali
Tassiulas, Leandros
Krishnaswamy, Smita
Yang, Menglin
Ganguly, Sukanta
Srinivasan, Kiran
Ying, Rex
author_facet Madhu, Hiren
Bui, Ngoc
Maatouk, Ali
Tassiulas, Leandros
Krishnaswamy, Smita
Yang, Menglin
Ganguly, Sukanta
Srinivasan, Kiran
Ying, Rex
contents Embedding geometry plays a fundamental role in retrieval quality, yet dense retrievers for retrieval-augmented generation (RAG) remain largely confined to Euclidean space. However, natural language exhibits hierarchical structure from broad topics to specific entities that Euclidean embeddings fail to preserve, causing semantically distant documents to appear spuriously similar and increasing hallucination risk. To address these limitations, we introduce hyperbolic dense retrieval, developing two model variants in the Lorentz model of hyperbolic space: HyTE-FH, a fully hyperbolic transformer, and HyTE-H, a hybrid architecture projecting pre-trained Euclidean embeddings into hyperbolic space. To prevent representational collapse during sequence aggregation, we introduce the Outward Einstein Midpoint, a geometry-aware pooling operator that provably preserves hierarchical structure. On MTEB, HyTE-FH outperforms equivalent Euclidean baselines, while on RAGBench, HyTE-H achieves up to 29% gains over Euclidean baselines in context relevance and answer relevance using substantially smaller models than current state-of-the-art retrievers. Our analysis also reveals that hyperbolic representations encode document specificity through norm-based separation, with over 20% radial increase from general to specific concepts, a property absent in Euclidean embeddings, underscoring the critical role of geometric inductive bias in faithful RAG systems.
format Preprint
id arxiv_https___arxiv_org_abs_2602_07739
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle HypRAG: Hyperbolic Dense Retrieval for Retrieval Augmented Generation
Madhu, Hiren
Bui, Ngoc
Maatouk, Ali
Tassiulas, Leandros
Krishnaswamy, Smita
Yang, Menglin
Ganguly, Sukanta
Srinivasan, Kiran
Ying, Rex
Information Retrieval
Artificial Intelligence
Embedding geometry plays a fundamental role in retrieval quality, yet dense retrievers for retrieval-augmented generation (RAG) remain largely confined to Euclidean space. However, natural language exhibits hierarchical structure from broad topics to specific entities that Euclidean embeddings fail to preserve, causing semantically distant documents to appear spuriously similar and increasing hallucination risk. To address these limitations, we introduce hyperbolic dense retrieval, developing two model variants in the Lorentz model of hyperbolic space: HyTE-FH, a fully hyperbolic transformer, and HyTE-H, a hybrid architecture projecting pre-trained Euclidean embeddings into hyperbolic space. To prevent representational collapse during sequence aggregation, we introduce the Outward Einstein Midpoint, a geometry-aware pooling operator that provably preserves hierarchical structure. On MTEB, HyTE-FH outperforms equivalent Euclidean baselines, while on RAGBench, HyTE-H achieves up to 29% gains over Euclidean baselines in context relevance and answer relevance using substantially smaller models than current state-of-the-art retrievers. Our analysis also reveals that hyperbolic representations encode document specificity through norm-based separation, with over 20% radial increase from general to specific concepts, a property absent in Euclidean embeddings, underscoring the critical role of geometric inductive bias in faithful RAG systems.
title HypRAG: Hyperbolic Dense Retrieval for Retrieval Augmented Generation
topic Information Retrieval
Artificial Intelligence
url https://arxiv.org/abs/2602.07739