一种高效知识蒸馏的知识图谱推荐方法、设备及介质

    公开(公告)号:CN119961519A

    公开(公告)日:2025-05-09

    申请号:CN202510061002.X

    申请日:2025-01-15

    Inventor: 许勇 李芷珊

    Abstract: 本发明公开了一种高效知识蒸馏的知识图谱推荐方法、设备及介质,其中方法包括:通过四项蒸馏任务,将知识从大型的基于知识图谱的教师模型转移到小型的单层图注意力网络学生模型中,其中四项蒸馏任务包括:1)用户嵌入和物品嵌入的蒸馏;2)用户嵌入和物品嵌入的角度蒸馏;3)用户和物品预测结果的蒸馏;4)用户和物品预测结果的蒸馏;获取输入数据,输入经过四项蒸馏任务获得的学生模型中,输出推荐结果。本发明除了在嵌入层面和预测层面进行知识蒸馏,还在角度属性和知识图谱注意力信息层面实现知识蒸馏,不仅能提高模型训练的收敛速度,还能更有效地提取知识图谱中的信息,缓解KG中的数据噪声问题和冷启动问题。

Patent Agency Ranking