research

Papers

Note: * stands for equal contributions; $^\dagger$ stands for students that I mentored.

Causal Discovery with Fewer Conditional Independence Tests
Kirankumar Shiragur*, Jiaqi Zhang*, Caroline Uhler. ICML, 2024.
[arXiv] [code] [bibtex]

Towards Causal Foundation Model: on Duality between Causal Inference and Attention
Jiaqi Zhang*, Joel Jennings, Agrin Hilmkil, Nick Pawlowski, Cheng Zhang, Chao Ma*. ICML, 2024.
[arXiv] [code] [bibtex]

Membership Testing in Markov Equivalence Classes via Independence Query Oracles
Jiaqi Zhang*, Kirankumar Shiragur*, Caroline Uhler. AISTATS (Oral Presentation, <3%), 2024.
[arXiv] [conference] [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]