About Me

I’m a fourth year Ph.D. student in MIT EECS, advised by Caroline Uhler. My research focuses on establishing statistical and algorithmic foundations for decision-making within systems created by underlying causal rules. In particular, I develop tools to understand causal relationships using data, extrapolate to predict the effects of unseen interventions, and select informative interventions for experimental design. Motivated by problems in cell biology, these tools allow us to learn from and better design large-scale interventional experiments (e.g., using genetic or chemical perturbations).

My research has been supported by the Eric and Wendy Schmidt Center PhD Fellowship and the Apple Scholarship. I was a research intern at Microsoft Research. I obtained my Bachelor’s degree in Mathematics from Peking University, where I worked with Zaiwen Wen, Mengdi Wang and Le Cong.



Membership Testing in Markov Equivalence Classes via Independence Query Oracles
Jiaqi Zhang*, Kirankumar Shiragur*, Caroline Uhler. AISTATS (Oral Presentation, <3%), 2024.
[arXiv] [bibtex]

Towards Causal Foundation Model: on Duality between Causal Inference and Attention
Jiaqi Zhang*, Joel Jennings, Cheng Zhang, Chao Ma*. arXiv preprint, 2023.
[arXiv] [code (coming soon)] [bibtex]

Meek Separators and Their Applications in Targeted Causal Discovery
Kirankumar Shiragur*, Jiaqi Zhang*, Caroline Uhler. NeurIPS, 2023.
[arXiv] [code] [conference] [bibtex]

Identifiability guarantees for causal disentanglement from soft interventions
Jiaqi Zhang, Kristjan Greenewald, Chandler Squires, Akash Srivastava, Karthikeyan Shanmugam, Caroline Uhler. NeurIPS, 2023.
[arXiv] [code] [conference] [bibtex]

Active learning for optimal intervention design in causal models
Jiaqi Zhang, Louis Cammarata, Chandler Squires, Themistoklis P Sapsis, Caroline Uhler. Nature Machine Intelligence, 2023.
[arXiv] [code] [journal] [bibtex]

Machine-learning-optimized Cas12a barcoding enables the recovery of single-cell lineages and transcriptional profiles
Nicholas W Hughes, Yuanhao Qu*, Jiaqi Zhang*, …, Monte M Winslow, Mengdi Wang, Le Cong. Molecular Cell, 2022.
[code] [journal] [bibtex]

Stochastic Augmented Projected Gradient Methods for the Large-Scale Precoding Matrix Indicator Selection Problem
Jiaqi Zhang, Zeyu Jin, Bo Jiang, Zaiwen Wen. IEEE Transactions on Wireless Communications, 2022.
[journal] [bibtex]

Matching a desired causal state via shift interventions
Jiaqi Zhang, Chandler Squires, Caroline Uhler. NeurIPS, 2021.
[arXiv] [code] [conference] [bibtex]