KICE: A Knowledge Consolidation and Expansion Framework for Relation Extraction
DOI:
https://doi.org/10.1609/aaai.v37i11.26565Keywords:
SNLP: Information Extraction, DMKM: Rule Mining & Pattern Mining, HAI: Human-in-the-Loop Machine Learning, ML: Classification and RegressionAbstract
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