Knowledge graph embedding (KGE) is a simple and effective method for knowledge graph completion. The higher the embedding dimension, the higher the model performance. However, the increase in embedding dimension leads to a significant rise in storage and computation. In this paper, we propose CombDE, a knowledge distillation framework for KGE models to address this problem. First, CombDE introduces self-distillation, so that the student model could retain more performance of the teacher model in each distillation. Secondly, considering the influence of unreliable knowledge transfer on distillation performance, a dynamic adjustment mechanism of the soft label is proposed, which dynamically allocated weights through dynamic distillation temperature and truth value judgment to optimize distillation. Experiments show that our CombDE achieves optimal distillation performance compared to the baseline model. Meanwhile, on average, CombDE can reduce the training time by more than 70%.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CombDE: Direct-Distillation Combined with Self-distillation for Knowledge Graph Embeddings

  • Yusi Chen,
  • Hongtao Zhou,
  • Housheng Su

摘要

Knowledge graph embedding (KGE) is a simple and effective method for knowledge graph completion. The higher the embedding dimension, the higher the model performance. However, the increase in embedding dimension leads to a significant rise in storage and computation. In this paper, we propose CombDE, a knowledge distillation framework for KGE models to address this problem. First, CombDE introduces self-distillation, so that the student model could retain more performance of the teacher model in each distillation. Secondly, considering the influence of unreliable knowledge transfer on distillation performance, a dynamic adjustment mechanism of the soft label is proposed, which dynamically allocated weights through dynamic distillation temperature and truth value judgment to optimize distillation. Experiments show that our CombDE achieves optimal distillation performance compared to the baseline model. Meanwhile, on average, CombDE can reduce the training time by more than 70%.