← Search

Kangrui Wang

12 accepted papers

2026

ROSETTA: Constructing Code-Based Reward from Unconstrained Language Preference

ICLR 2026poster

Intelligent embodied agents not only need to accomplish preset tasks, but also learn to align with individual human needs and preferences. Extracting reward signals from human language preferences allows an embodied agent to adapt through reinforcement learning. However, human language preferences a…

Cited by 0SourcecodeScholar
2026

Understanding Reasoning Collapse in LLM Agent Reinforcement Learning

ICML 2026oral

In closed-loop multi-turn agent reinforcement learning, LLM agents exhibit reasoning collapse, where reasoning shift toward generic templates, weakly coupled to the inputs. We firstly identify that such collapse is easy to miss with entropy or surface diversity metrics since reasoning text still var…

Cited by 0SourceScholar
2026

Understanding VLMs Spatial Mental Modeling Capability from Limited Views

ICLR 2026poster

Can Vision Language Models (VLMs) imagine the full scene from just a few views, like humans do? Humans form spatial mental models, internal representations of unseen space, to reason about layout, perspective, and motion. Our new MindCube benchmark with 21,154 questions across 3,268 images exposes t…

Cited by 0SourcecodeScholar
2026

WorldAgen: Unified State-Action Prediction with Test-Time World Model Training

AAAI 2026technical

How can vision-language-action (VLA) models adapt to new environments where world dynamics shift? While recent research has combined world modeling and action prediction to improve VLA performance, existing methods largely rely on pretraining in static datasets, without mechanisms for active adaptat

Cited by 0SourcePDFScholar
2025

EmbodiedBench: Comprehensive Benchmarking Multi-modal Large Language Models for Vision-Driven Embodied Agents

ICML 2025oral

Leveraging Multi-modal Large Language Models (MLLMs) to create embodied agents offers a promising avenue for tackling real-world tasks. While language-centric embodied agents have garnered substantial attention, MLLM-based embodied agents remain underexplored due to the lack of comprehensive evaluat…

2025

VAGEN: Reinforcing World Model Reasoning for Multi-Turn VLM Agents

NeurIPS 2025poster

A major challenge in training VLM agents, compared to LLM agents, is that states shift from simple texts to complex visual observations, which introduces partial observability and demands robust world modeling. We ask: can VLM agents build internal world models through explicit visual state reasonin…

Cited by 0SourceScholar
2024

Embodied Agent Interface: Benchmarking LLMs for Embodied Decision Making

NeurIPS 2024oral

We aim to evaluate Large Language Models (LLMs) for embodied decision making. While a significant body of work has been leveraging LLMs for decision making in embodied environments, we still lack a systematic understanding of their performance because they are usually applied in different domains, f…

Cited by 33SourcePDFScholar
2023

Language Models Can Improve Event Prediction by Few-Shot Abductive Reasoning

NeurIPS 2023poster

Large language models have shown astonishing performance on a wide range of reasoning tasks. In this paper, we investigate whether they could reason about real-world events and help improve the prediction performance of event sequence models. We design LAMP, a framework that integrates a large langu…

Cited by 52SourcePDFScholar
2023

Towards the Difficulty for a Deep Neural Network to Learn Concepts of Different Complexities

NeurIPS 2023poster

This paper theoretically explains the intuition that simple concepts are more likely to be learned by deep neural networks (DNNs) than complex concepts. In fact, recent studies have observed [24, 15] and proved [26] the emergence of interactive concepts in a DNN, i.e., it is proven that a DNN usuall…

Cited by 19SourcePDFScholar
2019

Multi-resolution Multi-task Gaussian Processes

NeurIPS 2019poster

We consider evidence integration from potentially dependent observation processes under varying spatio-temporal sampling resolutions and noise levels. We offer a multi-resolution multi-task (MRGP) framework that allows for both inter-task and intra-task multi-resolution and multi-fidelity. We develo…