Add model card for TORA
Browse filesHi! I'm Niels from the Hugging Face community science team. I've opened this PR to add a model card for the TORA model.
This model card includes:
- Links to the paper, project page, and GitHub repository.
- Relevant metadata including the `pipeline_tag` and `license`.
- A summary of the TORA framework.
- Sample usage instructions for evaluation derived from the official repository.
Please let me know if you'd like any changes!
README.md
CHANGED
|
@@ -1,3 +1,48 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
pipeline_tag: other
|
| 4 |
+
tags:
|
| 5 |
+
- 3d
|
| 6 |
+
- point-cloud
|
| 7 |
+
- shape-assembly
|
| 8 |
+
- flow-matching
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# TORA: Topological Representation Alignment for 3D Shape Assembly
|
| 12 |
+
|
| 13 |
+
[**Paper**](https://huggingface.co/papers/2604.04050) | [**Project Page**](https://nahyuklee.github.io/tora) | [**Code**](https://github.com/NahyukLEE/tora)
|
| 14 |
+
|
| 15 |
+
**TORA** is a topology-first representation alignment framework for 3D shape assembly. It distills relational structure from a frozen pretrained 3D encoder into a flow-matching backbone during training. By matching the similarity structure between student and teacher representations (using CKA loss), TORA achieves faster convergence and improved accuracy for assembling unposed 3D parts into complete objects.
|
| 16 |
+
|
| 17 |
+
## Overview
|
| 18 |
+
|
| 19 |
+
Flow-matching methods for 3D shape assembly learn point-wise velocity fields that transport parts toward assembled configurations. TORA introduces zero inference overhead while yielding two consistent benefits:
|
| 20 |
+
- **Faster convergence**: Up to 6.9x faster training.
|
| 21 |
+
- **Improved accuracy**: Better performance in-distribution and greater robustness under domain shift.
|
| 22 |
+
- **Zero overhead**: The teacher alignment occurs only during training.
|
| 23 |
+
|
| 24 |
+
## Usage
|
| 25 |
+
|
| 26 |
+
For detailed setup and training instructions, please refer to the [official GitHub repository](https://github.com/NahyukLEE/tora).
|
| 27 |
+
|
| 28 |
+
### Evaluation
|
| 29 |
+
|
| 30 |
+
To run evaluation on a trained TORA model, you can use the following command:
|
| 31 |
+
|
| 32 |
+
```bash
|
| 33 |
+
python sample.py \
|
| 34 |
+
ckpt_path=./output/TORA_base/best.ckpt \
|
| 35 |
+
data_root="../dataset" \
|
| 36 |
+
data=main/bbad_everyday
|
| 37 |
+
```
|
| 38 |
+
|
| 39 |
+
## Citation
|
| 40 |
+
|
| 41 |
+
```bibtex
|
| 42 |
+
@article{lee2026tora,
|
| 43 |
+
title = {TORA: Topological Representation Alignment for 3D Shape Assembly},
|
| 44 |
+
author = {Lee, Nahyuk and Chen, Zhiang and Pollefeys, Marc and Hong, Sunghwan},
|
| 45 |
+
journal = {arXiv preprint arXiv:2604.04050},
|
| 46 |
+
year = {2026}
|
| 47 |
+
}
|
| 48 |
+
```
|