← Search

Yuhang Jiang

15 accepted papers

2026

Adversarial Latent Embedding Repair for LLM Continual Learning

ICML 2026poster

Research on continual learning for LLMs seeks to acquire new skills without catastrophic forgetting of established prior knowledge. However, domain-specific fine-tuning still triggers severe, long-tailed forgetting issues even under narrow updates, particularly when the pre-training data is inaccess…

Cited by 0SourceScholar
2026

Small Generalizable Prompt Predictive Models Can Steer Efficient RL Post-Training of Large Reasoning Models

ICML 2026poster

Reinforcement learning enhances the reasoning capabilities of large language models but often involves high computational costs due to rollout-intensive optimization. Online prompt selection presents a plausible solution by prioritizing informative prompts to improve training efficiency. However, cu…

Cited by 0SourceScholar
2025

Latent Reward: LLM-Empowered Credit Assignment in Episodic Reinforcement Learning

AAAI 2025technical

Reinforcement learning (RL) often encounters delayed and sparse feedback in real-world applications, even with only episodic rewards. Previous approaches have made some progress in reward redistribution for credit assignment but still face challenges, including training difficulties due to redundan…

2024

Doubly Mild Generalization for Offline Reinforcement Learning

NeurIPS 2024poster

Offline Reinforcement Learning (RL) suffers from the extrapolation error and value overestimation. From a generalization perspective, this issue can be attributed to the over-generalization of value functions or policies towards out-of-distribution (OOD) actions. Significant efforts have been devote…

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

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

Near-Optimal Regret Bounds for Multi-batch Reinforcement Learning

NeurIPS 2022accept

In this paper, we study the episodic reinforcement learning (RL) problem modeled by finite-horizon Markov Decision Processes (MDPs) with constraint on the number of batches. The multi-batch reinforcement learning framework, where the agent is required to provide a time schedule to update policy befo…

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