论文标题

关于图形生成模型的评估指标

On Evaluation Metrics for Graph Generative Models

论文作者

Thompson, Rylee, Knyazev, Boris, Ghalebi, Elahe, Kim, Jungtaek, Taylor, Graham W.

论文摘要

在图像生成中,可以通过视觉检查模型输出自然评估生成模型。但是,对于图形生成模型(GGM)而言,情况并非总是如此,这使得他们的评估具有挑战性。当前,评估GGM的标准过程受到三个临界局限性:i)它不会产生单个分数,这使模型选择具有挑战性,ii)在许多情况下,它无法考虑基础的边缘和节点特征,iii)的执行缓慢。在这项工作中,我们通过搜索标量,域,不可稳定和可扩展指标来评估和排名GGM来缓解这些问题。为此,我们研究了现有的GGM指标和基于神经网络的指标,从使用特定于任务网络提取的嵌入的图像的生成模型中出现了。由某些图形神经网络(GNN)的力量提取有意义的图表表示无需任何培训的动力,我们根据未经训练的随机GNN提取的特征介绍了几个指标。我们设计实验,以彻底测试其测量生成图的多样性和保真度的能力,以及它们的样本和计算效率。根据样品的数量,我们建议我们表明的两个基于随机-GNN的指标之一,而不是先前存在的指标。尽管我们专注于将这些指标应用于GGM评估,但实际上,这使得无论域如何,都可以轻松计算任何两组图之间的差异。我们的代码在以下网址发布:https://github.com/uoguelph-mlrg/ggm-metrics。

In image generation, generative models can be evaluated naturally by visually inspecting model outputs. However, this is not always the case for graph generative models (GGMs), making their evaluation challenging. Currently, the standard process for evaluating GGMs suffers from three critical limitations: i) it does not produce a single score which makes model selection challenging, ii) in many cases it fails to consider underlying edge and node features, and iii) it is prohibitively slow to perform. In this work, we mitigate these issues by searching for scalar, domain-agnostic, and scalable metrics for evaluating and ranking GGMs. To this end, we study existing GGM metrics and neural-network-based metrics emerging from generative models of images that use embeddings extracted from a task-specific network. Motivated by the power of certain Graph Neural Networks (GNNs) to extract meaningful graph representations without any training, we introduce several metrics based on the features extracted by an untrained random GNN. We design experiments to thoroughly test metrics on their ability to measure the diversity and fidelity of generated graphs, as well as their sample and computational efficiency. Depending on the quantity of samples, we recommend one of two random-GNN-based metrics that we show to be more expressive than pre-existing metrics. While we focus on applying these metrics to GGM evaluation, in practice this enables the ability to easily compute the dissimilarity between any two sets of graphs regardless of domain. Our code is released at: https://github.com/uoguelph-mlrg/GGM-metrics.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源