arXiv:2606.11831cs.LGcs.AI2026-06KDD被引 1

用扩散模型学习图先验,让神经关系推断更准更可靠

From Uniform to Learned Graph Priors: Diffusion for Structure Discovery

论文配图:From Uniform to Learned Graph Priors: Diffusion for Structure Discovery
图 1 · 摘自论文原文
  • 用扩散模型构建可学习的图先验,动态校准边缘后验分布
  • 在多个标准数据集上提升结构推断性能,边缘后验更明确
  • 适合做复杂系统结构发现的研究者,尤其关注图推理可靠性

神经关系推断(NRI)方法通过变分推理从轨迹中发现交互图,但通常依赖过于简化的独立边先验,近似均匀分布,导致边缘后验分散且不确定,难以准确揭示真实结构。为此,我们提出Diff-prior:一种基于扩散模型的自适应先验,用于校准潜在图分布而非生成图。核心思想是将先验融合重构为可学习的去噪式校准过程,将分散、模糊的边缘后验整理为更可靠的全局结构。Diff-prior 在结构采样前作用于编码器输出的边分布,作为直接的去噪校准器,提供适用于结构变量的通用训练范式。在标准基准上的实验验证了该框架的有效性,结果表明,Diff-prior 提升了多类 NRI 架构的结构推断性能,并生成更决断的边缘后验。代码已开源:https://github.com/Hardy158118/Diffprior。

原文摘要 · Abstract (English)

Neural relational inference (NRI) methods discover interaction graphs from trajectories through variational reasoning on discrete potential edges. However, these methods typically rely on oversimplified, factorized graph priors. Such priors, typically nearing uniform distributions, treat edges as independent entities. This systemic misalignment does not match the real-world systems and yields diffuse and indecisive edge posteriors limiting the reliability of structural discovery. To address this, we propose \textit{Diff-prior}, a diffusion-parameterized adaptive prior used to calibrate latent graph distribution rather than generate graphs. Our core insight is to reframe prior integration as a learnable denoising-style calibration that organizes scattered, uncertain edge posteriors into a more reliable overall structure which can be trained by the diffusion model. Diff-prior learns an adaptive structure prior that performs structured calibration on the edge posteriors during inference, guiding it towards a distribution closer to the underlying structure. The diff-prior operates before structural sampling and acts as a denoising calibrator directly on the encoder edge distribution, which provides a generic training paradigm over structured variables. Experiments on standard benchmarks validated our framework, and the results indicate that Diff-prior improves the performance of structure inference and generates more decisive edge posteriors across multiple NRI-family architectures. The code is available on https://github.com/Hardy158118/Diffprior.

图结构发现扩散模型先验学习

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