Existence-Field Diffusion Model for Spatial Point Processes with Variable Cardinality
TL;DR
The existence-field diffusion model (EFDM), which associates each potential point with a continuous variable representing its degree of presence, is proposed, which uses Gaussian diffusion to jointly model point locations and cardinality through a simple continuous representation.
Abstract
We study generative modeling of spatial point processes (SPP), where both the number of points and their spatial configuration are governed by a joint distribution. While diffusion models have achieved strong performance in modeling complex distributions, extending them to variable-cardinality SPP remains challenging. Existing approaches either sample cardinality before generating locations conditionally, or introduce specialized discrete operations to change the number of points during generation. We propose the existence-field diffusion model (EFDM), which associates each potential point with a continuous variable representing its degree of presence. EFDM uses Gaussian diffusion to jointly model point locations and cardinality through a simple continuous representation. Existence probabilities can increase or decrease in both forward and reverse diffusion, without explicit point-addition or point-deletion steps. The same construction naturally accommodates categorical attributes, allowing locations, existence, and attributes to be modeled together. Experiments demonstrate improved molecular stability and validity over the compared baselines on the molecular dataset, alongside competitive performance on real-world spatial and synthetic datasets.