Abstract:Pretrained language models (PLMs) have demonstrated strong performance on many natural language processing (NLP) tasks. Despite their great success, these PLMs are typically pretrained only on unstructured free texts without leveraging existing structured knowledge bases that are readily available for many domains, especially scientific domains. As a result, these PLMs may not achieve satisfactory performance on knowledge-intensive tasks such as biomedical NLP. Comprehending a complex biomedical document without domain-specific knowledge is challenging, even for humans. Inspired by this observation, we propose a general framework for incorporating various types of domain knowledge from multiple sources into biomedical PLMs. We encode domain knowledge using lightweight adapter modules, bottleneck feed-forward networks that are inserted into different locations of a backbone PLM. For each knowledge source of interest, we pretrain an adapter module to capture the knowledge in a self-supervised way. We design a wide range of self-supervised objectives to accommodate diverse types of knowledge, ranging from entity relations to description sentences. Once a set of pretrained adapters is available, we employ fusion layers to combine the knowledge encoded within these adapters for downstream tasks. Each fusion layer is a parameterized mixer of the available trained adapters that can identify and activate the most useful adapters for a given input. Our method diverges from prior work by including a knowledge consolidation phase, during which we teach the fusion layers to effectively combine knowledge from both the original PLM and newly-acquired external knowledge using a large collection of unannotated texts. After the consolidation phase, the complete knowledge-enhanced model can be fine-tuned for any downstream task of interest to achieve optimal performance. Extensive experiments on many biomedical NLP datasets show that our proposed framework consistently improves the performance of the underlying PLMs on various downstream tasks such as natural language inference, question answering, and entity linking. These results demonstrate the benefits of using multiple sources of external knowledge to enhance PLMs and the effectiveness of the framework for incorporating knowledge into PLMs. While primarily focused on the biomedical domain in this work, our framework is highly adaptable and can be easily applied to other domains, such as the bioenergy sector.

Biomedical Knowledge Representation Learning

A Knowledge Representation Method Based on Two-Layer Modeling for Constructing Medical Knowledge Base

Improving Biomedical Pretrained Language Models with Knowledge

KEBLM: Knowledge-Enhanced Biomedical Language Models

A Survey on Approaches and Applications of Knowledge Representation Learning

Continual knowledge infusion into pre-trained biomedical language models

Application and investigation of knowledge graph in biomedical field

Knowledge Representation Learning: A Quantitative Review

Knowledge-based Biomedical Data Science 2019

Neurosymbolic AI for Reasoning on Biomedical Knowledge Graphs

A Biomedical Open Knowledge Network Harnesses the Power of AI to Understand Deep Human Biology

Representation Learning and NLP

Benchmark and Best Practices for Biomedical Knowledge Graph Embeddings

AI for Biomedicine in the Era of Large Language Models

Relation-Aware Deep Neural Network Enables More Efficient Biomedical Knowledge Acquisition from Massive Literature

A Representation Model For Biological Entities By Fusing Structured Axioms With Unstructured Texts

Knowledge Enhanced Lstm For Coreference Resolution On Biomedical Texts

Deep Learning, Natural Language Processing, and Explainable Artificial Intelligence in the Biomedical Domain

Knowledge Graph Embeddings in the Biomedical Domain: Are They Useful? A Look at Link Prediction, Rule Learning, and Downstream Polypharmacy Tasks

Extracting Biomedical Factual Knowledge Using Pretrained Language Model and Electronic Health Record Context

Representation-Enhanced Neural Knowledge Integration with Application to Large-Scale Medical Ontology Learning