← Search

Bohang Zhang

17 accepted papers

2025

Beyond Atoms: Enhancing Molecular Pretrained Representations with 3D Space Modeling

ICML 2025poster

Molecular pretrained representations (MPR) has emerged as a powerful approach for addressing the challenge of limited supervised data in applications such as drug discovery and material design. While early MPR methods relied on 1D sequences and 2D graphs, recent advancements have incorporated 3D co…

Cited by 1SourcePDFScholar
2025

From Sequence to Structure: Uncovering Substructure Reasoning in Transformers

NeurIPS 2025poster

Recent studies suggest that large language models (LLMs) possess the capability to solve graph reasoning tasks. Notably, even when graph structures are embedded within textual descriptions, LLMs can still effectively answer related questions. This raises a fundamental question: How can a decoder-onl…

Cited by 0SourceScholar
2025

Homomorphism Expressivity of Spectral Invariant Graph Neural Networks

ICLR 2025oral

Graph spectra are an important class of structural features on graphs that have shown promising results in enhancing Graph Neural Networks (GNNs). Despite their widespread practical use, the theoretical understanding of the power of spectral invariants --- particularly their contribution to GNNs ---…

Cited by 0SourcePDFScholar
2025

How Do Large Language Models Understand Graph Patterns? A Benchmark for Graph Pattern Comprehension

ICLR 2025poster

Benchmarking the capabilities and limitations of large language models (LLMs) in graph-related tasks is becoming an increasingly popular and crucial area of research. Recent studies have shown that LLMs exhibit a preliminary ability to understand graph structures and node features. However, the pote…

Cited by 4SourcePDFScholar
2024

Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness

ICLR 2024oral

Designing expressive Graph Neural Networks (GNNs) is a fundamental topic in the graph learning community. So far, GNN expressiveness has been primarily assessed via the Weisfeiler-Lehman (WL) hierarchy. However, such an expressivity measure has notable limitations: it is inherently coarse, qualitati…

2024

Can Graph Learning Improve Planning in LLM-based Agents?

NeurIPS 2024poster

Task planning in language agents is emerging as an important research topic alongside the development of large language models (LLMs). It aims to break down complex user requests in natural language into solvable sub-tasks, thereby fulfilling the original requests. In this context, the sub-tasks can…

2024

Do Efficient Transformers Really Save Computation?

ICML 2024poster

As transformer-based language models are trained on increasingly large datasets and with vast numbers of parameters, finding more efficient alternatives to the standard Transformer has become very valuable. While many efficient Transformers and Transformer alternatives have been proposed, none provi…

Cited by 16SourcePDFScholar
2024

On the Expressive Power of Spectral Invariant Graph Neural Networks

ICML 2024poster

Incorporating spectral information to enhance Graph Neural Networks (GNNs) has shown promising results but raises a fundamental challenge due to the inherent ambiguity of eigenvectors. Various architectures have been proposed to address this ambiguity, referred to as spectral invariant architectures…

2023

A Complete Expressiveness Hierarchy for Subgraph GNNs via Subgraph Weisfeiler-Lehman Tests

ICML 2023poster

Recently, subgraph GNNs have emerged as an important direction for developing expressive graph neural networks (GNNs). While numerous architectures have been proposed, so far there is still a limited understanding of how various design paradigms differ in terms of expressive power, nor is it clear w…

2023

Finding Generalization Measures by Contrasting Signal and Noise

ICML 2023poster

Generalization is one of the most fundamental challenges in deep learning, aiming to predict model performances on unseen data. Empirically, such predictions usually rely on a validation set, while recent works showed that an unlabeled validation set also works. Without validation sets, it is extrem…

2023

Rethinking the Expressive Power of GNNs via Graph Biconnectivity

ICLR 2023top-5%

Designing expressive Graph Neural Networks (GNNs) is a central topic in learning graph-structured data. While numerous approaches have been proposed to improve GNNs with respect to the Weisfeiler-Lehman (WL) test, for most of them, there is still a lack of deep understanding of what additional power…

2023

Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspective

NeurIPS 2023oral

Recent studies have discovered that Chain-of-Thought prompting (CoT) can dramatically improve the performance of Large Language Models (LLMs), particularly when dealing with complex tasks involving mathematics or reasoning. Despite the enormous empirical success, the underlying mechanisms behind CoT…

Cited by 247SourcePDFScholar
2022

Rethinking Lipschitz Neural Networks and Certified Robustness: A Boolean Function Perspective

NeurIPS 2022accept

Designing neural networks with bounded Lipschitz constant is a promising way to obtain certifiably robust classifiers against adversarial examples. However, the relevant progress for the important $\ell_\infty$ perturbation setting is rather limited, and a principled understanding of how to design e…

2021

Non-convex Distributionally Robust Optimization: Non-asymptotic Analysis

NeurIPS 2021poster

Distributionally robust optimization (DRO) is a widely-used approach to learn models that are robust against distribution shift. Compared with the standard optimization setting, the objective function in DRO is more difficult to optimize, and most of the existing theoretical results make strong assu…

Cited by 60SourcePDFScholar
2021

Towards Certifying L-infinity Robustness using Neural Networks with L-inf-dist Neurons

ICML 2021spotlight

It is well-known that standard neural networks, even with a high classification accuracy, are vulnerable to small $\ell_\infty$-norm bounded adversarial perturbations. Although many attempts have been made, most previous works either can only provide empirical verification of the defense to a partic…

2020

Improved Analysis of Clipping Algorithms for Non-convex Optimization

NeurIPS 2020poster

Gradient clipping is commonly used in training deep neural networks partly due to its practicability in relieving the exploding gradient problem. Recently, \citet{zhang2019gradient} show that clipped (stochastic) Gradient Descent (GD) converges faster than vanilla GD via introducing a new assumpt…