← Search

Jianzhun Shao

11 accepted papers

2024

LLM-Empowered State Representation for Reinforcement Learning

ICML 2024poster

Conventional state representations in reinforcement learning often omit critical task-related details, presenting a significant challenge for value networks in establishing accurate mappings from states to task rewards. Traditional methods typically depend on extensive sample learning to enrich stat…

2023

Complementary Attention for Multi-Agent Reinforcement Learning

ICML 2023poster

In cooperative multi-agent reinforcement learning, centralized training with decentralized execution (CTDE) shows great promise for a trade-off between independent Q-learning and joint action learning. However, vanilla CTDE methods assumed a fixed number of agents could hardly adapt to real-world sc…

Cited by 10SourcePDFScholar
2023

Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning

NeurIPS 2023poster

Offline multi-agent reinforcement learning is challenging due to the coupling effect of both distribution shift issue common in offline setting and the high dimension issue common in multi-agent setting, making the action out-of-distribution (OOD) and value overestimation phenomenon excessively seve…

2023

DARL: Distance-Aware Uncertainty Estimation for Offline Reinforcement Learning

AAAI 2023technical

To facilitate offline reinforcement learning, uncertainty estimation is commonly used to detect out-of-distribution data. By inspecting, we show that current explicit uncertainty estimators such as Monte Carlo Dropout and model ensemble are not competent to provide trustworthy uncertainty estimation…

Cited by 4SourcePDFScholar
2023

Hokoff: Real Game Dataset from Honor of Kings and its Offline Reinforcement Learning Benchmarks

NeurIPS 2023poster

The advancement of Offline Reinforcement Learning (RL) and Offline Multi-Agent Reinforcement Learning (MARL) critically depends on the availability of high-quality, pre-collected offline datasets that represent real-world complexities and practical applications. However, existing datasets often fall…

2022

SPD: Synergy Pattern Diversifying Oriented Unsupervised Multi-agent Reinforcement Learning

NeurIPS 2022accept

Reinforcement learning typically relies heavily on a well-designed reward signal, which gets more challenging in cooperative multi-agent reinforcement learning. Alternatively, unsupervised reinforcement learning (URL) has delivered on its promise in the recent past to learn useful skills and explore…

Cited by 2SourcePDFScholar
2022

Self-Organized Group for Cooperative Multi-agent Reinforcement Learning

NeurIPS 2022accept

Centralized training with decentralized execution (CTDE) has achieved great success in cooperative multi-agent reinforcement learning (MARL) in practical applications. However, CTDE-based methods typically suffer from poor zero-shot generalization ability with dynamic team composition and varying pa…

Cited by 31SourcePDFScholar
2022

State Deviation Correction for Offline Reinforcement Learning

AAAI 2022technical

Offline reinforcement learning aims to maximize the expected cumulative rewards with a fixed collection of data. The basic principle of current offline reinforcement learning methods is to restrict the policy to the offline dataset action space. However, they ignore the case where the dataset's traj…

Cited by 22SourcePDFScholar
2022

Wasserstein Unsupervised Reinforcement Learning

AAAI 2022technical

Unsupervised reinforcement learning aims to train agents to learn a handful of policies or skills in environments without external reward. These pre-trained policies can accelerate learning when endowed with external reward, and can also be used as primitive options in hierarchical reinforcement lea…

Cited by 30SourcePDFScholar
2020

PFRL: Pose-Free Reinforcement Learning for 6D Pose Estimation

CVPR 2020poster

6D pose estimation from a single RGB image is a challenging and vital task in computer vision. The current mainstream deep model methods resort to 2D images annotated with real-world ground-truth 6D object poses, whose collection is fairly cumbersome and expensive, even unavailable in many cases. In…

Cited by 44PDFScholar
2020

Self6D: Self-Supervised Monocular 6D Object Pose Estimation

ECCV 2020poster

6D object pose estimation is a fundamental problem in computer vision. Convolutional Neural Networks (CNNs) have recently proven to be capable of predicting reliable 6D pose estimates even from monocular images. Nonetheless, CNNs are identified as being extremely data-driven, and acquiring adequate…