← Search

Chengxing Jia

11 accepted papers

2026

ADM-v2: Pursuing Full-Horizon Roll-out in Dynamics Models for Offline Policy Learning and Evaluation

ICLR 2026poster

Model-based methods for offline Reinforcement Learning transfer extensive policy exploration and evaluation to data-driven dynamics models, effectively saving real-world samples in the offline setting. We expect the dynamics model to allow the policy to roll out full-horizon episodes, which is cruci…

Cited by 0SourcecodeScholar
2026

Clipping Low-Probability Tokens in SFT Yields a Generalizable Initialization for RL

ICML 2026poster

Supervised Fine-Tuning (SFT) is a critical step for adapting Large Language Models (LLMs) to specialized domains, often serving as an initialization for subsequent reinforcement learning (RL). However, SFT can overfit a small set of expert data, harming generalization and eroding prior knowledge. Th…

Cited by 0SourceScholar
2025

Any-step Dynamics Model Improves Future Predictions for Online and Offline Reinforcement Learning

ICLR 2025poster

Model-based methods in reinforcement learning offer a promising approach to enhance data efficiency by facilitating policy exploration within a dynamics model. However, accurately predicting sequential steps in the dynamics model remains a challenge due to the bootstrapping prediction, which attribu…

2025

Controlling Large Language Model with Latent Action

ICML 2025poster

Adapting Large Language Models (LLMs) to downstream tasks using Reinforcement Learning (RL) has proven to be an effective approach. However, LLMs do not inherently define the structure of an agent for RL training, particularly in terms of specifying the action space. This paper studies learning a co…

Cited by 0SourcePDFScholar
2025

Q-Adapter: Customizing Pre-trained LLMs to New Preferences with Forgetting Mitigation

ICLR 2025poster

Large Language Models (LLMs), trained on a large amount of corpus, have demonstrated remarkable abilities. However, it may not be sufficient to directly apply open-source LLMs like Llama to certain real-world scenarios, since most of them are trained for \emph{general} purposes. Thus, the demands fo…

2024

Debiased Offline Representation Learning for Fast Online Adaptation in Non-stationary Dynamics

ICML 2024poster

Developing policies that can adapt to non-stationary environments is essential for real-world reinforcement learning applications. Nevertheless, learning such adaptable policies in offline settings, with only a limited set of pre-collected trajectories, presents significant challenges. A key difficu…

2024

Offline Transition Modeling via Contrastive Energy Learning

ICML 2024poster

Learning a high-quality transition model is of great importance for sequential decision-making tasks, especially in offline settings. Nevertheless, the complex behaviors of transition dynamics in real-world environments pose challenges for the standard forward models because of their inductive bias…

Cited by 3SourcePDFScholar
2024

Policy Rehearsing: Training Generalizable Policies for Reinforcement Learning

ICLR 2024poster

Human beings can make adaptive decisions in a preparatory manner, i.e., by making preparations in advance, which offers significant advantages in scenarios where both online and offline experiences are expensive and limited. Meanwhile, current reinforcement learning methods commonly rely on numerous…

Cited by 4SourcePDFScholar
2023

Discovering Generalizable Multi-agent Coordination Skills from Multi-task Offline Data

ICLR 2023poster

Cooperative multi-agent reinforcement learning (MARL) faces the challenge of adapting to multiple tasks with varying agents and targets. Previous multi-task MARL approaches require costly interactions to simultaneously learn or fine-tune policies in different tasks. However, the situation that an ag…

Cited by 31SourcePDFScholar
2023

Fast Teammate Adaptation in the Presence of Sudden Policy Change

UAI 2023poster

Cooperative multi-agent reinforcement learning (MARL), where agents coordinates with teammate(s) for a shared goal, may sustain non-stationary caused by the policy change of teammates. Prior works mainly concentrate on the policy change cross episodes, ignoring the fact that teammates may suffer f…

Cited by 13SourcePDFScholar
2023

Model-Bellman Inconsistency for Model-based Offline Reinforcement Learning

ICML 2023poster

For offline reinforcement learning (RL), model-based methods are expected to be data-efficient as they incorporate dynamics models to generate more data. However, due to inevitable model errors, straightforwardly learning a policy in the model typically fails in the offline setting. Previous studies…