← Search

Chongjie Zhang

66 accepted papers

2026

Correlated Policy Optimization in Multi-Agent Subteams

ICLR 2026poster

In cooperative multi-agent reinforcement learning, agents often face scalability challenges due to the exponential growth of the joint action and observation spaces. Inspired by the structure of human teams, we explore subteam-based coordination, where agents are partitioned into fully correlated su…

Cited by 0SourceScholar
2026

OPRIDE: Efficient Offline Preference-based Reinforcement Learning via In-Dataset Exploration

ICLR 2026poster

Preference-based reinforcement learning (PbRL) can help avoid sophisticated reward designs and align better with human intentions, showing great promise in various real-world applications. However, obtaining human feedback for preferences can be expensive and time-consuming, which forms a strong bar…

Cited by 0SourceScholar
2026

Think Twice Before You Act: Protecting LLM Agents Against Tool Description Poisoning via Isolated Planning

ICML 2026poster

The integration of external tools has substantially expanded the capabilities of large language model (LLM) agents, but also introduced new attack surfaces beyond prompt injection. In particular, cross-tool description poisoning can manipulate planner-visible tool metadata to steer an agent’s trajec…

Cited by 0SourceScholar
2026

Translating Flow to Policy via Hindsight Online Imitation

ICLR 2026poster

Recent advances in hierarchical robot systems leverage a high-level planner to propose task plans and a low-level policy to generate robot actions. This design allows training the planner on action-free or even non-robot data sources (e.g., videos), providing transferable high-level guidance. Nevert…

Cited by 0SourceScholar
2025

A Bayesian Fast-Slow Framework to Mitigate Interference in Non-Stationary Reinforcement Learning

NeurIPS 2025poster

Given the ever-changing nature of the world and its inhabitants, agents must possess the ability to adapt and evolve over time. Recent research in Given the ever-changing nature of the world and its inhabitants, agents must possess the ability to adapt and evolve over time. Recent research in non-st…

Cited by 0SourceScholar
2025

EcoLoRA: Communication-Efficient Federated Fine-Tuning of Large Language Models

EMNLP 2025

To address data locality and privacy restrictions, Federated Learning (FL) has recently been adopted to fine-tune large language models (LLMs), enabling improved performance on various downstream tasks without requiring aggregated data. However, the repeated exchange of model updates in FL can resul

Cited by 0SourcePDFScholar
2025

Episodic Novelty Through Temporal Distance

ICLR 2025poster

Exploration in sparse reward environments remains a significant challenge in reinforcement learning, particularly in Contextual Markov Decision Processes (CMDPs), where environments differ across episodes. Existing episodic intrinsic motivation methods for CMDPs primarily rely on count-based approac…

Cited by 0SourcePDFScholar
2025

Fewer May Be Better: Enhancing Offline Reinforcement Learning with Reduced Dataset

ICLR 2025poster

Research in offline reinforcement learning (RL) marks a paradigm shift in RL. However, a critical yet under-investigated aspect of offline RL is determining the subset of the offline dataset, which is used to improve algorithm performance while accelerating algorithm training. Moreover, the size of…

Cited by 0SourcePDFScholar
2025

Learning Policy Committees for Effective Personalization in MDPs with Diverse Tasks

ICML 2025poster

Many dynamic decision problems, such as robotic control, involve a series of tasks, many of which are unknown at training time. Typical approaches for these problems, such as multi-task and meta reinforcement learning, do not generalize well when the tasks are diverse. On the other hand, approaches…

2025

Learning to Plan Before Answering: Self-Teaching LLMs to Learn Abstract Plans for Problem Solving

ICLR 2025poster

In the field of large language model (LLM) post-training, the effectiveness of utilizing synthetic data generated by the LLM itself has been well-presented. However, a key question remains unaddressed: what essential information should such self-generated data encapsulate? Existing approaches only p…

Cited by 0SourcePDFScholar
2025

Revisiting Multi-Agent World Modeling from a Diffusion-Inspired Perspective

NeurIPS 2025poster

World models have recently attracted growing interest in Multi-Agent Reinforcement Learning (MARL) due to their ability to improve sample efficiency for policy learning. However, accurately modeling environments in MARL is challenging due to the exponentially large joint action space and highly unce…

Cited by 0SourcecodeScholar
2024

Bayesian Design Principles for Offline-to-Online Reinforcement Learning

ICML 2024poster

Offline reinforcement learning (RL) is crucial for real-world applications where exploration can be costly or unsafe. However, offline learned policies are often suboptimal, and further online fine-tuning is required. In this paper, we tackle the fundamental dilemma of offline-to-online fine-tuning:…

2024

Efficient Multi-agent Reinforcement Learning by Planning

ICLR 2024poster

Multi-agent reinforcement learning (MARL) algorithms have accomplished remarkable breakthroughs in solving large-scale decision-making tasks. Nonetheless, most existing MARL algorithms are model-free, limiting sample efficiency and hindering their applicability in more challenging scenarios. In cont…

2024

GOMAA-Geo: GOal Modality Agnostic Active Geo-localization

NeurIPS 2024poster

We consider the task of active geo-localization (AGL) in which an agent uses a sequence of visual cues observed during aerial navigation to find a target specified through multiple possible modalities. This could emulate a UAV involved in a search-and-rescue operation navigating through an area, obs…

2024

Imitation Learning from Observation with Automatic Discount Scheduling

ICLR 2024poster

Humans often acquire new skills through observation and imitation. For robotic agents, learning from the plethora of unlabeled video demonstration data available on the Internet necessitates imitating the expert without access to its action, presenting a challenge known as Imitation Learning from Ob…

2024

Planning, Fast and Slow: Online Reinforcement Learning with Action-Free Offline Data via Multiscale Planners

ICML 2024poster

The surge in volumes of video data offers unprecedented opportunities for advancing reinforcement learning (RL). This growth has motivated the development of passive RL, seeking to convert passive observations into actionable insights. This paper explores the prerequisites and mechanisms through whi…

Cited by 0SourcePDFScholar
2024

STAR: Spatio-Temporal State Compression for Multi-Agent Tasks with Rich Observations

IJCAI 2024poster

This paper focuses on the problem of learning compressed state representations for multi-agent tasks. Under the assumption of rich observation, we pinpoint that the state representations should be compressed both spatially and temporally to enable efficient prioritization of task-relevant features,…

Cited by 0SourcePDFScholar
2024

Stylized Offline Reinforcement Learning: Extracting Diverse High-Quality Behaviors from Heterogeneous Datasets

ICLR 2024poster

Previous literature on policy diversity in reinforcement learning (RL) either focuses on the online setting or ignores the policy performance. In contrast, offline RL, which aims to learn high-quality policies from batched data, has yet to fully leverage the intrinsic diversity of the offline datase…

Cited by 9SourcePDFScholar
2024

Towards Robust Offline Reinforcement Learning under Diverse Data Corruption

ICLR 2024spotlight

Offline reinforcement learning (RL) presents a promising approach for learning reinforced policies from offline datasets without the need for costly or unsafe interactions with the environment. However, datasets collected by humans in real-world environments are often noisy and may even be malicious…

2023

Conservative Offline Policy Adaptation in Multi-Agent Games

NeurIPS 2023poster

Prior research on policy adaptation in multi-agent games has often relied on online interaction with the target agent in training, which can be expensive and impractical in real-world scenarios. Inspired by recent progress in offline reinforcement learn- ing, this paper studies offline policy adapta…

Cited by 2SourcePDFScholar
2023

Flow to Control: Offline Reinforcement Learning with Lossless Primitive Discovery

AAAI 2023technical

Offline reinforcement learning (RL) enables the agent to effectively learn from logged data, which significantly extends the applicability of RL algorithms in real-world scenarios where exploration can be expensive or unsafe. Previous works have shown that extracting primitive skills from the recurr…

Cited by 18SourcePDFScholar
2023

Learning to Solve Tasks with Exploring Prior Behaviours

IROS 2023poster

Demonstrations are widely used in Deep Reinforcement Learning (DRL) for facilitating solving tasks with sparse rewards. However, the tasks in real-world scenarios can often have varied initial conditions from the demonstration, which would require additional prior behaviours. For example, consider w…

Cited by 3SourcecodeScholar
2023

Offline Meta Reinforcement Learning with In-Distribution Online Adaptation

ICML 2023poster

Recent offline meta-reinforcement learning (meta-RL) methods typically utilize task-dependent behavior policies (e.g., training RL agents on each individual task) to collect a multi-task dataset. However, these methods always require extra information for fast adaptation, such as offline context for…

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…

2023

The Provable Benefit of Unsupervised Data Sharing for Offline Reinforcement Learning

ICLR 2023poster

Self-supervised methods have become crucial for advancing deep learning by leveraging data itself to reduce the need for expensive annotations. However, the question of how to conduct self-supervised offline reinforcement learning (RL) in a principled way remains unclear. In this paper, we address t…

Cited by 17SourcePDFScholar
2023

Unsupervised Behavior Extraction via Random Intent Priors

NeurIPS 2023poster

Reward-free data is abundant and contains rich prior knowledge of human behaviors, but it is not well exploited by offline reinforcement learning (RL) algorithms. In this paper, we propose UBER, an unsupervised approach to extract useful behaviors from offline reward-free datasets via diversified re…

Cited by 8SourcePDFScholar
2023

What is Essential for Unseen Goal Generalization of Offline Goal-conditioned RL?

ICML 2023poster

Offline goal-conditioned RL (GCRL) offers a way to train general-purpose agents from fully offline datasets. In addition to being conservative within the dataset, the generalization ability to achieve unseen goals is another fundamental challenge for offline GCRL. However, to the best of our knowled…

2022

Active Hierarchical Exploration with Stable Subgoal Representation Learning

ICLR 2022poster

Goal-conditioned hierarchical reinforcement learning (GCHRL) provides a promising approach to solving long-horizon tasks. Recently, its success has been extended to more general settings by concurrently learning hierarchical policies and subgoal representations. Although GCHRL possesses superior exp…

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

Individual Reward Assisted Multi-Agent Reinforcement Learning

ICML 2022spotlight

In many real-world multi-agent systems, the sparsity of team rewards often makes it difficult for an algorithm to successfully learn a cooperative team policy. At present, the common way for solving this problem is to design some dense individual rewards for the agents to guide the cooperation. Howe…

2022

LAPO: Latent-Variable Advantage-Weighted Policy Optimization for Offline Reinforcement Learning

NeurIPS 2022accept

Offline reinforcement learning methods hold the promise of learning policies from pre-collected datasets without the need to query the environment for new samples. This setting is particularly well-suited for continuous control robotic applications for which online data collection based on trial-and…

Cited by 25SourcePDFScholar
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

Multi-Agent Concentrative Coordination with Decentralized Task Representation

IJCAI 2022poster

Value-based multi-agent reinforcement learning (MARL) methods hold the promise of promoting coordination in cooperative settings. Popular MARL methods mainly focus on the scalability or the representational capacity of value functions. Such a learning paradigm can reduce agents' uncertainties and pr…

2022

Multi-Agent Incentive Communication via Decentralized Teammate Modeling

AAAI 2022technical

Effective communication can improve coordination in cooperative multi-agent reinforcement learning (MARL). One popular communication scheme is exchanging agents' local observations or latent embeddings and using them to augment individual local policy input. Such a communication paradigm can reduce…

2022

Offline Reinforcement Learning with Value-based Episodic Memory

ICLR 2022poster

Offline reinforcement learning (RL) shows promise of applying RL to real-world problems by effectively utilizing previously collected data. Most existing offline RL algorithms use regularization or constraints to suppress extrapolation error for actions outside the dataset. In this paper, we adopt a…

Cited by 50SourcePDFScholar
2022

On the Role of Discount Factor in Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning (RL) enables effective learning from previously collected data without exploration, which shows great promise in real-world applications when exploration is expensive or even infeasible. The discount factor, $\gamma$, plays a vital role in improving online RL sample ef…

Cited by 25SourcePDFScholar
2022

RORL: Robust Offline Reinforcement Learning via Conservative Smoothing

NeurIPS 2022accept

Offline reinforcement learning (RL) provides a promising direction to exploit massive amount of offline data for complex decision-making tasks. Due to the distribution shift issue, current offline RL algorithms are generally designed to be conservative in value estimation and action selection. Howev…

2022

Rethinking Goal-Conditioned Supervised Learning and Its Connection to Offline RL

ICLR 2022poster

Solving goal-conditioned tasks with sparse rewards using self-supervised learning is promising because of its simplicity and stability over current reinforcement learning (RL) algorithms. A recent work, called Goal-Conditioned Supervised Learning (GCSL), provides a new learning framework by iterativ…

2022

Safe Opponent-Exploitation Subgame Refinement

NeurIPS 2022accept

In zero-sum games, an NE strategy tends to be overly conservative confronted with opponents of limited rationality, because it does not actively exploit their weaknesses. From another perspective, best responding to an estimated opponent model is vulnerable to estimation errors and lacks safety guar…

Cited by 9SourcePDFScholar
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

Episodic Multi-agent Reinforcement Learning with Curiosity-driven Exploration

NeurIPS 2021poster

Efficient exploration in deep cooperative multi-agent reinforcement learning (MARL) still remains challenging in complex coordination problems. In this paper, we introduce a novel Episodic Multi-agent reinforcement learning with Curiosity-driven exploration, called EMC. We leverage an insight of pop…

Cited by 101SourcePDFScholar
2021

Generalizable Episodic Memory for Deep Reinforcement Learning

ICML 2021spotlight

Episodic memory-based methods can rapidly latch onto past successful strategies by a non-parametric memory and improve sample efficiency of traditional reinforcement learning. However, little effort is put into the continuous domain, where a state is never visited twice, and previous episodic method…

2021

MetaCURE: Meta Reinforcement Learning with Empowerment-Driven Exploration

ICML 2021spotlight

Meta reinforcement learning (meta-RL) extracts knowledge from previous tasks and achieves fast adaptation to new tasks. Despite recent progress, efficient exploration in meta-RL remains a key challenge in sparse-reward tasks, as it requires quickly finding informative task-relevant experiences in bo…

2021

Model-Based Reinforcement Learning via Imagination with Derived Memory

NeurIPS 2021poster

Model-based reinforcement learning aims to improve the sample efficiency of policy learning by modeling the dynamics of the environment. Recently, the latent dynamics model is further developed to enable fast planning in a compact space. It summarizes the high-dimensional experiences of an agent, wh…

Cited by 9SourcePDFScholar
2021

Offline Reinforcement Learning with Reverse Model-based Imagination

NeurIPS 2021poster

In offline reinforcement learning (offline RL), one of the main challenges is to deal with the distributional shift between the learning policy and the given dataset. To address this problem, recent offline RL methods attempt to introduce conservatism bias to encourage learning in high-confidence a…

Cited by 71SourcePDFScholar
2021

On the Estimation Bias in Double Q-Learning

NeurIPS 2021poster

Double Q-learning is a classical method for reducing overestimation bias, which is caused by taking maximum estimated values in the Bellman operation. Its variants in the deep Q-learning paradigm have shown great promise in producing reliable value prediction and improving learning performance. Howe…

2021

QPLEX: Duplex Dueling Multi-Agent Q-Learning

ICLR 2021poster

We explore value-based multi-agent reinforcement learning (MARL) in the popular paradigm of centralized training with decentralized execution (CTDE). CTDE has an important concept, Individual-Global-Max (IGM) principle, which requires the consistency between joint and local action selections to supp…

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
2021

Towards Understanding Cooperative Multi-Agent Q-Learning with Value Factorization

NeurIPS 2021poster

Value factorization is a popular and promising approach to scaling up multi-agent reinforcement learning in cooperative settings, which balances the learning scalability and the representational capacity of value functions. However, the theoretical understanding of such methods is limited. In this p…

Cited by 43SourcePDFScholar
2020

Bridging Imagination and Reality for Model-Based Deep Reinforcement Learning

NeurIPS 2020poster

Sample efficiency has been one of the major challenges for deep reinforcement learning. Recently, model-based reinforcement learning has been proposed to address this challenge by performing planning on imaginary trajectories with a learned world model. However, world model learning may suffer from…

2020

Learning Nearly Decomposable Value Functions Via Communication Minimization

ICLR 2020poster

Reinforcement learning encounters major challenges in multi-agent settings, such as scalability and non-stationarity. Recently, value function factorization learning emerges as a promising way to address these challenges in collaborative multi-agent systems. However, existing methods have been focus…

Cited by 178SourcecodeScholar
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…

2019

Hierarchical Reinforcement Learning with Advantage-Based Auxiliary Rewards

NeurIPS 2019poster

Hierarchical Reinforcement Learning (HRL) is a promising approach to solving long-horizon problems with sparse and delayed rewards. Many existing HRL algorithms either use pre-trained low-level skills that are unadaptable, or require domain-specific information to define low-level rewards. In this p…

Cited by 102SourcePDFScholar