Learning Adaptive Propagation for Knowledge Graph Reasoning
Due to the success of Graph Neural Networks (GNNs) in learning from
graph-structured data, various GNN-based methods have been introduced to learn
from knowledge graphs (KGs). In this paper, to reveal the key factors
underneath existing GNN-based methods, we revisit exemplar works from the lens
of the propagation path. We find that the answer entity can be close to queried
one, but the information dependency can be long. Thus, better reasoning
performance can be obtained by exploring longer propagation paths. However,
identifying such a long-range dependency in KG is hard since the number of
involved entities grows exponentially. This motivates us to learn an adaptive
propagation path that filters out irrelevant entities while preserving
promising targets during the propagation. First, we design an incremental
sampling mechanism where the close and promising target can be preserved.
Second, we design a learning-based sampling distribution to identify the
targets with fewer involved entities. In this way, GNN can go deeper to capture
long-range information. Extensive experiments show that our method is efficient
and achieves state-of-the-art performances in both transductive and inductive
reasoning settings, benefiting from the deeper propagation.