FLAME: Empowering Frozen LLMs for Knowledge Graph Completion
Not on the map. Why?
Abstract (arXiv)
Traditional knowledge graph completion (KGC) methods rely solely on structural information and struggle with sparsity, while Large Language Models (LLMs) address these limitations through rich world knowledge and strong context modeling. Fine-tuning LLMs is effective but costly, while non-fine-tuned LLMs are efficient but suboptimal. To address this trade-off, we propose FLAME, a framework that extracts context-aware hidden states from intermediate layers of frozen LLMs to train data-efficient KGC classifiers. We bridge LLM-KG semantic gaps via subgraph-based entity descriptions and employ sliced mutual information (SMI) to quantify task-relevant information in representations. Experiments demonstrate that FLAME achieves 47% improvement over non-fine-tuned LLM baselines and, to our knowledge, is the first to achieve fine-tuned performance with 188× memory efficiency and 26.11× speedup.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 30 references that are in the map. First posted on arXiv 13 Aug 2024.
Builds on
19 of 30 references.
References · 19
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2023Mistral 7Bcited by 3,065
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
- 2023LIMA: Less Is More for Alignmentcited by 1,080
- 2014Embedding Entities and Relations for Learning and Inference in Knowledge Basescited by 929
- 2016Complex Embeddings for Simple Link Predictioncited by 897
- 2019RotatE: Knowledge Graph Embedding by Relational Rotation in Complex Spacecited by 893
- 2024Gemma: Open Models Based on Gemini Research and Technologycited by 889
- 2017Convolutional 2D Knowledge Graph Embeddingscited by 729
- 2020Knowledge Graphscited by 619
- 2019KG-BERT: BERT for Knowledge Graph Completioncited by 323
- 2022Sequence-to-Sequence Knowledge Graph Completion and Question Answeringcited by 75
7 more
- 2023Making Large Language Models Perform Better in Knowledge Graph Completioncited by 48
- 2023Exploring Large Language Models for Knowledge Graph Completioncited by 43
- 2024KICGPT: Large Language Model with Knowledge in Context for Knowledge Graph Completioncited by 36
- 2022Knowledge Is Flat: A Seq2Seq Generative Framework for Various Knowledge Graph Completioncited by 32
- 2022Joint Language Semantic and Structure Embedding for Knowledge Graph Completioncited by 15
- 2023Enhancing Text-based Knowledge Graph Completion with Zero-Shot Large Language Models: A Focus on Semantic Enhancementcited by 14
- 2024Rule or Story, Which is a Better Commonsense Expression for Talking with Large Language Models?cited by 1
Built on by
Not cited yet.