当前位置:
X-MOL 学术
›
arXiv.cs.SI
›
论文详情
Our official English website, www.x-mol.net, welcomes your
feedback! (Note: you will need to create a separate account there.)
Design Space for Graph Neural Networks
arXiv - CS - Social and Information Networks Pub Date : 2020-11-17 , DOI: arxiv-2011.08843 Jiaxuan You, Rex Ying, Jure Leskovec
arXiv - CS - Social and Information Networks Pub Date : 2020-11-17 , DOI: arxiv-2011.08843 Jiaxuan You, Rex Ying, Jure Leskovec
The rapid evolution of Graph Neural Networks (GNNs) has led to a growing
number of new architectures as well as novel applications. However, current
research focuses on proposing and evaluating specific architectural designs of
GNNs, as opposed to studying the more general design space of GNNs that
consists of a Cartesian product of different design dimensions, such as the
number of layers or the type of the aggregation function. Additionally, GNN
designs are often specialized to a single task, yet few efforts have been made
to understand how to quickly find the best GNN design for a novel task or a
novel dataset. Here we define and systematically study the architectural design
space for GNNs which consists of 315,000 different designs over 32 different
predictive tasks. Our approach features three key innovations: (1) A general
GNN design space; (2) a GNN task space with a similarity metric, so that for a
given novel task/dataset, we can quickly identify/transfer the best performing
architecture; (3) an efficient and effective design space evaluation method
which allows insights to be distilled from a huge number of model-task
combinations. Our key results include: (1) A comprehensive set of guidelines
for designing well-performing GNNs; (2) while best GNN designs for different
tasks vary significantly, the GNN task space allows for transferring the best
designs across different tasks; (3) models discovered using our design space
achieve state-of-the-art performance. Overall, our work offers a principled and
scalable approach to transition from studying individual GNN designs for
specific tasks, to systematically studying the GNN design space and the task
space. Finally, we release GraphGym, a powerful platform for exploring
different GNN designs and tasks. GraphGym features modularized GNN
implementation, standardized GNN evaluation, and reproducible and scalable
experiment management.
中文翻译:
图神经网络的设计空间
图神经网络 (GNN) 的快速发展催生了越来越多的新架构和新应用。然而,当前的研究侧重于提出和评估 GNN 的特定架构设计,而不是研究由不同设计维度的笛卡尔积组成的 GNN 更一般的设计空间,例如层数或聚合函数的类型. 此外,GNN 设计通常专门针对单个任务,但很少有人努力了解如何为新任务或新数据集快速找到最佳 GNN 设计。在这里,我们定义并系统地研究了 GNN 的架构设计空间,该空间由 32 种不同预测任务的 315,000 种不同设计组成。我们的方法具有三个关键创新:(1)通用 GNN 设计空间;(2) 具有相似性度量的 GNN 任务空间,以便对于给定的新任务/数据集,我们可以快速识别/转移性能最佳的架构;(3) 一种高效且有效的设计空间评估方法,可以从大量模型-任务组合中提炼见解。我们的主要结果包括:(1)一套用于设计性能良好的 GNN 的综合指南;(2) 虽然针对不同任务的最佳 GNN 设计差异很大,但 GNN 任务空间允许在不同任务之间转移最佳设计;(3) 使用我们的设计空间发现的模型实现了最先进的性能。总的来说,我们的工作提供了一种有原则和可扩展的方法,可以从研究针对特定任务的单个 GNN 设计过渡到系统地研究 GNN 设计空间和任务空间。最后,我们发布了 GraphGym,这是一个用于探索不同 GNN 设计和任务的强大平台。GraphGym 具有模块化 GNN 实现、标准化 GNN 评估以及可重复和可扩展的实验管理。
更新日期:2020-11-18
中文翻译:
图神经网络的设计空间
图神经网络 (GNN) 的快速发展催生了越来越多的新架构和新应用。然而,当前的研究侧重于提出和评估 GNN 的特定架构设计,而不是研究由不同设计维度的笛卡尔积组成的 GNN 更一般的设计空间,例如层数或聚合函数的类型. 此外,GNN 设计通常专门针对单个任务,但很少有人努力了解如何为新任务或新数据集快速找到最佳 GNN 设计。在这里,我们定义并系统地研究了 GNN 的架构设计空间,该空间由 32 种不同预测任务的 315,000 种不同设计组成。我们的方法具有三个关键创新:(1)通用 GNN 设计空间;(2) 具有相似性度量的 GNN 任务空间,以便对于给定的新任务/数据集,我们可以快速识别/转移性能最佳的架构;(3) 一种高效且有效的设计空间评估方法,可以从大量模型-任务组合中提炼见解。我们的主要结果包括:(1)一套用于设计性能良好的 GNN 的综合指南;(2) 虽然针对不同任务的最佳 GNN 设计差异很大,但 GNN 任务空间允许在不同任务之间转移最佳设计;(3) 使用我们的设计空间发现的模型实现了最先进的性能。总的来说,我们的工作提供了一种有原则和可扩展的方法,可以从研究针对特定任务的单个 GNN 设计过渡到系统地研究 GNN 设计空间和任务空间。最后,我们发布了 GraphGym,这是一个用于探索不同 GNN 设计和任务的强大平台。GraphGym 具有模块化 GNN 实现、标准化 GNN 评估以及可重复和可扩展的实验管理。