← Search

Yu Guang Wang

11 accepted papers

2026

How Out-of-Distribution Detection Learning Theory Enhances Transformer: Learnability and Reliability

ICML 2026poster

Transformers excel in natural language processing and computer vision tasks. However, they still face challenges in generalizing to Out-of-Distribution (OOD) datasets, i.e. data whose distribution differs from that seen during training. OOD detection aims to distinguish outliers while preserving in-…

Cited by 0SourceScholar
2026

Linear Causal Representation Learning by Topological Ordering, Pruning, and Disentanglement

ICML 2026spotlight

Causal representation learning (CRL) has garnered increasing interests from the causal inference and artificial intelligence community, due to its capability of disentangling potentially complex data-generating mechanism into causally interpretable latent features, by leveraging the heterogeneity of…

Cited by 0SourceScholar
2025

How Particle System Theory Enhances Hypergraph Message Passing

NeurIPS 2025poster

Hypergraphs effectively model higher-order relationships in natural phenomena, capturing complex interactions beyond pairwise connections. We introduce a novel hypergraph message passing framework inspired by interacting particle systems, where hyperedges act as fields inducing shared node dynamics.…

Cited by 0SourcecodeScholar
2024

How Universal Polynomial Bases Enhance Spectral Graph Neural Networks: Heterophily, Over-smoothing, and Over-squashing

ICML 2024poster

Spectral Graph Neural Networks (GNNs), alternatively known as *graph filters*, have gained increasing prevalence for heterophily graphs. Optimal graph filters rely on Laplacian eigendecomposition for Fourier transform. In an attempt to avert prohibitive computations, numerous polynomial filters have…

2023

ACMP: Allen-Cahn Message Passing with Attractive and Repulsive Forces for Graph Neural Networks

ICLR 2023top-25%

Neural message passing is a basic feature extraction unit for graph-structured data considering neighboring node features in network propagation from one layer to the next. We model such process by an interacting particle system with attractive and repulsive forces and the Allen-Cahn force arising i…

2023

EqMotion: Equivariant Multi-Agent Motion Prediction With Invariant Interaction Reasoning

CVPR 2023poster

Learning to predict agent motions with relationship reasoning is important for many applications. In motion prediction tasks, maintaining motion equivariance under Euclidean geometric transformations and invariance of agent interaction is a critical and fundamental principle. However, such equivaria…

2023

Graph Denoising Diffusion for Inverse Protein Folding

NeurIPS 2023poster

Inverse protein folding is challenging due to its inherent one-to-many mapping characteristic, where numerous possible amino acid sequences can fold into a single, identical protein backbone. This task involves not only identifying viable sequences but also representing the sheer diversity of potent…

2023

How Powerful are Shallow Neural Networks with Bandlimited Random Weights?

ICML 2023poster

We investigate the expressive power of depth-2 bandlimited random neural networks. A random net is a neural network where the hidden layer parameters are frozen with random assignment, and only the output layer parameters are trained by loss minimization. Using random weights for a hidden layer is a…

Cited by 10SourcePDFScholar
2021

Weisfeiler and Lehman Go Cellular: CW Networks

NeurIPS 2021poster

Graph Neural Networks (GNNs) are limited in their expressive power, struggle with long-range interactions and lack a principled way to model higher-order structures. These problems can be attributed to the strong coupling between the computational graph and the input graph structure. The recently pr…

2020

Path Integral Based Convolution and Pooling for Graph Neural Networks

NeurIPS 2020poster

Graph neural networks (GNNs) extends the functionality of traditional neural networks to graph-structured data. Similar to CNNs, an optimized design of graph convolution and pooling is key to success. Borrowing ideas from physics, we propose a path integral based graph neural networks (PAN) for clas…