Search Results for author: Yujun Yan

Found 9 papers, 6 papers with code

Interpretable Sparsification of Brain Graphs: Better Practices and Effective Designs for Graph Neural Networks

1 code implementation26 Jun 2023 Gaotang Li, Marlena Duda, Xiang Zhang, Danai Koutra, Yujun Yan

Based on these insights, we propose a new model, Interpretable Graph Sparsification (IGS), which enhances graph classification performance by up to 5. 1% with 55. 0% fewer edges.

Graph Classification

Size Generalization of Graph Neural Networks on Biological Data: Insights and Practices from the Spectral Perspective

no code implementations24 May 2023 Gaotang Li, Danai Koutra, Yujun Yan

Our empirical results reveal that our proposed size-insensitive attention strategy substantially enhances graph classification performance on large test graphs, which are 2-10 times larger than the training graphs, resulting in an improvement in F1 scores by up to 8%.

Graph Classification

Augmentations in Graph Contrastive Learning: Current Methodological Flaws & Towards Better Practices

no code implementations5 Nov 2021 Puja Trivedi, Ekdeep Singh Lubana, Yujun Yan, Yaoqing Yang, Danai Koutra

Unsupervised graph representation learning is critical to a wide range of applications where labels may be scarce or expensive to procure.

Contrastive Learning Data Augmentation +5

Improving Semi-supervised Federated Learning by Reducing the Gradient Diversity of Models

1 code implementation26 Aug 2020 Zhengming Zhang, Yaoqing Yang, Zhewei Yao, Yujun Yan, Joseph E. Gonzalez, Michael W. Mahoney

Replacing BN with the recently-proposed Group Normalization (GN) can reduce gradient diversity and improve test accuracy.

Federated Learning

Beyond Homophily in Graph Neural Networks: Current Limitations and Effective Designs

4 code implementations NeurIPS 2020 Jiong Zhu, Yujun Yan, Lingxiao Zhao, Mark Heimann, Leman Akoglu, Danai Koutra

We investigate the representation power of graph neural networks in the semi-supervised node classification task under heterophily or low homophily, i. e., in networks where connected nodes may have different class labels and dissimilar features.

Node Classification on Non-Homophilic (Heterophilic) Graphs

Neural Execution Engines: Learning to Execute Subroutines

1 code implementation NeurIPS 2020 Yujun Yan, Kevin Swersky, Danai Koutra, Parthasarathy Ranganathan, Milad Hashemi

A significant effort has been made to train neural networks that replicate algorithmic reasoning, but they often fail to learn the abstract concepts underlying these algorithms.

Learning to Execute

NEURAL EXECUTION ENGINES

no code implementations ICLR 2020 Yujun Yan, Kevin Swersky, Danai Koutra, Parthasarathy Ranganathan, Milad Hashemi

Turing complete computation and reasoning are often regarded as necessary pre- cursors to general intelligence.

Cannot find the paper you are looking for? You can Submit a new open access paper.