arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.04597cs.AIcs.CLcs.LG

Probabilistic Box Embeddings for Uncertain Knowledge Graph Reasoning

UMass Amherst(马萨诸塞大学阿默斯特分校);USC(南加州大学);Information Sciences Institute, USC(南加州大学信息科学研究所);UCLA(加州大学洛杉矶分校)

Xuelu Chen, Michael Boratko, Muhao Chen, Shib Sankar Dasgupta, Xiang Lorraine Li, Andrew McCallum

英文摘要:

Knowledge bases often consist of facts which are harvested from a variety of sources, many of which are noisy and some of which conflict, resulting in a level of uncertainty for each triple. Knowledge bases are also often incomplete, prompting the use of embedding methods to generalize from known facts, however, existing embedding methods only model triple-level uncertainty, and reasoning results lack global consistency. To address these shortcomings, we propose BEUrRE, a novel uncertain knowledge graph embedding method with calibrated probabilistic semantics. BEUrRE models each entity as a box (i.e. axis-aligned hyperrectangle) and relations between two entities as affine transforms on the head and tail entity boxes. The geometry of the boxes allows for efficient calculation of intersections and volumes, endowing the model with calibrated probabilistic semantics and facilitating the incorporation of relational constraints. Extensive experiments on two benchmark datasets show that BEUrRE consistently outperforms baselines on confidence prediction and fact ranking due to its probabilistic calibration and ability to capture high-order dependencies among facts.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑