Search papers, labs, and topics across Lattice.
This paper introduces DiffTSP, a discrete diffusion model that reformulates the Triple Set Prediction (TSP) task in Knowledge Graph Completion (KGC) as a generative process, allowing for the inference of missing triples without prior knowledge of any elements. By progressively adding noise to the knowledge graph and employing a structure-aware denoising network, DiffTSP captures the dependencies among predicted triples, ensuring consistency in the generated outputs. The model achieves state-of-the-art performance across three public datasets, demonstrating its effectiveness in addressing the limitations of traditional TSP methods that predict triples individually.
Generating complete sets of triples in one pass, DiffTSP outperforms existing methods by effectively capturing interdependencies among predictions.
Knowledge Graphs (KGs) are composed of triples, and the goal of Knowledge Graph Completion (KGC) is to infer the missing factual triples. Traditional KGC tasks predict missing elements in a triple given one or two of its elements. As a more realistic task, the Triple Set Prediction (TSP) task aims to infer the set of missing triples conditioned only on the observed knowledge graph, without assuming any partial information about the missing triples. Existing TSP methods predict the set of missing triples in a triple-by-triple manner, falling short in capturing the dependencies among the predicted triples to ensure consistency. To address this issue, we propose a novel discrete diffusion model termed DiffTSP that treats TSP as a generative task. DiffTSP progressively adds noise to the KG through a discrete diffusion process, achieved by masking relational edges. The reverse process then gradually recovers the complete KG conditioned on the incomplete graph. To this end, we design a structure-aware denoising network that integrates a relational context encoder with a relational graph diffusion transformer for knowledge graph generation. DiffTSP can generate the complete set of triples in a one-pass manner while ensuring the dependencies among the predicted triples. Our approach achieves state-of-the-art performance on three public datasets. Code: https://github.com/ADMIS-TONGJI/DiffTSP.
