Latent Graph Diffusion: A Unified Framework for Generation and Prediction on Graphs
CoRR(2024)
摘要
In this paper, we propose the first framework that enables solving graph
learning tasks of all levels (node, edge and graph) and all types (generation,
regression and classification) with one model. We first propose Latent Graph
Diffusion (LGD), a generative model that can generate node, edge, and
graph-level features of all categories simultaneously. We achieve this goal by
embedding the graph structures and features into a latent space leveraging a
powerful encoder which can also be decoded, then training a diffusion model in
the latent space. LGD is also capable of conditional generation through a
specifically designed cross-attention mechanism. Then we formulate prediction
tasks including regression and classification as (conditional) generation,
which enables our LGD to solve tasks of all levels and all types with provable
guarantees. We verify the effectiveness of our framework with extensive
experiments, where our models achieve state-of-the-art or highly competitive
results across generation and regression tasks.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要