← Search

Yaodong Yang

117 accepted papers

2026

Align Once, Benefit Multilingually: Enforcing Multilingual Consistency for LLM Safety Alignment

ICLR 2026poster

The widespread deployment of large language models (LLMs) across linguistic communities necessitates reliable multilingual safety alignment. However, recent efforts to extend alignment to other languages often require substantial resources, either through large-scale, high-quality supervision in the…

Cited by 0SourceScholar
2026

Debate with Images: Detecting Deceptive Behaviors in Multimodal Large Language Models

ICML 2026poster

As frontier AI systems become increasingly capable, concerns about deceptive behaviors have intensified. Unlike hallucinations, which stem from capability limitations, deception involves strategically misleading responses despite correct internal representations. While prior work has primarily studi…

Cited by 0SourceScholar
2026

DexGraspVLA: A Vision-Language-Action Framework Towards General Dexterous Grasping

AAAI 2026technical

Dexterous grasping remains a fundamental yet challenging problem in robotics. A general-purpose robot must be capable of grasping diverse objects in arbitrary scenarios. However, existing research typically relies on restrictive assumptions, such as single-object settings or limited environments, sh

Cited by 0SourcePDFScholar
2026

DexKnot: Generalizable Visuomotor Policy Learning for Dexterous Bag-Knotting Manipulation

ICRA 2026poster

Knotting plastic bags is a common task in daily life, yet it is challenging for robots due to the bags' infinite degrees of freedom and complex physical dynamics. Existing methods often struggle in generalization to unseen bag instances or deformations. To address this, we present DexKnot, a framewo…

2026

Fault Tolerant Multi-Agent Learning with Adversarial Budget Constraints

ICML 2026poster

We study robustness to agent malfunctions in cooperative multi-agent reinforcement learning (MARL), a failure mode that is critical in practice yet underexplored in existing theory. We introduce MARTA, a plug-and-play robustness layer that augments standard MARL algorithms with a {\fontfamily{cmss}\…

Cited by 0SourceScholar
2026

MVR: Multi-view Video Reward Shaping for Reinforcement Learning

ICLR 2026poster

Reward design is of great importance for solving complex tasks with reinforcement learning. Recent studies have explored using image-text similarity produced by vision-language models (VLMs) to augment rewards of a task with visual feedback. A common practice linearly adds VLM scores to task or succ…

Cited by 0SourceScholar
2026

On Robustness of Vision-Language-Action Model against Multi-Modal Perturbations

ICLR 2026poster

In Vision–Language–Action (VLA) models, robustness to real-world perturbations is critical for deployment. Existing methods target simple visual disturbances, overlooking the broader multi-modal perturbations that arise in actions, instructions, environments, and observations. Here, we first evaluat…

Cited by 0SourcecodeScholar
2026

PoliCon: Evaluating LLMs on Achieving Diverse Political Consensus Objectives

ICLR 2026poster

Achieving political consensus is crucial yet challenging for the effective functioning of social governance. However, although frontier AI systems represented by large language models (LLMs) have developed rapidly in recent years, their capabilities in this scope are still understudied. In this pape…

Cited by 0SourcecodeScholar
2026

SafeLab: An Interactive High-Fidelity Benchmark for Embodied Safety in Scientific Robotics

ICML 2026poster

Laboratory automation driven by scientific embodied agents represents a critical frontier in modern laboratories. Unlike conventional robotic domains, laboratory environments impose zero-tolerance constraints on manipulation precision and collision, as minor deviations can lead to irreversible chemi…

Cited by 0SourceScholar
2026

ScaleMoE: Mixture-of-Experts for Scalable Continuous Control in Actor-Critic Reinforcement Learning

ICML 2026spotlight

Scaling network remains a bottleneck in deep reinforcement learning (RL): simply enlarging actor–critic networks destabilizes training and soon saturates performance. Although recent monolithic architectures such as SimBa and BRC have shown that carefully designed inductive biases can enable positiv…

Cited by 0SourceScholar
2026

UniCode: Augmenting Evaluation for Code Reasoning

ICML 2026poster

Current coding benchmarks often inflate Large Language Model (LLM) capabilities due to static paradigms and data contamination, enabling models to exploit statistical shortcuts rather than genuine reasoning. To address this, we introduce \textbf{UniCode}, a generative evaluation framework that syste…

Cited by 0SourceScholar
2026

VLA-Arena: An Open-Source Framework for Benchmarking Vision-Language-Action Models

ICML 2026poster

While Vision-Language-Action models (VLAs) are rapidly advancing toward generalist robot policies, quantitatively characterizing their capability boundaries and failure modes remains challenging. To address this, we introduce **VLA-Arena**, a comprehensive benchmark. It features a novel structured t…

Cited by 0SourceScholar
2026

Vulnerable Agent Identification in Large-Scale Multi-Agent Reinforcement Learning

ICML 2026poster

Partial agent failure becomes inevitable when systems scale up, making it crucial to identify the subset of agents whose failure causes worst-case system performance degradations. We study this Vulnerable Agent Identification (VAI) problem in large-scale multi-agent reinforcement learning (MARL). We…

Cited by 0SourceScholar
2025

Adaptive Visuo-Tactile Fusion with Predictive Force Attention for Dexterous Manipulation

IROS 2025

Effectively utilizing multi-sensory data is important for robots to generalize across diverse tasks. However, the heterogeneous nature of these modalities makes fusion challenging. Existing methods propose strategies to obtain comprehensively fused features but often ignore the fact that each modali

Cited by 12SourcecodeScholar
2025

Amulet: ReAlignment During Test Time for Personalized Preference Adaptation of LLMs

ICLR 2025poster

How to align large language models (LLMs) with user preferences from a static general dataset has been frequently studied. However, user preferences are usually personalized, changing, and diverse. This leads to the problem that the actual user preferences often do not coincide with those trained by…

Cited by 1SourcePDFScholar
2025

Benchmarking Multi-National Value Alignment for Large Language Models

ACL 2025finding

Do Large Language Models (LLMs) hold positions that conflict with your country’s values? Occasionally they do! However, existing works primarily focus on ethical reviews, failing to capture the diversity of national values, which encompass broader policy, legal, and moral considerations. Furthermore…

2025

Boosting Policy and Process Reward Models with Monte Carlo Tree Search in Open-Domain QA

ACL 2025finding

The recent introduction of OpenAI’s O1/O3 model represents a significant milestone in developing strong reasoning capabilities in Large Language Models (LLMs). By introducing more computational budget during test-time, LLMs have the potential to explore more accurate and higher-quality solutions. Ho…

2025

ClutterDexGrasp: A Sim-to-Real System for General Dexterous Grasping in Cluttered Scenes

CoRL 2025oral

Dexterous grasping in cluttered scenes presents significant challenges due to diverse object geometries, occlusions, and potential collisions. Existing methods primarily focus on single-object grasping or grasp-pose prediction without interaction, which are insufficient for complex, cluttered scenes…

Cited by 0SourceScholar
2025

DexFlyWheel: A Scalable and Self-improving Data Generation Framework for Dexterous Manipulation

NeurIPS 2025spotlight

Dexterous manipulation is critical for advancing robot capabilities in real-world applications, yet diverse and high-quality datasets remain scarce. Existing data collection methods either rely on human teleoperation or require significant human engineering, or generate data with limited diversity,…

Cited by 0SourceScholar
2025

Differentiable Information Enhanced Model-Based Reinforcement Learning

AAAI 2025technical

Differentiable environments have heralded new possibilities for learning control policies by offering rich differentiable information that facilitates gradient-based methods. In comparison to prevailing model-free reinforcement learning approaches, model-based reinforcement learning (MBRL) methods e…

Cited by 0SourcePDFScholar
2025

Emerging Safety Attack and Defense in Federated Instruction Tuning of Large Language Models

ICLR 2025poster

Federated learning (FL) enables multiple parties to collaboratively fine-tune an large language model (LLM) without the need of direct data sharing. Ideally, by training on decentralized data that is aligned with human preferences and safety principles, federated instruction tuning (FedIT) can resul…

2025

Empirical Study on Robustness and Resilience in Cooperative Multi-Agent Reinforcement Learning

NeurIPS 2025poster

In cooperative Multi-Agent Reinforcement Learning (MARL), it is a common practice to tune hyperparameters in ideal simulated environments to maximize cooperative performance. However, policies tuned for cooperation often fail to maintain robustness and resilience under real-world uncertainties. Buil…

Cited by 0SourceScholar
2025

Enhancing LLM-Based Social Bot via an Adversarial Learning Framework

EMNLP 2025

Developing Large Language Model (LLM) agents that exhibit human-like behavior, encompassing not only individual heterogeneity rooted in unique user profiles but also adaptive response to socially connected neighbors, is a significant research challenge. Social media platforms, with their diverse use

2025

Falcon: Fast Visuomotor Policies via Partial Denoising

ICML 2025poster

Diffusion policies are widely adopted in complex visuomotor tasks for their ability to capture multimodal action distributions. However, the multiple sampling steps required for action generation significantly harm real-time inference efficiency, which limits their applicability in real-time decisio…

Cited by 0SourcePDFScholar
2025

Generative RLHF-V: Learning Principles from Multi-modal Human Preference

NeurIPS 2025poster

Training multi-modal large language models (MLLMs) that align with human intentions is a long-term challenge. Traditional score-only reward models for alignment suffer from low accuracy, weak generalization, and poor interpretability, blocking the progress of alignment methods, \textit{e.g.,} reinfo…

Cited by 0SourceScholar
2025

In-Context Editing: Learning Knowledge from Self-Induced Distributions

ICLR 2025poster

In scenarios where language models must incorporate new information efficiently without extensive retraining, traditional fine-tuning methods are prone to overfitting, degraded generalization, and unnatural language generation. To address these limitations, we introduce Consistent In-Context Editing…

2025

InterMT: Multi-Turn Interleaved Preference Alignment with Human Feedback

NeurIPS 2025spotlight

As multimodal large models (MLLMs) continue to advance across challenging tasks, a key question emerges: \textbf{\textit{What essential capabilities are still missing? }} A critical aspect of human learning is continuous interaction with the environment -- not limited to language, but also involving…

Cited by 0SourceScholar
2025

Language Models Resist Alignment: Evidence From Data Compression

ACL 2025long

Large language models (LLMs) may exhibit unintended or undesirable behaviors. Recent works have concentrated on aligning LLMs to mitigate harmful outputs. Despite these efforts, some anomalies indicate that even a well-conducted alignment process can be easily circumvented, whether intentionally or…

2025

Libra-Leaderboard: Towards Responsible AI through a Balanced Leaderboard of Safety and Capability

NAACL 2025system demonstrations

As large language models (LLMs) continue to evolve, leaderboards play a significant role in steering their development. Existing leaderboards often prioritize model capabilities while overlooking safety concerns, leaving a significant gap in responsible AI development. To address this gap, we introd…

2025

Magnetic Preference Optimization: Achieving Last-iterate Convergence for Language Model Alignment

ICLR 2025poster

Self-play methods have demonstrated remarkable success in enhancing model capabilities across various domains. In the context of Reinforcement Learning from Human Feedback (RLHF), self-play not only boosts Large Language Model (LLM) performance but also overcomes the limitations of traditional Bradl…

Cited by 3SourcePDFScholar
2025

Mitigating Reward Over-Optimization in RLHF via Behavior-Supported Regularization

ICLR 2025poster

Reinforcement learning from human feedback (RLHF) is an effective method for aligning large language models (LLMs) with human values. However, reward over-optimization remains an open challenge leading to discrepancies between the performance of LLMs under the reward model and the true human objecti…

Cited by 0SourcePDFScholar
2025

PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preference

ACL 2025long

In this work, we introduce the PKU-SafeRLHF dataset, designed to promote research on safety alignment in large language models (LLMs). As a sibling project to SafeRLHF and BeaverTails, we separate annotations of helpfulness and harmlessness for question-answering pairs, providing distinct perspectiv…

2025

RAT: Adversarial Attacks on Deep Reinforcement Agents for Targeted Behaviors

AAAI 2025technical

Evaluating deep reinforcement learning (DRL) agents against targeted behavior attacks is critical for assessing their robustness. These attacks aim to manipulate the victim into specific behaviors that align with the attacker’s objectives, often bypassing traditional reward-based defenses. Prior met…

2025

Reward Generalization in RLHF: A Topological Perspective

ACL 2025finding

Existing alignment methods share a common topology of information flow, where reward information is collected from humans, modeled with preference learning, and used to tune language models. However, this shared topology has not been systematically characterized, nor have its alternatives been thoro…

Cited by 0SourcePDFScholar
2025

Risk-aware Direct Preference Optimization under Nested Risk Measure

NeurIPS 2025poster

When fine-tuning pre-trained Large Language Models (LLMs) to align with human values and intentions, maximizing the estimated reward can lead to superior performance, but it also introduces potential risks due to deviations from the reference model's intended behavior. Most existing methods typicall…

Cited by 0SourcecodeScholar
2025

STAR: Efficient Preference-based Reinforcement Learning via Dual Regularization

NeurIPS 2025poster

Preference-based reinforcement learning (PbRL) bypasses complex reward engineering by learning from human feedback. However, due to the high cost of obtaining feedback, PbRL typically relies on a limited set of preference-labeled samples. This data scarcity introduces two key inefficiencies: (1) the…

Cited by 0SourceScholar
2025

Safe RLHF-V: Safe Reinforcement Learning from Multi-modal Human Feedback

NeurIPS 2025poster

Multimodal large language models (MLLMs) are essential for building general-purpose AI assistants; however, they pose increasing safety risks. How can we ensure safety alignment of MLLMs to prevent undesired behaviors? Going further, it is critical to explore how to fine-tune MLLMs to preserve capab…

Cited by 0SourceScholar
2025

SafeLawBench: Towards Safe Alignment of Large Language Models

ACL 2025finding

With the growing prevalence of large language models (LLMs), the safety of LLMs has raised significant concerns. However, there is still a lack of definitive standards for evaluating their safety due to the subjective nature of current safety benchmarks. To address this gap, we conducted the first e…

2025

SafeVLA: Towards Safety Alignment of Vision-Language-Action Model via Constrained Learning

NeurIPS 2025spotlight

Vision-language-action models (VLAs) show potential as generalist robot policies. However, these models pose extreme safety challenges during real-world deployment, including the risk of harm to the environment, the robot itself, and humans. *How can safety constraints be explicitly integrated into…

Cited by 0SourceScholar
2025

Sequence to Sequence Reward Modeling: Improving RLHF by Language Feedback

AAAI 2025technical

Aligning the behavior of Large language models (LLMs) with human intentions and values remains a critical challenge. Reinforcement learning from human feedback (RLHF) aligns LLMs by training a reward model (RM) on human preferences and fine-tuning the LLMs to maximize RM feedback. Despite its effect…

Cited by 4SourcePDFScholar
2025

Social World Model-Augmented Mechanism Design Policy Learning

NeurIPS 2025poster

Designing adaptive mechanisms to align individual and collective interests remains a central challenge in artificial social intelligence. Existing methods often struggle with modeling heterogeneous agents possessing persistent latent traits (e.g., skills, preferences) and dealing with complex multi-…

Cited by 0SourceScholar
2025

Stream Aligner: Efficient Sentence-Level Alignment via Distribution Induction

AAAI 2025technical

The rapid advancement of large language models (LLMs) has led to significant improvements in their capabilities, but also to increased concerns about their alignment with human values and intentions. Current alignment strategies, including adaptive training and inference-time methods, have demonstra…

2025

Towards Efficient Collaboration via Graph Modeling in Reinforcement Learning

AAAI 2025technical

In multi-agent reinforcement learning, a commonly considered paradigm is centralized training with decentralized execution. However, in this framework, decentralized execution restricts the development of coordinated policies due to the local observation limitation. In this paper, we consider the co…

Cited by 1SourcePDFScholar
2025

World Models Should Prioritize the Unification of Physical and Social Dynamics

NeurIPS 2025poster

World models, which explicitly learn environmental dynamics to lay the foundation for planning, reasoning, and decision-making, are rapidly advancing in predicting both physical dynamics and aspects of social behavior, yet predominantly in separate silos. This division results in a systemic failure…

Cited by 0SourceScholar
2024

A Perspective of Q-value Estimation on Offline-to-Online Reinforcement Learning

AAAI 2024technical

Offline-to-online Reinforcement Learning (O2O RL) aims to improve the performance of offline pretrained policy using only a few online samples. Built on offline RL algorithms, most O2O methods focus on the balance between RL objective and pessimism, or the utilization of offline and online samples.…

2024

Aligner: Efficient Alignment by Learning to Correct

NeurIPS 2024oral

With the rapid development of large language models (LLMs) and ever-evolving practical requirements, finding an efficient and effective alignment method has never been more critical. However, the tension between the complexity of current alignment methods and the need for rapid iteration in deployme…

Cited by 20SourcePDFScholar
2024

AnySkill: Learning Open-Vocabulary Physical Skill for Interactive Agents

CVPR 2024poster

Traditional approaches in physics-based motion generation centered around imitation learning and reward shaping often struggle to adapt to new scenarios. To tackle this limitation we propose AnySkill a novel hierarchical method that learns physically plausible interactions following open-vocabulary…

Cited by 20SourcePDFScholar
2024

Byzantine Robust Cooperative Multi-Agent Reinforcement Learning as a Bayesian Game

ICLR 2024poster

In this study, we explore the robustness of cooperative multi-agent reinforcement learning (c-MARL) against Byzantine failures, where any agent can enact arbitrary, worst-case actions due to malfunction or adversarial attack. To address the uncertainty that any agent can be adversarial, we propose a…

2024

CivRealm: A Learning and Reasoning Odyssey in Civilization for Decision-Making Agents

ICLR 2024spotlight

The generalization of decision-making agents encompasses two fundamental elements: learning from past experiences and reasoning in novel contexts. However, the predominant emphasis in most interactive environments is on learning, often at the expense of complexity in reasoning. In this paper, we int…

2024

Efficient Adaptation in Mixed-Motive Environments via Hierarchical Opponent Modeling and Planning

ICML 2024poster

Despite the recent successes of multi-agent reinforcement learning (MARL) algorithms, efficiently adapting to co-players in mixed-motive environments remains a significant challenge. One feasible approach is to hierarchically model co-players' behavior based on inferring their characteristics. Howev…

Cited by 1SourcePDFScholar
2024

End-to-End Neuro-Symbolic Reinforcement Learning with Textual Explanations

ICML 2024spotlight

Neuro-symbolic reinforcement learning (NS-RL) has emerged as a promising paradigm for explainable decision-making, characterized by the interpretability of symbolic policies. NS-RL entails structured state representations for tasks with visual observations, but previous methods cannot refine the str…

2024

Maximum Entropy Heterogeneous-Agent Reinforcement Learning

ICLR 2024spotlight

*Multi-agent reinforcement learning* (MARL) has been shown effective for cooperative games in recent years. However, existing state-of-the-art methods face challenges related to sample complexity, training instability, and the risk of converging to a suboptimal Nash Equilibrium. In this paper, we pr…

Cited by 18SourcePDFScholar
2024

Neural Attention Field: Emerging Point Relevance in 3D Scenes for One-Shot Dexterous Grasping

CoRL 2024poster

One-shot transfer of dexterous grasps to novel scenes with object and context variations has been a challenging problem. While distilled feature fields from large vision models have enabled semantic correspondences across 3D scenes, their features are point-based and restricted to object surfaces, l…

Cited by 2SourceScholar
2024

Off-Agent Trust Region Policy Optimization

IJCAI 2024poster

Leveraging the experiences of other agents offers a powerful mechanism to enhance policy optimization in multi-agent reinforcement learning (MARL). However, contemporary MARL algorithms often neglect experience sharing possibilities or adopt a simple approach via direct parameter sharing. Our work e…

Cited by 0SourcePDFScholar
2024

Panacea: Pareto Alignment via Preference Adaptation for LLMs

NeurIPS 2024poster

Current methods for large language model alignment typically use scalar human preference labels. However, this convention tends to oversimplify the multi-dimensional and heterogeneous nature of human preferences, leading to reduced expressivity and even misalignment. This paper presents Panacea, an…

Cited by 32SourcePDFScholar
2024

ProAgent: Building Proactive Cooperative Agents with Large Language Models

AAAI 2024technical

Building agents with adaptive behavior in cooperative tasks stands as a paramount goal in the realm of multi-agent systems. Current approaches to developing cooperative agents rely primarily on learning-based methods, whose policy generalization depends heavily on the diversity of teammates they int…

2024

ProgressGym: Alignment with a Millennium of Moral Progress

NeurIPS 2024spotlight

Frontier AI systems, including large language models (LLMs), hold increasing influence over the epistemology of human users. Such influence can reinforce prevailing societal values, potentially contributing to the lock-in of misguided moral beliefs and, consequently, the perpetuation of problematic…

2024

STAS: Spatial-Temporal Return Decomposition for Solving Sparse Rewards Problems in Multi-agent Reinforcement Learning

AAAI 2024technical

Centralized Training with Decentralized Execution (CTDE) has been proven to be an effective paradigm in cooperative multi-agent reinforcement learning (MARL). One of the major challenges is credit assignment, which aims to credit agents by their contributions. They lack the functionality to model co…

2024

Safe RLHF: Safe Reinforcement Learning from Human Feedback

ICLR 2024spotlight

With the development of large language models (LLMs), striking a balance between the performance and safety of AI systems has never been more critical. However, the inherent tension between the objectives of helpfulness and harmlessness presents a significant challenge during LLM training. To addres…

2024

Safe Reinforcement Learning using Finite-Horizon Gradient-based Estimation

ICML 2024poster

A key aspect of Safe Reinforcement Learning (Safe RL) involves estimating the constraint condition for the next policy, which is crucial for guiding the optimization of safe policy updates. However, the existing *Advantage-based Estimation* (ABE) method relies on the infinite-horizon discounted adva…

Cited by 1SourcePDFScholar
2024

SafeDreamer: Safe Reinforcement Learning with World Models

ICLR 2024poster

The deployment of Reinforcement Learning (RL) in real-world applications is constrained by its failure to satisfy safety criteria. Existing Safe Reinforcement Learning (SafeRL) methods, which rely on cost functions to enforce safety, often fail to achieve zero-cost performance in complex scenarios,…

2024

SafeSora: Towards Safety Alignment of Text2Video Generation via a Human Preference Dataset

NeurIPS 2024poster

To mitigate the risk of harmful outputs from large vision models (LVMs), we introduce the *SafeSora* dataset to promote research on aligning text-to-video generation with human values. This dataset encompasses human preferences in text-to-video generation tasks along two primary dimensions: helpfuln…

2024

Sample-Efficient Multiagent Reinforcement Learning with Reset Replay

ICML 2024poster

The popularity of multiagent reinforcement learning (MARL) is growing rapidly with the demand for real-world tasks that require swarm intelligence. However, a noticeable drawback of MARL is its low sample efficiency, which leads to a huge amount of interactions with the environment. Surprisingly, fe…

Cited by 0SourcePDFScholar
2024

Scalable Constrained Policy Optimization for Safe Multi-agent Reinforcement Learning

NeurIPS 2024poster

A challenging problem in seeking to bring multi-agent reinforcement learning (MARL) techniques into real-world applications, such as autonomous driving and drone swarms, is how to control multiple agents safely and cooperatively to accomplish tasks. Most existing safe MARL methods learn the centrali…

Cited by 1SourcePDFScholar
2023

A Game-Theoretic Framework for Managing Risk in Multi-Agent Systems

ICML 2023poster

In order for agents in multi-agent systems (MAS) to be safe, they need to take into account the risks posed by the actions of other agents. However, the dominant paradigm in game theory (GT) assumes that agents are not affected by risk from other agents and only strive to maximise their expected uti…

Cited by 19SourcePDFScholar
2023

ACE: Cooperative Multi-Agent Q-learning with Bidirectional Action-Dependency

AAAI 2023technical

Multi-agent reinforcement learning (MARL) suffers from the non-stationarity problem, which is the ever-changing targets at every iteration when multiple agents update their policies at the same time. Starting from first principle, in this paper, we manage to solve the non-stationarity problem by pro…

2023

BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Dataset

NeurIPS 2023poster

In this paper, we introduce the BeaverTails dataset, aimed at fostering research on safety alignment in large language models (LLMs). This dataset uniquely separates annotations of helpfulness and harmlessness for question-answering pairs, thus offering distinct perspectives on these crucial attribu…

Cited by 400SourcePDFScholar
2023

Boosting Multiagent Reinforcement Learning via Permutation Invariant and Permutation Equivariant Networks

ICLR 2023poster

The state space in Multiagent Reinforcement Learning (MARL) grows exponentially with the agent number. Such a curse of dimensionality results in poor scalability and low sample efficiency, inhibiting MARL for decades. To break this curse, we propose a unified agent permutation framework that exploit…

Cited by 30SourcePDFScholar
2023

Dynamic Handover: Throw and Catch with Bimanual Hands

CoRL 2023poster

Humans throw and catch objects all the time. However, such a seemingly common skill introduces a lot of challenges for robots to achieve: The robots need to operate such dynamic actions at high-speed, collaborate precisely, and interact with diverse objects. In this paper, we design a system with tw…

Cited by 50SourcecodeScholar
2023

GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models

ICML 2023poster

This paper introduces a distributed, GPU-centric experience replay system, GEAR, designed to perform scalable reinforcement learning (RL) with large sequence models (such as transformers). With such models, existing systems such as Reverb face considerable bottlenecks in memory, computation, and com…

2023

GenDexGrasp: Generalizable Dexterous Grasping

ICRA 2023poster

Generating dexterous grasping has been a long-standing and challenging robotic task. Despite recent progress, existing methods primarily suffer from two issues. First, most prior art focuses on a specific type of robot hand, lacking generalizable capability of handling unseen ones. Second, prior art…

Cited by 80SourcecodeScholar
2023

Learning to Shape Rewards Using a Game of Two Partners

AAAI 2023technical

Reward shaping (RS) is a powerful method in reinforcement learning (RL) for overcoming the problem of sparse or uninformative rewards. However, RS typically relies on manually engineered shaping-reward functions whose construc- tion is time-consuming and error-prone. It also requires domain knowledg…

Cited by 8SourcePDFScholar
2023

MANSA: Learning Fast and Slow in Multi-Agent Systems

ICML 2023poster

In multi-agent reinforcement learning (MARL), independent learning (IL) often shows remarkable performance and easily scales with the number of agents. Yet, using IL can be inefficient and runs the risk of failing to successfully train, particularly in scenarios that require agents to coordinate the…

Cited by 7SourcePDFScholar
2023

Multi-Agent First Order Constrained Optimization in Policy Space

NeurIPS 2023poster

In the realm of multi-agent reinforcement learning (MARL), achieving high performance is crucial for a successful multi-agent system. Meanwhile, the ability to avoid unsafe actions is becoming an urgent and imperative problem to solve for real-life applications. Whereas, it is still challenging to…

Cited by 3SourcePDFScholar
2023

Policy Space Diversity for Non-Transitive Games

NeurIPS 2023poster

Policy-Space Response Oracles (PSRO) is an influential algorithm framework for approximating a Nash Equilibrium (NE) in multi-agent non-transitive games. Many previous studies have been trying to promote policy diversity in PSRO. A major weakness with existing diversity metrics is that a more divers…

Cited by 18SourcePDFScholar
2023

Quality-Similar Diversity via Population Based Reinforcement Learning

ICLR 2023poster

Diversity is a growing research topic in Reinforcement Learning (RL). Previous research on diversity has mainly focused on promoting diversity to encourage exploration and thereby improve quality (the cumulative reward), maximizing diversity subject to quality constraints, or jointly maximizing qual…

Cited by 22SourcePDFScholar
2023

RLAfford: End-to-End Affordance Learning for Robotic Manipulation

ICRA 2023poster

Learning to manipulate 3D objects in an interactive environment has been a challenging problem in Reinforcement Learning (RL). In particular, it is hard to train a policy that can generalize over objects with different semantic categories, diverse shape geometry and versatile functionality. In this…

Cited by 73SourceScholar
2023

Regret-Minimizing Double Oracle for Extensive-Form Games

ICML 2023poster

By incorporating regret minimization, double oracle methods have demonstrated rapid convergence to Nash Equilibrium (NE) in normal-form games and extensive-form games, through algorithms such as online double oracle (ODO) and extensive-form double oracle (XDO), respectively. In this study, we furthe…

2023

Safety Gymnasium: A Unified Safe Reinforcement Learning Benchmark

NeurIPS 2023poster

Artificial intelligence (AI) systems possess significant potential to drive societal progress. However, their deployment often faces obstacles due to substantial safety concerns. Safe reinforcement learning (SafeRL) emerges as a solution to optimize policies while simultaneously adhering to multiple…

Cited by 73SourcePDFScholar
2023

Subspace-Aware Exploration for Sparse-Reward Multi-Agent Tasks

AAAI 2023technical

Exploration under sparse rewards is a key challenge for multi-agent reinforcement learning problems. One possible solution to this issue is to exploit inherent task structures for an acceleration of exploration. In this paper, we present a novel exploration approach, which encodes a special structur…

Cited by 8SourcePDFScholar
2023

Team-PSRO for Learning Approximate TMECor in Large Team Games via Cooperative Reinforcement Learning

NeurIPS 2023poster

Recent algorithms have achieved superhuman performance at a number of two-player zero-sum games such as poker and go. However, many real-world situations are multi-player games. Zero-sum two-team games, such as bridge and football, involve two teams where each member of the team shares the same rewa…

Cited by 14SourcePDFScholar
2023

UniDexGrasp++: Improving Dexterous Grasping Policy Learning via Geometry-Aware Curriculum and Iterative Generalist-Specialist Learning

ICCV 2023oral

We propose a novel, object-agnostic method for learning a universal policy for dexterous object grasping from realistic point cloud observations and proprioceptive information under a table-top setting, namely UniDexGrasp++. To address the challenge of learning the vision-based policy across thousan…

Cited by 88PDFScholar
2022

A Theoretical Understanding of Gradient Bias in Meta-Reinforcement Learning

NeurIPS 2022accept

Gradient-based Meta-RL (GMRL) refers to methods that maintain two-level optimisation procedures wherein the outer-loop meta-learner guides the inner-loop gradient-based reinforcement learner to achieve fast adaptations. In this paper, we develop a unified framework that describes variations of GMRL…

2022

A Unified Diversity Measure for Multiagent Reinforcement Learning

NeurIPS 2022accept

Promoting behavioural diversity is of critical importance in multi-agent reinforcement learning, since it helps the agent population maintain robust performance when encountering unfamiliar opponents at test time, or, when the game is highly non-transitive in the strategy space (e.g., Rock-Paper-Sc…

Cited by 16SourcePDFScholar
2022

Constrained Update Projection Approach to Safe Policy Optimization

NeurIPS 2022accept

Safe reinforcement learning (RL) studies problems where an intelligent agent has to not only maximize reward but also avoid exploring unsafe areas. In this study, we propose CUP, a novel policy optimization method based on Constrained Update Projection framework that enjoys rigorous safety guarantee…

2022

LIGS: Learnable Intrinsic-Reward Generation Selection for Multi-Agent Learning

ICLR 2022poster

Efficient exploration is important for reinforcement learners (RL) to achieve high rewards. In multi-agent systems, coordinated exploration and behaviour is critical for agents to jointly achieve optimal outcomes. In this paper, we introduce a new general framework for improving coordination and per…

Cited by 26SourcePDFScholar
2022

MATE: Benchmarking Multi-Agent Reinforcement Learning in Distributed Target Coverage Control

NeurIPS 2022accept

We introduce the Multi-Agent Tracking Environment (MATE), a novel multi-agent environment simulates the target coverage control problems in the real world. MATE hosts an asymmetric cooperative-competitive game consisting of two groups of learning agents--"cameras" and "targets"--with opposing intere…

2022

Meta-Reward-Net: Implicitly Differentiable Reward Learning for Preference-based Reinforcement Learning

NeurIPS 2022accept

Setting up a well-designed reward function has been challenging for many reinforcement learning applications. Preference-based reinforcement learning (PbRL) provides a new framework that avoids reward engineering by leveraging human preferences (i.e., preferring apples over oranges) as the reward si…

Cited by 69SourcePDFScholar
2022

Multi-Agent Reinforcement Learning is a Sequence Modeling Problem

NeurIPS 2022accept

Large sequence models (SM) such as GPT series and BERT have displayed outstanding performance and generalization capabilities in natural language process, vision and recently reinforcement learning. A natural follow-up question is how to abstract multi-agent decision making also as an sequence model…

2022

On the Convergence of Fictitious Play: A Decomposition Approach

IJCAI 2022poster

Fictitious play (FP) is one of the most fundamental game-theoretical learning frameworks for computing Nash equilibrium in n-player games, which builds the foundation for modern multi-agent learning algorithms. Although FP has provable convergence guarantees on zero-sum games and potential games, ma…

Cited by 4SourcePDFScholar
2022

Scalable Model-based Policy Optimization for Decentralized Networked Systems

IROS 2022poster

Reinforcement learning algorithms require a large amount of samples; this often limits their real-world applications on even simple tasks. Such a challenge is more outstanding in multi-agent tasks, as each step of operation is more costly, requiring communications or shifting or resources. This work…

Cited by 10SourcecodeScholar
2022

Towards Human-Level Bimanual Dexterous Manipulation with Reinforcement Learning

NeurIPS 2022accept

Achieving human-level dexterity is an important open problem in robotics. However, tasks of dexterous hand manipulation even at the baby level are challenging to solve through reinforcement learning (RL). The difficulty lies in the high degrees of freedom and the required cooperation among heterogen…

2022

Transformer-based Working Memory for Multiagent Reinforcement Learning with Action Parsing

NeurIPS 2022accept

Learning in real-world multiagent tasks is challenging due to the usual partial observability of each agent. Previous efforts alleviate the partial observability by historical hidden states with Recurrent Neural Networks, however, they do not consider the multiagent characters that either the multia…

Cited by 20SourcePDFScholar
2022

Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning

ICLR 2022poster

Trust region methods rigorously enabled reinforcement learning (RL) agents to learn monotonically improving policies, leading to superior performance on a variety of tasks. Unfortunately, when it comes to multi-agent reinforcement learning (MARL), the property of monotonic improvement may not simpl…

Cited by 329SourcePDFScholar
2022

What about Inputting Policy in Value Function: Policy Representation and Policy-Extended Value Function Approximator

AAAI 2022technical

We study Policy-extended Value Function Approximator (PeVFA) in Reinforcement Learning (RL), which extends conventional value function approximator (VFA) to take as input not only the state (and action) but also an explicit policy representation. Such an extension enables PeVFA to preserve values of…

Cited by 26SourcePDFScholar
2021

Foresee then Evaluate: Decomposing Value Estimation with Latent Future Prediction

AAAI 2021technical

Value function is the central notion of Reinforcement Learning (RL). Value estimation, especially with function approximation, can be challenging since it involves the stochasticity of environmental dynamics and reward signals that can be sparse and delayed in some cases. A typical model-free RL alg…

2021

Learning in Nonzero-Sum Stochastic Games with Potentials

ICML 2021spotlight

Multi-agent reinforcement learning (MARL) has become effective in tackling discrete cooperative game scenarios. However, MARL has yet to penetrate settings beyond those modelled by team and zero-sum games, confining it to a small subset of multi-agent systems. In this paper, we introduce a new gener…

Cited by 59SourcePDFScholar
2021

Modelling Behavioural Diversity for Learning in Open-Ended Games

ICML 2021oral

Promoting behavioural diversity is critical for solving games with non-transitive dynamics where strategic cycles exist, and there is no consistent winner (e.g., Rock-Paper-Scissors). Yet, there is a lack of rigorous treatment for defining diversity and constructing diversity-aware learning dynamics…

Cited by 79SourcePDFScholar
2021

Neural Auto-Curricula in Two-Player Zero-Sum Games

NeurIPS 2021poster

When solving two-player zero-sum games, multi-agent reinforcement learning (MARL) algorithms often create populations of agents where, at each iteration, a new agent is discovered as the best response to a mixture over the opponent population. Within such a process, the update rules of "who to compe…

2021

Settling the Variance of Multi-Agent Policy Gradients

NeurIPS 2021poster

Policy gradient (PG) methods are popular reinforcement learning (RL) methods where a baseline is often applied to reduce the variance of gradient estimates. In multi-agent RL (MARL), although the PG theorem can be naturally extended, the effectiveness of multi-agent PG (MAPG) methods degrades as th…

2021

Towards Unifying Behavioral and Response Diversity for Open-ended Learning in Zero-sum Games

NeurIPS 2021poster

Measuring and promoting policy diversity is critical for solving games with strong non-transitive dynamics where strategic cycles exist, and there is no consistent winner (e.g., Rock-Paper-Scissors). With that in mind, maintaining a pool of diverse policies via open-ended learning is an attractive s…

2020

Modelling Bounded Rationality in Multi-Agent Interactions by Generalized Recursive Reasoning

IJCAI 2020poster

Though limited in real-world decision making, most multi-agent reinforcement learning (MARL) models assume perfectly rational agents -- a property hardly met due to individual's cognitive limitation and/or the tractability of the decision problem. In this paper, we introduce generalized recursive…

2020

Multi-Agent Determinantal Q-Learning

ICML 2020poster

Centralized training with decentralized execution has become an important paradigm in multi-agent learning. Though practical, current methods rely on restrictive assumptions to decompose the centralized value function across agents for execution. In this paper, we eliminate this restriction by propo…

2020

Q-value Path Decomposition for Deep Multiagent Reinforcement Learning

ICML 2020poster

Recently, deep multiagent reinforcement learning (MARL) has become a highly active research area as many real-world problems can be inherently viewed as multiagent systems. A particularly interesting and widely applicable class of problems is the partially observable cooperative multiagent setting,…

Cited by 73SourcePDFScholar
2020

Replica-Exchange Nos\'e-Hoover Dynamics for Bayesian Learning on Large Datasets

NeurIPS 2020poster

In this paper, we present a new practical method for Bayesian learning that can rapidly draw representative samples from complex posterior distributions with multiple isolated modes in the presence of mini-batch noise. This is achieved by simulating a collection of replicas in parallel with differen…

2020

SMARTS: An Open-Source Scalable Multi-Agent RL Training School for Autonomous Driving

CoRL 2020

Interaction is fundamental in autonomous driving (AD). Despite more than a decade of intensive R&D in AD, how to dynamically interact with diverse road users in various contexts still remains unsolved. Multi-agent learning has recently seen big breakthroughs and has much to offer towards solving rea

2019

Probabilistic Recursive Reasoning for Multi-Agent Reinforcement Learning

ICLR 2019poster

Humans are capable of attributing latent mental contents such as beliefs, or intentions to others. The social skill is critical in everyday life to reason about the potential consequences of their behaviors so as to plan ahead. It is known that humans use this reasoning ability recursively, i.e. con…

Cited by 196SourcePDFScholar
2018

Mean Field Multi-Agent Reinforcement Learning

ICML 2018oral

Existing multi-agent reinforcement learning methods are limited typically to a small number of agents. When the agent number increases largely, the learning becomes intractable due to the curse of the dimensionality and the exponential growth of agent interactions. In this paper, we present Mean Fie…

2018

Thermostat-assisted continuously-tempered Hamiltonian Monte Carlo for Bayesian learning

NeurIPS 2018poster

In this paper, we propose a novel sampling method, the thermostat-assisted continuously-tempered Hamiltonian Monte Carlo, for the purpose of multimodal Bayesian learning. It simulates a noisy dynamical system by incorporating both a continuously-varying tempering variable and the Nos\'e-Hoover therm…