← Search

Zaixi Zhang

20 accepted papers

2026

DAWN: Distributed LLM Multi-Agent Workflow Synthesis

AAAI 2026technical

Large language models (LLMs) have recently empowered multi-agent systems (MAS) to achieve remarkable advances in collaborative reasoning and complex task automation. The effectiveness of these systems fundamentally depends on the design of adaptive communication graphs—the underlying workflows that

Cited by 0SourcePDFScholar
2026

PoseX: AI Defeats Physics-based Methods on Protein Ligand Cross-Docking

ICLR 2026poster

Recently, significant progress has been made in protein-ligand docking, especially in deep learning methods, and some benchmarks were proposed, such as PoseBench and PLINDER. However, these studies typically focus on the self-docking scenario, which is less practical in real-world applications. More…

Cited by 0SourcecodeScholar
2026

Systematic Biosafety Evaluation of DNA Language Models under Jailbreak Attacks

ICLR 2026poster

DNA, encoding genetic instructions for almost all living organisms, fuels groundbreaking advances in genomics and synthetic biology. Recently, DNA Language Models have achieved success in designing synthetic functional DNA sequences, even whole genomes of novel bacteriophage, verified with wet lab e…

Cited by 0SourcecodeScholar
2025

Securing the Language of Life: Inheritable Watermarks from DNA Language Models to Proteins

NeurIPS 2025poster

DNA language models have revolutionized our ability to design and manipulate DNA sequences—the fundamental language of life—with unprecedented precision, enabling transformative applications in therapeutics, synthetic biology, and gene editing. However, this capability also poses significant dual-us…

Cited by 0SourceScholar
2024

Binding-Adaptive Diffusion Models for Structure-Based Drug Design

AAAI 2024technical

Structure-based drug design (SBDD) aims to generate 3D ligand molecules that bind to specific protein targets. Existing 3D deep generative models including diffusion models have shown great promise for SBDD. However, it is complex to capture the essential protein-ligand interactions exactly in 3D sp…

2024

DeltaDock: A Unified Framework for Accurate, Efficient, and Physically Reliable Molecular Docking

NeurIPS 2024poster

Molecular docking, a technique for predicting ligand binding poses, is crucial in structure-based drug design for understanding protein-ligand interactions. Recent advancements in docking methods, particularly those leveraging geometric deep learning (GDL), have demonstrated significant efficiency a…

2024

Generalized Protein Pocket Generation with Prior-Informed Flow Matching

NeurIPS 2024spotlight

Designing ligand-binding proteins, such as enzymes and biosensors, is essential in bioengineering and protein biology. One critical step in this process involves designing protein pockets, the protein interface binding with the ligand. Current approaches to pocket generation often suffer from time-i…

Cited by 2SourcePDFScholar
2024

What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding

ICML 2024poster

Graph Transformers, which incorporate self-attention and positional encoding, have recently emerged as a powerful architecture for various graph learning tasks. Despite their impressive performance, the complex non-convex interactions across layers and the recursive graph structure have made it chal…

Cited by 16SourcePDFScholar
2023

AdaptSSR: Pre-training User Model with Augmentation-Adaptive Self-Supervised Ranking

NeurIPS 2023poster

User modeling, which aims to capture users' characteristics or interests, heavily relies on task-specific labeled data and suffers from the data sparsity issue. Several recent studies tackled this problem by pre-training the user model on massive user behavior sequences with a contrastive learning t…

2023

Backdoor Defense via Deconfounded Representation Learning

CVPR 2023poster

Deep neural networks (DNNs) are recently shown to be vulnerable to backdoor attacks, where attackers embed hidden backdoors in the DNN model by injecting a few poisoned examples into the training dataset. While extensive efforts have been made to detect and remove backdoors from backdoored DNNs, it…

2023

Full-Atom Protein Pocket Design via Iterative Refinement

NeurIPS 2023spotlight

The design of \emph{de novo} functional proteins that bind with specific ligand molecules is crucial in various domains like therapeutics and bio-engineering. One vital yet challenging step is to design the protein pocket, the cavity region of protein where the ligand binds with. Existing methods su…

2023

Knowledge Distillation for High Dimensional Search Index

NeurIPS 2023poster

Lightweight compressed models are prevalent in Approximate Nearest Neighbor Search (ANNS) and Maximum Inner Product Search (MIPS) owing to their superiority of retrieval efficiency in large-scale datasets. However, results given by compressed methods are less accurate due to the curse of dimension a…

Cited by 7SourcePDFScholar
2023

Molecule Generation For Target Protein Binding with Structural Motifs

ICLR 2023poster

Designing ligand molecules that bind to specific protein binding sites is a fundamental problem in structure-based drug design. Although deep generative models and geometric deep learning have made great progress in drug design, existing works either sample in the 2D graph space or fail to generate…

Cited by 56SourcePDFScholar
2023

Untargeted Attack against Federated Recommendation Systems via Poisonous Item Embeddings and the Defense

AAAI 2023technical

Federated recommendation (FedRec) can train personalized recommenders without collecting user data, but the decentralized nature makes it susceptible to poisoning attacks. Most previous studies focus on the targeted attack to promote certain items, while the untargeted attack that aims to degrade th…

2022

Hierarchical Graph Transformer with Adaptive Node Sampling

NeurIPS 2022accept

The Transformer architecture has achieved remarkable success in a number of domains including natural language processing and computer vision. However, when it comes to graph-structured data, transformers have not achieved competitive performance, especially on large graphs. In this paper, we identi…

2022

ProtGNN: Towards Self-Explaining Graph Neural Networks

AAAI 2022technical

Despite the recent progress in Graph Neural Networks (GNNs), it remains challenging to explain the predictions made by GNNs. Existing explanation methods mainly focus on post-hoc explanations where another explanatory model is employed to provide explanations for a trained GNN. The fact that post-h…

2021

GraphMI: Extracting Private Graph Data from Graph Neural Networks

IJCAI 2021poster

As machine learning becomes more widely used for critical applications, the need to study its implications in privacy becomes urgent. Given access to the target model and auxiliary information, model inversion attack aims to infer sensitive features of the training dataset, which leads to great pr…

2021

Motif-based Graph Self-Supervised Learning for Molecular Property Prediction

NeurIPS 2021poster

Predicting molecular properties with data-driven methods has drawn much attention in recent years. Particularly, Graph Neural Networks (GNNs) have demonstrated remarkable success in various molecular generation and prediction tasks. In cases where labeled data is scarce, GNNs can be pre-trained on u…