← Search

Stone Tao

11 accepted papers

2025

Demonstrating GPU Parallelized Robot Simulation and Rendering for Generalizable Embodied AI with ManiSkill3

RSS 2025poster

Simulation has enabled unprecedented compute-scalable approaches to robot learning. However, many existing simulation frameworks typically support a narrow range of scenes/tasks and lack features critical for scaling generalizable robotics and sim2real. We introduce and open source ManiSkill3, the f…

Cited by 0PDFScholar
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

ManiSkill-HAB: A Benchmark for Low-Level Manipulation in Home Rearrangement Tasks

ICLR 2025poster

High-quality benchmarks are the foundation for embodied AI research, enabling significant advancements in long-horizon navigation, manipulation and rearrangement tasks. However, as frontier tasks in robotics get more advanced, they require faster simulation speed, more intricate test environments, a…

2025

Multi-Stage Manipulation with Demonstration-Augmented Reward, Policy, and World Model Learning

ICML 2025poster

Long-horizon tasks in robotic manipulation present significant challenges in reinforcement learning (RL) due to the difficulty of designing dense reward functions and effectively exploring the expansive state-action space. However, despite a lack of dense rewards, these tasks often have a multi-stag…

2025

MyoChallenge 2024: A New Benchmark for Physiological Dexterity and Agility in Bionic Humans

NeurIPS 2025poster

Recent advancements in bionic prosthetic technology offer transformative opportunities to restore mobility and functionality for individuals with missing limbs. Users of bionic limbs, or bionic humans, learn to seamlessly integrate prosthetic extensions into their motor repertoire, regaining critica…

Cited by 0SourceScholar
2025

Policy Decorator: Model-Agnostic Online Refinement for Large Policy Model

ICLR 2025poster

Recent advancements in robot learning have used imitation learning with large models and extensive demonstrations to develop effective policies. However, these models are often limited by the quantity quality, and diversity of demonstrations. This paper explores improving offline-trained imitation l…

2025

Staggered Environment Resets Improve Massively Parallel On-Policy Reinforcement Learning

NeurIPS 2025poster

Massively parallel GPU simulation environments have accelerated reinforcement learning (RL) research by enabling fast data collection for on-policy RL algorithms like Proximal Policy Optimization (PPO). To maximize throughput, it is common to use short rollouts per policy update, increasing the upda…

Cited by 0SourceScholar
2024

Reverse Forward Curriculum Learning for Extreme Sample and Demo Efficiency

ICLR 2024poster

Reinforcement learning (RL) presents a promising framework to learn policies through environment interaction, but often requires an infeasible amount of interaction data to solve complex tasks from sparse rewards. One direction includes augmenting RL with offline data demonstrating desired tasks, bu…

2023

Abstract-to-Executable Trajectory Translation for One-Shot Task Generalization

ICML 2023poster

Training long-horizon robotic policies in complex physical environments is essential for many applications, such as robotic manipulation. However, learning a policy that can generalize to unseen tasks is challenging. In this work, we propose to achieve one-shot task generalization by decoupling plan…

2023

ManiSkill2: A Unified Benchmark for Generalizable Manipulation Skills

ICLR 2023poster

Generalizable manipulation skills, which can be composed to tackle long-horizon and complex daily chores, are one of the cornerstones of Embodied AI. However, existing benchmarks, mostly composed of a suite of simulatable environments, are insufficient to push cutting-edge research works because the…

2021

ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrations

NeurIPS 2021poster

Object manipulation from 3D visual inputs poses many challenges on building generalizable perception and policy models. However, 3D assets in existing benchmarks mostly lack the diversity of 3D shapes that align with real-world intra-class complexity in topology and geometry. Here we propose SAPIEN…

Cited by 145SourcecodeScholar