Timezone: »
We propose a unified framework for adaptive connection sampling in graph neural networks (GNNs) that generalizes existing stochastic regularization methods for training GNNs. The proposed framework not only alleviates over-smoothing and over-fitting tendencies of deep GNNs, but also enables learning with uncertainty in graph analytic tasks with GNNs. Instead of using fixed sampling rates or hand-tuning themas model hyperparameters in existing stochastic regularization methods, our adaptive connection sampling can be trained jointly with GNN model parameters in both global and local fashions. GNN training with adaptive connection sampling is shown to be mathematically equivalent to an efficient approximation of training BayesianGNNs. Experimental results with ablation studies on benchmark datasets validate that adaptively learning the sampling rate given graph training data is the key to boost the performance of GNNs in semi-supervised node classification, less prone to over-smoothing and over-fitting with more robust prediction.
Author Information
Arman Hasanzadeh (Texas A&M University)
I am a machine learning researcher and a PhD student at Texas A&M University. My main areas of interest are learning with graph structured data, generative models, representation learning, Bayesian inference, and deep learning. My works have been applied to a variety of applications including graph analytics (e.g. network embedding, node classification, (dynamic) link prediction), drug generation, text generation, time series analysis, data integration, and anomaly detection. I have multiple publications in top-tier conferences and journals such as NeurIPS, ICML, ICASSP (best paper finalist), ISMB, and Bioinformatics.
Ehsan Hajiramezanali (Texas A&M University)
Shahin Boluki (Texas A&M University)
Mingyuan Zhou (University of Texas at Austin)
Nick Duffield (Texas A&M University)
Krishna Narayanan (Texas A&M University)
Xiaoning Qian (Texas A&M University)
More from the Same Authors
-
2023 : Reinstating Continuous Climate Patterns From Small and Discretized Data »
Xihaier Luo · Xiaoning Qian · Nathan Urban · Byung-Jun Yoon -
2023 Poster: Prototype-oriented unsupervised anomaly detection for multivariate time series »
yuxin li · Wenchao Chen · Bo Chen · Dongsheng Wang · Long Tian · Mingyuan Zhou -
2023 Poster: Bayesian Progressive Deep Topic Model with Knowledge Informed Textual Data Coarsening Process »
Zhibin Duan · Xinyang Liu · Yudi Su · Yishi Xu · Bo Chen · Mingyuan Zhou -
2023 Poster: Efficient Approximations of Complete Interatomic Potentials for Crystal Property Prediction »
Yuchao Lin · Keqiang Yan · Youzhi Luo · Yi Liu · Xiaoning Qian · Shuiwang Ji -
2023 Poster: Efficient and Equivariant Graph Networks for Predicting Quantum Hamiltonian »
Haiyang Yu · Zhao Xu · Xiaofeng Qian · Xiaoning Qian · Shuiwang Ji -
2023 Poster: Learning to Jump: Thinning and Thickening Latent Counts for Generative Modeling »
Tianqi Chen · Mingyuan Zhou -
2023 Poster: POUF: Prompt-Oriented Unsupervised Fine-tuning for Large Pre-trained Models »
Korawat Tanwisuth · Shujian Zhang · Huangjie Zheng · Pengcheng He · Mingyuan Zhou -
2022 Poster: VariGrow: Variational Architecture Growing for Task-Agnostic Continual Learning based on Bayesian Novelty »
Randy Ardywibowo · Zepeng Huo · Zhangyang “Atlas” Wang · Bobak Mortazavi · Shuai Huang · Xiaoning Qian -
2022 Poster: Deep Variational Graph Convolutional Recurrent Network for Multivariate Time Series Anomaly Detection »
Wenchao Chen · Long Tian · Bo Chen · Liang Dai · Zhibin Duan · Mingyuan Zhou -
2022 Poster: Bayesian Deep Embedding Topic Meta-Learner »
Zhibin Duan · Yishi Xu · Jianqiao Sun · Bo Chen · Wenchao Chen · CHAOJIE WANG · Mingyuan Zhou -
2022 Spotlight: Bayesian Deep Embedding Topic Meta-Learner »
Zhibin Duan · Yishi Xu · Jianqiao Sun · Bo Chen · Wenchao Chen · CHAOJIE WANG · Mingyuan Zhou -
2022 Spotlight: Deep Variational Graph Convolutional Recurrent Network for Multivariate Time Series Anomaly Detection »
Wenchao Chen · Long Tian · Bo Chen · Liang Dai · Zhibin Duan · Mingyuan Zhou -
2022 Spotlight: VariGrow: Variational Architecture Growing for Task-Agnostic Continual Learning based on Bayesian Novelty »
Randy Ardywibowo · Zepeng Huo · Zhangyang “Atlas” Wang · Bobak Mortazavi · Shuai Huang · Xiaoning Qian -
2022 Poster: Regularizing a Model-based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning »
Shentao Yang · Yihao Feng · Shujian Zhang · Mingyuan Zhou -
2022 Spotlight: Regularizing a Model-based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning »
Shentao Yang · Yihao Feng · Shujian Zhang · Mingyuan Zhou -
2021 Poster: Bayesian Attention Belief Networks »
Shujian Zhang · Xinjie Fan · Bo Chen · Mingyuan Zhou -
2021 Spotlight: Bayesian Attention Belief Networks »
Shujian Zhang · Xinjie Fan · Bo Chen · Mingyuan Zhou -
2021 Poster: Sawtooth Factorial Topic Embeddings Guided Gamma Belief Network »
Zhibin Duan · Dongsheng Wang · Bo Chen · CHAOJIE WANG · Wenchao Chen · yewen li · Jie Ren · Mingyuan Zhou -
2021 Poster: ARMS: Antithetic-REINFORCE-Multi-Sample Gradient for Binary Variables »
Alek Dimitriev · Mingyuan Zhou -
2021 Spotlight: ARMS: Antithetic-REINFORCE-Multi-Sample Gradient for Binary Variables »
Alek Dimitriev · Mingyuan Zhou -
2021 Spotlight: Sawtooth Factorial Topic Embeddings Guided Gamma Belief Network »
Zhibin Duan · Dongsheng Wang · Bo Chen · CHAOJIE WANG · Wenchao Chen · yewen li · Jie Ren · Mingyuan Zhou -
2020 Poster: Thompson Sampling via Local Uncertainty »
Zhendong Wang · Mingyuan Zhou -
2020 Poster: NADS: Neural Architecture Distribution Search for Uncertainty Awareness »
Randy Ardywibowo · Shahin Boluki · Xinyu Gong · Zhangyang “Atlas” Wang · Xiaoning Qian -
2020 Poster: Recurrent Hierarchical Topic-Guided RNN for Language Generation »
Dandan Guo · Bo Chen · Ruiying Lu · Mingyuan Zhou -
2019 Poster: ARSM: Augment-REINFORCE-Swap-Merge Estimator for Gradient Backpropagation Through Categorical Variables »
Mingzhang Yin · Yuguang Yue · Mingyuan Zhou -
2019 Oral: ARSM: Augment-REINFORCE-Swap-Merge Estimator for Gradient Backpropagation Through Categorical Variables »
Mingzhang Yin · Yuguang Yue · Mingyuan Zhou -
2019 Poster: Convolutional Poisson Gamma Belief Network »
CHAOJIE WANG · Bo Chen · SUCHENG XIAO · Mingyuan Zhou -
2019 Poster: Locally Private Bayesian Inference for Count Models »
Aaron Schein · Steven Wu · Alexandra Schofield · Mingyuan Zhou · Hanna Wallach -
2019 Oral: Convolutional Poisson Gamma Belief Network »
CHAOJIE WANG · Bo Chen · SUCHENG XIAO · Mingyuan Zhou -
2019 Oral: Locally Private Bayesian Inference for Count Models »
Aaron Schein · Steven Wu · Alexandra Schofield · Mingyuan Zhou · Hanna Wallach -
2018 Poster: Inter and Intra Topic Structure Learning with Word Embeddings »
He Zhao · Lan Du · Wray Buntine · Mingyuan Zhou -
2018 Oral: Inter and Intra Topic Structure Learning with Word Embeddings »
He Zhao · Lan Du · Wray Buntine · Mingyuan Zhou -
2018 Poster: Semi-Implicit Variational Inference »
Mingzhang Yin · Mingyuan Zhou -
2018 Oral: Semi-Implicit Variational Inference »
Mingzhang Yin · Mingyuan Zhou -
2017 Poster: Deep Latent Dirichlet Allocation with Topic-Layer-Adaptive Stochastic Gradient Riemannian MCMC »
Yulai Cong · Bo Chen · Hongwei Liu · Mingyuan Zhou -
2017 Talk: Deep Latent Dirichlet Allocation with Topic-Layer-Adaptive Stochastic Gradient Riemannian MCMC »
Yulai Cong · Bo Chen · Hongwei Liu · Mingyuan Zhou