activity
20122023
most citedCompressing Neural Networks using the Variational Information Bottleneck

122 citations · 894 across the 44 of their papers we have counts for

collaborators

44 papers

cs.LG2023★ 1 cited

On the Initialization of Graph Neural Networks

Jiahang Li, Yakun Song, Xiang Song +1

Graph Neural Networks (GNNs) have displayed considerable promise in graph representation learning across various applications. The core learning process requires the initialization…

cs.LG2023

From Hypergraph Energy Functions to Hypergraph Neural Networks

Yuxin Wang, Quan Gan, Xipeng Qiu +2

Hypergraphs are a powerful abstraction for representing higher-order interactions between entities of interest. To exploit these relationships in making downstream predictions, a v…

cs.LG2023★ 75 cited

NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification

Qitian Wu, Wentao Zhao, Zenan Li +2

Graph neural networks have been extensively studied for learning with inter-connected data. Despite this, recent evidence has revealed GNNs' deficiencies related to over-squashing,…

cs.LG2023★ 10 cited

Learning Manifold Dimensions with Conditional Variational Autoencoders

Yijia Zheng, Tong He, Yixuan Qiu +1

Although the variational autoencoder (VAE) and its conditional extension (CVAE) are capable of state-of-the-art results across multiple domains, their precise behavior is still not…

cs.LG2023★ 1 cited

FreshGNN: Reducing Memory Access via Stable Historical Embeddings for Graph Neural Network Training

Kezhao Huang, Haitian Jiang, Minjie Wang +7

A key performance bottleneck when training graph neural network (GNN) models on large, real-world graphs is loading node features onto a GPU. Due to limited GPU memory, expensive d…

cs.LG2022★ 1 cited

Refined Edge Usage of Graph Neural Networks for Edge Prediction

Jiarui Jin, Yangkun Wang, Weinan Zhang +5

Graph Neural Networks (GNNs), originally proposed for node classification, have also motivated many recent works on edge prediction (a.k.a., link prediction). However, existing met…