← Search

Zuobai Zhang

18 accepted papers

2026

Fast Proteome-Scale Protein Interaction Retrieval via Residue-Level Factorization

ICLR 2026poster

Protein-protein interactions (PPIs) are mediated at the residue level. Most sequence-based PPI models consider residue-residue interactions across two proteins, which can yield accurate interaction scores but are too slow to scale. At proteome scale, identifying candidate PPIs requires evaluating ne…

Cited by 0SourcecodeScholar
2026

La-Proteina: Atomistic Protein Generation via Partially Latent Flow Matching

ICLR 2026poster

Recently, many generative models for de novo protein structure design have emerged. Yet, only few tackle the difficult task of directly generating fully atomistic structures jointly with the underlying amino acid sequence. This is challenging, for instance, because the model must reason over side ch…

Cited by 0SourcecodeScholar
2026

PerturbDiff: Functional Diffusion for Single-Cell Perturbation Modeling

ICML 2026poster

Building _Virtual Cells_ that can accurately simulate cellular responses to perturbations is a long-standing goal in systems biology. A fundamental challenge is that high-throughput single-cell sequencing is destructive: the same cell cannot be observed both before and after a perturbation. Thus, pe…

Cited by 0SourceScholar
2026

Scaling Atomistic Protein Binder Design with Generative Pretraining and Test-Time Compute

ICLR 2026oral

Protein interaction modeling is central to protein design, which has been transformed by machine learning with broad applications in drug discovery and beyond. In this landscape, structure-based de novo binder design is most often cast as either conditional generative modeling or sequence optimizati…

Cited by 0SourcecodeScholar
2026

Towards All-Atom Foundation Models for Biomolecular Binding Affinity Prediction

ICLR 2026poster

Biomolecular interactions play a critical role in biological processes. While recent breakthroughs like AlphaFold 3 have enabled accurate modeling of biomolecular complex structures, predicting binding affinity remains challenging mainly due to limited high-quality data. Recent methods are often spe…

Cited by 0SourcecodeScholar
2025

Proteina: Scaling Flow-based Protein Structure Generative Models

ICLR 2025oral

Recently, diffusion- and flow-based generative models of protein structures have emerged as a powerful tool for de novo protein design. Here, we develop *Proteina*, a new large-scale flow-based protein backbone generator that utilizes hierarchical fold class labels for conditioning and relies on a t…

2024

Cell ontology guided transcriptome foundation model

NeurIPS 2024spotlight

Transcriptome foundation models (TFMs) hold great promises of deciphering the transcriptomic language that dictate diverse cell functions by self-supervised learning on large-scale single-cell gene expression data, and ultimately unraveling the complex mechanisms of human diseases. However, current…

2024

Evaluating Representation Learning on the Protein Structure Universe

ICLR 2024poster

We introduce ProteinWorkshop, a comprehensive benchmark suite for representation learning on protein structures with Geometric Graph Neural Networks. We consider large-scale pre-training and downstream tasks on both experimental and predicted structures to enable the systematic evaluation of the qua…

2024

Multi-Scale Representation Learning for Protein Fitness Prediction

NeurIPS 2024poster

Designing novel functional proteins crucially depends on accurately modeling their fitness landscape. Given the limited availability of functional annotations from wet-lab experiments, previous methods have primarily relied on self-supervised models trained on vast, unlabeled protein sequence or str…

2024

Str2Str: A Score-based Framework for Zero-shot Protein Conformation Sampling

ICLR 2024poster

The dynamic nature of proteins is crucial for determining their biological functions and properties, for which Monte Carlo (MC) and molecular dynamics (MD) simulations stand as predominant tools to study such phenomena. By utilizing empirically derived force fields, MC or MD simulations explore the…

2023

DiffPack: A Torsional Diffusion Model for Autoregressive Protein Side-Chain Packing

NeurIPS 2023poster

Proteins play a critical role in carrying out biological functions, and their 3D structures are essential in determining their functions. Accurately predicting the conformation of protein side-chains given their backbones is important for applications in protein structure prediction, design and pro…

2023

FusionRetro: Molecule Representation Fusion via In-Context Learning for Retrosynthetic Planning

ICML 2023poster

Retrosynthetic planning aims to devise a complete multi-step synthetic route from starting materials to a target molecule. Current strategies use a decoupled approach of single-step retrosynthesis models and search algorithms, taking only the product as the input to predict the reactants for each pl…

2023

Pre-Training Protein Encoder via Siamese Sequence-Structure Diffusion Trajectory Prediction

NeurIPS 2023spotlight

Self-supervised pre-training methods on proteins have recently gained attention, with most approaches focusing on either protein sequences or structures, neglecting the exploration of their joint distribution, which is crucial for a comprehensive understanding of protein functions by integrating co-…

2023

Protein Representation Learning by Geometric Structure Pretraining

ICLR 2023poster

Learning effective protein representations is critical in a variety of tasks in biology such as predicting protein function or structure. Existing approaches usually pretrain protein language models on a large number of unlabeled amino acid sequences and then finetune the models with some labeled da…

2022

Neural-Symbolic Models for Logical Queries on Knowledge Graphs

ICML 2022spotlight

Answering complex first-order logic (FOL) queries on knowledge graphs is a fundamental task for multi-hop reasoning. Traditional symbolic methods traverse a complete knowledge graph to extract the answers, which provides good interpretation for each step. Recent neural methods learn geometric embedd…

2022

PEER: A Comprehensive and Multi-Task Benchmark for Protein Sequence Understanding

NeurIPS 2022accept

We are now witnessing significant progress of deep learning methods in a variety of tasks (or datasets) of proteins. However, there is a lack of a standard benchmark to evaluate the performance of different methods, which hinders the progress of deep learning in this field. In this paper, we propose…

2022

Structured Multi-task Learning for Molecular Property Prediction

AISTATS 2022poster

Multi-task learning for molecular property prediction is becoming increasingly important in drug discovery. However, in contrast to other domains, the performance of multi-task learning in drug discovery is still not satisfying as the number of labeled data for each task is too limited, which calls…

2021

Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Prediction

NeurIPS 2021poster

Link prediction is a very fundamental task on graphs. Inspired by traditional path-based methods, in this paper we propose a general and flexible representation learning framework based on paths for link prediction. Specifically, we define the representation of a pair of nodes as the generalized sum…