Research

Working Papers

TBA

Publications

\( \cdot \) Adaptive Multi-Round Allocation with Stochastic Arrivals
Yuqi Pan*, Davin Choo*, Haichuan Wang, Milind Tambe, Alastair van Heerden, Cheryl Johnson.
ICML 2026

\( \cdot \) Reward Shaping for Inference-Time Alignment: A Stackelberg Game Perspective
Haichuan Wang, Tao Lin, Lingkai Kong, Ce Li, Hezi Jiang, Milind Tambe.
ICML 2026 [paper]

\( \cdot \) Rule-Bottleneck Reinforcement Learning: Joint Explanation and Decision Optimization for Resource Allocation with Language Agents
Mauricio Tec*, Guojun Xiong*, Haichuan Wang, Francesca Dominici, Milind Tambe.
ICJAI 2026 [paper]

\( \cdot \) The Publication Choice Problem
(α–β) Haichuan Wang, Yifan Wu, Haifeng Xu.
This preliminary version of this paper won The David S. Hu Thesis Awards in Economics at UChicago [link]
AAAI 2026 (Oral)[paper]

\( \cdot \) Generative AI Against Poaching: Latent Composite Flow Matching for Wildlife Conservation
Lingkai Kong*, Haichuan Wang*, Charles A. Emogor, Vincent Börsch-Supan, Lily Xu, Milind Tambe
IAAI 2026 [paper]

\( \cdot \) Composite Flow Matching for Reinforcement Learning with Shifted-Dynamics Data
Lingkai Kong*, Haichuan Wang*, Tonghan Wang*, Guojun Xiong, Milind Tambe.
NeurIPS 2025 (Spotlight) [paper][code]

\( \cdot \) Robust Optimization with Diffusion Models for Green Security
Lingkai Kong, Haichuan Wang, Yuqi Pan, Cheol Woo Kim, Mingxiao Song, Alayna Nguyen, Tonghan Wang, Haifeng Xu, Milind Tambe.
UAI 2025 [paper]

\( \cdot \) Finite-Horizon Single-Pull Restless Bandits: An Efficient Index Policy For Scarce Resource Allocation
Guojun Xiong, Haichuan Wang, Yuqi Pan, Saptarshi Mandal, Sanket Shah, Niclas Boehmer, Milind Tambe.
AAMAS 2025 [paper]

Other Research

\( \cdot \) Decentralized Paymaster
Calvin Cai, Kaili Wang, Joe Fang, Rachel Mayer, Michael Grant, Haichuan Wang.
Patent Pending Application.

Service

Reviewer: AIES 2025