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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shukla, Arth, Tao, Stone, Su, Hao
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913712243212288
author Shukla, Arth
Tao, Stone
Su, Hao
author_facet Shukla, Arth
Tao, Stone
Su, Hao
contents 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, and larger demonstration datasets. To this end, we present MS-HAB, a holistic benchmark for low-level manipulation and in-home object rearrangement. First, we provide a GPU-accelerated implementation of the Home Assistant Benchmark (HAB). We support realistic low-level control and achieve over 3x the speed of prior magical grasp implementations at a fraction of the GPU memory usage. Second, we train extensive reinforcement learning (RL) and imitation learning (IL) baselines for future work to compare against. Finally, we develop a rule-based trajectory filtering system to sample specific demonstrations from our RL policies which match predefined criteria for robot behavior and safety. Combining demonstration filtering with our fast environments enables efficient, controlled data generation at scale.
format Preprint
id arxiv_https___arxiv_org_abs_2412_13211
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ManiSkill-HAB: A Benchmark for Low-Level Manipulation in Home Rearrangement Tasks
Shukla, Arth
Tao, Stone
Su, Hao
Robotics
Artificial Intelligence
Computer Vision and Pattern Recognition
Machine Learning
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, and larger demonstration datasets. To this end, we present MS-HAB, a holistic benchmark for low-level manipulation and in-home object rearrangement. First, we provide a GPU-accelerated implementation of the Home Assistant Benchmark (HAB). We support realistic low-level control and achieve over 3x the speed of prior magical grasp implementations at a fraction of the GPU memory usage. Second, we train extensive reinforcement learning (RL) and imitation learning (IL) baselines for future work to compare against. Finally, we develop a rule-based trajectory filtering system to sample specific demonstrations from our RL policies which match predefined criteria for robot behavior and safety. Combining demonstration filtering with our fast environments enables efficient, controlled data generation at scale.
title ManiSkill-HAB: A Benchmark for Low-Level Manipulation in Home Rearrangement Tasks
topic Robotics
Artificial Intelligence
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2412.13211