← Search

Jiayuan Mao

51 accepted papers

2026

Adversarial Game-Theoretic Algorithm for Dexterous Grasp Synthesis

ICRA 2026poster

For many complex tasks, multi-finger robot hands are poised to revolutionize how we interact with the world, but reliably grasping objects remains a significant challenge. We focus on the problem of synthesizing grasps for multi-finger robot hands that, given an target object's geometry and pose, co…

2026

Composition-Grounded Instruction Synthesis for Visual Reasoning

ICLR 2026poster

Pretrained multi-modal large language models (MLLMs) demonstrate strong performance on diverse multimodal tasks, but remain limited in reasoning capabilities for domains where annotations are difficult to collect. In this work, we focus on artificial image domains such as charts, rendered documents,…

Cited by 0SourcecodeScholar
2026

Discovering Hybrid World Representations with Co-Evolving Foundation Models

AAAI 2026technical

This perspective article discusses an emerging research direction: to what extent can foundation models yield usable structure for modeling the physical world? We offer a Markovian formulation of structured world models and outline the notion of multi-level hybrid world representations that support

Cited by 0SourcePDFScholar
2026

Learning Composable Skills by Discovering Spatial and Temporal Structure with Foundation Models

ICRA 2026poster

We present STACK, a framework for discovering and learning composable manipulation skills from unsegmented demonstrations by leveraging spatial and temporal structure extracted from foundation models. STACK automatically extracts temporal structure by segmenting raw demonstrations into short-horizon…

Cited by 0codeScholar
2026

Multi-Modal Manipulation Via Multi-Modal Policy Consensus

ICRA 2026poster

Effectively integrating diverse sensory modalities is crucial for robotic manipulation. However, the typical approach of feature concatenation is often suboptimal: dominant modalities such as vision can overwhelm sparse but critical signals like touch in contact-rich tasks, and monolithic architectu…

2026

SIMPACT: Simulation-Enabled Action Planning using Vision-Language Models

CVPR 2026

Vision-Language Models (VLMs) exhibit remarkable common-sense and semantic reasoning capabilities. However, they lack a grounded understanding of physical dynamics. This limitation arises from training VLMs on static internet-scale visual-language data that contain no causal interactions or action-c

Cited by 0SourceScholar
2025

Creativity or Brute Force? Using Brainteasers as a Window into the Problem-Solving Abilities of Large Language Models

NeurIPS 2025poster

Accuracy remains a standard metric for evaluating AI systems, but it offers limited insight into how models arrive at their solutions. In this work, we introduce a benchmark based on brainteasers written in long narrative form to probe more deeply into the types of reasoning strategies that models…

Cited by 0SourceScholar
2025

Infer Human’s Intentions Before Following Natural Language Instructions

AAAI 2025technical

For AI agents to be helpful to humans, they should be able to follow natural language instructions to complete everyday cooperative tasks in human environments. However, real human instructions inherently possess ambiguity, because the human speakers assume sufficient prior knowledge about their hid…

2025

KALM: Keypoint Abstraction Using Large Models for Object-Relative Imitation Learning

ICRA 2025

Generalization to novel object configurations and instances across diverse tasks and environments is a critical challenge in robotics. Keypoint-based representations have been proven effective as a succinct representation for capturing essential object features, and for establishing a reference fram

Cited by 30SourcecodeScholar
2025

Learning Linear Attention in Polynomial Time

NeurIPS 2025oral

Previous research has explored the expressivity of Transformer models in simulating Boolean circuits or Turing machines. However, the efficient learnability of Transformers from data has remained an open question. Our study addresses this gap by providing the first polynomial-time learnability resu…

Cited by 0SourceScholar
2025

One-Shot Manipulation Strategy Learning by Making Contact Analogies

ICRA 2025

We present a novel approach, MAGIC (manipulation analogies for generalizable intelligent contacts), for one-shot learning of manipulation strategies with fast and extensive generalization to novel objects. By leveraging a reference action trajectory, MAGIC effectively identifies similar contact poin

Cited by 8SourcecodeScholar
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
2024

Grounding Language Plans in Demonstrations Through Counterfactual Perturbations

ICLR 2024spotlight

Grounding the common-sense reasoning of Large Language Models in physical domains remains a pivotal yet unsolved problem for embodied AI. Whereas prior works have focused on leveraging LLMs directly for planning in symbolic spaces, this work uses LLMs to guide the search of task structures and const…

2024

Learning Compositional Behaviors from Demonstration and Language

CoRL 2024poster

We introduce Behavior from Language and Demonstration (BLADE), a framework for long-horizon robotic manipulation by integrating imitation learning and model-based planning. BLADE leverages language-annotated demonstrations, extracts abstract action knowledge from large language models (LLMs), and co…

Cited by 3SourceScholar
2024

Learning Grounded Action Abstractions from Language

ICLR 2024poster

Effective planning in the real world requires not only world knowledge, but the ability to leverage that knowledge to build the right representation of the task at hand. Decades of hierarchical planning techniques have used domain-specific temporal action abstractions to support efficient and accura…

Cited by 5SourcePDFScholar
2024

Learning to Act from Actionless Videos through Dense Correspondences

ICLR 2024spotlight

In this work, we present an approach to construct a video-based robot policy capable of reliably executing diverse tasks across different robots and environments from few video demonstrations without using any action annotations. Our method leverages images as a task-agnostic representation, encodin…

Cited by 56SourcePDFScholar
2024

Set It Up!: Functional Object Arrangement with Compositional Generative Models

RSS 2024poster

This paper studies the challenge of developing robots capable of understanding under-specified instructions for creating functional object arrangements, such as "set up a dining table for two"; previous arrangement approaches have focused on much more explicit instructions, such as "put object A on…

2023

Compositional Diffusion-Based Continuous Constraint Solvers

CoRL 2023poster

This paper introduces an approach for learning to solve continuous constraint satisfaction problems (CCSP) in robotic reasoning and planning. Previous methods primarily rely on hand-engineering or learning generators for specific constraint types and then rejecting the value assignments when other c…

Cited by 29SourceScholar
2023

Learning Rational Subgoals from Demonstrations and Instructions

AAAI 2023technical

We present a framework for learning useful subgoals that support efficient long-term planning to achieve novel goals. At the core of our framework is a collection of rational subgoals (RSGs), which are essentially binary classifiers over the environmental states. RSGs can be learned from weakly-anno…

Cited by 6SourcePDFScholar
2023

Learning Reusable Manipulation Strategies

CoRL 2023poster

Humans demonstrate an impressive ability to acquire and generalize manipulation "tricks." Even from a single demonstration, such as using soup ladles to reach for distant objects, we can apply this skill to new scenarios involving different object positions, sizes, and categories (e.g., forks and ha…

Cited by 11SourceScholar
2023

Programmatically Grounded, Compositionally Generalizable Robotic Manipulation

ICLR 2023top-25%

Robots operating in the real world require both rich manipulation skills as well as the ability to semantically reason about when to apply those skills. Towards this goal, recent works have integrated semantic representations from large-scale pretrained vision-language (VL) models into manipulation…

2023

What Planning Problems Can A Relational Neural Network Solve?

NeurIPS 2023spotlight

Goal-conditioned policies are generally understood to be "feed-forward" circuits, in the form of neural networks that map from the current state and the goal specification to the next action to take. However, under what circumstances such a policy can be learned and how efficient the policy will be…

2023

What’s Left? Concept Grounding with Logic-Enhanced Foundation Models

NeurIPS 2023poster

Recent works such as VisProg and ViperGPT have smartly composed foundation models for visual reasoning—using large language models (LLMs) to produce programs that can be executed by pre-trained vision-language models. However, they operate in limited domains, such as 2D images, not fully exploiting…

2022

CLEVRER-Humans: Describing Physical and Causal Events the Human Way

NeurIPS 2022accept

Building machines that can reason about physical events and their causal relationships is crucial for flexible interaction with the physical world. However, most existing physical and causal reasoning benchmarks are exclusively based on synthetically generated events and synthetic natural language d…

Cited by 23SourcePDFScholar
2022

FALCON: Fast Visual Concept Learning by Integrating Images, Linguistic descriptions, and Conceptual Relations

ICLR 2022poster

We present a meta-learning framework for learning new visual concepts quickly, from just one or a few examples, guided by multiple naturally occurring data streams: simultaneously looking at images, reading sentences that describe the objects in the scene, and interpreting supplemental sentences tha…

Cited by 24SourcePDFScholar
2022

HandMeThat: Human-Robot Communication in Physical and Social Environments

NeurIPS 2022accept

We introduce HandMeThat, a benchmark for a holistic evaluation of instruction understanding and following in physical and social environments. While previous datasets primarily focused on language grounding and planning, HandMeThat considers the resolution of human instructions with ambiguities base…

Cited by 21SourcePDFScholar
2022

IKEA-Manual: Seeing Shape Assembly Step by Step

NeurIPS 2022accept

Human-designed visual manuals are crucial components in shape assembly activities. They provide step-by-step guidance on how we should move and connect different parts in a convenient and physically-realizable way. While there has been an ongoing effort in building agents that perform assembly tasks…

Cited by 19SourcePDFScholar
2022

PDSketch: Integrated Domain Programming, Learning, and Planning

NeurIPS 2022accept

This paper studies a model learning and online planning approach towards building flexible and general robots. Specifically, we investigate how to exploit the locality and sparsity structures in the underlying environmental transition model to improve model generalization, data-efficiency, and runti…

Cited by 40SourcePDFScholar
2022

Translating a Visual LEGO Manual to a Machine-Executable Plan

ECCV 2022poster

"We study the problem of translating an image-based, step-by-step assembly manual created by human designers into machine-interpretable instructions. We formulate this problem as a sequential prediction task: at each step, our model reads the manual, locates the components to be added to the current…

Cited by 22SourcePDFScholar
2021

Grammar-Based Grounded Lexicon Learning

NeurIPS 2021poster

We present Grammar-Based Grounded Language Learning (G2L2), a lexicalist approach toward learning a compositional and grounded meaning representation of language from grounded data, such as paired images and texts. At the core of G2L2 is a collection of lexicon entries, which map each word to a tupl…

Cited by 17SourcePDFScholar
2021

Grounding Physical Concepts of Objects and Events Through Dynamic Visual Reasoning

ICLR 2021poster

We study the problem of dynamic visual reasoning on raw videos. This is a challenging problem; currently, state-of-the-art models often require dense supervision on physical object properties and events from simulation, which are impractical to obtain in real life. In this paper, we present the Dyna…

Cited by 104SourcePDFScholar
2021

Temporal and Object Quantification Networks

IJCAI 2021poster

We present Temporal and Object Quantification Networks (TOQ-Nets), a new class of neuro-symbolic networks with a structural bias that enables them to learn to recognize complex relational-temporal events. This is done by including reasoning layers that implement finite-domain quantification over obj…

Cited by 7SourcePDFScholar
2020

Multi-Plane Program Induction with 3D Box Priors

NeurIPS 2020poster

We consider two important aspects in understanding and editing images: modeling regular, program-like texture or patterns in 2D planes, and 3D posing of these planes in the scene. Unlike prior work on image-based program synthesis, which assumes the image contains a single visible 2D plane, we prese…

Cited by 15SourcePDFScholar
2020

Perspective Plane Program Induction From a Single Image

CVPR 2020poster

We study the inverse graphics problem of inferring a holistic representation for natural images. Given an input image, our goal is to induce a neuro-symbolic, program-like representation that jointly models camera poses, object locations, and global scene structures. Such high-level, holistic scene…

Cited by 15PDFScholar
2019

Program-Guided Image Manipulators

ICCV 2019poster

Humans are capable of building holistic representations for images at various levels, from local objects, to pairwise relations, to global structures. The interpretation of structures involves reasoning over repetition and symmetry of the objects in the image. In this paper, we present the Program-G…

Cited by 24PDFScholar
2019

The Neuro-Symbolic Concept Learner: Interpreting Scenes, Words, and Sentences From Natural Supervision

ICLR 2019oral

We propose the Neuro-Symbolic Concept Learner (NS-CL), a model that learns visual concepts, words, and semantic parsing of sentences without explicit supervision on any of them; instead, our model learns by simply looking at images and reading paired questions and answers. Our model builds an object…

2019

Unified Visual-Semantic Embeddings: Bridging Vision and Language With Structured Meaning Representations

CVPR 2019oral

We propose the Unified Visual-Semantic Embeddings (Unified VSE) for learning a joint space of visual representation and textual semantics. The model unifies the embeddings of concepts at different levels: objects, attributes, relations, and full scenes. We view the sentential semantics as a combinat…

Cited by 221PDFcodeScholar
2018

Acquisition of Localization Confidence for Accurate Object Detection

ECCV 2018poster

Modern CNN-based object detectors rely on bounding box regression and non-maximum suppression to localize objects. While the probabilities for class labels naturally reflect classification confidence, localization confidence is absent. This makes properly localized bounding boxes degenerate during i…