← Search

Yifei Zhou

16 accepted papers

2026

Holistic Agent Leaderboard: The Missing Infrastructure for AI Agent Evaluation

ICLR 2026poster

AI agents have been developed for complex real-world tasks from coding to customer service. But AI agent evaluations suffer from many challenges that undermine our understanding of how well agents really work (Figure 1). We introduce the Holistic Agent Leaderboard (HAL) to address these challenges.…

Cited by 0SourcecodeScholar
2025

Digi-Q: Learning VLM Q-Value Functions for Training Device-Control Agents

ICLR 2025poster

While a number of existing approaches for building foundation model agents rely on prompting or fine-tuning with human demonstrations, it is not sufficient in dynamic environments (e.g., mobile device control). On-policy reinforcement learning (RL) should address these limitations, but collecting ac…

2025

KALIE: Fine-Tuning Vision-Language Models for Open-World Manipulation Without Robot Data

ICRA 2025

Building generalist robotic systems involves effectively endowing robots with the capabilities to handle novel objects in an open-world setting. Inspired by the advances of large pre-trained models, we propose Keypoint Affordance Learning from Imagined Environments (KALIE), which adapts pre-trained

Cited by 15SourceScholar
2025

Proposer-Agent-Evaluator (PAE): Autonomous Skill Discovery For Foundation Model Internet Agents

ICML 2025poster

A generalist foundation model agent needs to have a large and diverse skill repertoire, such as finding directions between two travel locations and buying specific items from the Internet. If each skill needs to be specified manually through a fixed set of human-annotated instructions, the agent’s s…

Cited by 11SourcePDFScholar
2025

Thinking vs. Doing: Improving Agent Reasoning by Scaling Test-Time Interaction

NeurIPS 2025poster

Test-time scaling in agentic tasks often relies on generating long reasoning traces ("think" more) before acting, but this does not allow agents to acquire new information from the environment or adapt behavior over time. In this work, we propose scaling test-time interaction, an untapped dimension…

Cited by 0SourceScholar
2024

Aligning Large Language Models with Representation Editing: A Control Perspective

NeurIPS 2024poster

Aligning large language models (LLMs) with human objectives is crucial for real-world applications. However, fine-tuning LLMs for alignment often suffers from unstable training and requires substantial computing resources. Test-time alignment techniques, such as prompting and guided decoding, do not…

2024

ArCHer: Training Language Model Agents via Hierarchical Multi-Turn RL

ICML 2024poster

Large language models (LLMs) have the potential to tackle sequential decision-making problems due to their generalist capabilities. Instead of optimizing ``myopic'' surrogate objectives such as human preferences within a single turn, in such problems, we wish to directly optimize long-term objective…

2024

DigiRL: Training In-The-Wild Device-Control Agents with Autonomous Reinforcement Learning

NeurIPS 2024poster

Pre-trained vision language models (VLMs), though powerful, typically lack training on decision-centric data, rendering them sub-optimal for decision-making tasks such as in-the-wild device control through Graphical User Interfaces (GUIs) when used off-the-shelf. While training with static demonstra…

Cited by 40SourcePDFScholar
2024

Fine-Tuning Large Vision-Language Models as Decision-Making Agents via Reinforcement Learning

NeurIPS 2024poster

Large vision-language models (VLMs) fine-tuned on specialized visual instruction-following data have exhibited impressive language reasoning capabilities across various scenarios. However, this fine-tuning paradigm may not be able to efficiently learn optimal decision-making agents in multi-step goa…

Cited by 68SourcePDFScholar
2024

Offline Data Enhanced On-Policy Policy Gradient with Provable Guarantees

ICLR 2024poster

Hybrid RL is the setting where an RL agent has access to both offline data and online data by interacting with the real-world environment. In this work, we propose a new hybrid RL algorithm that combines an on-policy actor-critic method with offline data. On-policy methods such as policy gradient an…

2023

BT^2: Backward-compatible Training with Basis Transformation

ICCV 2023poster

Modern retrieval system often requires recomputing the representation of every piece of data in the gallery when updating to a better representation model. This process is known as backfilling and can be especially costly in the real world where the gallery often contains billions of samples. Recent…

Cited by 6PDFcodeScholar
2023

Hybrid RL: Using both offline and online data can make RL efficient

ICLR 2023poster

We consider a hybrid reinforcement learning setting (Hybrid RL), in which an agent has access to an offline dataset and the ability to collect experience via real-world online interaction. The framework mitigates the challenges that arise in both pure offline and online RL settings, allowing for the…

2023

Test-Time Distribution Normalization for Contrastively Learned Visual-language Models

NeurIPS 2023poster

Advances in the field of visual-language contrastive learning have made it possible for many downstream applications to be carried out efficiently and accurately by simply taking the dot product between image and text representations. One of the most representative approaches proposed recently know…

2022

GAPX: Generalized Autoregressive Paraphrase-Identification X

NeurIPS 2022accept

Paraphrase Identification is a fundamental task in Natural Language Processing. While much progress has been made in the field, the performance of many state-of- the-art models often suffer from distribution shift during inference time. We verify that a major source of this performance drop comes fr…