arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.07186cs.LGcs.AI
收藏

Relation-aware Graph Attention Model With Adaptive Self-adversarial Training

IBM Almaden Research Center(IBM阿尔马登研究中心);IBM Thomas J. Watson Research Center(IBM托马斯·J·沃森研究中心)

Xiao Qin, Nasrullah Sheikh, Berthold Reinwald, Lingfei Wu

英文摘要:

This paper describes an end-to-end solution for the relationship prediction task in heterogeneous, multi-relational graphs. We particularly address two building blocks in the pipeline, namely heterogeneous graph representation learning and negative sampling. Existing message passing-based graph neural networks use edges either for graph traversal and/or selection of message encoding functions. Ignoring the edge semantics could have severe repercussions on the quality of embeddings, especially when dealing with two nodes having multiple relations. Furthermore, the expressivity of the learned representation depends on the quality of negative samples used during training. Although existing hard negative sampling techniques can identify challenging negative relationships for optimization, new techniques are required to control false negatives during training as false negatives could corrupt the learning process. To address these issues, first, we propose RelGNN -- a message passing-based heterogeneous graph attention model. In particular, RelGNN generates the states of different relations and leverages them along with the node states to weigh the messages. RelGNN also adopts a self-attention mechanism to balance the importance of attribute features and topological features for generating the final entity embeddings. Second, we introduce a parameter-free negative sampling technique -- adaptive self-adversarial (ASA) negative sampling. ASA reduces the false-negative rate by leveraging positive relationships to effectively guide the identification of true negative samples. Our experimental evaluation demonstrates that RelGNN optimized by ASA for relationship prediction improves state-of-the-art performance across established benchmarks as well as on a real industrial dataset.

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

↑