← Search

Chenjun Xiao

24 accepted papers

2026

An MRP Formulation for Supervised Learning: Generalized Temporal Difference Learning Models (Abstract Reprint)

AAAI 2026technical

Background: Traditional supervised learning (SL) assumes data points are independently and identically distributed (i.i.d.), which overlooks dependencies in real-world data. Reinforcement learning (RL), in contrast, models dependencies through state transitions. Objectives: This study aims to bridge

Cited by 0SourcePDFScholar
2026

Bottleneck-Guided Spectral Subgoals For Offline Goal-Conditioned RL

ICML 2026poster

Offline goal-conditioned RL (OGCRL) learns to reach arbitrary goals from offline dataset, but long-horizon performance hinges on crossing a handful of hard-to-cross bottlenecks. These bottlenecks not only dictate the feasible paths toward the goal but also act as critical keypoints, marking the tran…

Cited by 0SourceScholar
2026

ScaleMoE: Mixture-of-Experts for Scalable Continuous Control in Actor-Critic Reinforcement Learning

ICML 2026spotlight

Scaling network remains a bottleneck in deep reinforcement learning (RL): simply enlarging actor–critic networks destabilizes training and soon saturates performance. Although recent monolithic architectures such as SimBa and BRC have shown that carefully designed inductive biases can enable positiv…

Cited by 0SourceScholar
2025

Behavior-Regularized Diffusion Policy Optimization for Offline Reinforcement Learning

ICML 2025poster

Behavior regularization, which constrains the policy to stay close to some behavior policy, is widely used in offline reinforcement learning (RL) to manage the risk of hazardous exploitation of unseen actions. Nevertheless, existing literature on behavior-regularized RL primarily focuses on explicit…

Cited by 1SourcePDFScholar
2024

Diffusion Spectral Representation for Reinforcement Learning

NeurIPS 2024poster

Diffusion-based models have achieved notable empirical successes in reinforcement learning (RL) due to their expressiveness in modeling complex distributions. Despite existing methods being promising, the key challenge of extending existing methods for broader real-world applications lies in the com…

Cited by 1SourcePDFScholar
2024

Exploiting the Replay Memory Before Exploring the Environment: Enhancing Reinforcement Learning Through Empirical MDP Iteration

NeurIPS 2024poster

Reinforcement learning (RL) algorithms are typically based on optimizing a Markov Decision Process (MDP) using the optimal Bellman equation. Recent studies have revealed that focusing the optimization of Bellman equations solely on in-sample actions tends to result in more stable optimization, espec…

Cited by 0SourcePDFScholar
2024

HarmonyDream: Task Harmonization Inside World Models

ICML 2024poster

Model-based reinforcement learning (MBRL) holds the promise of sample-efficient learning by utilizing a world model, which models how the environment works and typically encompasses components for two tasks: observation modeling and reward modeling. In this paper, through a dedicated empirical inves…

2024

Iteratively Refined Behavior Regularization for Offline Reinforcement Learning

NeurIPS 2024poster

One of the fundamental challenges for offline reinforcement learning (RL) is ensuring robustness to data distribution. Whether the data originates from a near-optimal policy or not, we anticipate that an algorithm should demonstrate its ability to learn an effective control policy that seamlessly al…

Cited by 7SourcePDFScholar
2024

Multiagent Gumbel MuZero: Efficient Planning in Combinatorial Action Spaces

AAAI 2024technical

AlphaZero and MuZero have achieved state-of-the-art (SOTA) performance in a wide range of domains, including board games and robotics, with discrete and continuous action spaces. However, to obtain an improved policy, they often require an excessively large number of simulations, especially for doma…

2024

Provable Representation with Efficient Planning for Partially Observable Reinforcement Learning

ICML 2024poster

In most real-world reinforcement learning applications, state information is only partially observable, which breaks the Markov decision process assumption and leads to inferior performance for algorithms that conflate observations with state. Partially Observable Markov Decision Processes (POMDPs),…

Cited by 8SourcePDFScholar
2024

Rethinking Decision Transformer via Hierarchical Reinforcement Learning

ICML 2024poster

Decision Transformer (DT) is an innovative algorithm leveraging recent advances of the transformer architecture in reinforcement learning (RL). However, a notable limitation of DT is its reliance on recalling trajectories from datasets, losing the capability to seamlessly stitch sub-optimal trajecto…

Cited by 16SourcePDFScholar
2024

Target Networks and Over-parameterization Stabilize Off-policy Bootstrapping with Function Approximation

ICML 2024spotlight

We prove that the combination of a target network and over-parameterized linear function approximation establishes a weaker convergence condition for bootstrapped value estimation in certain cases, even with off-policy data. Our condition is naturally satisfied for expected updates over the entire s…

2023

Conditionally optimistic exploration for cooperative deep multi-agent reinforcement learning

UAI 2023poster

Efficient exploration is critical in cooperative deep Multi-Agent Reinforcement Learning (MARL). In this work, we propose an exploration method that effectively encourages cooperative exploration based on the idea of sequential action-computation scheme. The high-level intuition is that to perform o…

2023

Energy-based Predictive Representations for Partially Observed Reinforcement Learning

UAI 2023poster

In real-world applications, handling partial observability is a common requirement for reinforcement learning algorithms, which is not captured by a Markov decision process (MDP). Although partially observable Markov decision processes (POMDPs) have been specifically designed to address this require…

Cited by 4SourcePDFScholar
2023

Latent Variable Representation for Reinforcement Learning

ICLR 2023poster

Deep latent variable models have achieved significant empirical successes in model-based reinforcement learning (RL) due to their expressiveness in modeling complex transition dynamics. On the other hand, it remains unclear theoretically and empirically how latent variable models may facilitate lear…

Cited by 12SourcePDFScholar
2023

Replay Memory as An Empirical MDP: Combining Conservative Estimation with Experience Replay

ICLR 2023poster

Experience replay, which stores transitions in a replay memory for repeated use, plays an important role of improving sample efficiency in reinforcement learning. Existing techniques such as reweighted sampling, episodic learning and reverse sweep update further process the information in the replay…

Cited by 11SourcePDFScholar
2023

The In-Sample Softmax for Offline Reinforcement Learning

ICLR 2023top-25%

Reinforcement learning (RL) agents can leverage batches of previously collected data to extract a reasonable control policy. An emerging issue in this offline RL setting, however, is that the bootstrapping update underlying many of our methods suffers from insufficient action-coverage: standard max…

2022

The Curse of Passive Data Collection in Batch Reinforcement Learning

AISTATS 2022poster

In high stake applications, active experimentation may be considered too risky and thus data are often collected passively. While in simple cases, such as in bandits, passive and active data collection are similarly effective, the price of passive sampling can be much higher when collecting data fro…

Cited by 20SourcePDFScholar
2022

Understanding and Leveraging Overparameterization in Recursive Value Estimation

ICLR 2022poster

The theory of function approximation in reinforcement learning (RL) typically considers low capacity representations that incur a tradeoff between approximation error, stability and generalization. Current deep architectures, however, operate in an overparameterized regime where approximation error…

Cited by 18SourcePDFScholar
2021

On the Optimality of Batch Policy Optimization Algorithms

ICML 2021spotlight

Batch policy optimization considers leveraging existing data for policy construction before interacting with an environment. Although interest in this problem has grown significantly in recent years, its theoretical foundations remain under-developed. To advance the understanding of this problem, we…

Cited by 37SourcePDFScholar
2021

Understanding the Effect of Stochasticity in Policy Optimization

NeurIPS 2021poster

We study the effect of stochasticity in on-policy policy optimization, and make the following four contributions. \emph{First}, we show that the preferability of optimization methods depends critically on whether stochastic versus exact gradients are used. In particular, unlike the true gradient set…

Cited by 28SourcePDFScholar
2020

Escaping the Gravitational Pull of Softmax

NeurIPS 2020oral

The softmax is the standard transformation used in machine learning to map real-valued vectors to categorical distributions. Unfortunately, this transform poses serious drawbacks for gradient descent (ascent) optimization. We reveal this difficulty by establishing two negative results: (1) optimizin…

Cited by 66SourcePDFScholar
2020

On the Global Convergence Rates of Softmax Policy Gradient Methods

ICML 2020poster

We make three contributions toward better understanding policy gradient methods in the tabular setting. First, we show that with the true gradient, policy gradient with a softmax parametrization converges at a $O(1/t)$ rate, with constants depending on the problem and initialization. This result sig…

Cited by 349SourcePDFScholar
2019

Maximum Entropy Monte-Carlo Planning

NeurIPS 2019poster

We develop a new algorithm for online planning in large scale sequential decision problems that improves upon the worst case efficiency of UCT. The idea is to augment Monte-Carlo Tree Search (MCTS) with maximum entropy policy optimization, evaluating each search node by softmax values back-propagat…

Cited by 45SourcePDFScholar