You Can Have Better Graph Neural Networks by Not Training Weights at All: Finding Untrained Graph Tickets

Tianjin Huang*, Tianlong Chen, Meng Fang, Vlado Menkovski, Jiaxu Zhao, Lu Yin, Yulong Pei, Decebal Constantin Mocanu, Zhangyang Wang, Mykola Pechenizkiy, Shiwei Liu

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionAcademicpeer-review

56 Downloads (Pure)

Abstract

Recent works have impressively demonstrated that there exists a subnetwork in randomly initialized convolutional neural networks (CNNs) that can match the performance of the fully trained dense networks at initialization, without any optimization of the weights of the network (i.e., untrained networks). However, the presence of such untrained subnetworks in graph neural networks (GNNs) still remains mysterious. In this paper we carry out the first-of-its-kind exploration of discovering matching untrained GNNs. With sparsity as the core tool, we can find \textit{untrained sparse subnetworks} at the initialization, that can match the performance of \textit{fully trained dense} GNNs. Besides this already encouraging finding of comparable performance, we show that the found untrained subnetworks can substantially mitigate the GNN over-smoothing problem, hence becoming a powerful tool to enable deeper GNNs without bells and whistles. We also observe that such sparse untrained subnetworks have appealing performance in out-of-distribution detection and robustness of input perturbations. We evaluate our method across widely-used GNN architectures on various popular datasets including the Open Graph Benchmark (OGB).
Original languageEnglish
Title of host publicationProceedings of the First Learning on Graphs Conference (LoG 2022)
PublisherOpenReview
Publication statusPublished - 9 Dec 2022
Event1st Learning on Graphs Conference, LOG 2022 - Virtual Conference
Duration: 9 Dec 202212 Dec 2022
Conference number: 1
https://logconference.org/

Conference

Conference1st Learning on Graphs Conference, LOG 2022
Abbreviated titleLOG 2022
CityVirtual Conference
Period9/12/2212/12/22
Internet address

Fingerprint

Dive into the research topics of 'You Can Have Better Graph Neural Networks by Not Training Weights at All: Finding Untrained Graph Tickets'. Together they form a unique fingerprint.
  • Best Paper Award at Learning on Graphs (LoG 2022) conference

    Huang, T. (Recipient), Chen, T. (Recipient), Fang, M. (Recipient), Menkovski, V. (Recipient), Zhao, J. (Recipient), Yin, L. (Recipient), Pei, Y. (Recipient), Mocanu, D. C. (Recipient), Wang, Z. (Recipient), Pechenizkiy, M. (Recipient) & Liu, S. (Recipient), 12 Dec 2022

    Prize

Cite this