atlas

Research map / 3D vision / 3D generation

SUGAR: Pre-training 3D Visual Representations for Robotics

CVPR 2024 · 3D vision › 3D generation · approach: Semantic scene completion · 14 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes SUGAR, a transformer-based 3D point cloud pre-training framework that jointly learns semantics, geometry and affordance from single- and multi-object simulated scenes via five tasks, evaluated on zero-shot 3D recognition, referring expression grounding, and RLBench manipulation.

Problem

Existing visual representation pre-training for robotics is mostly 2D, and existing 3D pre-training is limited to single complete objects, ignoring occlusion, clutter and object affordances.

Approach

A prompt-based transformer (point cloud encoder + prompt decoder) pre-trained with five tasks (masked point modeling, cross-modal distillation, grasping pose synthesis, instance segmentation, referring expression grounding) on automatically generated multi-object simulated scenes with free grasp and segmentation labels, using curriculum learning from single to multi-object data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Pre-train on ensembled single-object data instead of ShapeNet-only single-object dataSUGAR · ScanObjectNN OBJ BGTop1 accuracy: 50.3 -> 68.0Helped
Further train on multi-object scenesSUGAR · ScanObjectNN OBJ BGTop1 accuracy: 68.0 -> 65.6Hurt
Further train on multi-object scenesSUGAR · ModelNet40Top1 accuracy: 84.3 -> 83.8Hurt
Use ensembled pre-training data instead of ShapeNet-only dataSUGAR single-object pre-training · Objaverse-LVISTop1 accuracy: 27.3 -> 42.1Helped
Pre-train on ensembled multi-object dataSUGAR referring expression grounding · OCID-Ref testAcc@0.25: 90.77 -> 97.74Helped
Increase the key-point count from Ne=768 to Ne=1536 (ablation)SUGAR without pre-training · RoboRefit testBAcc@0.5: 50.64 -> 55.62Helped
Pre-train on single-object data only (ablation)SUGAR referring expression grounding · RoboRefit testBAcc@0.5: 55.62 -> 52.90Hurt
Add multi-object scene pre-training (ablation)SUGAR referring expression grounding · RoboRefit testBAcc@0.5: 55.62 -> 61.85Helped
Add grasping pose synthesis to pre-training (ablation)SUGAR referring expression grounding · RoboRefit testBAcc@0.5: 59.70 -> 61.85Helped
Add ensembled multi-object pre-training data (ablation)SUGAR referring expression grounding · RoboRefit testBAcc@0.5: 61.85 -> 65.04Helped
Pre-train on ShapeNet single-object data (ablation)SUGAR manipulation policy · RLBench 10 tasksAvg. success rate: 85.9 -> 88.1Helped
Pre-train on ShapeNet multi-object scenes instead of single-object data (ablation)SUGAR manipulation policy · RLBench 10 tasksAvg. success rate: 88.1 -> 93.0Helped

Surprising findings

Limitations

Experimental setup

Datasets: ShapeNet 3D-FUTURE ABO Objaverse ACRONYM ModelNet40 ScanObjectNN Objaverse-LVIS OCID-Ref RoboRefit RLBench

Benchmarks: ModelNet40 ScanObjectNN Objaverse-LVIS OCID-Ref RoboRefit RLBench

Metrics: Top1 accuracy Top3 accuracy Top5 accuracy Acc@0.25 Acc@0.5 mIoU success rate

Baselines: ReCon CLIP2Point ULIP-PointBERT OpenShape-PointBERT R3M MVP CLIP V-Cond RefTR Hiveformer PolarNet Auto-λ

Builds on: OpenShape ReCon Hiveformer DETR PointNet

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy