Do Not Train It: A Linear Neural Architecture Search of Graph Neural Networks

Peng Xu, Lin Zhang, Xuanzhou Liu, Jiaqi Sun, Yue Zhao +2 more
5/23/2023
cs.LGcs.AI

No official code found. Best implementation: shipxu123/NAC (5 stars, last commit 2023-06-15).

Abstract

Neural architecture search (NAS) for Graph neural networks (GNNs), called NAS-GNNs, has achieved significant performance over manually designed GNN architectures. However, these methods inherit issues from the conventional NAS methods, such as high computational cost and optimization difficulty. More importantly, previous NAS methods have ignored the uniqueness of GNNs, where GNNs possess expressive power without training. With the randomly-initialized weights, we can then seek the optimal architecture parameters via the sparse coding objective and derive a novel NAS-GNNs method, namely neural architecture coding (NAC). Consequently, our NAC holds a no-update scheme on GNNs and can efficiently compute in linear time. Empirical evaluations on multiple GNN benchmark datasets demonstrate that our approach leads to state-of-the-art performance, which is up to $200\times$ faster and $18.8\%$ more accurate than the strong baselines.

View on arXivView PDF

Code Implementations(1)

shipxu123/NACCommunity100%

This repo is the official implementation of "Do Not Train It: A Linear Neural Architecture Search of Graph Neural Networks" (Xu et al., ICML 2023)

51Shell, PythonJun 15, 20233 years ago

Similar Papers

Cite this paper

@article{xu2023train,
  title  = {Do Not Train It: A Linear Neural Architecture Search of Graph Neural Networks},
  author = {Peng Xu and Lin Zhang and Xuanzhou Liu and Jiaqi Sun and Yue Zhao and Haiqin Yang and Bei Yu},
  year   = {2023},
  eprint = {2305.14065},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2305.14065v3},
  journal = {ICML 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2305.14065.svg)](https://codeofpaper.com/paper/2305.14065)

Discussion