当前位置: 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

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
down
wechat
bug