← Search

Yuanheng Zhu

12 accepted papers

2026

Empowering Multi-Robot Cooperation via Sequential World Models

ICLR 2026poster

Model-based reinforcement learning (MBRL) has achieved remarkable success in robotics due to its high sample efficiency and planning capability. However, extending MBRL to physical multi-robot cooperation remains challenging due to the complexity of joint dynamics. To address this challenge, we prop…

Cited by 0SourcecodeScholar
2026

R2PS: Worst-Case Robust Real-Time Pursuit Strategies under Partial Observability

ICLR 2026poster

Computing worst-case robust strategies in pursuit-evasion games (PEGs) is time-consuming, especially when real-world factors like partial observability are considered. While important for general security purposes, real-time applicable pursuit strategies for graph-based PEGs are currently missing wh…

Cited by 0SourceScholar
2026

SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning

ICLR 2026poster

Large language models (LLMs) have achieved remarkable progress in reasoning tasks, yet optimally integrating Supervised Fine-Tuning (SFT) and Reinforcement Learning (RL) remains a fundamental challenge. Through a comprehensive analysis of token distributions, learning dynamics, and integration mecha…

Cited by 0SourcecodeScholar
2025

Constrained Exploitability Descent: An Offline Reinforcement Learning Method for Finding Mixed-Strategy Nash Equilibrium

ICML 2025poster

This paper proposes Constrained Exploitability Descent (CED), a model-free offline reinforcement learning (RL) algorithm for solving adversarial Markov games (MGs). CED combines the game-theoretical approach of Exploitability Descent (ED) with policy constraint methods from offline RL. While policy…

Cited by 0SourcePDFScholar
2025

DipLLM: Fine-Tuning LLM for Strategic Decision-making in Diplomacy

ICML 2025poster

Diplomacy is a complex multiplayer game that re- quires both cooperation and competition, posing significant challenges for AI systems. Traditional methods rely on equilibrium search to generate extensive game data for training, which demands substantial computational resources. Large Lan- guage Mod…

Cited by 0SourcePDFScholar
2025

Divergence-Regularized Discounted Aggregation: Equilibrium Finding in Multiplayer Partially Observable Stochastic Games

ICLR 2025poster

This paper presents Divergence-Regularized Discounted Aggregation (DRDA), a multi-round learning system for solving partially observable stochastic games (POSGs). DRDA is based on action values and applicable to multiplayer POSGs, which can unify normal-form games (NFGs), extensive-form games (EFGs)…

Cited by 0SourcePDFScholar
2025

Empowering LLM Agents with Zero-Shot Optimal Decision-Making through Q-learning

ICLR 2025poster

Large language models (LLMs) are trained on extensive text data to gain general comprehension capability. Current LLM agents leverage this ability to make zero- or few-shot decisions without reinforcement learning (RL) but fail in making optimal decisions, as LLMs inherently perform next-token predi…

Cited by 3SourcePDFScholar
2025

Equilibrium Policy Generalization: A Reinforcement Learning Framework for Cross-Graph Zero-Shot Generalization in Pursuit-Evasion Games

NeurIPS 2025poster

Equilibrium learning in adversarial games is an important topic widely examined in the fields of game theory and reinforcement learning (RL). Pursuit-evasion game (PEG), as an important class of real-world games from the fields of robotics and security, requires exponential time to be accurately sol…

Cited by 0SourceScholar
2025

INS: Interaction-aware Synthesis to Enhance Offline Multi-agent Reinforcement Learning

ICLR 2025poster

Data scarcity in offline multi-agent reinforcement learning (MARL) is a key challenge for real-world applications. Recent advances in offline single-agent reinforcement learning (RL) demonstrate the potential of data synthesis to mitigate this issue. However, in multi-agent systems, interactions bet…

Cited by 0SourcePDFScholar
2025

Learning and Planning Multi-Agent Tasks via an MoE-based World Model

NeurIPS 2025poster

Multi-task multi-agent reinforcement learning (MT-MARL) aims to develop a single model capable of solving a diverse set of tasks. However, existing methods often fall short due to the substantial variation in optimal policies across tasks, making it challenging for a single policy model to generaliz…

Cited by 0SourcecodeScholar
2025

RLAE: Reinforcement Learning-Assisted Ensemble for LLMs

EMNLP 2025

Ensembling large language models (LLMs) can effectively combine diverse strengths of different models, offering a promising approach to enhance performance across various tasks. However, existing methods typically rely on fixed weighting strategies that fail to adapt to the dynamic, context-dependen

Cited by 0SourcePDFScholar
2024

Meta-DT: Offline Meta-RL as Conditional Sequence Modeling with World Model Disentanglement

NeurIPS 2024poster

A longstanding goal of artificial general intelligence is highly capable generalists that can learn from diverse experiences and generalize to unseen tasks. The language and vision communities have seen remarkable progress toward this trend by scaling up transformer-based models trained on massive d…