← Search

Lei Yuan

34 accepted papers

2026

Decentralized and Disentangled Task–Role Representation Learning for Generalizable Offline Multi-Agent Meta Reinforcement Learning

ICML 2026poster

Offline meta reinforcement learning (RL) enables agents to learn a unified policy from multi-task offline data to support generalization in out-of-distribution (OOD) tasks. Recent approaches in single-agent RL tackle this by learning an efficient task representation to distinguish between tasks, sho…

Cited by 0SourceScholar
2026

Hierarchical Value-Decomposed Offline Reinforcement Learning for Whole-Body Control

ICLR 2026poster

Scaling imitation learning to high-DoF whole-body robots is fundamentally limited by the \textbf{curse of dimensionality} and the prohibitive cost of collecting expert demonstrations. We argue that the core bottleneck is paradigmatic: real-world supervision for whole-body control is inherently imper…

Cited by 0SourceScholar
2026

Learning Disentangled Multi-Agent World Model for Decentralized Control

ICML 2026poster

World models enable learning policies via latent imagination, offering benefits such as history compression and sample efficiency. The primary challenge in applying world models to multi-agent tasks is that modeling multi-agent dynamics in latent space requires integrating information from different…

Cited by 0SourceScholar
2026

Multi-agent In-context Coordination via Decentralized Memory Retrieval

AAAI 2026technical

Large transformer models, trained on diverse datasets, have demonstrated impressive few-shot performance on previously unseen tasks without requiring parameter updates. This capability has also been explored in Reinforcement Learning (RL), where agents interact with the environment to retrieve conte

Cited by 0SourcePDFScholar
2026

Multi-view Consistent Latent Action Learning for World Modeling and Control

ICML 2026poster

The scalability of world models is currently bottlenecked by the scarcity of action annotations. While self-supervised latent action learning offers a potential solution, existing single-view paradigms—relying on information bottlenecks or Vector Quantization (VQ)—often conflate superficial 2D pixel…

Cited by 0SourceScholar
2026

Offline Multi-agent Continual Cooperation via Skill Partition and Reuse

ICML 2026poster

Extracting skills from multi-agent offline dataset improves learning efficiency via sharing task-invariant coordination skills among tasks. In settings where tasks occur sequentially and the space of skills grows exponentially, existing approaches that rely on heuristically designed and fixed-sized …

Cited by 0SourceScholar
2026

Provably Efficient Policy-Reward Co-Pretraining for Adversarial Imitation Learning

ICML 2026poster

Adversarial imitation learning (AIL) demonstrates superior expert sample efficiency compared to behavioral cloning (BC), yet requires substantial online environment interaction. While recent empirical work has explored initializing AIL algorithms with BC-pretrained policies to address this limitatio…

Cited by 0SourceScholar
2026

Speedup Patch: Learning a Plug-and-Play Policy to Accelerate Embodied Manipulation

ICML 2026poster

While current embodied policies exhibit remarkable manipulation skills, their execution remains unsatisfactorily slow as they inherit the tardy pacing of human demonstrations. Existing acceleration methods typically require policy retraining or costly online interactions, limiting their scalability …

Cited by 0SourceScholar
2026

Towards Complete Multi-Agent Coordination Policy Learning via Denoising Maximum Entropy Optimization

ICML 2026poster

Parameter sharing is a widely used technique in Multi-Agent Reinforcement Learning (MARL) that enhances sample efficiency by equipping agents with a unified policy. While effective in homogeneous settings, it often struggles in heterogeneous environments where agents possess diverse capabilities. Co…

Cited by 0SourceScholar
2025

Adaptable Safe Policy Learning from Multi-task Data with Constraint Prioritized Decision Transformer

NeurIPS 2025poster

Learning safe reinforcement learning (RL) policies from offline multi-task datasets without direct environmental interaction is crucial for efficient and reliable deployment of RL agents. Benefiting from their scalability and strong in-context learning capabilities, recent approaches attempt to util…

Cited by 0SourceScholar
2025

Efficient Multi-agent Offline Coordination via Diffusion-based Trajectory Stitching

ICLR 2025poster

Learning from offline data without interacting with the environment is a promising way to fully leverage the intelligent decision-making capabilities of multi-agent reinforcement learning (MARL). Previous approaches have primarily focused on developing learning techniques, such as conservative metho…

Cited by 1SourcePDFScholar
2025

Explainable Text Classification with LLMs: Enhancing Performance through Dialectical Prompting and Explanation-Guided Training

EMNLP 2025

Large Language Models (LLMs) have achieved impressive success across a range of natural language processing tasks. However, they still underperform in text classification tasks compared to fine-tuned small models. This can be linked to complexities in addressing context-dependent expressions and com

2025

LLM-Assisted Semantically Diverse Teammate Generation for Efficient Multi-agent Coordination

ICML 2025poster

Training with diverse teammates is the key for learning generalizable agents. Typical approaches aim to generate diverse teammates by utilizing techniques like randomization, designing regularization terms, or reducing policy compatibility, etc. However, such teammates lack semantic information, res…

2025

Learning to Reuse Policies in State Evolvable Environments

ICML 2025poster

The policy trained via reinforcement learning (RL) makes decisions based on sensor-derived state features. It is common for state features to evolve for reasons such as periodic sensor maintenance or the addition of new sensors for performance improvement. The deployed policy fails in new state spac…

Cited by 0SourcePDFScholar
2025

Multi-Agent Imitation by Learning and Sampling from Factorized Soft Q-Function

NeurIPS 2025poster

Learning from multi-agent expert demonstrations, known as Multi-Agent Imitation Learning (MAIL), provides a promising approach to sequential decision-making. However, existing MAIL methods including Behavior Cloning (BC) and Adversarial Imitation Learning (AIL) face significant challenges: BC suffer…

Cited by 0SourcecodeScholar
2025

Q-Adapter: Customizing Pre-trained LLMs to New Preferences with Forgetting Mitigation

ICLR 2025poster

Large Language Models (LLMs), trained on a large amount of corpus, have demonstrated remarkable abilities. However, it may not be sufficient to directly apply open-source LLMs like Llama to certain real-world scenarios, since most of them are trained for \emph{general} purposes. Thus, the demands fo…

2025

Sequential Multi-Agent Dynamic Algorithm Configuration

NeurIPS 2025poster

The performance of an algorithm often critically depends on its hyperparameter configuration. Dynamic algorithm configuration (DAC) is a recent trend in automated machine learning, which can dynamically adjust the algorithm’s configuration during the execution process and relieve users from tedious…

Cited by 0SourcecodeScholar
2025

SkillTree: Explainable Skill-Based Deep Reinforcement Learning for Long-Horizon Control Tasks

AAAI 2025technical

Deep reinforcement learning (DRL) has achieved remarkable success in various domains, yet its reliance on neural networks results in a lack of transparency, which limits its practical applications in safety-critical and human-agent interaction domains. Decision trees, known for their notable explain…

2024

Continual Multi-Objective Reinforcement Learning via Reward Model Rehearsal

IJCAI 2024poster

Multi-objective reinforcement learning (MORL) approaches address real-world problems with multiple objectives by learning policies maximizing returns weighted by different user preferences. Typical methods assume the objectives remain unchanged throughout the agent's lifetime. However, in some real-…

Cited by 0SourcePDFScholar
2024

Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics

ICML 2024poster

Developing policies that can adapt to non-stationary environments is essential for real-world reinforcement learning applications. Nevertheless, learning such adaptable policies in offline settings, with only a limited set of pre-collected trajectories, presents significant challenges. A key difficu…

2024

Multi-Agent Domain Calibration with a Handful of Offline Data

NeurIPS 2024poster

The shift in dynamics results in significant performance degradation of policies trained in the source domain when deployed in a different target domain, posing a challenge for the practical application of reinforcement learning (RL) in real-world scenarios. Domain transfer methods aim to bridge thi…

Cited by 0SourcePDFScholar
2024

Policy Rehearsing: Training Generalizable Policies for Reinforcement Learning

ICLR 2024poster

Human beings can make adaptive decisions in a preparatory manner, i.e., by making preparations in advance, which offers significant advantages in scenarios where both online and offline experiences are expensive and limited. Meanwhile, current reinforcement learning methods commonly rely on numerous…

Cited by 4SourcePDFScholar
2023

Discovering Generalizable Multi-agent Coordination Skills from Multi-task Offline Data

ICLR 2023poster

Cooperative multi-agent reinforcement learning (MARL) faces the challenge of adapting to multiple tasks with varying agents and targets. Previous multi-task MARL approaches require costly interactions to simultaneously learn or fine-tune policies in different tasks. However, the situation that an ag…

Cited by 31SourcePDFScholar
2023

Fast Teammate Adaptation in the Presence of Sudden Policy Change

UAI 2023poster

Cooperative multi-agent reinforcement learning (MARL), where agents coordinates with teammate(s) for a shared goal, may sustain non-stationary caused by the policy change of teammates. Prior works mainly concentrate on the policy change cross episodes, ignoring the fact that teammates may suffer f…

Cited by 13SourcePDFScholar
2023

Robust Multi-Agent Coordination via Evolutionary Generation of Auxiliary Adversarial Attackers

AAAI 2023technical

Cooperative Multi-agent Reinforcement Learning (CMARL) has shown to be promising for many real-world applications. Previous works mainly focus on improving coordination ability via solving MARL-specific challenges (e.g., non-stationarity, credit assignment, scalability), but ignore the policy pertur…

2022

A Neural Network Architecture for Program Understanding Inspired by Human Behaviors

ACL 2022long

Program understanding is a fundamental task in program language processing. Despite the success, existing works fail to take human behaviors as reference in understanding programs. In this paper, we consider human behaviors and propose the PGNN-EK model that consists of two main components. On the o…

2022

Efficient Multi-Agent Communication via Shapley Message Value

IJCAI 2022poster

Utilizing messages from teammates is crucial in cooperative multi-agent tasks due to the partially observable nature of the environment. Naively asking messages from all teammates without pruning may confuse individual agents, hindering the learning process and impairing the whole system's performan…

2022

Efficient Multi-agent Communication via Self-supervised Information Aggregation

NeurIPS 2022accept

Utilizing messages from teammates can improve coordination in cooperative Multi-agent Reinforcement Learning (MARL). To obtain meaningful information for decision-making, previous works typically combine raw messages generated by teammates with local information as inputs for policy. However, neglec…

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

Multi-agent Dynamic Algorithm Configuration

NeurIPS 2022accept

Automated algorithm configuration relieves users from tedious, trial-and-error tuning tasks. A popular algorithm configuration tuning paradigm is dynamic algorithm configuration (DAC), in which an agent learns dynamic configuration policies across instances by reinforcement learning (RL). However, i…

2021

Rank the Episodes: A Simple Approach for Exploration in Procedurally-Generated Environments

ICLR 2021poster

Exploration under sparse reward is a long-standing challenge of model-free reinforcement learning. The state-of-the-art methods address this challenge by introducing intrinsic rewards to encourage exploration in novel states or uncertain environment dynamics. Unfortunately, methods based on intrinsi…