← Search

Tonghan Wang

19 accepted papers

2026

Translate Policy to Language: Flow Matching Generated Rewards for LLM Explanations

ICLR 2026poster

As humans increasingly share environments with diverse agents powered by RL, LLMs, and beyond, the ability to explain agent policies in natural language is vital for reliable coexistence. We introduce a general-purpose framework that trains explanation-generating LLMs via reinforcement learning from…

Cited by 0SourceScholar
2025

Adaptive Frontier Exploration on Graphs with Applications to Network-Based Disease Testing

NeurIPS 2025poster

We study a sequential decision-making problem on a $n$-node graph $\mathcal{G}$ where each node has an unknown label from a finite set $\mathbf{\Omega}$, drawn from a joint distribution $\mathcal{P}$ that is Markov with respect to $\mathcal{G}$. At each step, selecting a node reveals its label and y…

Cited by 0SourceScholar
2025

BundleFlow: Deep Menus for Combinatorial Auctions by Diffusion-Based Optimization

NeurIPS 2025poster

Differentiable economics—the use of deep learning for auction design—has driven progress in multi-item auction design with additive and unit-demand valuations. However, there has been little progress for combinatorial auctions (CAs), even in the simplest and yet important single bidder case, due to…

Cited by 0SourceScholar
2025

Composite Flow Matching for Reinforcement Learning with Shifted-Dynamics Data

NeurIPS 2025spotlight

Incorporating pre-collected offline data from a source environment can significantly improve the sample efficiency of reinforcement learning (RL), but this benefit is often challenged by discrepancies between the transition dynamics of the source and target environments. Existing methods typically a…

Cited by 0SourceScholar
2025

Robust Optimization with Diffusion Models for Green Security

UAI 2025

In green security, defenders must forecast adversarial behavior-such as poaching, illegal logging, and illegal fishing-to plan effective patrols. These behavior are often highly uncertain and complex. Prior work has leveraged game theory to design robust patrol strategies to handle uncertainty, but

Cited by 0SourcePDFScholar
2025

The Bandit Whisperer: Communication Learning for Restless Bandits

AAAI 2025technical

Applying Reinforcement Learning (RL) to Restless Multi-Arm Bandits (RMABs) offers a promising avenue for addressing allocation problems with resource constraints and temporal dynamics. However, classic RMAB models largely overlook the challenges of (systematic) data errors - a common occurrence in r…

Cited by 6SourcePDFScholar
2024

Multi-Sender Persuasion: A Computational Perspective

ICML 2024poster

We consider *multiple senders* with informational advantage signaling to convince a single self-interested actor to take certain actions. Generalizing the seminal *Bayesian Persuasion* framework, such settings are ubiquitous in computational economics, multi-agent learning, and machine learning with…

Cited by 11SourcePDFScholar
2024

Position: Social Environment Design Should be Further Developed for AI-based Policy-Making

ICML 2024poster

Artificial Intelligence (AI) holds promise as a technology that can be used to improve government and economic policy-making. This paper proposes a new research agenda towards this end by introducing **Social Environment Design**, a general framework for the use of AI in automated policy-making that…

Cited by 5SourcePDFScholar
2023

Deep Contract Design via Discontinuous Networks

NeurIPS 2023poster

Contract design involves a principal who establishes contractual agreements about payments for outcomes that arise from the actions of an agent. In this paper, we initiate the study of deep learning for the automated design of optimal contracts. We introduce a novel representation: the Discontinuous…

Cited by 15SourcePDFScholar
2023

Symmetry-Aware Robot Design with Structured Subgroups

ICML 2023poster

Robot design aims at learning to create robots that can be easily controlled and perform tasks efficiently. Previous works on robot design have proven its ability to generate robots for various tasks. However, these works searched the robots directly from the vast design space and ignored common str…

2022

Context-Aware Sparse Deep Coordination Graphs

ICLR 2022spotlight

Learning sparse coordination graphs adaptive to the coordination dynamics among agents is a long-standing problem in cooperative multi-agent learning. This paper studies this problem and proposes a novel method using the variance of payoff functions to construct context-aware sparse coordination top…

2022

Low-Rank Modular Reinforcement Learning via Muscle Synergy

NeurIPS 2022accept

Modular Reinforcement Learning (RL) decentralizes the control of multi-joint robots by learning policies for each actuator. Previous work on modular RL has proven its ability to control morphologically different agents with a shared actuator policy. However, with the increase in the Degree of Freedo…

2022

Self-Organized Polynomial-Time Coordination Graphs

ICML 2022spotlight

Coordination graph is a promising approach to model agent collaboration in multi-agent reinforcement learning. It conducts a graph-based value factorization and induces explicit coordination among agents to complete complicated tasks. However, one critical challenge in this paradigm is the complexit…

2021

Celebrating Diversity in Shared Multi-Agent Reinforcement Learning

NeurIPS 2021poster

Recently, deep multi-agent reinforcement learning (MARL) has shown the promise to solve complex cooperative tasks. Its success is partly because of parameter sharing among agents. However, such sharing may lead agents to behave similarly and limit their coordination capacity. In this paper, we aim t…

Cited by 189SourcePDFScholar
2021

DOP: Off-Policy Multi-Agent Decomposed Policy Gradients

ICLR 2021poster

Multi-agent policy gradient (MAPG) methods recently witness vigorous progress. However, there is a significant performance discrepancy between MAPG methods and state-of-the-art multi-agent value-based approaches. In this paper, we investigate causes that hinder the performance of MAPG algorithms and…

Cited by 158SourcePDFScholar
2021

RODE: Learning Roles to Decompose Multi-Agent Tasks

ICLR 2021poster

Role-based learning holds the promise of achieving scalable multi-agent learning by decomposing complex tasks using roles. However, it is largely unclear how to efficiently discover such a set of roles. To solve this problem, we propose to first decompose joint action spaces into restricted role act…

Cited by 260SourcePDFScholar
2020

Incorporating Pragmatic Reasoning Communication into Emergent Language

NeurIPS 2020spotlight

Emergentism and pragmatics are two research fields that study the dynamics of linguistic communication along quite different timescales and intelligence levels. From the perspective of multi-agent reinforcement learning, they correspond to stochastic games with reinforcement training and stage games…

Cited by 26SourcePDFScholar
2020

ROMA: Multi-Agent Reinforcement Learning with Emergent Roles

ICML 2020poster

The role concept provides a useful tool to design and understand complex multi-agent systems, which allows agents with a similar role to share similar behaviors. However, existing role-based methods use prior domain knowledge and predefine role structures and behaviors. In contrast, multi-agent rein…