← Search

Longtao Zheng

9 accepted papers

2026

Dr. Kernel: Reinforcement Learning Done Right for Triton Kernel Generations

ICML 2026poster

High-quality kernel is critical for scalable AI systems, and enabling LLMs to generate such code would advance AI development. However, training LLMs for this task requires sufficient data, a robust environment, and the process is often vulnerable to _reward hacking_ and _lazy optimization_. In thes…

Cited by 0SourceScholar
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

Cradle: Empowering Foundation Agents towards General Computer Control

ICML 2025poster

Despite their success in specific scenarios, existing foundation agents still struggle to generalize across various virtual scenarios, mainly due to the dramatically different encapsulations of environments with manually designed observation and action spaces. To handle this issue, we propose the Ge…

2025

Towards Efficient Online Tuning of VLM Agents via Counterfactual Soft Reinforcement Learning

ICML 2025poster

Online fine-tuning vision-language model (VLM) agents with reinforcement learning (RL) has shown promise for equipping agents with multi-step, goal-oriented capabilities in dynamic environments. However, their open-ended textual action space and non-end-to-end nature of action generation present sig…

2024

Synapse: Trajectory-as-Exemplar Prompting with Memory for Computer Control

ICLR 2024poster

Building agents with large language models (LLMs) for computer control is a burgeoning research area, where the agent receives computer states and performs actions to complete complex tasks. Previous computer agents have demonstrated the benefits of in-context learning (ICL); however, their performa…

2024

True Knowledge Comes from Practice: Aligning Large Language Models with Embodied Environments via Reinforcement Learning

ICLR 2024poster

Despite the impressive performance across numerous tasks, large language models (LLMs) often fail in solving simple decision-making tasks due to the misalignment of the knowledge in LLMs with environments. On the contrary, reinforcement learning (RL) agents learn policies from scratch, which makes t…

Cited by 47SourcePDFScholar
2023

Controlling Type Confounding in Ad Hoc Teamwork with Instance-wise Teammate Feedback Rectification

ICML 2023poster

Ad hoc teamwork requires an agent to cooperate with unknown teammates without prior coordination. Many works propose to abstract teammate instances into high-level representation of types and then pre-train the best response for each type. However, most of them do not consider the distribution of te…

Cited by 2SourcePDFScholar