← Search

Keyulu Xu

9 accepted papers

2021

GraphNorm: A Principled Approach to Accelerating Graph Neural Network Training

ICML 2021spotlight

Normalization is known to help the optimization of deep neural networks. Curiously, different architectures require specialized normalization methods. In this paper, we study what normalization is effective for Graph Neural Networks (GNNs). First, we adapt and evaluate the existing methods from othe…

Cited by 230SourcePDFScholar
2021

How Neural Networks Extrapolate: From Feedforward to Graph Neural Networks

ICLR 2021oral

We study how neural networks trained by gradient descent extrapolate, i.e., what they learn outside the support of the training distribution. Previous works report mixed empirical results when extrapolating with neural networks: while feedforward neural networks, a.k.a. multilayer perceptrons (MLPs…

2021

How does a Neural Network's Architecture Impact its Robustness to Noisy Labels?

NeurIPS 2021poster

Noisy labels are inevitable in large real-world datasets. In this work, we explore an area understudied by previous works --- how the network's architecture impacts its robustness to noisy labels. We provide a formal framework connecting the robustness of a network to the alignments between its arch…

Cited by 27SourcePDFScholar
2021

Information Obfuscation of Graph Neural Networks

ICML 2021spotlight

While the advent of Graph Neural Networks (GNNs) has greatly improved node and graph representation learning in many applications, the neighborhood aggregation scheme exposes additional vulnerabilities to adversaries seeking to extract node-level information about sensitive attributes. In this paper…

2021

Optimization of Graph Neural Networks: Implicit Acceleration by Skip Connections and More Depth

ICML 2021spotlight

Graph Neural Networks (GNNs) have been studied through the lens of expressive power and generalization. However, their optimization properties are less well understood. We take the first step towards analyzing GNN training by studying the gradient dynamics of GNNs. First, we analyze linearized GNNs…

Cited by 107SourcePDFScholar
2020

What Can Neural Networks Reason About?

ICLR 2020spotlight

Neural networks have succeeded in many reasoning tasks. Empirically, these tasks require specialized network structures, e.g., Graph Neural Networks (GNNs) perform well on many such tasks, but less structured networks fail. Theoretically, there is limited understanding of why and when a network stru…

Cited by 325SourcecodeScholar
2019

Graph Neural Tangent Kernel: Fusing Graph Neural Networks with Graph Kernels

NeurIPS 2019poster

While graph kernels (GKs) are easy to train and enjoy provable theoretical guarantees, their practical performances are limited by their expressive power, as the kernel function often depends on hand-crafted combinatorial features of graphs. Compared to graph kernels, graph neural networks (GNNs) us…

2018

Distributional Adversarial Networks

ICLR 2018workshop

In most current formulations of adversarial training, the discriminators can be expressed as single-input operators, that is, the mapping they define is separable over observations. In this work, we argue that this property might help explain the infamous mode collapse phenomenon in adversarially-tr…

Cited by 32SourcecodeScholar
2018

Representation Learning on Graphs with Jumping Knowledge Networks

ICML 2018oral

Recent deep learning approaches for representation learning on graphs follow a neighborhood aggregation procedure. We analyze some important properties of these models, and propose a strategy to overcome those. In particular, the range of "neighboring" nodes that a node’s representation draws from s…

Cited by 2591SourcePDFScholar