Search papers, labs, and topics across Lattice.
This paper introduces Diff-prior, a novel adaptive prior that utilizes diffusion processes to enhance the reliability of edge posteriors in neural relational inference (NRI) methods. By reframing prior integration as a learnable denoising calibration, Diff-prior organizes uncertain edge posteriors into a more coherent structure, significantly improving the performance of structure inference across various NRI architectures. Experimental results demonstrate that Diff-prior yields more decisive edge posteriors, addressing the limitations of traditional uniform priors that treat edges as independent entities.
Transforming edge posteriors from diffuse uncertainty into structured clarity, Diff-prior revolutionizes how we infer interaction graphs in complex systems.
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.
