← Search

Zhenghai Xue

10 accepted papers

2026

SimpleTIR: End-to-End Reinforcement Learning for Multi-Turn Tool-Integrated Reasoning

ICLR 2026poster

Large Language Models (LLMs) can enhance their reasoning by interacting with external tools, a paradigm known as Tool-Integrated Reasoning (TIR). However, extending TIR to multi-turn settings using Reinforcement Learning (RL) often exhibits training instability and degraded performance. We attribute…

Cited by 0SourcecodeScholar
2026

The Optimal Token Baseline: Variance Reduction for Long-Horizon LLM-RL

ICML 2026poster

Reinforcement Learning for Large Language Models (LLMs) often suffers from training collapse in long-horizon tasks due to exploding gradient variance. To mitigate this, baseline is commonly introduced for advantage computation; however, traditional value models remain difficult to optimize, and stan…

Cited by 0SourceScholar
2025

AgentStudio: A Toolkit for Building General Virtual Agents

ICLR 2025poster

General virtual agents need to handle multimodal observations, master complex action spaces, and self-improve in dynamic, open-domain environments. However, existing environments are often domain-specific and require complex setups, which limits agent development and evaluation in real-world setting…

Cited by 31SourcePDFScholar
2025

Policy Optimization under Imperfect Human Interactions with Agent-Gated Shared Autonomy

ICLR 2025poster

We introduce AGSA, an Agent-Gated Shared Autonomy framework that learns from high-level human feedback to tackle the challenges of reward-free training, safe exploration, and imperfect low-level human control. Recent human-in-the loop learning methods enable human participants to intervene a learnin…

Cited by 0SourcePDFScholar
2025

Policy Regularization on Globally Accessible States in Cross-Dynamics Reinforcement Learning

ICML 2025spotlight

To learn from data collected in diverse dynamics, Imitation from Observation (IfO) methods leverage expert state trajectories based on the premise that recovering expert state distributions in other dynamics facilitates policy learning in the current one. However, Imitation Learning inherently impos…

Cited by 0SourcePDFScholar
2024

S$2$AC: Energy-Based Reinforcement Learning with Stein Soft Actor Critic

ICLR 2024poster

Learning expressive stochastic policies instead of deterministic ones has been proposed to achieve better stability, sample complexity and robustness. Notably, in Maximum Entropy reinforcement learning (MaxEnt RL), the policy is modeled as an expressive energy-based model (EBM) over the Q-values. Ho…

2023

Guarded Policy Optimization with Imperfect Online Demonstrations

ICLR 2023top-25%

The Teacher-Student Framework (TSF) is a reinforcement learning setting where a teacher agent guards the training of a student agent by intervening and providing online demonstrations. Assuming optimal, the teacher policy has the perfect timing and capability to intervene in the learning process of…

2023

State Regularized Policy Optimization on Data with Dynamics Shift

NeurIPS 2023poster

In many real-world scenarios, Reinforcement Learning (RL) algorithms are trained on data with dynamics shift, i.e., with different underlying environment dynamics. A majority of current methods address such issue by training context encoders to identify environment parameters. Data with dynamics shi…

Cited by 17SourcePDFScholar
2021

Regret Minimization Experience Replay in Off-Policy Reinforcement Learning

NeurIPS 2021poster

In reinforcement learning, experience replay stores past samples for further reuse. Prioritized sampling is a promising technique to better utilize these samples. Previous criteria of prioritization include TD error, recentness and corrective feedback, which are mostly heuristically designed. In thi…