← Search

Baoxiang Wang

38 accepted papers

2026

Epistemic Gain, Aleatoric Cost: Uncertainty Decomposition in Multi-Agent Debate for Math Reasoning

ICML 2026poster

Multi-Agent Debate (MAD) has shown promise in leveraging collective intelligence to improve reasoning and reduce hallucinations, yet it remains unclear how information exchange shapes the underlying ability. Empirically, MAD exhibits paradoxical phenomena, such as accuracy improvement accompanied by…

Cited by 0SourceScholar
2026

Offline Multi-Agent Reinforcement Learning via Sequential Score Decomposition

ICML 2026poster

Offline cooperative multi-agent reinforcement learning (MARL) faces unique challenges due to the distribution shift between online and offline data collection. While online MARL typically converges to a single coordinated joint policy, offline datasets are often mixtures of diverse cooperative behav…

Cited by 0SourceScholar
2026

Talk, Judge, Cooperate: Gossip-Driven Indirect Reciprocity in Self-Interested LLM Agents

ICML 2026poster

Indirect reciprocity, which means helping those who help others, is difficult to sustain among decentralized, self-interested LLM agents without reliable reputation systems. We introduce Agentic Linguistic Gossip Network (ALIGN), an automated framework where agents strategically share open-ended gos…

Cited by 0SourceScholar
2026

The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL

ICML 2026poster

Reinforcement Learning for Large Language Models (LLMs) often suffers from training collapse in long-horizon tasks due to exploding gradient variance. To mitigate this, baseline is commonly introduced for advantage computation; however, traditional value models remain difficult to optimize, and stan…

Cited by 0SourceScholar
2026

Trust Region Masking for Long-Horizon LLM Reinforcement Learning

ICML 2026poster

Policy gradient methods for Large Language Models (LLMs) optimize a policy $\pi_\theta$ via a surrogate objective computed from samples of a rollout policy $\pi_{\text{roll}}$. However, modern LLM-RL pipelines suffer from unavoidable implementation divergences—such as backend discrepancies, Mixture-…

Cited by 0SourceScholar
2025

A Comprehensive Framework for Analyzing the Convergence of Adam: Bridging the Gap with SGD

ICML 2025poster

Adaptive moment estimation (Adam) is a cornerstone optimization algorithm in deep learning, widely recognized for its flexibility with adaptive learning rates and efficiency in handling large-scale data. However, despite its practical success, the theoretical understanding of Adam's convergence has…

Cited by 0SourcePDFScholar
2025

ADG: Ambient Diffusion-Guided Dataset Recovery for Corruption-Robust Offline Reinforcement Learning

NeurIPS 2025poster

Real-world datasets collected from sensors or human inputs are prone to noise and errors, posing significant challenges for applying offline reinforcement learning (RL). While existing methods have made progress in addressing corrupted actions and rewards, they remain insufficient for handling corru…

Cited by 0SourceScholar
2025

Improved Approximation Algorithms for $k$-Submodular Maximization via Multilinear Extension

ICLR 2025spotlight

We investigate a generalized form of submodular maximization, referred to as $k$-submodular maximization, with applications across the domains of social networks and machine learning. In this work, we propose the multilinear extension of $k$-submodular functions and unified Frank-Wolfe-type framewor…

Cited by 0SourcePDFScholar
2025

Learning Imperfect Information Extensive-form Games with Last-iterate Convergence under Bandit Feedback

ICML 2025poster

We investigate learning approximate Nash equilibrium (NE) policy profiles in two-player zero-sum imperfect information extensive-form games (IIEFGs) with last-iterate convergence guarantees. Existing algorithms either rely on full-information feedback or provide only asymptotic convergence rates. In…

Cited by 0SourcePDFScholar
2025

Learning to Communicate Through Implicit Communication Channels

ICLR 2025poster

Effective communication is an essential component in collaborative multi-agent systems. Situations where explicit messaging is not feasible have been common in human society throughout history, which motivate the study of implicit communication. Previous works on learning implicit communication most…

Cited by 1SourcePDFScholar
2025

Learning to Negotiate via Voluntary Commitment

AISTATS 2025poster

The partial alignment and conflict of autonomous agents lead to mixed-motive scenarios in many real-world applications. However, agents may fail to cooperate in practice even when cooperation yields a better outcome. One well known reason for this failure comes from non-credible commitments. To faci…

Cited by 0SourcecodeScholar
2025

Logarithmic Regret for Linear Markov Decision Processes with Adversarial Corruptions

AAAI 2025technical

In this work, we study the logarithmic regret for reinforcement learning (RL) with linear function approximation and adversarial corruptions, in the formulation of linear Markov decision processes (MDPs). Specifically, we consider the case where there exist adversarial corruptions over the reward fu…

Cited by 0SourcePDFScholar
2025

Multi-Agent Credit Assignment with Pretrained Language Models

AISTATS 2025poster

The difficulty of appropriately assigning credit is particularly heightened in cooperative MARL with sparse reward, due to the concurrent time and structural scales involved. Automatic subgoal generation (ASG) has recently emerged as a viable MARL approach inspired by utilizing subgoals in intrinsic…

Cited by 0SourceScholar
2025

Reward Translation via Reward Machine in Semi-Alignable MDPs

ICML 2025poster

Addressing reward design complexities in deep reinforcement learning is facilitated by knowledge transfer across different domains. To this end, we define \textit{reward translation} to describe the cross-domain reward transfer problem. However, current methods struggle with non-pairable and non-tim…

Cited by 0SourcePDFScholar
2025

Tackling Data Corruption in Offline Reinforcement Learning via Sequence Modeling

ICLR 2025poster

Learning policy from offline datasets through offline reinforcement learning (RL) holds promise for scaling data-driven decision-making while avoiding unsafe and costly online interactions. However, real-world data collected from sensors or humans often contains noise and errors, posing a significan…

2024

Carbon Market Simulation with Adaptive Mechanism Design

IJCAI 2024poster

A carbon market is a market-based tool that incentivizes economic agents to align individual profits with the global utility, i.e., reducing carbon emissions to tackle climate change. Cap and trade stands as a critical principle based on allocating and trading carbon allowances (carbon emission cred…

2024

Convergence to Nash Equilibrium and No-regret Guarantee in (Markov) Potential Games

AISTATS 2024poster

In this work, we study potential games and Markov potential games under stochastic cost and bandit feedback. We propose a variant of the Frank-Wolfe algorithm with sufficient exploration and recursive gradient estimation, which provably converges to the Nash equilibrium while attaining sublinear reg…

Cited by 0SourcePDFScholar
2024

Few-Shot Diffusion Models Escape the Curse of Dimensionality

NeurIPS 2024poster

While diffusion models have demonstrated impressive performance, there is a growing need for generating samples tailored to specific user-defined concepts. The customized requirements promote the development of few-shot diffusion models, which use limited $n_{ta}$ target samples to fine-tune a pre-t…

Cited by 1SourcePDFScholar
2024

Online Control with Adversarial Disturbance for Continuous-time Linear Systems

NeurIPS 2024poster

We study online control for continuous-time linear systems with finite sampling rates, where the objective is to design an online procedure that learns under non-stochastic noise and performs comparably to a fixed optimal linear controller. We present a novel two-level online algorithm, by integrat…

Cited by 0SourcePDFScholar
2024

Online Policy Optimization for Robust Markov Decision Process

UAI 2024poster

Reinforcement learning (RL) has exceeded human performance in many synthetic settings such as video games and Go. However, real-world deployment of end-to-end RL models is less common, as RL models can be very sensitive to perturbations in the environment. The robust Markov decision process (MDP) fr…

2024

Relative Policy-Transition Optimization for Fast Policy Transfer

AAAI 2024technical

We consider the problem of policy transfer between two Markov Decision Processes (MDPs). We introduce a lemma based on existing theoretical results in reinforcement learning to measure the relativity gap between two arbitrary MDPs, that is the difference between any two cumulative expected returns d…

Cited by 0SourcePDFScholar
2023

DPMAC: Differentially Private Communication for Cooperative Multi-Agent Reinforcement Learning

IJCAI 2023poster

Communication lays the foundation for cooperation in human society and in multi-agent reinforcement learning (MARL). Humans also desire to maintain their privacy when communicating with others, yet such privacy concern has not been considered in existing works in MARL. We propose the differentially…

2023

Information Design in Multi-Agent Reinforcement Learning

NeurIPS 2023poster

Reinforcement learning (RL) is inspired by the way human infants and animals learn from the environment. The setting is somewhat idealized because, in actual tasks, other agents in the environment have their own goals and behave adaptively to the ego agent. To thrive in those environments, the agent…

2023

Learning Adversarial Linear Mixture Markov Decision Processes with Bandit Feedback and Unknown Transition

ICLR 2023poster

We study reinforcement learning (RL) with linear function approximation, unknown transition, and adversarial losses in the bandit feedback setting. Specifically, the unknown transition probability function is a linear mixture model \citep{AyoubJSWY20,ZhouGS21,HeZG22} with a given feature mapping, an…

Cited by 13SourcePDFScholar
2023

Learning Adversarial Low-rank Markov Decision Processes with Unknown Transition and Full-information Feedback

NeurIPS 2023poster

In this work, we study the low-rank MDPs with adversarially changed losses in the full-information feedback setting. In particular, the unknown transition probability kernel admits a low-rank matrix decomposition \citep{REPUCB22}, and the loss functions may change adversarially but are revealed to t…

Cited by 5SourcePDFScholar
2023

Learning from Good Trajectories in Offline Multi-Agent Reinforcement Learning

AAAI 2023technical

Offline multi-agent reinforcement learning (MARL) aims to learn effective multi-agent policies from pre-collected datasets, which is an important step toward the deployment of multi-agent systems in real-world applications. However, in practice, each individual behavior policy that generates multi-a…

Cited by 16SourcePDFScholar
2023

Two Heads are Better Than One: A Simple Exploration Framework for Efficient Multi-Agent Reinforcement Learning

NeurIPS 2023poster

Exploration strategy plays an important role in reinforcement learning, especially in sparse-reward tasks. In cooperative multi-agent reinforcement learning~(MARL), designing a suitable exploration strategy is much more challenging due to the large state space and the complex interaction among agent…

Cited by 3SourcePDFScholar
2022

Deconfounded Value Decomposition for Multi-Agent Reinforcement Learning

ICML 2022spotlight

Value decomposition (VD) methods have been widely used in cooperative multi-agent reinforcement learning (MARL), where credit assignment plays an important role in guiding the agents’ decentralized execution. In this paper, we investigate VD from a novel perspective of causal inference. We first sho…

Cited by 23SourcePDFScholar