activity
20242026
most citedLearning Iterative Reasoning through Energy Diffusion

2 citations · 4 across the 7 of their papers we have counts for

collaborators

9 papers

cs.AI2026

Planning Commitment of LLM under Partial Observability

Cassie Huang, Zhanwei Cao, Yeawon Lee +7

LLM-based planning agents commit to plans of different lengths before interacting with a partially observable environment. Some aggressively commit to a long plan based on assumpti…

cs.RO2025

"Set It Up": Functional Object Arrangement with Compositional Generative Models (Journal Version)

Yiqing Xu, Jiayuan Mao, Linfeng Li +4

Functional object arrangement (FORM) is the task of arranging objects to fulfill a function, e.g., "set up a dining table for two". One key challenge here is that the instructions…

cs.AI2025

Building Intelligent Agents with Neuro-Symbolic Concepts

Jiayuan Mao, Joshua B. Tenenbaum, Jiajun Wu

This article presents a concept-centric paradigm for building agents that can learn continually and reason flexibly. The concept-centric agent utilizes a vocabulary of neuro-symbol…

cs.RO2024

One-Shot Manipulation Strategy Learning by Making Contact Analogies

Yuyao Liu, Jiayuan Mao, Joshua Tenenbaum +2

We present a novel approach, MAGIC (manipulation analogies for generalizable intelligent contacts), for one-shot learning of manipulation strategies with fast and extensive general…

cs.RO2024★ 1 cited

Keypoint Abstraction using Large Models for Object-Relative Imitation Learning

Xiaolin Fang, Bo-Ruei Huang, Jiayuan Mao +4

Generalization to novel object configurations and instances across diverse tasks and environments is a critical challenge in robotics. Keypoint-based representations have been prov…

cs.LG2024

Learning Linear Attention in Polynomial Time

Morris Yau, Ekin Akyürek, Jiayuan Mao +3

Previous research has explored the computational expressivity of Transformer models in simulating Boolean circuits or Turing machines. However, the learnability of these simulators…

cs.LG2024★ 2 cited

Learning Iterative Reasoning through Energy Diffusion

Yilun Du, Jiayuan Mao, Joshua B. Tenenbaum

We introduce iterative reasoning through energy diffusion (IRED), a novel framework for learning to reason for a variety of tasks by formulating reasoning and decision-making probl…

cs.CL2024★ 1 cited

Finding structure in logographic writing with library learning

Guangyuan Jiang, Matthias Hofer, Jiayuan Mao +3

One hallmark of human language is its combinatoriality -- reusing a relatively small inventory of building blocks to create a far larger inventory of increasingly complex structure…

cs.RO2024

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

Yiqing Xu, Jiayuan Mao, Yilun Du +3

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 ta…