NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
Synthetic Graph Generation for DGL-PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
Synthetic Graph Generation for DGL-PyTorch

The Synthetic Graph Generation tool enables users to generate arbitrary graphs based on provided real data.

This resource is using open-source code maintained in github (see the quick-start-guide section) and available for download from NGC

Synthetic data generation has become pervasive with imploding amounts of data and demand to deploy machine learning models leveraging such data. There has been an increasing interest in leveraging graph-based neural network model on graph datasets, though many public datasets are of a much smaller scale than that used in real-world applications. Synthetic Graph Generation is a common problem in multiple domains for various applications, including the generation of big graphs with similar properties to original or anonymizing data that cannot be shared. The Synthetic Graph Generation tool enables users to generate arbitrary graphs based on provided real data.

Synthetic Graph Generation architecture

The tool has the following architecture.

Synthetic Graph Generation architecture

The module is composed of three parts: a structural generator, which fits the graph structure, feature generator, which fits the feature distribution contained in the graph; and finally, an aligner, which aligns the generated features with the generated graph structure

Graph structural generator

The graph structural generator fits graph structure and generate a corresponding graph containing the nodes and edges.

Feature generator

The feature generator fits the feature distribution contained in the graph and generates the corresponding features. There is the option to allow users to generate features associated with nodes, edges, or both.

Aligner

The aligner aligns the generated features taken from the feature generator with the graph structure generated by a graph structural generator.

Default configuration

By default, the synthetic graph generation tool generates a random graph with random features specified by the user.

Feature support matrix

This tool supports the following features:

FeatureSynthetic Graph Generation
Non-partite graph generationYes
Bipartite graph generationYes
N-partite graph generationNo
Undirected graph generationYes
Directed graph generationYes
Self-loops generationYes
Edge features generationYes
Node features generationYes

Features

  • Non-partite graph generation is a task to generate a graph that doesn't contain any explicit partites (disjoint and independent sets of nodes).

  • Bipartite graph generation is a task to generate a graph that consists of two partites.

  • N-partite graph generation is a task to generate a graph that consists of an arbitrary number of partites.

  • Undirected graph generation is a task to generate a graph made up of a set of vertices connected by not ordered edges.

  • Directed graph generation is a task to generate a graph made up of a set of vertices connected by directed edges.

  • Self-loops generation is a task to generate edges that connect a vertex to itself.

  • Edge features generation is a task to generate features associated with an edge.

  • Node features generation is a task to generate features associated with a node.

Models

Structural graph generation

- RMAT
- Random (Erdos-Renyi)

Tabular features

- CTGAN (Conditional GAN)
- CTAB
- KDE
- Gaussian
- Random (Uniform)

Aligner

- XGBoost
- Random
Publisher
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
Latest Version22.12.0
UpdatedApril 6, 2023 UTC
Compressed Size29.72 KB