activity
20192023
most citedLabeling Trick: A Theory of Using Graph Neural Networks for Multi-Node Representation Learning

59 citations · 387 across the 30 of their papers we have counts for

collaborators

30 papers

cs.LG2023

Distance-Restricted Folklore Weisfeiler-Leman GNNs with Provable Cycle Counting Power

Junru Zhou, Jiarui Feng, Xiyuan Wang +1

The ability of graph neural networks (GNNs) to count certain graph substructures, especially cycles, is important for the success of GNNs on a wide range of tasks. It has been rece…

cs.LG2023★ 1 cited

VQGraph: Rethinking Graph Representation Space for Bridging GNNs and MLPs

Ling Yang, Ye Tian, Minkai Xu +7

GNN-to-MLP distillation aims to utilize knowledge distillation (KD) to learn computationally-efficient multi-layer perceptron (student MLP) on graph data by mimicking the output re…

cs.CL2023★ 15 cited

Large Language Models are In-Context Semantic Reasoners rather than Symbolic Reasoners

Xiaojuan Tang, Zilong Zheng, Jiaqi Li +4

The emergent few-shot reasoning capabilities of Large Language Models (LLMs) have excited the natural language and machine learning community over recent years. Despite of numerous…

cs.LG2023★ 1 cited

From Relational Pooling to Subgraph GNNs: A Universal Framework for More Expressive Graph Neural Networks

Cai Zhou, Xiyuan Wang, Muhan Zhang

Relational pooling is a framework for building more expressive and permutation-invariant graph neural networks. However, there is limited understanding of the exact enhancement in…

cs.CL2023★ 3 cited

Code Prompting: a Neural Symbolic Method for Complex Reasoning in Large Language Models

Yi Hu, Haotong Yang, Zhouchen Lin +1

Large language models (LLMs) have scaled up to unlock a wide range of complex reasoning tasks with the aid of various prompting methods. However, current prompting methods generate…

cs.LG2023

Improving Graph Neural Networks on Multi-node Tasks with the Labeling Trick

Xiyuan Wang, Pan Li, Muhan Zhang

In this paper, we study using graph neural networks (GNNs) for \textit{multi-node representation learning}, where a representation for a set of more than one node (such as a link)…

cs.LG2023★ 1 cited

An Empirical Study of Realized GNN Expressiveness

Yanbo Wang, Muhan Zhang

Research on the theoretical expressiveness of Graph Neural Networks (GNNs) has developed rapidly, and many methods have been proposed to enhance the expressiveness. However, most m…

cs.LG2023

An Efficient Subgraph GNN with Provable Substructure Counting Power

Zuoyu Yan, Junru Zhou, Liangcai Gao +2

We investigate the enhancement of graph neural networks' (GNNs) representation power through their ability in substructure counting. Recent advances have seen the adoption of subgr…

cs.LG2023★ 11 cited

SUREL+: Moving from Walks to Sets for Scalable Subgraph-based Graph Representation Learning

Haoteng Yin, Muhan Zhang, Jianguo Wang +1

Subgraph-based graph representation learning (SGRL) has recently emerged as a powerful tool in many prediction tasks on graphs due to its advantages in model expressiveness and gen…

cs.LG2023★ 3 cited

Is Distance Matrix Enough for Geometric Deep Learning?

Zian Li, Xiyuan Wang, Yinan Huang +1

Graph Neural Networks (GNNs) are often used for tasks involving the 3D geometry of a given graph, such as molecular dynamics simulation. While incorporating Euclidean distance into…

cs.LG2023★ 12 cited

Neural Common Neighbor with Completion for Link Prediction

Xiyuan Wang, Haotong Yang, Muhan Zhang

In this work, we propose a novel link prediction model and further boost it by studying graph incompleteness. First, we introduce MPNN-then-SF, an innovative architecture leveragin…

stat.ML2022

1st ICLR International Workshop on Privacy, Accountability, Interpretability, Robustness, Reasoning on Structured Data (PAIR^2Struct)

Hao Wang, Wanyu Lin, Hao He +3

Recent years have seen advances on principles and guidance relating to accountable and ethical use of artificial intelligence (AI) spring up around the globe. Specifically, Data Pr…

cs.AI2022★ 1 cited

RulE: Knowledge Graph Reasoning with Rule Embedding

Xiaojuan Tang, Song-Chun Zhu, Yitao Liang +1

Knowledge graph (KG) reasoning is an important problem for knowledge graphs. In this paper, we propose a novel and principled framework called \textbf{RulE} (stands for {Rul}e {E}m…

cs.LG2022★ 3 cited

Boosting the Cycle Counting Power of Graph Neural Networks with I-GNNs

Yinan Huang, Xingang Peng, Jianzhu Ma +1

Message Passing Neural Networks (MPNNs) are a widely used class of Graph Neural Networks (GNNs). The limited representational power of MPNNs inspires the study of provably powerful…

cs.LG2022★ 9 cited

Geodesic Graph Neural Network for Efficient Graph Representation Learning

Lecheng Kong, Yixin Chen, Muhan Zhang

Graph Neural Networks (GNNs) have recently been applied to graph learning tasks and achieved state-of-the-art (SOTA) results. However, many competitive methods run GNNs multiple ti…

cs.AI2022★ 4 cited

Rethinking Knowledge Graph Evaluation Under the Open-World Assumption

Haotong Yang, Zhouchen Lin, Muhan Zhang

Most knowledge graphs (KGs) are incomplete, which motivates one important research topic on automatically complementing knowledge graphs. However, evaluation of knowledge graph com…

cs.LG2022★ 2 cited

Graph Neural Network with Local Frame for Molecular Potential Energy Surface

Xiyuan Wang, Muhan Zhang

Modeling molecular potential energy surface is of pivotal importance in science. Graph Neural Networks have shown great success in this field. However, their message passing scheme…

cs.LG2022

Two-Dimensional Weisfeiler-Lehman Graph Neural Networks for Link Prediction

Yang Hu, Xiyuan Wang, Zhouchen Lin +2

Link prediction is one important application of graph neural networks (GNNs). Most existing GNNs for link prediction are based on one-dimensional Weisfeiler-Lehman (1-WL) test. 1-W…

cs.LG2022★ 36 cited

How Powerful are Spectral Graph Neural Networks

Xiyuan Wang, Muhan Zhang

Spectral Graph Neural Network is a kind of Graph Neural Network (GNN) based on graph signal filters. Some models able to learn arbitrary spectral filters have emerged recently. How…

cs.LG2022★ 37 cited

How Powerful are K-hop Message Passing Graph Neural Networks

Jiarui Feng, Yixin Chen, Fuhai Li +2

The most popular design paradigm for Graph Neural Networks (GNNs) is 1-hop message passing -- aggregating information from 1-hop neighbors repeatedly. However, the expressive power…

cs.LG2022★ 22 cited

Equivariant and Stable Positional Encoding for More Powerful Graph Neural Networks

Haorui Wang, Haoteng Yin, Muhan Zhang +1

Graph neural networks (GNN) have shown great advantages in many graph-based learning tasks but often fail to predict accurately for a task-based on sets of nodes such as link/motif…

cs.LG2022★ 29 cited

Algorithm and System Co-design for Efficient Subgraph-based Graph Representation Learning

Haoteng Yin, Muhan Zhang, Yanbang Wang +2

Subgraph-based graph representation learning (SGRL) has been recently proposed to deal with some fundamental challenges encountered by canonical graph neural networks (GNNs), and h…

cs.LG2022★ 54 cited

Decoupling the Depth and Scope of Graph Neural Networks

Hanqing Zeng, Muhan Zhang, Yinglong Xia +6

State-of-the-art Graph Neural Networks (GNNs) have limited scalability with respect to the graph and model sizes. On large graphs, increasing the model depth often means exponentia…

cs.LG2021

Efficient Dynamic Graph Representation Learning at Scale

Xinshi Chen, Yan Zhu, Haowen Xu +4

Dynamic graphs with ordered sequences of events between nodes are prevalent in real-world industrial applications such as e-commerce and social platforms. However, representation l…