Triple set prediction (TSP) is to recover missing relational triples in a knowledge graph by predicting the triple set instead of isolated links.
Existing approaches predominantly rely on the subgraph-based completion paradigm, where the original graph is first partitioned into local subgraphs which are then independently completed. Such a paradigm suffers from:
- fragmented structural reasoning,
- limited cross subgraph dependency modeling, and
- missing predictions across subgraphs.
提出的方法:LatentDiff
This paper proposes LatentDiff, a novel unified latent generative framework for TSP, which reformulates knowledge graph completion as latent space graph generation.
LatentDiff first learns a compact latent representation using a relation-aware residual-quantized knowledge graph autoencoder, which:
- encodes the whole graph into relation-aware latent embeddings,
- quantizes them with chunked residual codebooks, and
- reconstructs triples through a factorized decoder.
观测锚定的潜在图扩散
On top of this latent space, we introduce an observation-anchored latent graph diffusion model, where the target latent graph representation is perturbed toward the observed latent. The reverse process recovers missing triples through progressive latent denoising.
实验结果
Experiments on three benchmark datasets show that LatentDiff outperforms the SOTA in terms of Ftsp by 5.4 percentage points on average.