← Search

Zhewei Wei

28 accepted papers

2026

GDGB: A Benchmark for Generative Dynamic Text-Attributed Graph Learning

ICLR 2026poster

Dynamic Text-Attributed Graphs (DyTAGs), which intricately integrate structural, temporal, and textual attributes, are crucial for modeling complex real-world systems. However, most existing DyTAG datasets exhibit poor textual quality, which severely limits their utility for generative DyTAG tasks r…

Cited by 0SourcecodeScholar
2026

Position: The Turing-Completeness of Real-World Autoregressive Transformers Relies Heavily on Context Management

ICML 2026poster

Many works make the eye-catching claim that Transformers are Turing-complete. However, the literature often conflates two distinct settings: (i) a *fixed Transformer system* setting, in which a fixed autoregressive Transformer is coupled with a fixed context-management method to process inputs of di…

Cited by 0SourceScholar
2026

Revisiting Matrix Sketching in Linear Bandits: Achieving Sublinear Regret via Dyadic Block Sketching

ICLR 2026poster

Linear bandits have become a cornerstone of online learning and sequential decision-making, providing solid theoretical foundations for balancing exploration and exploitation. Within this domain, matrix sketching serves as a critical component for achieving computational efficiency, especially when…

Cited by 0SourceScholar
2026

Robustness in Text-Attributed Graph Learning: Insights, Trade-offs, and New Defenses

ICLR 2026poster

While Graph Neural Networks (GNNs) and Large Language Models (LLMs) are powerful approaches for learning on Text-Attributed Graphs (TAGs), a comprehensive understanding of their robustness remains elusive. Current evaluations are fragmented, failing to systematically investigate the distinct effect…

Cited by 0SourcecodeScholar
2026

TetraGT: Tetrahedral Geometry-Driven Explicit Token Interactions with Graph Transformer for Molecular Representation Learning

ICLR 2026poster

Molecular representations that fully capture geometric parameters such as bond angles and torsion angles are crucial for accurately predicting important molecular properties including enzyme catalytic activity, drug bioactivity, and molecular spectral characteristics, as demonstrated by extensive st…

Cited by 0SourceScholar
2025

Advancing Retrosynthesis with Retrieval-Augmented Graph Generation

AAAI 2025technical

Diffusion-based molecular graph generative models have achieved significant success in template-free, single-step retrosynthesis prediction. However, these models typically generate reactants from scratch, often overlooking the fact that the scaffold of a product molecule typically remains unchanged…

2025

Dimension-Free Adaptive Subgradient Methods with Frequent Directions

ICML 2025poster

In this paper, we investigate the acceleration of adaptive subgradient methods through frequent directions (FD), a widely-used matrix sketching technique. The state-of-the-art regret bound exhibits a _linear_ dependence on the dimensionality $d$, leading to unsatisfactory guarantees for high-dimensi…

Cited by 0SourcePDFScholar
2025

Fast Second-Order Online Kernel Learning Through Incremental Matrix Sketching and Decomposition

IJCAI 2025

Second-order Online Kernel Learning (OKL) has attracted considerable research interest due to its promising predictive performance in streaming environments. However, existing second-order OKL approaches suffer from at least quadratic time complexity with respect to the pre-set budget, rendering the

Cited by 0SourcePDFScholar
2025

Future Link Prediction Without Memory or Aggregation

NeurIPS 2025poster

Future link prediction on temporal graphs is a fundamental task with wide applicability in real-world dynamic systems. These scenarios often involve both recurring (seen) and novel (unseen) interactions, requiring models to generalize effectively across both types of edges. However, existing methods…

Cited by 0SourcecodeScholar
2025

LLM-Based Multi-Agent Systems are Scalable Graph Generative Models

ACL 2025finding

The structural properties of naturally arising social graphs are extensively studied to understand their evolution. Prior approaches for modeling network dynamics typically rely on rule-based models, which lack realism and generalizability, or deep learning-based models, which require large-scale tr…

2025

Position: Spectral GNNs Rely Less on Graph Fourier Basis than Conceived

ICML 2025poster

Spectral graph learning builds upon two foundations: Graph Fourier basis as its theoretical cornerstone,with polynomial approximation to enable practical implementation. While this framework has led to numerous successful designs, we argue that its effectiveness might stem from mechanisms different…

Cited by 0SourcePDFScholar
2025

Scalable and Certifiable Graph Unlearning: Overcoming the Approximation Error Barrier

ICLR 2025spotlight

Graph unlearning has emerged as a pivotal research area for ensuring privacy protection, given the widespread adoption of Graph Neural Networks (GNNs) in applications involving sensitive user data. Among existing studies, certified graph unlearning is distinguished by providing robust privacy guaran…

2025

TGB-Seq Benchmark: Challenging Temporal GNNs with Complex Sequential Dynamics

ICLR 2025poster

Future link prediction is a fundamental challenge in various real-world dynamic systems. To address this, numerous temporal graph neural networks (temporal GNNs) and benchmark datasets have been developed. However, these datasets often feature excessive repeated edges and lack complex sequential dyn…

2025

Towards Effective and Efficient Continual Pre-training of Large Language Models

ACL 2025long

Continual pre-training (CPT) has been an important approach for adapting language models to specific domains or tasks. In this paper, we comprehensively study its key designs to balance the new abilities while retaining the original abilities, and present an effective CPT method that can greatly imp…

2024

EquiPocket: an E(3)-Equivariant Geometric Graph Neural Network for Ligand Binding Site Prediction

ICML 2024oral

Predicting the binding sites of target proteins plays a fundamental role in drug discovery. Most existing deep-learning methods consider a protein as a 3D image by spatially clustering its atoms into voxels and then feed the voxelized protein into a 3D CNN for prediction. However, the CNN-based meth…

2024

Intruding with Words: Towards Understanding Graph Injection Attacks at the Text Level

NeurIPS 2024poster

Graph Neural Networks (GNNs) excel across various applications but remain vulnerable to adversarial attacks, particularly Graph Injection Attacks (GIAs), which inject malicious nodes into the original graph and pose realistic threats. Text-attributed graphs (TAGs), where nodes are associated with te…

2024

PolyGCL: GRAPH CONTRASTIVE LEARNING via Learnable Spectral Polynomial Filters

ICLR 2024spotlight

Recently, Graph Contrastive Learning (GCL) has achieved significantly superior performance in self-supervised graph representation learning. However, the existing GCL technique has inherent smooth characteristics because of its low-pass GNN encoder and objective based on homophily assumption, which…

Cited by 23SourcePDFScholar
2024

S-MolSearch: 3D Semi-supervised Contrastive Learning for Bioactive Molecule Search

NeurIPS 2024poster

Virtual Screening is an essential technique in the early phases of drug discovery, aimed at identifying promising drug candidates from vast molecular libraries. Recently, ligand-based virtual screening has garnered significant attention due to its efficacy in conducting extensive database screening…

Cited by 1SourcePDFScholar
2024

SRAP-Agent: Simulating and Optimizing Scarce Resource Allocation Policy with LLM-based Agent

EMNLP 2024finding

Public scarce resource allocation plays a crucial role in economics as it directly influences the efficiency and equity in society. Traditional studies including theoretical model-based, empirical study-based and simulation-based methods encounter limitations due to the idealized assumption of compl…

2023

Preformer: Predictive Transformer with Multi-Scale Segment-Wise Correlations for Long-Term Time Series Forecasting

ICASSP 2023accepted

In long-term time series forecasting, most Transformer-based methods adopt the standard point-wise attention mechanism, which not only has high complexity but also cannot explicitly capture the predictive dependencies from contexts since the corresponding key and value are transformed from the same…

Cited by 0SourceScholar
2023

Uni-Mol: A Universal 3D Molecular Representation Learning Framework

ICLR 2023poster

Molecular representation learning (MRL) has gained tremendous attention due to its critical role in learning from limited supervised data for applications like drug design. In most MRL methods, molecules are treated as 1D sequential tokens or 2D topology graphs, limiting their ability to incorporate…

2022

Convolutional Neural Networks on Graphs with Chebyshev Approximation, Revisited

NeurIPS 2022accept

Designing spectral convolutional networks is a challenging problem in graph learning. ChebNet, one of the early attempts, approximates the spectral graph convolutions using Chebyshev polynomials. GCN simplifies ChebNet by utilizing only the first two Chebyshev polynomials while still outperforming i…

2022

EvenNet: Ignoring Odd-Hop Neighbors Improves Robustness of Graph Neural Networks

NeurIPS 2022accept

Graph Neural Networks (GNNs) have received extensive research attention for their promising performance in graph machine learning. Despite their extraordinary predictive accuracy, existing approaches, such as GCN and GPRGNN, are not robust in the face of homophily changes on test graphs, rendering t…

2021

BernNet: Learning Arbitrary Graph Spectral Filters via Bernstein Approximation

NeurIPS 2021poster

Many representative graph neural networks, $e.g.$, GPR-GNN and ChebNet, approximate graph convolutions with graph spectral filters. However, existing work either applies predefined filter weights or learns them without necessary constraints, which may lead to oversimplified or ill-posed filters. To…

2021

Graph Neural Networks Inspired by Classical Iterative Algorithms

ICML 2021oral

Despite the recent success of graph neural networks (GNN), common architectures often exhibit significant limitations, including sensitivity to oversmoothing, long-range dependencies, and spurious edges, e.g., as can occur as a result of graph heterophily or adversarial attacks. To at least partiall…

2020

Scalable Graph Neural Networks via Bidirectional Propagation

NeurIPS 2020poster

Graph Neural Networks (GNN) are an emerging field for learning on non-Euclidean data. Recently, there has been increased interest in designing GNN that scales to large graphs. Most existing methods use "graph sampling" or "layer-wise sampling" techniques to reduce training time; However, these metho…

2020

Simple and Deep Graph Convolutional Networks

ICML 2020poster

Graph convolutional networks (GCNs) are a powerful deep learning approach for graph-structured data. Recently, GCNs and subsequent variants have shown superior performance in various application areas on real-world datasets. Despite their success, most of the current GCN models are shallow, due to t…