Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Graph is Worth 1-bit Spikes: When Graph Contrastive Learning Meets Spiking Neural Networks

About

While contrastive self-supervised learning has become the de-facto learning paradigm for graph neural networks, the pursuit of higher task accuracy requires a larger hidden dimensionality to learn informative and discriminative full-precision representations, raising concerns about computation, memory footprint, and energy consumption burden (largely overlooked) for real-world applications. This work explores a promising direction for graph contrastive learning (GCL) with spiking neural networks (SNNs), which leverage sparse and binary characteristics to learn more biologically plausible and compact representations. We propose SpikeGCL, a novel GCL framework to learn binarized 1-bit representations for graphs, making balanced trade-offs between efficiency and performance. We provide theoretical guarantees to demonstrate that SpikeGCL has comparable expressiveness with its full-precision counterparts. Experimental results demonstrate that, with nearly 32x representation storage compression, SpikeGCL is either comparable to or outperforms many fancy state-of-the-art supervised and self-supervised methods across several graph benchmarks.

Jintang Li, Huizhe Zhang, Ruofan Wu, Zulun Zhu, Baokun Wang, Changhua Meng, Zibin Zheng, Liang Chen• 2023

Related benchmarks

TaskDatasetResultRank
Node ClassificationIMDB
Macro F1 Score0.602
179
Node ClassificationPhoto
Mean Accuracy92.5
165
Node ClassificationPhysics
Accuracy95.21
145
Node ClassificationComputers
Mean Accuracy89.04
143
Node ClassificationCS
Accuracy91.77
128
Node ClassificationACM
Macro F190.5
104
Node ClassificationDBLP
Micro-F191.5
24
Link PredictionPhoto
AUC-ROC95.58
19
Link PredictionPhoto (test)
AP95.16
19
Link PredictionComputers
AUC-ROC92.72
19
Showing 10 of 18 rows

Other info

Follow for update