arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2203.00638cs.LGcs.AI
收藏

PaSca:一个可扩展范式下的图神经结构搜索系统

PaSca: a Graph Neural Architecture Search System under the Scalable Paradigm

Peking University(北京大学);Tencent Inc.(腾讯公司)

Wentao Zhang, Yu Shen, Zheyu Lin, Yang Li, Xiaosen Li, Wen Ouyang, Yangyu Tao, Zhi Yang, Bin Cui

摘要:

图神经网络(GNN)在各种基于图的任务中已经实现了最先进的性能。然而,由于主流GNN是基于神经消息传递机制设计的,因此它们不能很好地扩展到数据大小和消息传递步骤。尽管人们对可扩展GNN的设计产生了浓厚的兴趣,但目前的研究集中在特定的GNN设计上,而不是一般的设计空间,限制了潜在的可扩展GNN模型的发现。本文提出了PasCa,这是一种新的范式和系统,它提供了一种原则性的方法来系统地构建和探索可扩展GNN的设计空间,而不是研究单个设计。通过解构消息传递机制,PasCa提出了一种新的可扩展图神经架构范式(Scalable Graph Neural Architecture Paradigm,SGAP),并给出了一个包含15万种不同设计的通用架构设计空间。遵循该范例,我们实现了一个自动搜索引擎,该引擎可以自动搜索性能良好且可扩展的GNN架构,以平衡多个标准之间的权衡(例如,准确性和效率)。对10个基准数据集的实证研究表明,代表性实例(即,PasCa-V1,V2和V3)在竞争基线之间实现了一致的性能。具体来说,PasCa-V3在我们的大型行业数据集上的预测准确性方面优于最先进的GNN方法JK-Net 0.4\%,同时实现了高达$28.3\times$的训练加速。

英文摘要:

Graph neural networks (GNNs) have achieved state-of-the-art performance in various graph-based tasks. However, as mainstream GNNs are designed based on the neural message passing mechanism, they do not scale well to data size and message passing steps. Although there has been an emerging interest in the design of scalable GNNs, current researches focus on specific GNN design, rather than the general design space, limiting the discovery of potential scalable GNN models. This paper proposes PasCa, a new paradigm and system that offers a principled approach to systemically construct and explore the design space for scalable GNNs, rather than studying individual designs. Through deconstructing the message passing mechanism, PasCa presents a novel Scalable Graph Neural Architecture Paradigm (SGAP), together with a general architecture design space consisting of 150k different designs. Following the paradigm, we implement an auto-search engine that can automatically search well-performing and scalable GNN architectures to balance the trade-off between multiple criteria (e.g., accuracy and efficiency) via multi-objective optimization. Empirical studies on ten benchmark datasets demonstrate that the representative instances (i.e., PasCa-V1, V2, and V3) discovered by our system achieve consistent performance among competitive baselines. Concretely, PasCa-V3 outperforms the state-of-the-art GNN method JK-Net by 0.4\% in terms of predictive accuracy on our large industry dataset while achieving up to $28.3\times$ training speedups.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑