← Search

Hao Bai

7 accepted papers

2026

Patching Gaps In LLM Reasoning With Interventional Training

ICLR 2026poster

Reinforcement learning (RL) training of large language models (LLMs) is limited by the policy's ability to generate rollouts with non-zero rewards: without such rewards, the policy is not updated and learning is stalled on hard problems, which are problems that the policy consistently fails to sampl…

Cited by 0SourceScholar
2026

WebGym: Scaling Training Environments for Long-Horizon Visual Web Agents with Realistic Tasks

CVPR 2026

We present WebGym, the largest-to-date open-source environment for training realistic visual web agents. Real websites are non-stationary and diverse, making artificial or small-scale task sets insufficient for robust policy learning. WebGym contains nearly 300,000 tasks with rubric-based evaluation

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

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

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
2023

Social Commonsense-Guided Search Query Generation for Open-Domain Knowledge-Powered Conversations

EMNLP 2023short findings

Open-domain dialog involves generating search queries that help obtain relevant knowledge for holding informative conversations. However, it can be challenging to determine what information to retrieve when the user is passive and does not express a clear need or request. To tackle this issue, we pr…

Cited by 0SourceScholar