← Search

Shangtong Zhang

27 accepted papers

2026

Asymptotic and Finite Sample Analysis of Nonexpansive Stochastic Approximations with Markovian Noise

AAAI 2026technical

Stochastic approximation is a powerful class of algorithms with celebrated success. However, a large body of previous analysis focuses on stochastic approximations driven by contractive operators, which is not applicable in some important reinforcement learning settings like the average reward setti

Cited by 9SourcePDFScholar
2026

Convergence of Two-Timescale Stochastic Approximation with Markovian Samples and Applications in Reinforcement Learning

ICML 2026poster

Stochastic approximations (SA)--algorithms which derive their power through the use of random, incremental updates--are at the heart of reinforcement learning (RL). Expanding the theory of SA has established rigorous results concerning the most important algorithms in RL, including stochastic gradie…

Cited by 0SourceScholar
2026

MathlibLemma: Folklore Lemma Generation and Benchmark for Formal Mathematics

ICML 2026poster

While the ecosystem of Lean and Mathlib has enjoyed celebrated success in formal mathematical reasoning with the help of large language models (LLMs), the absence of many folklore lemmas in Mathlib remains a persistent barrier that limits Lean's usability as an everyday tool for mathematicians like …

Cited by 0SourceScholar
2026

Reward Is Enough: LLMs Are In-Context Reinforcement Learners

ICLR 2026poster

Reinforcement learning (RL) is a human-designed framework for solving sequential decision-making problems. In this work, we demonstrate that, surprisingly, RL emerges in LLMs at inference time – a phenomenon known as in-context RL (ICRL). To reveal this capability, we introduce a simple multi-round…

Cited by 0SourceScholar
2025

Counterfactual Explanations for Continuous Action Reinforcement Learning

IJCAI 2025

Reinforcement Learning (RL) has shown great promise in domains like healthcare and robotics but often struggles with adoption due to its lack of interpretability. Counterfactual explanations, which address ``what if” scenarios, provide a promising avenue for understanding RL decisions but remain und

2025

Efficient Policy Evaluation with Safety Constraint for Reinforcement Learning

ICLR 2025poster

In reinforcement learning, classic on-policy evaluation methods often suffer from high variance and require massive online data to attain the desired accuracy. Previous studies attempt to reduce evaluation variance by searching for or designing proper behavior policies to collect data. However, thes…

Cited by 3SourcePDFScholar
2025

Finite Sample Analysis of Linear Temporal Difference Learning with Arbitrary Features

NeurIPS 2025poster

Linear TD($\lambda$) is one of the most fundamental reinforcement learning algorithms for policy evaluation. Previously, convergence rates are typically established under the assumption of linearly independent features, which does not hold in many practical scenarios. This paper instead establishes…

Cited by 0SourceScholar
2025

Linear $Q$-Learning Does Not Diverge in $L^2$: Convergence Rates to a Bounded Set

ICML 2025poster

$Q$-learning is one of the most fundamental reinforcement learning algorithms. It is widely believed that $Q$-learning with linear function approximation (i.e., linear $Q$-learning) suffers from possible divergence until the recent work Meyn (2024) which establishes the ultimate almost sure boundedn…

Cited by 0SourcePDFScholar
2025

Transformers Can Learn Temporal Difference Methods for In-Context Reinforcement Learning

ICLR 2025poster

Traditionally, reinforcement learning (RL) agents learn to solve new tasks by updating their neural network parameters through interactions with the task environment. However, recent works demonstrate that some RL agents, after certain pretraining procedures, can learn to solve unseen new tasks with…

Cited by 8SourcePDFScholar
2023

On the Convergence of SARSA with Linear Function Approximation

ICML 2023poster

SARSA, a classical on-policy control algorithm for reinforcement learning, is known to chatter when combined with linear function approximation: SARSA does not diverge but oscillates in a bounded region. However, little is known about how fast SARSA converges to that region and how large the region…

Cited by 13SourcePDFScholar
2022

Learning Expected Emphatic Traces for Deep RL

AAAI 2022technical

Off-policy sampling and experience replay are key for improving sample efficiency and scaling model-free temporal difference learning methods. When combined with function approximation, such as neural networks, this combination is known as the deadly triad and is potentially unstable. Recently, it h…

Cited by 16SourcePDFScholar
2021

Average-Reward Off-Policy Policy Evaluation with Function Approximation

ICML 2021spotlight

We consider off-policy policy evaluation with function approximation (FA) in average-reward MDPs, where the goal is to estimate both the reward rate and the differential value function. For this problem, bootstrapping is necessary and, along with off-policy learning and FA, results in the deadly tri…

2021

Mean-Variance Policy Iteration for Risk-Averse Reinforcement Learning

AAAI 2021technical

We present a mean-variance policy iteration (MVPI) framework for risk-averse control in a discounted infinite horizon MDP optimizing the variance of a per-step reward random variable. MVPI enjoys great flexibility in that any policy evaluation method and risk-neutral control method can be dropped in…

2020

GradientDICE: Rethinking Generalized Offline Estimation of Stationary Values

ICML 2020poster

We present GradientDICE for estimating the density ratio between the state distribution of the target policy and the sampling distribution in off-policy reinforcement learning. GradientDICE fixes several problems of GenDICE (Zhang et al., 2020), the current state-of-the-art for estimating such densi…

2020

Learning Retrospective Knowledge with Reverse Reinforcement Learning

NeurIPS 2020poster

We present a Reverse Reinforcement Learning (Reverse RL) approach for representing retrospective knowledge. General Value Functions (GVFs) have enjoyed great success in representing predictive knowledge, i.e., answering questions about possible future outcomes such as “how much fuel will be consumed…

2020

Provably Convergent Two-Timescale Off-Policy Actor-Critic with Function Approximation

ICML 2020poster

We present the first provably convergent two-timescale off-policy actor-critic algorithm (COF-PAC) with function approximation. Key to COF-PAC is the introduction of a new critic, the emphasis critic, which is trained via Gradient Emphasis Learning (GEM), a novel combination of the key ideas of Grad…