mlsanityData to 14 Oct 2026

Beyond Prefixes: Graph-as-Memory Cross-Attention for Knowledge Graph Completion with Large Language Models

arXiv 2510.08966 ↗First posted 10 Oct 2025cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Fusing Knowledge Graphs with Large Language Models (LLMs) is crucial for knowledge-intensive tasks like knowledge graph completion. Existing LLM-based approaches typically inject graph information via prefix concatenation, resulting in shallow interactions that fail to support fine-grained evidence retrieval during generation. Beyond prefixes, we propose Graph-as-Memory Tuning (GMT), a new paradigm that represents local graph structure as explicit graph memory and injects it into LLMs via deep, token-wise cross-attention. Specifically, GMT first employs a Semantic Graph Module to encode context-aware semantics from local neighborhoods guided by knowledge-enhanced relations, and compresses them into a fixed number of graph memory tokens. A Graph-as-Memory Cross-Attention Fusion Module then integrates these tokens into multiple Transformer layers, allowing LLM hidden state to dynamically retrieve relevant graph evidence. To enable efficient adaptation, GMT applies LoRA only to the memory cross-attention while keeping the base LLM frozen. Extensive experiments show that GMT significantly outperforms prefix-tuning and other strong baselines, providing more potent signals for robust reasoning. The code is published at https://github.com/tongruiliu/GMT.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 39 references that are in the map. First posted on arXiv 10 Oct 2025.

Builds on

23 of 39 references.

References · 23

  1. 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
  2. 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 932
  3. 2016Complex Embeddings for Simple Link Predictioncited by 902
  4. 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 898
  5. 2017Convolutional 2D Knowledge Graph Embeddingscited by 730
  6. 2023Unifying Large Language Models and Knowledge Graphs: A Roadmapcited by 668
  7. 2019Composition-based Multi-Relational Graph Convolutional Networkscited by 396
  8. 2019KG-BERT: BERT for Knowledge Graph Completioncited by 324
  9. 2021Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Predictioncited by 275
  10. 2019Learning Hierarchy-Aware Knowledge Graph Embeddings for Link Predictioncited by 133
  11. 2020BoxE: A Box Embedding Model for Knowledge Base Completioncited by 105
  12. 2022SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Modelscited by 91
11 more
  1. 2023Making Large Language Models Perform Better in Knowledge Graph Completioncited by 48
  2. 2024KICGPT: Large Language Model with Knowledge in Context for Knowledge Graph Completioncited by 36
  3. 2023A Universal Question-Answering Platform for Knowledge Graphscited by 17
  4. 2023Data Scarcity in Recommendation Systems: A Surveycited by 17
  5. 2024MKGL: Mastery of a Three-Word Languagecited by 17
  6. 2024KG-FIT: Knowledge Graph Fine-Tuning Upon Open-World Knowledgecited by 16
  7. 2024Path-based Explanation for Knowledge Graph Completioncited by 11
  8. 2025Self-supervised Quantized Representation for Seamlessly Integrating Knowledge Graphs with Large Language Modelscited by 11
  9. 2024Comprehending Knowledge Graphs with Large Language Models for Recommender Systemscited by 9
  10. 2025GLTW: Joint Improved Graph Transformer and LLM via Three-Word Language for Knowledge Graph Completioncited by 2
  11. 2025Flow-Modulated Scoring for Semantic-Aware Knowledge Graph Completioncited by 1

Built on by

Not cited yet.