nielsr HF Staff commited on
Commit
2cf6840
·
verified ·
1 Parent(s): e144fb3

Add model card for TORA

Browse files

Hi! 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!

Files changed (1) hide show
  1. README.md +48 -3
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
+ ```