fmLRE: A Low-Resource Relation Extraction Model Based on Feature Mapping Similarity Calculation

Authors

  • Peng Wang Southeast University
  • Tong Shao Southeast University
  • Ke Ji Southeast University
  • Guozheng Li Southeast University
  • Wenjun Ke Southeast University Beijing Institute of Computer Technology and Application

DOI:

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

Keywords:

SNLP: Information Extraction, DMKM: Linked Open Data, Knowledge Graphs & KB Completion

Abstract

Low-resource relation extraction (LRE) aims to extract relations from limited labeled corpora. Existing work takes advantages of self-training or distant supervision to expand the limited labeled data in the data-driven approaches, while the selection bias of pseudo labels may cause the error accumulation in subsequent relation classification. To address this issue, this paper proposes fmLRE, an iterative feedback method based on feature mapping similarity calculation to improve the accuracy of pseudo labels. First, it calculates the similarities between pseudo-label and real-label data of the same category in a feature mapping space based on semantic features of labeled dataset after feature projection. Then, it fine-tunes initial model according to the iterative process of reinforcement learning. Finally, the similarity is used as a threshold for screening high-precision pseudo-labels and the basis for setting different rewards, which also acts as a penalty term for the loss function of relation classifier. Experimental results demonstrate that fmLRE achieves the state-of-the-art performance compared with strong baselines on two public datasets.

Downloads

Published

2023-06-26

How to Cite

Wang, P., Shao, T., Ji, K., Li, G., & Ke, W. (2023). fmLRE: A Low-Resource Relation Extraction Model Based on Feature Mapping Similarity Calculation. Proceedings of the AAAI Conference on Artificial Intelligence, 37(11), 13700-13708. https://doi.org/10.1609/aaai.v37i11.26605

Issue

Section

AAAI Technical Track on Speech & Natural Language Processing