Improve model card: Add pipeline tag, links, license, and usage info
#1
by nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,3 +1,77 @@
|
|
| 1 |
-
---
|
| 2 |
-
license:
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
pipeline_tag: robotics
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
+
# Much Ado About Noising: Dispelling the Myths of Generative Robotic Control
|
| 7 |
+
|
| 8 |
+
This repository contains the code for the paper [Much Ado About Noising: Dispelling the Myths of Generative Robotic Control](https://huggingface.co/papers/2512.01809). It provides a PyTorch-based framework for behavior cloning with flow matching and related generative models, incorporating best practices from diffusion model training. The research investigates the factors underlying the success of generative control policies (GCPs) in robotics, suggesting their advantage stems from iterative computation with intermediate supervision and stochasticity, rather than multimodal action distributions or complex observation-to-action mappings.
|
| 9 |
+
|
| 10 |
+
- **Project Page:** https://simchowitzlabpublic.github.io/much-ado-about-noising-project/
|
| 11 |
+
- **Code Repository:** https://github.com/simchowitzlabpublic/much-ado-about-noising
|
| 12 |
+
|
| 13 |
+
## Features
|
| 14 |
+
|
| 15 |
+
- 🧩 **Clean & Modular**: Composable components for losses, samplers, networks, and encoders
|
| 16 |
+
- ⚡ **Fast**: Optimized with torch.compile and CUDA graphs for maximum throughput
|
| 17 |
+
- 📊 **Best Practices**: EMA, warmup scheduling, auto-resume, and proven training techniques
|
| 18 |
+
- 🎯 **Diverse Algorithms**: Support for flow matching, consistency models, shortcut models, and regression
|
| 19 |
+
- 🤖 **Robot-Ready**: Pre-configured for Robomimic, Kitchen, and PushT tasks
|
| 20 |
+
|
| 21 |
+
## Installation
|
| 22 |
+
|
| 23 |
+
```bash
|
| 24 |
+
uv sync
|
| 25 |
+
# install robomimic dependencies
|
| 26 |
+
uv sync --extras robomimic
|
| 27 |
+
# install kitchen dependencies
|
| 28 |
+
uv sync --extras kitchen
|
| 29 |
+
```
|
| 30 |
+
|
| 31 |
+
## Quick Start
|
| 32 |
+
|
| 33 |
+
### Training
|
| 34 |
+
|
| 35 |
+
```bash
|
| 36 |
+
# Train Robomimic (state observations)
|
| 37 |
+
uv run examples/train_robomimic.py \
|
| 38 |
+
task=lift_ph_state \
|
| 39 |
+
network=chiunet \
|
| 40 |
+
optimization.loss_type=flow \
|
| 41 |
+
log.wandb_mode=online
|
| 42 |
+
|
| 43 |
+
# Train Robomimic (image observations)
|
| 44 |
+
uv run examples/train_robomimic.py \
|
| 45 |
+
task=lift_ph_image \
|
| 46 |
+
network=chiunet \
|
| 47 |
+
optimization.batch_size=256
|
| 48 |
+
|
| 49 |
+
# Train Kitchen
|
| 50 |
+
uv run examples/train_kitchen.py task=kitchen_state
|
| 51 |
+
|
| 52 |
+
# Train PushT
|
| 53 |
+
uv run examples/train_pusht.py task=pusht_state
|
| 54 |
+
```
|
| 55 |
+
|
| 56 |
+
### Evaluation
|
| 57 |
+
|
| 58 |
+
```bash
|
| 59 |
+
# Evaluate trained model
|
| 60 |
+
uv run examples/train_robomimic.py \
|
| 61 |
+
mode=eval \
|
| 62 |
+
optimization.model_path="/path/to/checkpoint.pt"
|
| 63 |
+
```
|
| 64 |
+
|
| 65 |
+
## Citation
|
| 66 |
+
|
| 67 |
+
If you find our work helpful, please cite the paper:
|
| 68 |
+
|
| 69 |
+
```bibtex
|
| 70 |
+
@article{anon2025much,
|
| 71 |
+
title={Much Ado About Noising: Dispelling the Myths of Generative Robotic Control},
|
| 72 |
+
author={Anonymous},
|
| 73 |
+
journal={arXiv preprint arXiv:2512.01809},
|
| 74 |
+
year={2025},
|
| 75 |
+
url={https://arxiv.org/abs/2512.01809}
|
| 76 |
+
}
|
| 77 |
+
```
|