Training and Evaluating Graph Generative Models

Loading...
Thumbnail Image

Date

Authors

Thompson, Rylee

Journal Title

Journal ISSN

Volume Title

Publisher

University of Guelph

Abstract

In this thesis-by-articles we make several contributions related to graph generative models (GGMs) and their applications.

In our first article, we investigate the use of GGMs for the sequential design of 3D structures. We propose LEGO as a toy problem that is complex enough to approximate real-world design and develop a method for representing simple LEGO structures as a graph to train a GGM. We extend a popular GGM approach, Deep Generative Models of Graphs (DGMG), to operate on these LEGO structures and propose several evaluation metrics originally developed for generative models of images that utilize a relevant pretrained network.

In our second article, we dive deeper into the proposed metrics to identify a single metric that can be used to evaluate GGMs regardless of domain. We propose replacing the pretrained network used in evaluation with a randomly initialized one to reduce overhead requirements, and design several experiments to objectively score each metric on several criteria. Through this process, we show that pretraining the network is not required: a network with random parameters is sufficient for evaluation. We further identify several strong metrics that can be used to easily evaluate GGMs across domains.

Description

Keywords

Machine learning, Artificial intelligence, graph generative models, deep learning, graph neural networks

Citation

Thompson, R., Ghalebi, E., DeVries, T., and Taylor, G. W. (2020). Building LEGO using deep generative models of graphs. In NeurIPS Workshop on Machine Learning for Engineering Modeling, Simulation, and Design (ML4Eng), https://doi.org/10.48550/arXiv.2012.11543
Thompson, R., Knyazev, B., Ghalebi, E., Kim, J., and Taylor, G. W. (2022). On evaluation metrics for graph generative models. In International Conference on Learning Representations., https://doi.org/10.48550/arXiv.2201.09871