Selected Publications

(✉︎) Corresponding author;  (‡) Equal contribution; (α-β order) alphabetical authorship ordering. You can also find my articles on my Google Scholar profile.

1. Partial Identification with Proxy of Latent Confoundings via Sum-of-ratios Fractional Programming: Extended
   Zhiheng Zhang ^✉︎.
   Informs Journal on Optimization (IJOO), 2026.
2. Adjusting Auxiliary Variables Under Approximate Neighborhood Interference
   Xin Lu ^✉︎, Yuhao Wang, Zhiheng Zhang (α-β order).
   Journal of the American Statistical Association (JASA), 2026.
3. A Minimax Learning Approach for Causal Inference Under Unmeasured Confounding with Negative Controls
   Nathan Kallus, Xiaojie Mao, Masatoshi Uehara, Zhiheng Zhang (α-β order).
   Journal of Machine Learning Research (JMLR), 2026 (Reject with encouragement to resubmit).

1. Wasserstein Policy Learning for Distributional Outcomes
   Yiyan Huang, Cheuk Hang Leung, Qi Wu, Zhiheng Zhang (α-β order).
   COLT, 2026 (Top conference in learning theory).
2. Causal Representation Learning with Optimal Compression and Complex Treatments
   Wanting Liang, Haoang Chi, Zhiheng Zhang ^✉︎.
   ICML, 2026.
3. Causal Matrix Completion under Multiple Treatments via Mixed Synthetic Nearest Neighbors
   Minrui Luo, Zhiheng Zhang ^✉︎.
   ICML, 2026.
4. Feasible Fusion: Constrained Joint Estimation under Structural Non-Overlap
   Yuxi Du, Zhiheng Zhang ^✉︎, Haoxuan Li, Cong Fang, Jixing Xu, Peng Zhen, Jiecheng Guo.
   ICML, 2026.
5. Budgeted Active Experimentation for Treatment Effect Estimation from Observational and Randomized Data
   Jiacan Gao, Xinyan Su (α-β order), Mingyuan Ma, Yiyan Huang, Xiao Xu, Xinrui Wan, Tianqi Gu, Enyun Yu, Jiecheng Guo, Zhiheng Zhang ^✉︎.
   ICML, 2026.
6. Off-Policy Evaluation Beyond Overlap under Network Interference
   Ziyan Wang, Yiran Liu, Zhiheng Zhang ^✉︎.
   ICML, 2026.
7. Partial Identification under High-Dimensional Potential Outcomes and Confounders via Optimal Transport
   Yunfeng Wang, Zhiheng Zhang ^✉︎, Zijun Gao.
   ICML, 2026.
8. MLUBench: A Benchmark for Lifelong Unlearning Evaluation in MLLMs
   He Li, Haoang Chi, Qizhou Wang, Yunxin Mao, Zhiheng Zhang, Jie Tan, Tongliang Liu, Wenjing Yang, Bo Han.
   ICML, 2026.
9. Treatment Responder Classification with Abstention
   Haoxiang Wang, Haoxuan Li, Ziyan Wang, Zhiheng Zhang, Aoqi Zuo, Erdun Gao, Kun Zhang, Mingming Gong.
   ICML, 2026 (Spotlight, <2%).
10. SpaCellAgent: A Self-evolving LLM-Based Multi-Agent Framework for Trajectory Analysis
    Songhan Wang, Haoang Chi, He Li, Zhiheng Zhang, Jiayan Yuan, Cheems Wang, Hao Peng, Xinwang Liu, Wenjing Yang.
    KDD, 2026.
11. Online Experimental Design With Estimation–Regret Trade-off Under Network Interference
    Zhiheng Zhang ^{✉︎ ‡}, Zichen Wang ^‡.
    NeurIPS, 2025. Invited talk at POMSHK2026.
12. Design-Based Bandits Under Network Interference: Trade-Off Between Regret and Statistical Inference
    Zichen Wang ^‡, Haoyang Hong ^‡, Chuanhao Li, Haoxuan Li, Zhiheng Zhang ^✉︎, Huazheng Wang.
    NeurIPS, 2025. Invited talk at POMSHK2026.
13. Unveiling Environmental Sensitivity of Individual Gains in Influence Maximization
    Xinyan Su, Zhiheng Zhang ^✉︎, Jiyan Qiu, Zhaojuan Yue, Jun Li ^✉︎.
    NeurIPS, 2025.
14. Active Treatment Effect Estimation via Limited Samples
    Zhiheng Zhang, Haoxiang Wang, Haoxuan Li ^✉︎, Zhouchen Lin ^✉︎.
    ICML, 2025.
15. Tight Partial Identification of Causal Effects with Marginal Distribution of Unmeasured Confounders
    Zhiheng Zhang ^✉︎.
    ICML, 2024 (Spotlight, <2%).
16. Partial Identification with Proxy of Latent Confoundings via Sum-of-ratios Fractional Programming
    Zhiheng Zhang ^✉︎, Xinyan Su.
    UAI, 2024.
17. Robust Causal Inference for Recommender System to Overcome Noisy Confounders
    Zhiheng Zhang, Quanyu Dai ^✉︎, Xu Chen, Zhenhua Dong, Ruiming Tang.
    SIGIR, 2023.

1. Group Permutation Testing for Linear Models: Sharp Validity, Power Improvement, and Extension Beyond Exchangeability
   Zonghan Li ^{‡ ✉︎}, Hongyi Zhou ^‡, Zhiheng Zhang ^✉︎.
   Preprint, 2026.
2. Orthogonal Uplift Learning with Permutation-Invariant Representations for Combinatorial Treatments
   Xinyan Su ^‡, Jiacan Gao ^‡, Mingyuan Ma, Xiao Xu, Xinrui Wan, Tianqi Gu, Enyun Yu, Jiecheng Guo, Zhiheng Zhang ^✉︎.
   Preprint, 2026.
3. Individualized Causal Effects under Network Interference with Combinatorial Treatments
   Yunping Lu, Haoang Chi, Qirui Hu, Zhiheng Zhang ^✉︎.
   Preprint, 2026.
4. Bounding the Largest Intersection Number of Regular Simplicial Partitions
   Zhiheng Zhang ^✉︎.
   Working paper, 2025.
5. Dynamic Window-level Granger Causality of Multi-channel Time Series
   Zhiheng Zhang, Wenbo Hu ^✉︎, Tian Tian, Jun Zhu.
   Working paper, 2025.

1. Causal Inference under Real-world Constraints: Privacy, Robustness, and Pareto Balance
   Zhiheng Zhang, 2025.