RAG-GNN: Integrating Retrieved Knowledge with Graph Neural Networks for Precision Medicine
On the map: Retrieval-augmented generation → Medical and biomedical RAG
Abstract (arXiv)
Network topology excels at structural predictions but fails to capture functional semantics encoded in biomedical literature. We present RAG-GNN, an end-to-end trainable retrieval-augmented graph neural network framework that integrates GNN representations with dynamically retrieved literature-derived knowledge through a jointly optimized retrieval projection, gated fusion mechanism, and contrastive alignment. In a cancer signaling case study (379 proteins, 3,498 interactions, 14 functional categories), RAG-GNN improves functional clustering from silhouette = -0.237 ± 0.065 (GNN-only) to -0.144 ± 0.066, a consistent improvement of +0.093 ± 0.022 across 10 random seeds, while the learned retrieval achieves mean precision@10 = 0.242, a 152% improvement over the random baseline (0.096). Heuristic information decomposition with bootstrap confidence intervals reveals that topology and retrieval encode overwhelmingly shared information (95.6%), with retrieval improving both intra-cluster cohesion (silhouette) and cluster agreement (ARI +0.021 ± 0.015). Counterfactual experiments confirm that adversarial, absent, and random retrieval all degrade performance, validating that the gated fusion mechanism depends on document content. Benchmarking against eight established embedding methods demonstrates task-specific complementarity: topology-focused methods achieve strong link prediction, while retrieval augmentation consistently improves functional clustering within the controlled GNN-only ablation. DDR1 subnetwork analysis provides confirmatory validation consistent with established synthetic lethality relationships. These results establish that topology-only and retrieval-augmented approaches serve complementary purposes for precision medicine applications.
Key facts. Cited by 1 paper in the map (data through 13 Oct 2026). Placed in Medical and biomedical RAG, one of 15 directions in Retrieval-augmented generation. Builds on 25 of its 61 references that are in the map. First posted on arXiv 31 Jan 2026.
Builds on
25 of 61 references.
References · 25
- 2017Attention Is All You Needcited by 60,106
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,304
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,318
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,221
- 2017Graph Attention Networkscited by 8,888
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,712
- 2017Inductive Representation Learning on Large Graphscited by 6,303
- 2017Neural Message Passing for Quantum Chemistrycited by 3,471
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,196
- 2016node2vec: Scalable Feature Learning for Networkscited by 1,915
- 2014DeepWalk: Online Learning of Social Representationscited by 1,544
- 2021Improving language models by retrieving from trillions of tokenscited by 1,286
13 more
- 2020Domain-Specific Language Model Pretraining for Biomedical Natural Language Processingcited by 1,001
- 2019GraphSAINT: Graph Sampling Based Inductive Learning Methodcited by 618
- 2015LINE: Large-scale Information Network Embeddingcited by 582
- 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 260
- 2018Modeling polypharmacy side effects with graph convolutional networkscited by 237
- 2021ChemRL-GEM: Geometry Enhanced Molecular Representation Learning for Property Predictioncited by 204
- 2023BiomedGPT: A Generalist Vision-Language Foundation Model for Diverse Biomedical Taskscited by 201
- 2014Asymmetric LSH (ALSH) for Sublinear Time Maximum Inner Product Search (MIPS)cited by 109
- 2020Network Medicine Framework for Identifying Drug Repurposing Opportunities for COVID-19cited by 38
- 2023xTrimoGene: An Efficient and Scalable Representation Learner for Single-Cell RNA-Seq Datacited by 12
- 2025Hierarchical Molecular Language Models (HMLMs)cited by 4
- 2026Attention mechanisms in neural networkscited by 4
- 2026Resonant Sparse Geometry Networkscited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Selective Synchronization Attentioncited by 2
- 2026Selective Synchronization Attention16 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2024RAGraph: A General Retrieval-Augmented Graph Learning Framework
- 2025fastbmRAG: A Fast Graph-Based RAG Framework for Efficient Processing of Large-Scale Biomedical Literature
- 2025BioGraphFusion: Graph Knowledge Embedding for Biological Completion and Reasoning
- 2023Biomedical knowledge graph-optimized prompt generation for large language models
- 2024Graph-Based Retriever Captures the Long Tail of Biomedical Knowledge
- 2025K-Paths: Reasoning over Graph Paths for Drug Repurposing and Drug Interaction Prediction
- 2025DrKGC: Dynamic Subgraph Retrieval-Augmented LLMs for Knowledge Graph Completion across General and Biomedical Domains
- 2026GLM-RAG: Graph Language Models for Graph-Based Retrieval-Augmented Generation