arXiv:2501.02313cs.LGcs.AI2025-01中稿 · WSDM'2025被引 32

用扩散模型解决异构图噪声和关系语义捕捉难题

DiffGraph: Heterogeneous Graph Diffusion Model

  • 引入跨视图去噪机制,将异构数据映射到目标语义空间
  • 通过隐空间扩散过程实现去噪与关系过渡同步优化
  • 在链接预测和节点分类上超越现有方法,适合复杂图数据任务

图神经网络在建模图结构数据方面取得显著进展,但传统GNN难以应对现实场景中复杂的异构结构。现有方法仍面临两大挑战:噪声数据严重降低嵌入质量与学习性能,且无法有效捕捉异构关系间的复杂语义转换,影响下游预测。为此,我们提出异构图扩散模型(DiffGraph),首创一种跨视图去噪策略,将辅助异构数据转化为目标语义空间,精准提炼任务相关特征。其核心为先进的隐式异构图扩散机制,通过创新的前向与反向扩散过程实现优异的噪声管理。该方法可同时完成异构图去噪与跨类型语义过渡,且借助隐空间扩散显著简化图生成。在公开及工业数据集上的严格实验验证表明,DiffGraph在链接预测与节点分类任务中持续优于现有方法,树立了异构图处理在鲁棒性与效率方面的新基准。代码已开源:https://github.com/HKUDS/DiffGraph。

原文摘要 · Abstract (English)

Recent advances in Graph Neural Networks (GNNs) have revolutionized graph-structured data modeling, yet traditional GNNs struggle with complex heterogeneous structures prevalent in real-world scenarios. Despite progress in handling heterogeneous interactions, two fundamental challenges persist: noisy data significantly compromising embedding quality and learning performance, and existing methods' inability to capture intricate semantic transitions among heterogeneous relations, which impacts downstream predictions. To address these fundamental issues, we present the Heterogeneous Graph Diffusion Model (DiffGraph), a pioneering framework that introduces an innovative cross-view denoising strategy. This advanced approach transforms auxiliary heterogeneous data into target semantic spaces, enabling precise distillation of task-relevant information. At its core, DiffGraph features a sophisticated latent heterogeneous graph diffusion mechanism, implementing a novel forward and backward diffusion process for superior noise management. This methodology achieves simultaneous heterogeneous graph denoising and cross-type transition, while significantly simplifying graph generation through its latent-space diffusion capabilities. Through rigorous experimental validation on both public and industrial datasets, we demonstrate that DiffGraph consistently surpasses existing methods in link prediction and node classification tasks, establishing new benchmarks for robustness and efficiency in heterogeneous graph processing. The model implementation is publicly available at: https://github.com/HKUDS/DiffGraph.

异构图扩散模型去噪

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。