← Search

Tianmin Shu

40 accepted papers

2026

MindZero: Learning Online Mental Reasoning With Zero Annotations

ICML 2026poster

Effective real-world assistance requires AI agents with robust Theory of Mind (ToM): inferring human mental states from their behavior. Despite recent advances, several key challenges remain, including (1) online inference with robust uncertainty updates over multiple hypotheses; (2) efficient reaso…

Cited by 0SourceScholar
2026

Pragmatic Embodied Spoken Instruction Following in Human-Robot Collaboration with Theory of Mind

ICRA 2026poster

Spoken language instructions are ubiquitous in agent collaboration. However, in real-world human-robot collaboration, following human spoken instructions can be challenging due to various speaker and environmental factors, such as background noise or mispronunciation. When faced with noisy auditory …

2026

RealWebAssist: A Benchmark for Long-Horizon Web Assistance with Real-World Users

AAAI 2026technical

To achieve successful assistance with long-horizon web-based tasks, AI agents must be able to sequentially follow real-world user instructions over a long period. Unlike existing web-based agent benchmarks, sequential instruction following in the real world poses significant challenges beyond perfor

Cited by 0SourcePDFScholar
2026

WoW!: World Models in a Closed-Loop World

ICLR 2026oral

Generative world models (WMs) can now simulate worlds with striking visual realism, which naturally raises the question of whether they can endow embodied agents with predictive perception for decision making. Progress on this question has been limited by fragmented evaluation: most existing benchma…

Cited by 0SourcecodeScholar
2025

AutoToM: Scaling Model-based Mental Inference via Automated Agent Modeling

NeurIPS 2025spotlight

Theory of Mind (ToM), the ability to understand people's minds based on their behavior, is key to developing socially intelligent agents. Current approaches to ToM reasoning either rely on prompting Large Language Models (LLMs), which are prone to systematic errors, or use handcrafted, rigid agent m…

Cited by 0SourceScholar
2025

COMBO: Compositional World Models for Embodied Multi-Agent Cooperation

ICLR 2025poster

In this paper, we investigate the problem of embodied multi-agent cooperation, where decentralized agents must cooperate given only egocentric views of the world. To effectively plan in this setting, in contrast to learning world dynamics in a single-agent scenario, we must simulate world dynamics c…

2025

Do Vision-Language Models Have Internal World Models? Towards an Atomic Evaluation

ACL 2025finding

Internal world models (WMs) enable agents to understand the world’s state and predict transitions, serving as the basis for advanced deliberative reasoning.Recent large Vision-Language Models (VLMs), such as GPT-4o and Gemini, exhibit potential as general-purpose WMs. While the latest studies have e…

Cited by 0SourcePDFScholar
2025

MuMA-ToM: Multi-modal Multi-Agent Theory of Mind

AAAI 2025technical

Understanding people's social interactions in complex real-world scenarios often relies on intricate mental reasoning. To truly understand how and why people interact with one another, we must infer the underlying mental states that give rise to the social interactions, i.e., Theory of Mind reasonin…

2025

PartInstruct: Part-level Instruction Following for Fine-grained Robot Manipulation

RSS 2025poster

Fine-grained robot manipulation, such as lifting and rotating a bottle to display the label on the cap, requires robust reasoning about object parts and their relationships with intended tasks. Despite recent advances in training general-purpose robot manipulation policies guided by language instruc…

Cited by 0PDFScholar
2025

SimWorld: An Open-ended Simulator for Agents in Physical and Social Worlds

NeurIPS 2025spotlight

While LLM/VLM-powered AI agents have advanced rapidly in math, coding, and computer use, their applications in complex physical and social environments remain challenging. Building agents that can survive and thrive in the real world (e.g., by autonomously earning income) requires massive-scale inte…

Cited by 0SourcecodeScholar
2025

Synthesizing Photorealistic and Dynamic Urban Environments for Multimodal Robot Navigation and Collaboration

NeurIPS 2025poster

Recent advances in foundation models have shown promising results in developing generalist robotics that can perform diverse tasks in open-ended scenarios given multimodal inputs. However, current work has been mainly focused on indoor, household scenarios. In this work, we present SimWorld-Robotics…

Cited by 0SourcecodeScholar
2024

Building Cooperative Embodied Agents Modularly with Large Language Models

ICLR 2024poster

In this work, we address challenging multi-agent cooperation problems with decentralized control, raw sensory observations, costly communication, and multi-objective tasks instantiated in various embodied environments. While previous research either presupposes a cost-free communication channel or r…

Cited by 238SourcePDFScholar
2024

Constrained Human-AI Cooperation: An Inclusive Embodied Social Intelligence Challenge

NeurIPS 2024poster

We introduce Constrained Human-AI Cooperation (CHAIC), an inclusive embodied social intelligence challenge designed to test social perception and cooperation in embodied agents. In CHAIC, the goal is for an embodied agent equipped with egocentric observations to assist a human who may be operating u…

2024

Few-Shot Task Learning through Inverse Generative Modeling

NeurIPS 2024poster

Learning the intents of an agent, defined by its goals or motion style, is often extremely challenging from just a few examples. We refer to this problem as task concept learning and present our approach, Few-Shot Task Learning through Inverse Generative Modeling (FTL-IGM), which learns new task con…

Cited by 1SourcePDFScholar
2024

GOMA: Proactive Embodied Cooperative Communication via Goal-Oriented Mental Alignment

IROS 2024poster

Verbal communication plays a crucial role in human cooperation, particularly when the partners only have incomplete information about the task, environment, and each other’s mental state. In this paper, we propose a novel cooperative communication framework, Goal-Oriented Mental Alignment (GOMA). GO…

Cited by 10SourceScholar
2024

MMToM-QA: Multimodal Theory of Mind Question Answering

ACL 2024long

Theory of Mind (ToM), the ability to understand people’s mental states, is an essential ingredient for developing machines with human-level social intelligence. Recent machine learning models, particularly large language models, seem to show some aspects of ToM understanding. However, existing ToM b…

2024

Neural Amortized Inference for Nested Multi-Agent Reasoning

AAAI 2024technical

Multi-agent interactions, such as communication, teaching, and bluffing, often rely on higher-order social inference, i.e., understanding how others infer oneself. Such intricate reasoning can be effectively modeled through nested multi-agent reasoning. Nonetheless, the computational complexity esca…

2024

Pragmatic Feature Preferences: Learning Reward-Relevant Preferences from Human Input

ICML 2024poster

Humans use context to specify preferences over behaviors, i.e. their reward functions. Yet, algorithms for inferring reward models from preference data do not take this social learning view into account. Inspired by pragmatic human communication, we study how to extract fine-grained data regarding w…

Cited by 2SourcePDFScholar
2024

Risk-Bounded Online Team Interventions via Theory of Mind

ICRA 2024poster

Despite advancements in human-robot teamwork, limited progress was made in developing AI assistants capable of advising teams online during task time, due to the challenges of modeling both individual and collective beliefs of the team members. Dynamic epistemic logic has proved to be a viable tool…

Cited by 1SourceScholar
2023

Diagnosis, Feedback, Adaptation: A Human-in-the-Loop Framework for Test-Time Policy Adaptation

ICML 2023poster

Policies often fail at test-time due to *distribution shifts*---changes in the state and reward that occur when an end user deploys the policy in environments different from those seen in training. Data augmentation can help models be more robust to such shifts by varying specific concepts in the st…

Cited by 18SourcePDFScholar
2023

Language Models Meet World Models: Embodied Experiences Enhance Language Models

NeurIPS 2023poster

While large language models (LMs) have shown remarkable capabilities across numerous tasks, they often struggle with simple reasoning and planning in physical environments, such as understanding object permanence or planning household activities. The limitation arises from the fact that LMs are trai…

2023

NOPA: Neurally-guided Online Probabilistic Assistance for Building Socially Intelligent Home Assistants

ICRA 2023poster

In this work, we study how to build socially intelligent robots to assist people in their homes. In particular, we focus on assistance with online goal inference, where robots must simultaneously infer humans' goals and how to help them achieve those goals. Prior assistance methods either lack the a…

Cited by 24SourceScholar
2023

Stateful Active Facilitator: Coordination and Environmental Heterogeneity in Cooperative Multi-Agent Reinforcement Learning

ICLR 2023poster

In cooperative multi-agent reinforcement learning, a team of agents works together to achieve a common goal. Different environments or tasks may require varying degrees of coordination among agents in order to achieve the goal in an optimal way. The nature of coordination will depend on properties o…

Cited by 10SourcePDFScholar
2023

Zero-Shot Linear Combinations of Grounded Social Interactions with Linear Social MDPs

AAAI 2023technical

Humans and animals engage in rich social interactions. It is often theorized that a relatively small number of basic social interactions give rise to the full range of behavior observed. But no computational theory explaining how social interactions combine together has been proposed before. We do s…

Cited by 1SourcePDFScholar
2022

Discovering Generalizable Spatial Goal Representations via Graph-based Active Reward Learning

ICML 2022spotlight

In this work, we consider one-shot imitation learning for object rearrangement tasks, where an AI agent needs to watch a single expert demonstration and learn to perform the same task in different environments. To achieve a strong generalization, the AI agent must infer the spatial goal specificatio…

2022

Incorporating Rich Social Interactions Into MDPs

ICRA 2022poster

Much of what we do as humans is engage socially with other agents, a skill that robots must also eventually possess. We demonstrate that a rich theory of social interactions originating from microsociology can be formalized by extending a nested MDP where agents reason about arbitrary functions of e…

Cited by 10SourceScholar
2022

RLPrompt: Optimizing Discrete Text Prompts with Reinforcement Learning

EMNLP 2022main

Prompting has shown impressive success in enabling large pre-trained language models (LMs) to perform diverse NLP tasks, especially with only few downstream data. Automatically finding the optimal prompt for each task, however, is challenging. Most existing work resorts to tuning *soft* prompts (e.g…

2022

Show Me What You Can Do: Capability Calibration on Reachable Workspace for Human-Robot Collaboration

RA-L 2022

Aligning humans’ assessment of what a robot can do with its true capability is crucial for establishing a common ground between human and robot partners when they collaborate on a joint task. In this work, we propose an approach to calibrate humans’ estimate of a robot’s reachable workspace through

Cited by 4SourceScholar
2021

AGENT: A Benchmark for Core Psychological Reasoning

ICML 2021spotlight

For machine agents to successfully interact with humans in real-world settings, they will need to develop an understanding of human mental life. Intuitive psychology, the ability to reason about hidden mental variables that drive observable actions, comes naturally to people: even pre-verbal infants…

Cited by 96SourcePDFScholar
2021

PHASE: PHysically-grounded Abstract Social Events for Machine Social Perception

AAAI 2021technical

The ability to perceive and reason about social interactions in the context of physical environments is core to human social intelligence and human-machine cooperation. However, no prior dataset or benchmark has systematically evaluated physically grounded perception of complex social interactions t…

Cited by 36SourcePDFScholar
2021

Watch-And-Help: A Challenge for Social Perception and Human-AI Collaboration

ICLR 2021spotlight

In this paper, we introduce Watch-And-Help (WAH), a challenge for testing social intelligence in agents. In WAH, an AI agent needs to help a human-like agent perform a complex household task efficiently. To succeed, the AI agent needs to i) understand the underlying goal of the task by watching a si…

2020

Active Visual Information Gathering for Vision-Language Navigation

ECCV 2020poster

Vision-language navigation (VLN) is the task of entailing an agent to carry out navigational instructions inside photo-realistic environments. One of the key challenges in VLN is how to conduct a robust navigation by mitigating the uncertainty caused by ambiguous instructions and insufficient observ…

2018

Hierarchical and Interpretable Skill Acquisition in Multi-task Reinforcement Learning

ICLR 2018poster

Learning policies for complex tasks that require multiple different skills is a major challenge in reinforcement learning (RL). It is also a requirement for its deployment in real-world scenarios. This paper proposes a novel framework for efficient multi-task reinforcement learning. Our framework tr…

Cited by 194SourcePDFScholar
2018

Where and Why Are They Looking? Jointly Inferring Human Attention and Intentions in Complex Tasks

CVPR 2018poster

This paper addresses a new problem - jointly inferring human attention, intentions, and tasks from videos. Given an RGB-D video where a human performs a task, we answer three questions simultaneously: 1) where the human is looking - attention prediction; 2) why the human is looking there - intention…

Cited by 82SourcePDFScholar
2017

CERN: Confidence-Energy Recurrent Network for Group Activity Recognition

CVPR 2017poster

This work is about recognizing human activities occurring in videos at distinct semantic levels, including individual actions, interactions, and group activities. The recognition is realized using a two-level hierarchy of Long Short-Term Memory (LSTM) networks, forming a feed-forward deep architectu…

Cited by 228PDFcodeScholar
2017

Learning social affordance grammar from videos: Transferring human interactions to human-robot interactions

ICRA 2017poster

In this paper, we present a general framework for learning social affordance grammar as a spatiotemporal AND-OR graph (ST-AOG) from RGB-D videos of human interactions, and transfer the grammar to humanoids to enable a real-time motion inference for human-robot interaction (HRI). Based on Gibbs sampl…

Cited by 54SourceScholar
2015

Joint Inference of Groups, Events and Human Roles in Aerial Videos

CVPR 2015poster

With the advent of drones, aerial video analysis becomes increasingly important; yet, it has received scant attention in the literature. This paper addresses a new problem of parsing low-resolution aerial videos of large spatial areas, in terms of 1) grouping, 2) recognizing events and 3) assigning…

Cited by 228SourcePDFScholar