← Search

Shaofei Cai

14 accepted papers

2026

CUARewardBench: Benchmark for Evaluating Reward Models on Computer-using Agent Trajectories

ICML 2026poster

Computer-using agents (CUAs) enable task completion through natural interaction with operating systems and software interfaces. While script-based verifiers are widely adopted for evaluation, they suffer from limited scalability and inability to provide step-wise assessment. Reward models offer prom…

Cited by 0SourceScholar
2026

Learn the Ropes, Then Trust the Wins: Self-imitation with Progressive Exploration for Agentic Reinforcement Learning

ICLR 2026poster

Reinforcement learning (RL) is the dominant paradigm for sharpening strategic tool use capabilities of LLMs on long-horizon, sparsely-rewarded agent tasks, yet it faces a fundamental challenge of exploration-exploitation trade-off. Existing studies stimulate exploration through the lens of policy en…

Cited by 0SourcecodeScholar
2026

Preference Goal Tuning: Post-Training as Latent Control for Frozen Policies

ICML 2026poster

Goal-conditioned policies enable decision-making models to execute diverse behaviors based on specified goals, yet their downstream performance is often highly sensitive to the choice of instructions or prompts. To bypass the limitations of discrete text prompts, we formulate post-training adaptatio…

Cited by 0SourceScholar
2026

Steering Visuomotor Policy in Open Worlds via Cross-View Goal Alignment

AAAI 2026technical

We aim to develop a goal specification method that is semantically clear, spatially sensitive, domain-agnostic, and intuitive for human users to guide agent interactions in 3D environments. Specifically, we propose a novel cross-view goal alignment framework that allows users to specify target objec

Cited by 0SourcePDFScholar
2026

UniCode: Augmenting Evaluation for Code Reasoning

ICML 2026poster

Current coding benchmarks often inflate Large Language Model (LLM) capabilities due to static paradigms and data contamination, enabling models to exploit statistical shortcuts rather than genuine reasoning. To address this, we introduce \textbf{UniCode}, a generative evaluation framework that syste…

Cited by 0SourceScholar
2025

GROOT-2: Weakly Supervised Multimodal Instruction Following Agents

ICLR 2025poster

Developing agents that can follow multimodal instructions remains a fundamental challenge in robotics and AI. Although large-scale pre-training on unlabeled datasets has enabled agents to learn diverse behaviors, these agents often struggle with following instructions. While augmenting the dataset w…

Cited by 2SourcePDFScholar
2025

Open-World Skill Discovery from Unsegmented Demonstration Videos

ICCV 2025poster

Learning skills in open-world environments is essential for developing agents capable of handling a variety of tasks by combining basic skills. Online demonstration videos are typically long but unsegmented, making them difficult to segment and label with skill identifiers. Unlike existing methods t…

2025

ROCKET-1: Mastering Open-World Interaction with Visual-Temporal Context Prompting

CVPR 2025poster

Vision-language models (VLMs) have excelled in multimodal tasks, but adapting them to embodied decision-making in open-world environments presents challenges. One critical issue is bridging the gap between discrete entities in low-level observations and the abstract concepts required for effective p…

2024

GROOT: Learning to Follow Instructions by Watching Gameplay Videos

ICLR 2024spotlight

We study the problem of building a controller that can follow open-ended instructions in open-world environments. We propose to follow reference videos as instructions, which offer expressive goal specifications while eliminating the need for expensive text-gameplay annotations. A new learning frame…

Cited by 37SourcePDFScholar
2024

OmniJARVIS: Unified Vision-Language-Action Tokenization Enables Open-World Instruction Following Agents

NeurIPS 2024poster

This paper presents OmniJARVIS, a novel Vision-Language-Action (VLA) model for open-world instruction-following agents in Minecraft. Compared to prior works that either emit textual goals to separate controllers or produce the control command directly, OmniJARVIS seeks a different path to ensure bot…

Cited by 2SourcePDFScholar
2023

Describe, Explain, Plan and Select: Interactive Planning with LLMs Enables Open-World Multi-Task Agents

NeurIPS 2023poster

In this paper, we study the problem of planning in Minecraft, a popular, democratized yet challenging open-ended environment for developing multi-task embodied agents. We've found two primary challenges of empowering such agents with planning: 1) planning in an open-ended world like Minecraft requir…

2023

Open-World Multi-Task Control Through Goal-Aware Representation Learning and Adaptive Horizon Prediction

CVPR 2023poster

We study the problem of learning goal-conditioned policies in Minecraft, a popular, widely accessible yet challenging open-ended environment for developing human-level multi-task agents. We first identify two main challenges of learning such policies: 1) the indistinguishability of tasks from the st…

2022

Automatic Relation-Aware Graph Network Proliferation

CVPR 2022oral

Graph neural architecture search has sparked much attention as Graph Neural Networks (GNNs) have shown powerful reasoning capability in many relational tasks. However, the currently used graph search space overemphasizes learning node features and neglects mining hierarchical relational information.…

Cited by 12PDFcodeScholar
2021

Rethinking Graph Neural Architecture Search From Message-Passing

CVPR 2021poster

Graph neural networks (GNNs) emerged recently as a standard toolkit for learning from data on graphs. Current GNN designing works depend on immense human expertise to explore different message-passing mechanisms, and require manual enumeration to determine the proper message-passing depth. Inspired…

Cited by 66PDFcodeScholar