← Search

Junshan Zhang

13 accepted papers

2026

VITA: Vision-to-Action Flow Matching Policy

ICLR 2026poster

Conventional flow matching and diffusion-based policies sample through iterative denoising from standard noise distributions (e.g., Gaussian), and require conditioning modules to repeatedly incorporate visual information during the generative process, incurring substantial time and memory overhead.…

Cited by 0SourcecodeScholar
2025

AdaWM: Adaptive World Model based Planning for Autonomous Driving

ICLR 2025poster

World model based reinforcement learning (RL) has emerged as a promising approach for autonomous driving, which learns a latent dynamics model and uses it to train a planning policy. To speed up the learning process, the pretrain-finetune paradigm is often used, where online RL is initialized by a…

Cited by 1SourcePDFScholar
2024

How to Leverage Diverse Demonstrations in Offline Imitation Learning

ICML 2024poster

Offline Imitation Learning (IL) with imperfect demonstrations has garnered increasing attention owing to the scarcity of expert data in many real-world domains. A fundamental problem in this scenario is *how to extract positive behaviors from noisy data*. In general, current approaches to the proble…

2024

OLLIE: Imitation Learning from Offline Pretraining to Online Finetuning

ICML 2024poster

In this paper, we study offline-to-online Imitation Learning (IL) that pretrains an imitation policy from static demonstration data, followed by fast finetuning with minimal environmental interaction. We find the naive combination of existing offline IL and online IL methods tends to behave poorly i…

2023

CLARE: Conservative Model-Based Reward Learning for Offline Inverse Reinforcement Learning

ICLR 2023poster

This work aims to tackle a major challenge in offline Inverse Reinforcement Learning (IRL), namely the reward extrapolation error, where the learned reward function may fail to explain the task correctly and misguide the agent in unseen environments due to the intrinsic covariate shift. Leveraging b…

Cited by 38SourcePDFScholar
2022

Beyond Not-Forgetting: Continual Learning with Backward Knowledge Transfer

NeurIPS 2022accept

By learning a sequence of tasks continually, an agent in continual learning (CL) can improve the learning performance of both a new task and `old' tasks by leveraging the forward knowledge transfer and the backward knowledge transfer, respectively. However, most existing CL methods focus on addressi…

Cited by 58SourcePDFScholar
2022

Long-term Spatio-Temporal Forecasting via Dynamic Multiple-Graph Attention

IJCAI 2022poster

Many real-world ubiquitous applications, such as parking recommendations and air pollution monitoring, benefit significantly from accurate long-term spatio-temporal forecasting (LSTF). LSTF makes use of long-term dependency structure between the spatial and temporal domains, as well as the contextua…

2022

Model-Based Offline Meta-Reinforcement Learning with Regularization

ICLR 2022poster

Existing offline reinforcement learning (RL) methods face a few major challenges, particularly the distributional shift between the learned policy and the behavior policy. Offline Meta-RL is emerging as a promising approach to address these challenges, aiming to learn an informative meta-policy from…

Cited by 24SourcePDFScholar
2022

TRGP: Trust Region Gradient Projection for Continual Learning

ICLR 2022spotlight

Catastrophic forgetting is one of the major challenges in continual learning. To address this issue, some existing methods put restrictive constraints on the optimization space of the new task for minimizing the interference to old tasks. However, this may lead to unsatisfactory performance for the…