KICE: A Knowledge Consolidation and Expansion Framework for Relation Extraction

Authors

  • Yilin Lu Zhejiang University
  • Xiaoqiang Wang Zhejiang University
  • Haofeng Yang Zhejiang University
  • Siliang Tang Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v37i11.26565

Keywords:

SNLP: Information Extraction, DMKM: Rule Mining & Pattern Mining, HAI: Human-in-the-Loop Machine Learning, ML: Classification and Regression

Abstract

Machine Learning is often challenged by insufficient labeled data. Previous methods employing implicit commonsense knowledge of pre-trained language models (PLMs) or pattern-based symbolic knowledge have achieved great success in mitigating manual annotation efforts. In this paper, we focus on the collaboration among different knowledge sources and present KICE, a Knowledge-evolving framework by Iterative Consolidation and Expansion with the guidance of PLMs and rule-based patterns. Specifically, starting with limited labeled data as seeds, KICE first builds a Rule Generator by prompt-tuning to stimulate the rich knowledge distributed in PLMs, generate seed rules, and initialize the rules set. Afterwards, based on the rule-labeled data, the task model is trained in a self-training pipeline where the knowledge in rules set is consolidated with self-learned high-confidence rules. Finally, for the low-confidence rules, KICE solicits human-enlightened understanding and expands the knowledge coverage for better task model training. Our framework is verified on relation extraction (RE) task, and the experiments on TACRED show that the model performance (F1) grows from 33.24% to 79.84% with the enrichment of knowledge, outperforming all the baselines including other knowledgeable methods.

Downloads

Published

2023-06-26

How to Cite

Lu, Y., Wang, X., Yang, H., & Tang, S. (2023). KICE: A Knowledge Consolidation and Expansion Framework for Relation Extraction. Proceedings of the AAAI Conference on Artificial Intelligence, 37(11), 13336-13343. https://doi.org/10.1609/aaai.v37i11.26565

Issue

Section

AAAI Technical Track on Speech & Natural Language Processing