← Search

Yangtian Zhang

7 accepted papers

2026

Learning Permutation Distributions via Reflected Diffusion on Ranks

ICML 2026poster

The finite symmetric group $S_n$ provides a natural domain for permutations, yet learning probability distributions on $S_n$ is challenging due to its factorially growing size and discrete, non-Euclidean structure. Recent permutation diffusion methods define forward noising via shuffle-based random …

Cited by 0SourceScholar
2026

STRIDE: Post-Training LLMs to Reason and Refine Bio-Sequences via Edit Trajectories

ICML 2026poster

Discrete biological sequence optimization demands iterative refinement while satisfying strict syntactic constraints. Diffusion-based approaches provide strong progressive refinement but are not naturally aligned with discrete, grammar-constrained edit operations, whereas autoregressive LLMs readily…

Cited by 0SourceScholar
2026

Variational Learning for Insertion-based Generation

ICML 2026spotlight

Non-monotonic sequence generation methods, such as masked diffusion models, provide a flexible alternative to left-to-right autoregressive modeling by allowing tokens to be generated in non-fixed and prescribed orders. Despite their practical advantages, most existing non-monotonic models are order-…

Cited by 0SourceScholar
2025

Non-Markovian Discrete Diffusion with Causal Language Models

NeurIPS 2025poster

Discrete diffusion models offer a flexible, controllable approach to structured sequence generation, yet they still lag behind causal language models in expressive power. A key limitation lies in their reliance on the Markovian assumption, which restricts each step to condition only on the current s…

Cited by 0SourceScholar
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

E3Bind: An End-to-End Equivariant Network for Protein-Ligand Docking

ICLR 2023poster

In silico prediction of the ligand binding pose to a given protein target is a crucial but challenging task in drug discovery. This work focuses on blind flexible self-docking, where we aim to predict the positions, orientations and conformations of docked molecules. Traditional physics-based method…

Cited by 44SourcePDFScholar
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…