Journal of Jilin University Science Edition ›› 2023, Vol. 61 ›› Issue (2): 317-324.

Previous Articles     Next Articles

Lightweight Relation Extraction Based on Positive Soft Labels

SONG Hanyu1,2, OUYANG Dantong1,3,  YE Yuxin1,3   

  1. 1. College of Computer Science and Technology, Jilin University, Changchun 130012, China;
    2. FAW-Volkswagen Automotive Co.Ltd, Changchun 130011, China;
    3. Key Laboratory of Symbolic Computation and Knowledge Engineering of Ministry of Education, Jilin University, Changchun 130012, China
  • Received:2022-01-04 Online:2023-03-26 Published:2023-03-26

Abstract: Aiming at  the problem that the scale of relation extraction model was getting larger and larger, and the time consumption was getting longer and longer, we proposed a knowledge filtering mechanism to construct a lightweight relation extraction model by using the positive soft labels selected. Firstly, knowledge distillation was used to extract knowledge and store knowledge in soft labels. In order to avoid the problem of difficult  absorption of knowledge caused by the large gap between  teachers and  students in knowledge distillation, we used teacher assistant knowledge distillation pattern. Secondly,  the cosine similarity of labels was used to filter the positive soft labels and the positive soft labels were dynamically given  higher weight in each step of the distillation, so as to  weaken the influence caused by  the wrong labels in the knowledge transfer. The experimental results on SemEval-2010 Task 8 dataset show that the proposed  mode can not only complete the task of lightweight relation extraction, but also improve the extraction accuracy.

Key words: lightweight relation extraction, knowledge filtering, positive soft label, knowledge distillation, cosine similarity

CLC Number: 

  • TP391.1