Ya Shi Zhang — accepted AI-conference papers

2026

PerturbDiff: Functional Diffusion for Single-Cell Perturbation Modeling

Xinyu Yuan, Xixian Liu, Ya Shi Zhang, Zuobai Zhang, Hongyu Guo, Jian Tang

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 0 Source ↗Scholar ↗

2024

Mind the GAP: Improving Robustness to Subpopulation Shifts with Group-Aware Priors

Tim G. J. Rudner, Ya Shi Zhang, Andrew Gordon Wilson, Julia Kempe

AISTATS 2024poster

Machine learning models often perform poorly under subpopulation shifts in the data distribution. Developing methods that allow machine learning models to better generalize to such shifts is crucial for safe deployment in real-world settings. In this paper, we develop a family of group-aware prior (…

Cited by 8 Source ↗PDF ↗code ↗Scholar ↗