A UTOREGRESSIVE D IFFUSION M ODEL FOR G RAPH G ENERATION
Anonymous authors
Paper under double-blind review
A BSTRACT
Diffusion-based graph generative models have recently obtained promising results for graph generation.
However, existing diffusion-based graph generative models are all one-shot generative models that apply Gaussian diffusion in the dequantized adjacency matrix space.
Such a strategy can suffer from difficulty in model training, slow sampling speed, and incapability of incorporating constraints.
We propose an autoregressive diffusion model for graph generation.
Unlike existing methods, we define a node-absorbing diffusion process that operates directly in the discrete graph space.
For forward diffusion, we design a diffusion ordering network , which learns an optimal node absorbing ordering from graph topology.
For reverse generation, we design a denoising network that uses the reverse node ordering to efficiently reconstruct the graph by predicting one row of the adjacency matrix at a time.
Based on permutation invariance of graph generation, we show that the two networks can be jointly trained by optimizing a simple lower bound of data likelihood.
Our experiments on six diverse datasets show that our model achieves better or comparable generation performance with previous stateof-the-art, and meanwhile enjoys fast generation speed.
I NTRODUCTION
Generating graphs from a target distribution is a fundamental problem in many domains such as drug discovery (Li et al., 2018), material design (Maziarka et al., 2020), social network analysis (Grover et al., 2019), and public health (Yu et al., 2020).
Deep generative models have recently led to promising advances in this problem.
Different from traditional random graph models (Erdos et al., 1960; Albert & Barab ´ asi, 2002), these methods fit graph data with powerful deep generative models including variational auto-encoders (Simonovsky & Komodakis, 2018), generative adversarial networks (Maziarka et al., 2020), normalizing flows (Madhawa et al., 2019), and energy-based models (Liu et al., 2021).
These models are learned to capture complex graph structural patterns and then generate new high-fidelity graphs with desired properties.
Recently, the emergence of probabilistic diffusion models has led to interest in diffusion-based graph generation (Jo et al., 2022).
Diffusion models decompose the full complex transformation between noise and real data into many small steps of simple diffusion.
Compared with prior deep generative models, diffusion models enjoy both flexibility in modeling architecture and tractability of the model’s probability distributions.
To the best of our knowledge, there are two existing works on diffusion-based graph generation: Niu et al.
(2020) model the adjacency matrices using score matching at different noise scales, and uses annealed Langevin dynamics to sample new adjacency matrices for generation; Jo et al.
(2022) propose a continuous-time graph diffusion model that jointly models adjacency matrices and node features through stochastic differential equations (SDEs).
However, existing diffusion-based graph generative models suffer from three key drawbacks: (1)
Generation Efficiency .
The sampling processes of Niu et al.
Jo et al.
(2022) are slow, as Niu et al.
(2020) requires a large number of diffusion noising levels and Jo et al.
(2022) needs to solve a complex system SDEs.
(2) Continuous Approximation.
They convert discrete graphs to continuous state spaces by adding real-valued noise to graph adjacency matrices.
Such dequantization can distort the distribution of the original discrete graph structures, thus increasing the difficulty of model training.
(3) Incorporating constraints .
They are both one-shot generation models and hence cannot easily incorporate constraints during the one-shot generation process.
We propose an autoregressive graph generative model named G RAPH A RM via autoregressive diffusion on graphs.
Autoregressive diffusion (ARM) (Hoogeboom et al., 2022a) is an absorbing diffusion process (Austin et al., 2021) for discrete data, where exactly one dimension of the data decays to the absorbing state at each diffusion step.
In G RAPH A RM , we design node-absorbing autoregressive diffusion for graphs, which diffuses a graph directly in the discrete graph space instead of in the dequantized adjacency matrix space.
The forward pass absorbs one node in each step by masking it along with its connecting edges, which is repeated until all the nodes are absorbed and the graph becomes empty.
We further design a diffusion ordering network in G RAPH A RM , which is jointly trained with the reverse generator to learn an optimal node ordering for diffusion.
Compared with random ordering as in prior ARM models (Hoogeboom et al., 2022a), the learned diffusion ordering not only provides a better stochastic approximation of the true marginal graph likelihood, but also eases generative model training by leveraging structural regularities.
The backward pass in G RAPH A RM recovers the graph structure by learning to reverse the nodeabsorbing diffusion process with a denoising network.
The reverse generative process is autoregressive, which makes G RAPH A RM easy to handle constraints during generation.
However, a key challenge is to learn the distribution of reverse node ordering for optimizing the data likelihood.
We show that this difficulty can be circumvented by just using the exact reverse node ordering and optimizing a simple lower bound of likelihood, based on the permutation invariance property of graph generation.
The likelihood lower bound allows for jointly training the denoising network and the diffusion ordering network using a reinforcement learning procedure and gradient descent.
The generation speed of G RAPH A RM is much faster than the existing graph diffusion models (Jo et al., 2022; Niu et al., 2020).
Due to the autoregressive diffusion process in the node space, the number of diffusion steps in G RAPH A RM is the same as the number of nodes, which is typically much smaller than the sampling steps in (Jo et al., 2022; Niu et al., 2020).
Furthermore, at each step of the backward pass, we design the denoising network to predict one row of the adjacency matrix at one time.
The edges to be predicted follow a mixture of Bernoulli distribution to ensure dependencies among each other.
This makes G RAPH A RM much more efficient than most previous autoregressive graph generation models.
Our key contributions are as follows: (1) To the best of our knowledge, our work is the first autoregressive diffusion-based graph generation model, underpinned by a new node self-absorbing diffusion process.
(2) G RAPH A RM learns an optimal node generation ordering and thus better leverages the structural regularities for autoregressive graph diffusion.
(3) We validate our method on both synthetic and real-world graph generation tasks, on which we show that G RAPH A RM outperforms existing graph generative models and is efficient in generation speed.
B ACKGROUND
Diffusion Model and Absorbing Diffusion Given a training instance x 0 ∈ R D sampled from the underlying distribution p data ( x 0 ) , a diffusion model defines a forward Markov transition kernel q ( x t | x t − 1 ) to gradually corrupt training data until the data distribution is transformed into a simple noisy distribution.
The model then learns to reverse this process by learning a denoising transition kernel parameterized by a neural network p θ ( x t − 1 | x t ) .
Most existing works on diffusion models use Gaussian diffusion for continuous-state data.
To apply
Gaussian diffusion on discrete data, one can use dequantization method by adding small noise to the data.
However, dequantization distorts the original discrete distribution, which can cause difficulty in training diffusion-based models.
For example, dequantization on graph adjacency matrices can destroy graph connectivity information and hurt message passing.
Austin et al.
; Hoogeboom et al.
(2021) introduce several discrete state space diffusion models using different Markov transition matrices.
Among them, the absorbing diffusion is the most promising one due to its simplicity and strong empirical performance.
Definition 1 (Absorbing Discrete Diffusion) .
An absorbing diffusion is a Markov destruction process defined in the discrete state space.
At transition time step t , each element x ( i ) t in dimension i is independently decayed into an absorbing state with probabilities α ( t ) .
The absorbing state can be a [MASK] token for texts or gray pixel for images (Austin et al., 2021).
The diffusion process will converge to a stationary distribution that has all the mass on the absorbing state.
The reverse of the absorbing diffusion is learned with categorical distribution to generate the original data.
Typically, the decaying probabilities α ( t ) need to be small and the diffusion steps T need to be large to attain good performance.
Autoregressive Diffusion Model
Although absorbing diffusion directly operates on the discrete data, it still needs a large number of diffusion steps and thus its generation process can be slow.
Autoregressive diffusion model (ARM) (Hoogeboom et al., 2022a) describes a fixed absorbing diffusion process where the generative steps is equal to the dimensionality of the data, e .
g ., the number of pixels in an image or the number of tokens in a piece of text.
Definition 2 (Autoregressive Diffusion) .
An autoregressive diffusion is a stochastic absorbing process where exactly one dimension decays to the absorbing state at a time.
The absorbing diffusion is repeated until all the D dimensions are absorbed.
An equivalent way to describe this process is to first sample a permutation σ ∈ S D , where S D represents the set of all permutations of the dimension indices 1 , · · · , D .
Then each dimension of the data decays in that order towards the absorbing state.
The corresponding generative process then models the variables in the exact opposite order of the permutation.
ARM amounts to absorbing diffusion with continuous time limit, as detailed in Appendix A.2.
While ARM offers an efficient and general diffusion framework for discrete data, two key questions remains to be addressed for applying ARM for graphs: (1) How do we define absorbing states for inter-dependent nodes and edges in graphs without losing the efficiency of ARM?
(2) While ARM imposes a uniform ordering for arriving at an order-agnostic variational lower bound (VLB) of likelihood, a random ordering fails to capture graph topology.
How do we obtain a data-dependent ordering that leverages graph structural regularities during generation?
In the next section, we address these two challenges in our proposed G RAPH A RM model.
M ETHOD
A graph is represented by the tuple G =
( V, E ) with node set V = { v 1 , · · · , v n } and edge set
E = { ( v i , v j ) | v i , v j ∈ V ) } .
We denote by n = | V | and m = | E | the number of nodes and edges in G respectively.
Our goal is to learn a graph generative model from a set of training graphs.
We assume the graphs are unattributed and focus on graph topology generation.
3. A UTOREGRESSIVE GRAPH DIFFUSION P ROCESS
Due to the dependency between nodes and edges, it is nontrivial to apply absorbing diffusion (Austin et al., 2021) in the discrete graph space.
We first define absorbing node state on graphs as follows:
Definition 3 (Absorbing Node State) .
When a node v i enters the absorbing state, (1) it will be masked and (2) it will be connected to all the other nodes in G by masked edges.
Instead of only masking the original edges, we connect the masked node v i to all the other nodes with masked edges as we cannot know v
i ’s original neighbors in the absorbing state.
With the absorbing node state defined, we then need a node decay ordering for the forward absorbing pass.
A na¨ıve strategy is to use a random ordering sampled from a uniform distribution as in (Hoogeboom et al., 2022a).
In the reverse generation process, the variables will be generated in the exact reverse order, which also follows a uniform distribution.
However, such a strategy is problematic for graphs.
First, different graph datasets have different structural regularities, and it is key to leverage such regularities to ease generative learning.
For example, community-structured graphs typically consist of dense subgraphs that are loosely overlapping.
For such graphs, it is an easier learning task to generate one community first and then add the others, but a random node ordering cannot leverage such local structural regularity, which makes generation more difficult.
Second, to compute the likelihood, we need to marginalize over all possible node orderings due to node permutation invariance.
It will be more sample efficient if we can use an optimized proposal ordering distribution and use importance sampling to compute the data likelihood.
To address this issue, we propose to use a diffusion ordering network q ϕ ( σ | G 0 ) such that, at each diffusion step t , we sample from this network to select a node v σ ( t ) to be absorbed and obtain the
Masked edge
Masked node    ... corresponding masked graph G t (Figure 1).
This leads to the following definition of our graph autoregressive diffusion process:
Definition 4 (Autoregressive Graph Diffusion Process) .
In autoregressive graph diffusion, the node decay ordering σ is sampled from a diffusion ordering network q ϕ ( σ | G 0 ) parameterized by ϕ .
Then, exactly one node decays to the absorbing state at a time according to the sampled diffusion ordering.
The process proceeds until all the nodes are absorbed.
The diffusion ordering network follows a recurrent structure q ϕ ( σ | G 0 ) =
(cid:81) t q ϕ ( σ t | G 0 , σ ( <t ) ) .
At each step t , the distribution of the t -th node σ t is conditioned on the original graph G 0 and the generated node ordering up to t − 1 , i .
e ., σ ( <t ) .
We use a graph neural network (GNN) to encode the structural information in the graph.
To capture the partial ordering, we add positional encodings into node features (Vaswani et al., 2017) as in (Chen et al., 2021).
We denote the updated node embedding of node v i after passing the GNN as h di , and parameterize q ϕ ( σ t | G 0 , σ ( <t ) ) as a categorical distribution:
[Equation]
where V M is the set of node indices that have been masked.
With q ϕ ( σ | G 0 ) , G RAPH A RM can learn to optimize node ordering for diffusion.
However, this also requires us to infer the reverse generation ordering in the backward pass.
Inferring such a reverse generation ordering is difficult since we do not have access to the original graph G 0 in intermediate backward steps.
In Section 3.3, we show that it is possible to circumvent inferring this generation ordering by leveraging the permutation invariance of graph generation.
3. T HE REVERSE GENERATIVE PROCESS
In the generative process, a denoising network p θ ( G t | G t +1 ) will denoise the masked graph in the reverse order of the diffusion process.
We design p θ ( G t | G t +1 ) as a graph attention network (GAT) (Veliˇckovi´c et al., 2018; Liao et al., 2019) parameterized by θ , so that the model can distinguish the masked and unmasked edges.
For clarity, we use the vallina GAT to illustrate the computing process.
However, one can adopt any advanced graph neural network with attentive message passing.
At time t , the input to the denoising network p θ ( G t | G t +1 ) is the previous masked graph G t +1 .
A direct way is to use G t +1 which contains all the masked nodes with their corresponding masked edges.
However, during the initial generation steps, the graph is nearly fully connected with masked edges.
This has two issues: (1) the message passing procedure will be dominated by the masked edges which makes the messages uninformative.
(2) Storing the dense adjacency matrix is memory expensive, which makes the model unscalable to large graphs.
Therefore, during each generation step, we only keep the masked node to be denoised with its associated masked edges, while ignoring the other masked nodes.
We refer the modified masked graph as G ′ t , as shown in Figure 2.
The denoising network first uses an embedding layer to encode each node v i into a continuous embedding space, i .
e .
, h i =
Embedding ( v i ) .
At l -th message passing, we update the embedding of
[Figure]
node v i by aggregating the attentive messages from its neighbor nodes:
[Equation]
where W is the weight matrix, a is the attention vector.
In Eq.2, the attention mechanism enables the model to distinguish if the message comes from a masked edge.
After L rounds of message passing, we obtain the final embedding h Li for each node, then we predict the edges between the new node v σ ( t ) with all previously denoised nodes { v σ ( >t ) } .
One choice is to sequentially predict these edges as in (You et al., 2018b; Shi et al., 2020).
However, this sequential generation process takes O ( n 2 ) steps, which is inefficient.
Instead, we predict the connections of the new node to all previous nodes at once using a mixture of Bernoulli distribution as in (Liao et al., 2019).
The mixture distribution can capture the dependencies among edges to be generated and meanwhile reduce the autoregressive generation steps to O ( n ) .
3. T RAINING OBJECTIVE
We use approximate maximum likelihood as the training objective for G RAPH A RM .
We first derive the variational lower bound (VLB) of likelihood as:
[Equation]
where G 0: n denotes all values of G t for t = 0 , · · · , n and p ( σ 1 , · · · , σ n | G n ) is the distribution of the generation ordering.
A detailed derivation of Eq.
3 is given in Appendix A.
As we can see from Eq. 3, introducing the diffusion ordering network requires to infer a reverse generation ordering p ( σ 1 , · · · , σ n | G n ) .
This is difficult as we do not have the original graph G 0 in the intermediate generation process.
However, since graph generation is node permutation invariant, we do not need to learn such a generation ordering explicitly.
Note that the first term in Eq.3 encourages the model to generate the nodes in the exact reverse ordering of the diffusion process.
This pushes the generation ordering p ( σ 1 , · · · , σ n | G n ) close to the diffusion ordering q ϕ ( σ 1 , · · · , σ n ) | G 0 ) during training and the KL-divergence term will converge to 0.
Therefore, we can ignore the KL-divergence term and finally arrive at a simple training objective:
[Equation]
Compared with the random diffusion ordering, our design has two benefits: (1) We can automatically learn the optimal node generation ordering which leverages the structural information in the graph.
(2) We can consider the diffusion ordering network as an optimized proposal distribution of importance sampling for computing the data likelihood, which is more sample-efficient than an uniform proposal distribution.
3. P ARAMETER O PTIMIZATION
Learning the parameters of G RAPH A RM is challenging, because we need to evaluate the expectation of the likelihood over the diffusion ordering network.
We use a reinforcement learning (RL) procedure by sampling multiple diffusion trajectories, thereby enabling training both the diffusion ordering network q ϕ ( σ | G 0 ) and the denoising network p θ ( G t | G t +1 ) using gradient descent.
Specifically, at each training iteration, we explore the diffusion ordering network by creating
M diffusion trajectories for each training graph G ( i ) 0 .
Each trajectory is a sequence of graphs { G ( i,m ) t } 1 ≤ t ≤ n where the node decay ordering is sampled from q ϕ ( σ | G 0 ) .
The denoising network p θ ( G t | G t +1 ) is then trained to minimize the negative VLB using stochastic gradient descent (SGD):
[Equation]
where B train is the a minibatch sampled from the training data.
To evaluate the current diffusion ordering network, we create M trajectories for each validation graph and compute the negative VLB of the denoising network to obtain the corresponding rewards
[Equation]
G
[Equation]
Then, the diffusion ordering network can be updated with common RL optimization methods, e .
g ., the REINFORCE algorithm (Williams, 1992):
[Equation]
In practice, one can also explore other advanced RL technique, e .
g ., PPO2, (Schulman et al., 2017)
but we find that the simple REINFORCE algorithm has already achieved outstanding performance.
The detailed training procedure is summarized in Algorithm 1 in Appendix.
A.4.
C OMPARISON WITH OM
OM (Chen et al., 2021) also models the node ordering for autoregressive graph generation, but our method differs from OM in two aspects.
(1) The motivations are different.
Our method is motivated by autoregressive diffusion and treats the graph sequences in the diffusion process as the latent variable; OM treats the node ordering as the latent variable and infers its posterior distribution in a way similar to Variational autoencoder (VAE).
(2) Our training objective is much simpler than
OM.
First, we do not need to compute the complicated graph automorphism, which requires some approximation algorithms to compute.
Second, our training objective does not involve the entropy of the node ordering distribution.
Existing works (Lucas et al., 2019b;a) have shown that the entropy term in the VAE objective can easily cause posterior collapse.
E XPERIMENTS 4. E XPERIMENTAL S ETUP
Dataset .
We evaluate the performance of G RAPH A RM on six diverse graph generation benchmarks, covering both synthetic and real-world graphs from different domains: (1) Community-small (You et al., 2018b), (2) Caveman (You, 2018), (3)
Cora (Sen et al., 2008), (4) Breast (Gonzalez-Malerva et al., 2011), (5) Enzymes (Schomburg et al., 2004) and (6) Ego-small (Sen et al., 2008).
For each dataset, we use 80% of the graphs as training set and the rest 20% as test sets.
Following (Liao et al., 2019), we randomly select 20% from the training data as the validation set.
We generate the same amount of samples as the test set for each dataset.
More details can be seen in Appendix.
A.8.
Baselines .
We compare G RAPH A RM with the following baselines: DeepGMG (Li et al., 2018) and GraphRNN (You et al., 2018b) are RNN-based autoregressive graph generation models.
GraphAF (Shi et al., 2020) and GraphDF (Luo et al., 2021) are flow-based autoregressive models.
GRAN (Liao et al., 2019) is an autoregressive model conditioned on blocks.
OM (Chen et al., 2021) is an autoregressive model that infers node ordering using variational inference.
GraphVAE (Simonovsky & Komodakis, 2018) is an one-shot model based on VAE.
EDP-GNN (Niu et al., 2020) and GDSS (Jo et al., 2022) are scored-based one-shot methods.
[Table]
Parameter Settings .
We use ADAM with β 1 = 0 .
9 and β = 0 .
999 as the optimizer.
The learning rate is set for 10 − 4 and 5 × 10 − 4 for the denoising network and diffusion ordering network respectively on all the datasets.
For fair comparison, our denoising network uses the same graph attention network architecture as the baseline GRAN; the diffusion ordering network uses the same graph attention network architecture as the baseline OM.
As OM is compatible with any existing autoregressive methods, we use the strongest autoregressive baseline GRAN as its backbone.
More implementation details and parameter settings are provided in in Appendix A.6.
Evaluation metrics .
Following previous work (You et al., 2018b; Jo et al., 2022), we measure generation quality using the maximum mean discrepancy (MMD) as a distribution distance between the generated graphs and the test graphs.
Specifically, we compute the MMD of degree distribution, clustering coefficient, and orbit occurrence numbers of 4 nodes between the generated set and the test set.
We also report the generation time of different methods.
4. E XPERIMENTAL R ESULTS
Main Results Table 1 shows the generation performance on the six benchmarks for all the methods.
(1) As we can see, our method can outperform or achieve competitive performance compared with the baselines.
In terms of efficiency, our model is on par with the most efficient autoregressive baseline GRAN, and 10-100X faster than other baselines except for the one-shot baseline GraphVAE.
However, GraphVAE’s generation quality is much worse than ours as it generates all edges independently from the latent embedding and cannot well capture complex edge dependencies.
GDSS is the strongest baseline.
However, its generation is extremely slow as it needs to solve a complex SDE system in the reserve process.
Our G RAPH A RM model is more than 100 times faster than GDSS in generation on almost all the datasets.
The other graph diffusion model EDP-GNN is even slower than GDSS as its generation process uses annealed Langevin Dynamics with many different noise levels.
(3) G RAPH A RM also outperforms existing autoregressive graph generative models by large margins.
This is because they adopt a fixed node ordering when training the generative model while G RAPH A RM automatically learns an optimal generation ordering for the graph.
Though OM also learns a node ordering distribution, G RAPH A RM consistently outperforms it on all the datasets.
This is because OM uses the VAE objective which may suffer from posterior collapse, while G RAPH A RM has a much simpler training objective based on autoregressive diffusion.
[Table]
Constrained Graph Generation
In this set of experiments, we study G RAPH A RM ’s capability in constrained graph generation by comparing it with the strongest one-shot and autoregressive baselines.
We use the Caveman dataset for constrained graph generation, with the constraint that the maximum node degree is no larger than 6.
The detailed setup is in Appendix
A.7.
Table.
2 shows the constrained generation performance on the Caveman dataset.
GDSS and EDP-GNN are both one-shot generative model and thus they cannot guarantee the constraints are satisfied during sampling.
We find that more than half of their generated samples are invalid.
For autoregressive baselines, we apply the degree checking procedure on the two strongest baselines, i .
e .,GRAN and OM.
As we can see, G RAPH A RM can generate constrained samples that are closer to the data distribution.
This is useful for many real-world applications.
For example, when designing the contact networks of patients and healthcare workers in hospitals, a constraint of degrees for healthcare workers can help avoid superspreaders and potential infectious disease outbreaks (Jang et al., 2019; Adhikari et al., 2019).
[Table]
Evaluation of Log likelihood We further evaluate the expected negative log-likelihood (NLL) across node permutations on the test sets.
For GRAN, we sample 1000 node permutations from the uniform distribution.
For
OM and G RAPH A RM , we sample 1000 node permutations from the ordering network.
Table 3 shows the expected NLL on the test sets for Community-small and Breast.
As shown, G RAPH A RM can achieve competitive results with OM and outperform GRAN by large margins.
This is because both OM and
G RAPH A RM learn an optimal node ordering distribution; sampling from this distribution is more sample efficient than the uniform distribution.
Though GDSS and EDP-GNN are also diffusion based graph generative model, they cannot provide the likelihood.
Note that GDSS involves a system of SDEs and we cannot directly use the probability flow ODE in (Song et al., 2021) to compute the likelihood.
This can be a drawback in some density-based downstream tasks, e .
g ., outlier detection.
[Table]
Effect of Diffusion Ordering We further validate the superiority of the learned diffusion ordering in G RAPH A RM over random node ordering for graph generation.
We compare
G RAPH A RM with an ablation OA-ARM, which uses random node ordering (Hoogeboom et al., 2022a) for graph generation.
Table 4 shows the generation performance and generation ordering of G RAPH A RM and OA-ARM on the
Community-small dataset.
As shown, with the random generation ordering, the generation performance drops significantly.
This is because a random generation ordering fails to leverage the structural regularities in the graphs and thus learning the generative model is much more difficult.
To evaluate the node generation ordering, we use the spectral graph clustering method to partition the nodes into two clusters for each generated graph and then count the cross-cluster steps during the generation procedure.
As we can see, the average number of nodes that cross different clusters of our method is much smaller than OA-ARM.
This demonstrates that our method tends first to generate one cluster and then adds another cluster while OA-ARM just randomly generates the graph.
Therefore, the generation ordering of G RAPH A RM can better capture graph topology regularity than OA-ARM.
Figure 3 (in Appendix A.1) visualize the graph generative process of G RAPH A RM and OA-ARM.
As we can see, G RAPH A RM first generates one community and then moves to another; while OA-ARM randomly generates the graph and fails to capture the underlying graph distribution.
A DDITIONAL R ELATED W ORK
Graph Generation.
One-shot graph generative models generate all edges between nodes at once.
Models based on VAEs (Simonovsky & Komodakis, 2018; Liu et al., 2018; Ma et al., 2018) and
GANs (Cao & Kipf, 2022; Maziarka et al., 2020) generate all edges independently from latent embeddings.
This independence assumption can hurt the quality of the generated graphs.
Normalizing flow models (Zang & Wang, 2020; Madhawa et al., 2019) are restricted to invertible model architectures for building a normalized probability.
The other class is autoregressive graph generative models , which generate a graph by sequentially adding nodes and edges.
Autoregressive generation can be achieved using recurrent networks (Li et al., 2018;
You et al., 2018b; Popova et al., 2019),
VAEs (Liu et al., 2018; Jin et al., 2018; 2020), normalizing flows (Shi et al., 2020; Luo et al., 2021), and RL (You et al., 2018a).
By breaking the problem into smaller parts, these methods are more apt at capturing complex structural patterns and can easily incorporate constraints during generation.
However, a key drawback of them is that their training is sensitive to node ordering.
Most existing works pre-define a fixed node ordering by ad-hoc such as random breadth-first search (BFS) ordering (You et al., 2018b; Shi et al., 2020; Luo et al., 2021), which cannot preserve permutation invariance and also be suboptimal for generation.
Diffusion and Score-Based Generation.
Diffusion models have emerged as a new family of powerful deep generative models.
Denoising diffusion probabilistic modeling (DDPM)
(Sohl-Dickstein et al., 2015; Ho et al., 2020) perturbs the data distribution into a Gaussian base distribution through an forward Markov noising process, and then learns to recover data distribution via the reverse transition of the Markov chain.
Closely related to DDPM is score-based generation (Song & Ermon, 2019), which perturbs data with gradually increasing noise, and then learns to reverse the perturbation via score matching.
Song et al.
(2021) generalize diffusion models to continuous-time diffusion using forward and backward SDEs.
Diffusion and score-based models has been successfully developed for problems including image synthesis (Dhariwal & Nichol, 2021), text-to-image synthesis (Ramesh et al., 2022; Saharia et al., 2022), and molecular conformation modeling (Xu et al., 2021;
Hoogeboom et al., 2022b).
Few works have studied diffusion-based graph generation.
To the best of our knowledge, Niu et al.
Jo et al.
(2022) are the only two works in this line.
Different from these two works, our model is the first autoregressive diffusion model for graph generation, which defines diffusion directly in the discrete graph space.
L IMITATIONS AND D ISCUSSION
We have proposed a new autoregressive diffusion model for graph generation.
The proposed model
G RAPH A RM defines a node-absorbing diffusion process that directly operates in the discrete graph spaces.
We designed a diffusion ordering network that learns an optimal ordering for this diffusion process, coupled with a reverse denoising network that performs autoregressive graph reconstruction.
We derived a simple variational lower bound of the likelihood, and showed that the two networks can be jointly trained with reinforcement learning.
Our experiments have validated the generation performance and efficiency of G RAPH A RM .
We discuss several limitations and possible future extensions of G RAPH A RM : (1) Generating attributed graphs.
Our work has been focused on generating graph structures only.
However, it is possible to extend G RAPH A RM for attributed graph generation by changing the Bernoulli-mixture distribution to multinomial-mixture distribution.
(2) Handling more complex constraints.
We have shown that the autoregressive procedure of G RAPH A RM can handle constraints better than oneshot generative models.
However, practical graph generation applications can involve complex constraints on global-level properties such as graph spectrum.
How to handle such global-level constraints remains challenging for our model and existing graph generative models.
V ISUALIZATION OF GENERATION ORDERING
Figure 3 shows the graph generative process of GRAPHARM and OA-ARM.
As we can see,
GRAPHARM first generates one community and then moves to another; while OA-ARM randomly generates the graph and fails to capture the underlying graph distribution.
[Figure]
As we can see, G RAPH A RM first generates one community and then adds another one, which show that G RAPH A RM captures graph structural topology for generation.
In contrast, OA-ARM generates the graph with a random order.
A. C ONNECTION B ETWEEN A UTOREGRESSIVE D IFFUSION AND A BSORBING D IFFUSION
While ARM appears different from classic diffusion, it amounts to absorbing diffusion with continuous time limit.
Starting from state x 0 , we can define a continuous-time absorbing process, where each element x ( i ) t independently decays into an absorbing state with continuous-time probabilities α ( t ) .
This stochastic process is equivalent as using a finite set of D random transition times { τ i } Di = for recording the time where x ( i ) t was absorbed.
It was shown by Hoogeboom et al.
(2022a) that modeling the reverse generation of this process does not need to be conditioned on the precise values of the transition times τ i .
Hence, when training the reverse generative model, we only need to model x τ i based on x τ i +1 while ignoring τ i .
This allows for writing the variational lower bound (VLB) of likelihood as an expectation over an uniform ordering in an autoregressive form:
[Equation]
A. D ERIVATION OF E Q
Following (Chen et al., 2021), let us define Π ( G 1: n ) as the set of all node orderings that give the same graph sequence G 1: n , i.e., Π ( G 1: n ) = { σ 1: n : G t ( σ 1: t ) =
G t , ∀ t = 1 , · · · , n } where G t ( σ 1: t ) is the masked graph at time step t under the node ordering σ 1: t .
Following (Chen et al., 2021), the conditional p ( σ 1: n | G 1: n ) is a uniform distribution, i.e., p ( σ 1: n | G 1: n ) =
[Equation]
Π
[Equation]
G 1: n ) | , ∀ σ 1: n ∈ Π ( G 1: n ) .
Then, we have the following two equations (Chen et al., 2021):
[Equation]
To derive the VLB, we first prove that for any function of G 0: n , i.e., f ( G 0: n ) , its expectation over the diffused graph sequence is equal to the expectation over the node diffusion orderings.
That is:
[Equation]
Proof.
[Equation]
The third last row is because iterating over the double loops of all the graph sequences and the corresponding node orderings is equal to iterating over all possible node orderings.
Then, our VLB can be written as:
[Equation]
: n ) − log q ϕ ( G 1: n | G 0 ))
Eq.
= E q ϕ ( σ 1: n | G 0 ) (log | Π ( G 1: n ) | p θ ( G 1: n , σ 1: n ) + log p θ ( G 0 | G 1: n ) − log q ϕ ( G 1: n | G 0 ))
Eq.
9 =
E q ϕ ( σ 1: n | G 0 ) (log | Π ( G 1: n ) | p θ ( G 1: n , σ 1: n ) + log p θ ( G 0 | G 1: n ) − log | Π
( G 1: n ) | q ϕ ( σ 1: n | G 0 )) =
E q ϕ ( σ 1: n | G 0 ) (log | Π ( G 1: n ) | p θ ( G 1: n − 1 | σ 1: n ) p ( σ 1: n | G n ) p ( G n ) + log p θ ( G 0 | G 1: n ) − log | Π
( G 1: n ) | q ϕ ( σ 1: n | G 0 )) =
E q ϕ ( σ 1: n | G 0 ) (log p θ ( G 1: n − 1 | σ 1: n ) + log p θ ( G 0 | G 1: n ) + log p ( σ 1: n | G n ) + (cid:24)(cid:24)(cid:24)(cid:24)(cid:24)(cid:58) 0 log p ( G n ) − log q ϕ ( σ 1: n | G 0 )
+ log | Π ( G 1: n ) | − log | Π ( G 1: n ) | ) =
E q ϕ ( σ 1: n | G 0 ) (log p θ ( G 0: n − 1 | σ 1: n , G n ) + log p ( σ 1: n | G n ) − log q ϕ ( σ 1: n | G 0 ))
=
E q ϕ ( σ 1: n | G 0 ) (log p θ ( G 0: n − 1 ( σ 1: n ) | G n ) + log p ( σ 1: n | G n ) − log q ϕ ( σ 1: n | G 0 ))
=
E q ϕ ( σ 1: n | G 0 ) (log p θ ( G 0: n − 1 ( σ 1: n ) | G n ) + log p ( σ 1: n | G n ) − log q ϕ ( σ 1: n | G 0 ))
=
E q ϕ ( σ 1: n | G 0 ) n − 1 (cid:88) t
= log p θ ( G t ( σ 1: n )) | G t +1 ( σ 1: n ))
− KL ( q ϕ ( σ 1: n | G 0 ) | p ( σ 1: n | G n )) .
We use G t ( σ 1: n ) to denote the masked graph at time step t under the node ordering σ 1: n .
In the fifth last row, we have log p ( G n ) = 0 because G n is a deterministic graph wherein all the nodes are masked.
For brevity, we directly use p θ ( G t | G t +1 ) to represent p θ ( G t ( σ 1: n ) | G t +1 ( σ 1: n )) in Eq.since σ 1: n already appears in the expectation over q ϕ ( σ 1: n | G 0 ) .
A. T RAINING ALGORITHM OF G RAPH A RM
We use a reinforcement learning (RL) procedure by sampling multiple diffusion trajectories, thereby enabling training both the diffusion ordering network q ϕ ( σ | G 0 ) and the denoising network p θ ( G t | G t +1 ) using gradient descent.
Specifically, at each training iteration, we explore the diffusion ordering network by creating
M diffusion trajectories for each training graph G ( i ) 0 .
Each trajectory is a sequence of graphs
{ G ( i,m ) t } 1 ≤ t ≤ n where the node decay ordering is sampled from q ϕ ( σ | G 0 ) .
The denoising network p θ ( G t | G t +1 ) is then trained to minimize the negative VLB using stochastic gradient descent (SGD):
[Equation]
where B train is the a minibatch sampled from the training data.
To evaluate the current diffusion ordering network, we create M trajectories for each validation graph and compute the negative VLB of the denoising network to obtain the corresponding rewards
R mt = − (cid:80) i ∈B val (cid:80) t log p θ ( G ( i,m ) t | G ( i,m ) t +1 ) .
Then, the diffusion ordering network can be updated with common RL optimization methods, e .
g ., the REINFORCE algorithm (Williams, 1992):
[Equation]
The detailed training procedure is summarized in Algorithm 1.
Algorithm
Require:
Diffusion ordering network q ϕ ( σ | G 0 ) , Denoising network
[Equation]
G t | G t +1 ) 1: for # training iterations do 2:
Sample a minibatch B from the training set 3: for each i ∈ B train do 4: for each m ∈ [1 , M ] do 5: σ 1 , · · · , σ t ∼ q ϕ ( σ | G 0 ) 6:
Obtain corresponding diffusion trajectories { G ( i,m ) t } 1 ≤ t ≤ n 7: end for 8: θ j ← θ
[Equation]
1 −
[Equation]
∇ (cid:80)
[Equation]
train (cid:80) m (cid:80) t log p θ ( G ( i,m ) t | G ( i,m ) t +1 ) 9: end for 10:
Sample a minibatch from the validation set 11: for each i ∈ B val do 12: for each m ∈ [1 , M ] do 13: σ 1 , · · · , σ t ∼ q ϕ ( σ | G 0 ) 14:
Obtain corresponding diffusion trajectories { G ( i,m ) t } 1 ≤ t ≤ n 15: end for 16:
Compute the reward R mt = − (cid:80) i ∈D val (cid:80) t log p θ ( G ( i,m ) t | G ( i,m ) t +1 ) 17: ϕ
j ← ϕ j − 1 − η 2 M (cid:80) m R mt ∇ log q ϕ ( σ | G 0 ) .
18: end for 19: end for
A. E XTENDED RELATED WORK
Graph Generation.
One-shot graph generative models generate all edges between nodes at once.
Models based on VAEs (Simonovsky & Komodakis, 2018; Liu et al., 2018; Ma et al., 2018) and
GANs (Cao & Kipf, 2022; Maziarka et al., 2020) generate all edges independently from latent embeddings.
This independence assumption can hurt the quality of the generated graphs.
Normalizing flow models (Zang & Wang, 2020; Madhawa et al., 2019) are restricted to invertible model architectures for building a normalized probability.
Another drawback shared by all one-shot generation models is that they cannot incorporate constraints during generation due to the one-shot process.
The other class is autoregressive graph generative models , which generate a graph by sequentially adding nodes and edges.
Autoregressive generation can be achieved using recurrent networks (Li et al., 2018; You et al., 2018b; Popova et al., 2019), VAEs (Liu et al., 2018; Jin et al., 2018; 2020), normalizing flows (Shi et al., 2020; Luo et al., 2021), and RL (You et al., 2018a).
By breaking the problem into smaller parts, these methods are more apt at capturing complex structural patterns and can easily incorporate constraints during generation.
However, a key drawback of them is that their training is sensitive to node ordering.
Most existing works pre-define a fixed node ordering by ad-hoc such as random breadth-first search (BFS) ordering (You et al., 2018b; Shi et al., 2020; Luo et al., 2021), which cannot preserve permutation invariance and also be suboptimal for generation.
Chen et al.
(2021) propose to learn the node generation ordering using variational inference (VI).
Diffusion and Score-Based Generation.
Diffusion models have emerged as a new family of powerful deep generative models.
Denoising diffusion probabilistic modeling (DDPM)
(Sohl-Dickstein et al., 2015; Ho et al., 2020) perturbs the data distribution into a Gaussian base distribution through an forward Markov noising process, and then learns to recover data distribution via the reverse transition of the Markov chain.
Closely related to DDPM is score-based generation (Song & Ermon, 2019), which perturbs data with gradually increasing noise, and then learns to reverse the perturbation via score matching.
Song et al.
(2021) generalize diffusion models to continuous-time diffusion using forward and backward SDEs.
Diffusion and score-based models has been successfully developed for problems including image synthesis (Dhariwal & Nichol, 2021), text-to-image synthesis (Ramesh et al., 2022; Saharia et al., 2022), and molecular conformation modeling (Xu et al., 2021;
Hoogeboom et al., 2022b).
A few works have proposed discrete diffusion models that operate directly in discrete state spaces through multinomial diffusion that uses categorical noise (Hoogeboom et al., 2021), discrete absorbing diffusion (Austin et al., 2021), or autoregressive diffusion (Hoogeboom et al., 2022a).
Our
G RAPH A RM model extends autoregressive diffusion in two ways: (1) It defines autoregressive absorbing diffusion processes on graphs; (2) It learns an optimal absorbing node ordering via the permutation invariance property in graph generation.
Few works have studied diffusion-based graph generation.
To the best of our knowledge, Niu et al.
Jo et al.
(2022) are the only two works in this line.
Niu et al.
(2020) perturb the adjacency matrices with gradually increasing noise levels, and
Jo et al.
(2022) model diffusion on both the adjacency matrices and node features using continuoustime SDE-based diffusion.
Different from these two works, our model is the first autoregressive diffusion model for graph generation, which defines diffusion directly in the discrete graph space.
A. I MPLEMENTATION DETAILS
Model optimization : We use ADAM with β 1 = 0 .
9 and β = 0 .
999 as the optimizer.
The learning rate is set for 10 − 4 and 5 × 10 − 4 for the denoising network and diffusion ordering network respectively on all the datasets.
We perform model selection based on the average MMD of the three metrics on the validation set.
Network architecture : For fair comparison, our denoising network uses the same graph attention network architecture as the baseline GRAN which has 7 layers and hidden dimensions 128; the diffusion ordering network uses the same graph attention network architecture as the baseline OA which is a vallina GAT (Veli ˇ ckovi ´
c et al., 2018) that has 3 layers with 6 attention heads and residual connections.
The hidden dimension is set to 32.
As OM is compatible with any existing autoregressive methods, we use the strongest autoregressive baseline GRAN as its backbone.
Hyper-parameters : We set the number of trajectories M as 4 for all the datasets.
Both
G RAPH A RM and GRAN use 20 as the number of Bernoulli mixtures.
For GRAN, we use block size 1 and stride 1 to achieve the best generation performance.
For GDSS, we choose the best signal-to-noise ratio (SNR) from { 0 . 05 , 0 . 1 , 0 . 15 , 0 . 2 } and scale coefficient from { 0 . 1 , 0 . 2 , 0 . 3 , 0 . 4 , 0 . 5 , 0 . 6 , 0 . 7 , 0 . 8 , 0 . 9 , 1 . 0 } based on the average MMD of degree, clustering coefficient and orbit as in Jo et al.
For EDP-GNN, we use 6 noise levels { σ i } Li =
= [1 . 6 , 0 .
1] as suggested in the original work (Niu et al., 2020).
For OM, we use 4 as the sample size for variational inference as suggested in the original work (Chen et al., 2021)
A. E XPERIMENT S ETUP FOR C ONSTRAINED G RAPH G ENERATION
We use the Caveman dataset for the constrained generation experiment.
We set the constraint as that the maximum node degree is no larger than 6.
To generate graphs under this constraint with
G RAPH A RM , we add a degree checking into the generation process.
Specifically, when generating a new node and its connecting edges, we first check whether the constraint will be violated on previous nodes as new edges are added to them.
If so, the corresponding edges will be dropped; otherwise we proceed to check the constraint on the new generated node and randomly remove the extra edges that exceed the limit.
A. D ATASET S TATISTICS
We evaluate the performance of G RAPH A RM on six diverse graph generation benchmarks, covering both synthetic and real-world graphs from different domains: (1) Community-small (You et al., 2018b), containing 100 graphs randomly generated community graphs with 12 ≤ | V | ≤ 20 ; (2)
Caveman (You, 2018), containing 200 caveman graphs synthetically generated with 5 ≤ | V | ≤ 10 ; (3) Cora, containing 200 sub-graphs with 9 ≤ | V | ≤ 87 , extracted from Cora network (Sen et al., 2008) using random walk; (4) Breast, including 100 chemical graphs with 12 ≤ | V | ≤ 18 , sampled from Gonzalez-Malerva et al.
(2011); (5) Enzymes, including 563 protein graphs with 10 ≤ | V | ≤ 125 from BRENDA database (Schomburg et al., 2004); (6) Ego-small, containing 200 small subgraphs with 4 ≤ | V | ≤ 18 sampled from Citeseer Network Dataset (Sen et al., 2008).
For each dataset, we use 80% of the graphs as training set and the rest 20% as test sets.
Following (Liao et al., 2019), we randomly select 25% from the training data as the validation set.
A. A PPROXIMATION E RROR OF THE KL- DIVERGENCE T ERM IN E Q . 3.
To evaluate the approximation error, we compute the KL-divergence between the diffusion ordering and the generation ordering at each time step and then sum together, i.e.,
Approximation error= (cid:80) t KL (cid:0) q ( σ t | G 0 , σ ( <t )) | p ( σ t | G t ) (cid:1) .
Specifically, at each time step t , we use
G RAPH A RM to first generate a graph G 0 and record the corresponding node generation ordering σ .
Since we do not explicitly model distribution of the node index during the generation procedure, we determine the index of the generated node in G 0 by its connectivity to the unmasked nodes in
G t .
Finally, we forward the generation network multiple times to obtain an empirical distribution for p ( σ t | G t ) .
Table 5 provides the approximation error versus the training iterations.
As we can see, with the training progresses, the approximation error indeed becomes smaller and approaches to zero.
[Table]