MetaIE: Distilling a Meta Model from LLM for All Kinds of Information Extraction Tasks

Letian Peng,Zilong Wang,Feng Yao,Zihan Wang,Jingbo Shang
2024-03-31
Abstract:Information extraction (IE) is a fundamental area in natural language processing where prompting large language models (LLMs), even with in-context examples, cannot defeat small LMs tuned on very small IE datasets. We observe that IE tasks, such as named entity recognition and relation extraction, all focus on extracting important information, which can be formalized as a label-to-span matching. In this paper, we propose a novel framework MetaIE to build a small LM as meta-model by learning to extract "important information", i.e., the meta-understanding of IE, so that this meta-model can be adapted to all kind of IE tasks effectively and efficiently. Specifically, MetaIE obtains the small LM via a symbolic distillation from an LLM following the label-to-span scheme. We construct the distillation dataset via sampling sentences from language model pre-training datasets (e.g., OpenWebText in our implementation) and prompting an LLM to identify the typed spans of "important information". We evaluate the meta-model under the few-shot adaptation setting. Extensive results on 13 datasets from 6 IE tasks confirm that MetaIE can offer a better starting point for few-shot tuning on IE datasets and outperform other meta-models from (1) vanilla language model pre-training, (2) multi-IE-task pre-training with human annotations, and (3) single-IE-task symbolic distillation from LLM. Moreover, we provide comprehensive analyses of MetaIE, such as the size of the distillation dataset, the meta-model architecture, and the size of the meta-model.
Computation and Language
What problem does this paper attempt to address?
This paper mainly discusses the limitations of large-scale language models (LLMs) in information extraction (IE) tasks, even with the use of instance prompts, they cannot match the performance of small LMs fine-tuned on small-scale IE datasets. The authors observed that IE tasks such as named entity recognition (NER) and relation extraction (RE) focus on extracting important information, which can be formalized as a problem of matching labels to spans. Therefore, they proposed a new framework called MetaIE, which learns to extract "important information" from LLM through symbol distillation to construct a small LM as a meta-model, enabling this meta-model to adapt effectively to various IE tasks. MetaIE obtains a small LM through symbol distillation by following the scheme of matching labels to spans from LLM. Specifically, a distillation dataset is constructed by sampling sentences from the language model pretraining dataset and prompting LLM to recognize typed spans of "important information". The experimental evaluation assesses the performance of the meta-model in a few-shot adaptation setting and compares it with other methods, including original language model pretraining, pretraining with multiple IE tasks annotated by humans, and symbol distillation of LLM for a single IE task. The results show that MetaIE outperforms other methods in multiple IE tasks, especially in tasks that multi-task pretraining methods cannot handle. In addition, the paper provides a comprehensive analysis of MetaIE, studying the impact of factors such as distillation dataset size, meta-model architecture, and meta-model size. The success of MetaIE is not only attributed to LLMs but also to the scheme of matching labels to spans proposed by the authors. In conclusion, this paper addresses the problem of how to leverage the knowledge of LLMs to create a meta-model that can efficiently adapt to various information extraction tasks. Through symbol distillation, their proposed method, MetaIE, exhibits better performance than other methods with limited samples.