Knowledge-Aware Language Model Pretraining
Not on the map. Why?
Abstract (arXiv)
How much knowledge do pretrained language models hold? Recent research observed that pretrained transformers are adept at modeling semantics but it is unclear to what degree they grasp human knowledge, or how to ensure they do so. In this paper we incorporate knowledge-awareness in language model pretraining without changing the transformer architecture, inserting explicit knowledge layers, or adding external storage of semantic information. Rather, we simply signal the existence of entities to the input of the transformer in pretraining, with an entity-extended tokenizer; and at the output, with an additional entity prediction task. Our experiments show that solely by adding these entity signals in pretraining, significantly more knowledge is packed into the transformer parameters: we observe improved language modeling accuracy, factual correctness in LAMA knowledge probing tasks, and semantics in the hidden representations through edge probing.We also show that our knowledge-aware language model (KALM) can serve as a drop-in replacement for GPT-2 models, significantly improving downstream tasks like zero-shot question-answering with no task-related training.
Key facts. Cited by 52 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 12; 2022: 11; 2023: 12; 2024: 13; 2025: 3; 2026: 1.
52 papers.
Sort:
By area:
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2023Unifying Large Language Models and Knowledge Graphs: A Roadmapcited by 663
- 2021Pre-Trained Models: Past, Present and Futurecited by 319
- 2022Neural Theory-of-Mind? On the Limits of Social Intelligence in Large LMscited by 220
- 2022A Review on Language Models as Knowledge Basescited by 156
- 2022Deep Bidirectional Language-Knowledge Graph Pretrainingcited by 156
- 2021COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretrainingcited by 126
- 2023Can Knowledge Graphs Reduce Hallucinations in LLMs? : A Surveycited by 92
- 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Taskscited by 73
- 2023Give Us the Facts: Enhancing Large Language Models with Knowledge Graphs for Fact-aware Language Modelingcited by 56
- 2023FactKB: Generalizable Factuality Evaluation using Language Models Enhanced with Factual Knowledgecited by 55
- 2022Knowledge Graph Embedding: A Survey from the Perspective of Representation Spacescited by 41
38 more
- 2021Knowledge Enhanced Pretrained Language Models: A Compreshensive Surveycited by 33
- 2023Graph-Aware Language Model Pre-Training on a Large Graph Corpus Can Help Multiple Graph Applicationscited by 33
- 2024Graph Machine Learning in the Era of Large Language Models (LLMs)cited by 33
- 2021FlipDA: Effective and Robust Data Augmentation for Few-Shot Learningcited by 32
- 2021Knowledge-Grounded Dialogue Generation with a Unified Knowledge Representationcited by 32
- 2021Relational World Knowledge Representation in Contextual Language Models: A Reviewcited by 31
- 2024Chain-of-Action: Faithful and Multimodal Question Answering through Large Language Modelscited by 27
- 2021Combining pre-trained language models and structured knowledgecited by 25
- 2023KILM: Knowledge Injection into Encoder-Decoder Language Modelscited by 19
- 2024Conv-CoA: Improving Open-domain Question Answering in Large Language Models via Conversational Chain-of-Actioncited by 19
- 2021Knowledge Perceived Multi-modal Pretraining in E-commercecited by 18
- 2021XLM-K: Improving Cross-Lingual Language Model Pre-training with Multilingual Knowledgecited by 18
- 2025Large language models for artificial general intelligence (AGI): A survey of foundational principles and approachescited by 18
- 2024Semantic Sensitivities and Inconsistent Predictions: Measuring the Fragility of NLI Modelscited by 14
- 2021Billion-scale Pre-trained E-commerce Product Knowledge Graph Modelcited by 13
- 2025Sample Efficient Demonstration Selection for In-Context Learningcited by 12
- 2024KnowTuning: Knowledge-aware Fine-tuning for Large Language Modelscited by 11
- 2022Knowledge Mining with Scene Text for Fine-Grained Recognitioncited by 10
- 2024LLM-Powered Explanations: Unraveling Recommendations Through Subgraph Reasoningcited by 10
- 2024CuriousLLM: Elevating Multi-Document Question Answering with LLM-Enhanced Knowledge Graph Reasoningcited by 7
- 2024A Theory for Token-Level Harmonization in Retrieval-Augmented Generationcited by 7
- 2024Knowledge Graph Tuning: Real-time Large Language Model Personalization based on Human Feedbackcited by 6
- 2023ArK: Augmented Reality with Knowledge Interactive Emergent Abilitycited by 5
- 2023KGLens: Towards Efficient and Effective Knowledge Probing of Large Language Models with Knowledge Graphscited by 5
- 2024Knowledge-Centric Templatic Views of Documentscited by 5
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoningcited by 5
- 2022OPAL: Ontology-Aware Pretrained Language Model for End-to-End Task-Oriented Dialoguecited by 4
- 2024DEXTER: A Benchmark for open-domain Complex Question Answering using LLMscited by 4
- 2022Does Entity Abstraction Help Generative Transformers Reason?cited by 3
- 2022A Survey on Knowledge-Enhanced Pre-trained Language Modelscited by 3
- 2023In-Context Ability Transfer for Question Decomposition in Complex QAcited by 3
- 2023Furthest Reasoning with Plan Assessment: Stable Reasoning Path with Retrieval-Augmented Large Language Modelscited by 2
- 2025BALI: Enhancing Biomedical Language Representations through Knowledge Graph and Language Model Alignmentcited by 2
- 2021K-XLNet: A General Method for Combining Explicit Knowledge with Language Model Pretrainingcited by 1
- 2021Linking-Enhanced Pre-Training for Table Semantic Parsingcited by 1
- 2023Exploring Category Structure with Contextual Language Models and Lexical Semantic Networkscited by 1
- 2022PKGM: A Pre-trained Knowledge Graph Model for E-commerce Applicationcited by 0
- 2022ComDensE : Combined Dense Embedding of Relation-aware and Common Features for Knowledge Graph Completioncited by 0
50 most cited of 52 All 52 (JSON)
- 2026From Found to Designed: Concepts as a Design Axis for Large Language Models29 Jul 2026
- 2025BALI: Enhancing Biomedical Language Representations through Knowledge Graph and Language Model Alignment9 Sep 2025
- 2025Sample Efficient Demonstration Selection for In-Context Learning10 Jun 2025
- 2025Large language models for artificial general intelligence (AGI): A survey of foundational principles and approaches6 Jan 2025
- 2024Adapter-based Approaches to Knowledge-enhanced Language Models -- A Survey25 Nov 2024
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoning6 Nov 2024
- 2024DEXTER: A Benchmark for open-domain Complex Question Answering using LLMs24 Jun 2024
- 2024LLM-Powered Explanations: Unraveling Recommendations Through Subgraph Reasoning22 Jun 2024
- 2024A Theory for Token-Level Harmonization in Retrieval-Augmented Generation3 Jun 2024
- 2024Knowledge Graph Tuning: Real-time Large Language Model Personalization based on Human Feedback30 May 2024
- 2024Conv-CoA: Improving Open-domain Question Answering in Large Language Models via Conversational Chain-of-Action28 May 2024
- 2024Graph Machine Learning in the Era of Large Language Models (LLMs)23 Apr 2024
38 more
- 2024CuriousLLM: Elevating Multi-Document Question Answering with LLM-Enhanced Knowledge Graph Reasoning13 Apr 2024
- 2024Chain-of-Action: Faithful and Multimodal Question Answering through Large Language Models26 Mar 2024
- 2024KnowTuning: Knowledge-aware Fine-tuning for Large Language Models17 Feb 2024
- 2024Semantic Sensitivities and Inconsistent Predictions: Measuring the Fragility of NLI Models25 Jan 2024
- 2024Knowledge-Centric Templatic Views of Documents13 Jan 2024
- 2023KGLens: Towards Efficient and Effective Knowledge Probing of Large Language Models with Knowledge Graphs15 Dec 2023
- 2023Can Knowledge Graphs Reduce Hallucinations in LLMs? : A Survey14 Nov 2023
- 2023In-Context Ability Transfer for Question Decomposition in Complex QA26 Oct 2023
- 2023Furthest Reasoning with Plan Assessment: Stable Reasoning Path with Retrieval-Augmented Large Language Models22 Sep 2023
- 2023Give Us the Facts: Enhancing Large Language Models with Knowledge Graphs for Fact-aware Language Modeling20 Jun 2023
- 2023Unifying Large Language Models and Knowledge Graphs: A Roadmap14 Jun 2023
- 2023Graph-Aware Language Model Pre-Training on a Large Graph Corpus Can Help Multiple Graph Applications5 Jun 2023
- 2023FactKB: Generalizable Factuality Evaluation using Language Models Enhanced with Factual Knowledge14 May 2023
- 2023ArK: Augmented Reality with Knowledge Interactive Emergent Ability1 May 2023
- 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Tasks28 Apr 2023
- 2023KILM: Knowledge Injection into Encoder-Decoder Language Models17 Feb 2023
- 2023Exploring Category Structure with Contextual Language Models and Lexical Semantic Networks14 Feb 2023
- 2022A Survey on Knowledge-Enhanced Pre-trained Language Models27 Dec 2022
- 2022Knowledge Graph Embedding: A Survey from the Perspective of Representation Spaces7 Nov 2022
- 2022Neural Theory-of-Mind? On the Limits of Social Intelligence in Large LMs24 Oct 2022
- 2022Deep Bidirectional Language-Knowledge Graph Pretraining17 Oct 2022
- 2022Measuring and Narrowing the Compositionality Gap in Language Models7 Oct 2022
- 2022OPAL: Ontology-Aware Pretrained Language Model for End-to-End Task-Oriented Dialogue10 Sep 2022
- 2022ComDensE : Combined Dense Embedding of Relation-aware and Common Features for Knowledge Graph Completion29 Jun 2022
- 2022A Review on Language Models as Knowledge Bases12 Apr 2022
- 2022Knowledge Mining with Scene Text for Fine-Grained Recognition27 Mar 2022
- 2022PKGM: A Pre-trained Knowledge Graph Model for E-commerce Application2 Mar 2022
- 2022Does Entity Abstraction Help Generative Transformers Reason?5 Jan 2022
- 2021Knowledge-Grounded Dialogue Generation with a Unified Knowledge Representation15 Dec 2021
- 2021Linking-Enhanced Pre-Training for Table Semantic Parsing18 Nov 2021
- 2021Knowledge Enhanced Pretrained Language Models: A Compreshensive Survey16 Oct 2021
- 2021XLM-K: Improving Cross-Lingual Language Model Pre-training with Multilingual Knowledge26 Sep 2021
- 2021Knowledge Perceived Multi-modal Pretraining in E-commerce20 Aug 2021
- 2021FlipDA: Effective and Robust Data Augmentation for Few-Shot Learning13 Aug 2021
- 2021Pre-Trained Models: Past, Present and Future14 Jun 2021
- 2021Billion-scale Pre-trained E-commerce Product Knowledge Graph Model2 May 2021
- 2021Relational World Knowledge Representation in Contextual Language Models: A Review12 Apr 2021
- 2021K-XLNet: A General Method for Combining Explicit Knowledge with Language Model Pretraining25 Mar 2021
50 newest of 52 All 52 (JSON)
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,054
- 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Taskscited by 73
- 2024LLM-Powered Explanations: Unraveling Recommendations Through Subgraph Reasoningcited by 10
- 2024CuriousLLM: Elevating Multi-Document Question Answering with LLM-Enhanced Knowledge Graph Reasoningcited by 7
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoningcited by 5
- 2022Does Entity Abstraction Help Generative Transformers Reason?cited by 3
- 2023Furthest Reasoning with Plan Assessment: Stable Reasoning Path with Retrieval-Augmented Large Language Modelscited by 2
- 2024EXPLORA: Efficient Exemplar Subset Selection for Complex Reasoning6 Nov 2024
- 2024LLM-Powered Explanations: Unraveling Recommendations Through Subgraph Reasoning22 Jun 2024
- 2024CuriousLLM: Elevating Multi-Document Question Answering with LLM-Enhanced Knowledge Graph Reasoning13 Apr 2024
- 2023Furthest Reasoning with Plan Assessment: Stable Reasoning Path with Retrieval-Augmented Large Language Models22 Sep 2023
- 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Tasks28 Apr 2023
- 2022Measuring and Narrowing the Compositionality Gap in Language Models7 Oct 2022
- 2022Does Entity Abstraction Help Generative Transformers Reason?5 Jan 2022
- 2023Unifying Large Language Models and Knowledge Graphs: A Roadmapcited by 663
- 2021Pre-Trained Models: Past, Present and Futurecited by 319
- 2022A Review on Language Models as Knowledge Basescited by 156
- 2022Deep Bidirectional Language-Knowledge Graph Pretrainingcited by 156
- 2021COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretrainingcited by 126
- 2023Can Knowledge Graphs Reduce Hallucinations in LLMs? : A Surveycited by 92
- 2023Give Us the Facts: Enhancing Large Language Models with Knowledge Graphs for Fact-aware Language Modelingcited by 56
- 2023FactKB: Generalizable Factuality Evaluation using Language Models Enhanced with Factual Knowledgecited by 55
- 2022Knowledge Graph Embedding: A Survey from the Perspective of Representation Spacescited by 41
- 2021Knowledge Enhanced Pretrained Language Models: A Compreshensive Surveycited by 33
- 2023Graph-Aware Language Model Pre-Training on a Large Graph Corpus Can Help Multiple Graph Applicationscited by 33
- 2024Graph Machine Learning in the Era of Large Language Models (LLMs)cited by 33
12 most cited of 41 All 41 (JSON)
- 2026From Found to Designed: Concepts as a Design Axis for Large Language Models29 Jul 2026
- 2025BALI: Enhancing Biomedical Language Representations through Knowledge Graph and Language Model Alignment9 Sep 2025
- 2025Sample Efficient Demonstration Selection for In-Context Learning10 Jun 2025
- 2024Adapter-based Approaches to Knowledge-enhanced Language Models -- A Survey25 Nov 2024
- 2024DEXTER: A Benchmark for open-domain Complex Question Answering using LLMs24 Jun 2024
- 2024Knowledge Graph Tuning: Real-time Large Language Model Personalization based on Human Feedback30 May 2024
- 2024Conv-CoA: Improving Open-domain Question Answering in Large Language Models via Conversational Chain-of-Action28 May 2024
- 2024Graph Machine Learning in the Era of Large Language Models (LLMs)23 Apr 2024
- 2024KnowTuning: Knowledge-aware Fine-tuning for Large Language Models17 Feb 2024
- 2024Semantic Sensitivities and Inconsistent Predictions: Measuring the Fragility of NLI Models25 Jan 2024
- 2024Knowledge-Centric Templatic Views of Documents13 Jan 2024
- 2023KGLens: Towards Efficient and Effective Knowledge Probing of Large Language Models with Knowledge Graphs15 Dec 2023
12 newest of 41 All 41 (JSON)