pith. sign in

arxiv: 2109.10047 · v1 · pith:YFI3ILUGnew · submitted 2021-09-21 · 💻 cs.LG · cs.AI

Search For Deep Graph Neural Networks

classification 💻 cs.LG cs.AI
keywords searcharchitecturesdeepmodelsover-smoothproblemspaceaggregate
0
0 comments X
read the original abstract

Current GNN-oriented NAS methods focus on the search for different layer aggregate components with shallow and simple architectures, which are limited by the 'over-smooth' problem. To further explore the benefits from structural diversity and depth of GNN architectures, we propose a GNN generation pipeline with a novel two-stage search space, which aims at automatically generating high-performance while transferable deep GNN models in a block-wise manner. Meanwhile, to alleviate the 'over-smooth' problem, we incorporate multiple flexible residual connection in our search space and apply identity mapping in the basic GNN layers. For the search algorithm, we use deep-q-learning with epsilon-greedy exploration strategy and reward reshaping. Extensive experiments on real-world datasets show that our generated GNN models outperforms existing manually designed and NAS-based ones.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.