新实体嵌入初始化策略提升知识图谱持续学习效果
Improving Continual Learning of Knowledge Graph Embeddings via Informed Initialization
- 利用知识图谱模式与旧嵌入,按实体类别生成初始嵌入
- 显著减少灾难性遗忘,预测性能提升,训练时间缩短
- 兼容多种模型,适合频繁更新的小规模知识图谱
许多知识图谱(KG)频繁更新,要求其知识图谱嵌入(KGE)能适应变化。为解决此问题,现有持续学习方法在更新旧嵌入的同时引入新实体的嵌入。其中,嵌入初始化是关键步骤,直接影响最终嵌入精度和训练时长,尤其对小规模、频繁更新场景更为重要。本文提出一种新型知情初始化策略,可无缝集成到现有KGE持续学习方法中,通过利用KG模式及先前学习的嵌入,基于实体所属类别生成新实体的初始表示,从而在提升新知识获取能力的同时减少灾难性遗忘。大量实验表明,该策略不仅改善了结果嵌入的预测性能,还增强了知识保留能力,加速了知识获取过程,显著减少了增量学习所需训练轮次与时间。此外,该方法在多种KGE学习模型上均表现出良好效果。
原文摘要 · Abstract (English)
Many Knowledege Graphs (KGs) are frequently updated, forcing their Knowledge Graph Embeddings (KGEs) to adapt to these changes. To address this problem, continual learning techniques for KGEs incorporate embeddings for new entities while updating the old ones. One necessary step in these methods is the initialization of the embeddings, as an input to the KGE learning process, which can have an important impact in the accuracy of the final embeddings, as well as in the time required to train them. This is especially relevant for relatively small and frequent updates. We propose a novel informed embedding initialization strategy, which can be seamlessly integrated into existing continual learning methods for KGE, that enhances the acquisition of new knowledge while reducing catastrophic forgetting. Specifically, the KG schema and the previously learned embeddings are utilized to obtain initial representations for the new entities, based on the classes the entities belong to. Our extensive experimental analysis shows that the proposed initialization strategy improves the predictive performance of the resulting KGEs, while also enhancing knowledge retention. Furthermore, our approach accelerates knowledge acquisition, reducing the number of epochs, and therefore time, required to incrementally learn new embeddings. Finally, its benefits across various types of KGE learning models are demonstrated.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。