← Search

Jiafei Lyu

20 accepted papers

2026

Debiased Model-based Representations for Sample-efficient Continuous Control

ICML 2026poster

Model-based representations recently stand out as a promising framework that embeds latent dynamics information into the representations for downstream off-policy actor-critic learning. It implicitly combines the advantages of both model-free and model-based approaches while avoiding the training co…

Cited by 0SourceScholar
2026

Dual-Robust Cross-Domain Offline Reinforcement Learning Against Dynamics Shifts

ICLR 2026poster

Single-domain offline reinforcement learning (RL) often suffers from limited data coverage, while cross-domain offline RL handles this issue by leveraging additional data from other domains with dynamics shifts. However, existing studies primarily focus on train-time robustness (handling dynamics sh…

Cited by 0SourceScholar
2026

GenPRM: Scaling Test-Time Compute of Process Reward Models via Generative Reasoning

AAAI 2026technical

Recent advancements in Large Language Models (LLMs) have shown that it is promising to utilize Process Reward Models (PRMs) as verifiers to enhance the performance of LLMs. However, current PRMs face three key challenges: (1) limited process supervision and generalization capabilities, (2) dependenc

Cited by 0SourcePDFScholar
2026

Model-based Offline RL via Robust Value-Aware Model Learning with Implicitly Differentiable Adaptive Weighting

ICLR 2026poster

Model-based offline reinforcement learning (RL) aims to enhance offline RL with a dynamics model that facilitates policy exploration. However, model exploitation could occur due to inevitable model errors, which degrades algorithm performance. Adversarial model learning offers a theoretical framewor…

Cited by 0SourceScholar
2026

Unifying Value Alignment and Assignment in Cross-Domain Offline Reinforcement Learning with Heterogeneous Datasets

ICML 2026poster

Cross-domain offline reinforcement learning (RL) aims to train an agent that performs well in the target domain using a limited target domain dataset and a source domain dataset that exhibits a dynamics shift. Training directly on the original source dataset typically leads to performance collapse. …

Cited by 0SourceScholar
2025

ADG: Ambient Diffusion-Guided Dataset Recovery for Corruption-Robust Offline Reinforcement Learning

NeurIPS 2025poster

Real-world datasets collected from sensors or human inputs are prone to noise and errors, posing significant challenges for applying offline reinforcement learning (RL). While existing methods have made progress in addressing corrupted actions and rewards, they remain insufficient for handling corru…

Cited by 0SourceScholar
2025

Cross-Domain Offline Policy Adaptation with Optimal Transport and Dataset Constraint

ICLR 2025poster

We explore cross-domain offline reinforcement learning (RL) where offline datasets from another domain can be accessed to facilitate policy learning. However, the underlying environments of the two datasets may have dynamics mismatches, incurring inferior performance when simply merging the data of…

Cited by 1SourcePDFScholar
2025

Novelty-Guided Data Reuse for Efficient and Diversified Multi-Agent Reinforcement Learning

AAAI 2025technical

Recently, deep Multi-Agent Reinforcement Learning (MARL) has demonstrated its potential to tackle complex cooperative tasks, pushing the boundaries of AI in collaborative environments. However, the efficiency of these systems is often compromised by inadequate sample utilization and a lack of divers…

2025

SUMO: Search-Based Uncertainty Estimation for Model-Based Offline Reinforcement Learning

AAAI 2025technical

The performance of offline reinforcement learning (RL) suffers from the limited size and quality of static datasets. Model-based offline RL addresses this issue by generating synthetic samples through a dynamics model to enhance overall performance. To evaluate the reliability of the generated sampl…

2025

VLP: Vision-Language Preference Learning for Embodied Manipulation

EMNLP 2025

Reward engineering is one of the key challenges in Reinforcement Learning (RL). Preference-based RL effectively addresses this issue by learning from human feedback. However, it is both time-consuming and expensive to collect human preference labels. In this paper, we propose a novel V ision- L angu

2025

World Models with Hints of Large Language Models for Goal Achieving

NAACL 2025long

Reinforcement learning struggles in the face of long-horizon tasks and sparse goals due to the difficulty in manual reward specification. While existing methods address this by adding intrinsic rewards, they may fail to provide meaningful guidance in long-horizon decision-making tasks with large sta…

Cited by 2SourcePDFScholar
2024

Cross-Domain Policy Adaptation by Capturing Representation Mismatch

ICML 2024poster

It is vital to learn effective policies that can be transferred to different domains with dynamics discrepancies in reinforcement learning (RL). In this paper, we consider dynamics adaptation settings where there exists dynamics mismatch between the source domain and the target domain, and one can g…

2024

Exploration and Anti-Exploration with Distributional Random Network Distillation

ICML 2024poster

Exploration remains a critical issue in deep reinforcement learning for an agent to attain high returns in unknown environments. Although the prevailing exploration Random Network Distillation (RND) algorithm has been demonstrated to be effective in numerous environments, it often needs more discrim…

2024

ODRL: A Benchmark for Off-Dynamics Reinforcement Learning

NeurIPS 2024poster

We consider off-dynamics reinforcement learning (RL) where one needs to transfer policies across different domains with dynamics mismatch. Despite the focus on developing dynamics-aware algorithms, this field is hindered due to the lack of a standard benchmark. To bridge this gap, we introduce ODRL,…

2024

PEARL: Zero-shot Cross-task Preference Alignment and Robust Reward Learning for Robotic Manipulation

ICML 2024poster

In preference-based Reinforcement Learning (RL), obtaining a large number of preference labels are both time-consuming and costly. Furthermore, the queried human preferences cannot be utilized for the new tasks. In this paper, we propose Zero-shot Cross-task Preference Alignment and Robust Reward Le…

Cited by 5SourcePDFScholar
2024

SEABO: A Simple Search-Based Method for Offline Imitation Learning

ICLR 2024poster

Offline reinforcement learning (RL) has attracted much attention due to its ability in learning from static offline datasets and eliminating the need of interacting with the environment. Nevertheless, the success of offline RL relies heavily on the offline transitions annotated with reward labels. I…

2024

Using Human Feedback to Fine-tune Diffusion Models without Any Reward Model

CVPR 2024poster

Using reinforcement learning with human feedback (RLHF) has shown significant promise in fine-tuning diffusion models. Previous methods start by training a reward model that aligns with human preferences then leverage RL techniques to fine-tune the underlying models. However crafting an efficient re…

2022

Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imagination

NeurIPS 2022accept

The learned policy of model-free offline reinforcement learning (RL) methods is often constrained to stay within the support of datasets to avoid possible dangerous out-of-distribution actions or states, making it challenging to handle out-of-support region. Model-based RL methods offer a richer dat…

2022

Efficient Continuous Control with Double Actors and Regularized Critics

AAAI 2022technical

How to obtain good value estimation is a critical problem in Reinforcement Learning (RL). Current value estimation methods in continuous control, such as DDPG and TD3, suffer from unnecessary over- or under- estimation. In this paper, we explore the potential of double actors, which has been neglect…

2022

Mildly Conservative Q-Learning for Offline Reinforcement Learning

NeurIPS 2022accept

Offline reinforcement learning (RL) defines the task of learning from a static logged dataset without continually interacting with the environment. The distribution shift between the learned policy and the behavior policy makes it necessary for the value function to stay conservative such that out-o…