模型
DistMult(双线性嵌入模型)
DistMult是一种双线性知识图谱嵌入模型,通过将关系表示为对角矩阵,解决对称关系的建模问题。它计算高效,在链接预测任务中表现良好,但无法处理非对称关系,典型应用包括知识图谱补全和推荐系统。
详细解释
DistMult由Yang等人于2015年提出,其评分函数为f(h,r,t)=<h,r,t>,其中h,t为实体向量,r为关系向量,通过三向点积计算得分。该模型将关系矩阵约束为对角矩阵,大幅减少参数,但导致关系对称性假设,即无法区分头尾实体。它在WN18等对称关系多的数据集上表现优异,但在FB15k等含非对称关系的数据集上受限。后续工作如ComplEx通过引入复数扩展解决了该问题。
别名与相关写法
distmult