activity
20242026
collaborators
Showing cs.LGShow all

8 papers · 1 filter

cs.LG2026

MLREF: Efficient Module Reuse for Reward Design in Reinforcement Learning via Large Language Models

Chenglin Liu, Xun Wang, Ruishuo Chen +2

Reward function design remains a bottleneck in reinforcement learning. While large language models (LLMs) have enabled automated reward generation, existing methods generate and re…

cs.LG2026

Reparameterization Flow Policy Optimization

Hai Zhong, Zhuoran Li, Xun Wang +1

Reparameterization Policy Gradient (RPG) has emerged as a powerful paradigm for model-based reinforcement learning, enabling high sample efficiency by backpropagating gradients thr…

cs.LG2025

From Solo to Symphony: Orchestrating Multi-Agent Collaboration with Single-Agent Demos

Xun Wang, Zhuoran Li, Yanshan Lin +2

Training a team of agents from scratch in multi-agent reinforcement learning (MARL) is highly inefficient, much like asking beginners to play a symphony together without first prac…

cs.LG2025

OM2P: Offline Multi-Agent Mean-Flow Policy

Zhuoran Li, Xun Wang, Hai Zhong +3

Generative models, especially diffusion and flow-based models, have been promising in offline multi-agent reinforcement learning. However, integrating powerful generative models in…

cs.LG2025

Reparameterization Proximal Policy Optimization

Hai Zhong, Xun Wang, Zhuoran Li +1

By leveraging differentiable dynamics, Reparameterization Policy Gradient (RPG) achieves high sample efficiency. However, current approaches are hindered by two critical limitation…

cs.LG2025

Beyond the Proxy: Trajectory-Distilled Guidance for Offline GFlowNet Training

Ruishuo Chen, Xun Wang, Rui Hu +2

Generative Flow Networks (GFlowNets) excel at sampling diverse, high-reward objects. In many practical applications where active reward queries are infeasible, these models must be…

cs.LG2025

Beyond Shallow Behavior: Task-Efficient Value-Based Multi-Task Offline MARL via Skill Discovery

Xun Wang, Zhuoran Li, Hai Zhong +1

As a data-driven approach, offline MARL learns superior policies solely from offline datasets, ideal for domains rich in historical data but with high interaction costs and risks.…

cs.LG2024

Beyond Squared Error: Exploring Loss Design for Enhanced Training of Generative Flow Networks

Rui Hu, Yifan Zhang, Zhuoran Li +1

Generative Flow Networks (GFlowNets) are a novel class of generative models designed to sample from unnormalized distributions and have found applications in various important task…