Add model card

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +61 -0
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: text-to-3d
4
+ ---
5
+
6
+ # Memorization in 3D Shape Generation
7
+
8
+ This repository contains the official implementation of the paper [Memorization in 3D Shape Generation: An Empirical Study](https://huggingface.co/papers/2512.23628).
9
+
10
+ [**Project Page**](https://urrealhero.github.io/3DGenMemorizationWeb/) | [**GitHub Repository**](https://github.com/zlab-princeton/3d_mem)
11
+
12
+ ## Description
13
+ Generative models are increasingly used in 3D vision to synthesize novel shapes, yet it remains unclear whether their generation relies on memorizing training shapes. This work introduces an evaluation framework to quantify memorization ($Z_U$) in 3D generative models and studies the influence of different data and modeling designs on memorization. Through controlled experiments with a latent vector-set (Vecset) diffusion model, the authors provide analysis and strategies to reduce memorization without degrading generation quality.
14
+
15
+ ## Usage
16
+
17
+ For environment setup and instructions on data curation, please refer to the [official GitHub repository](https://github.com/zlab-princeton/3d_mem).
18
+
19
+ ### Inference
20
+
21
+ **Text-conditional generation**
22
+ ```bash
23
+ python inference.py \
24
+ --config configs/Baseline.yaml \
25
+ --ckpt PATH/TO/CHECKPOINT \
26
+ --out_dir outputs_text/ \
27
+ --num_samples 4 \
28
+ --text "a chair"
29
+ ```
30
+
31
+ **Class-conditional generation**
32
+ ```bash
33
+ python inference.py \
34
+ --config configs/Conditioning/LVIS-16-Category.yaml \
35
+ --ckpt PATH/TO/CHECKPOINT \
36
+ --out_dir outputs_lvis/ \
37
+ --num_samples 4 \
38
+ --class_id 0
39
+ ```
40
+
41
+ **Image-conditional generation**
42
+ ```bash
43
+ python inference.py \
44
+ --config configs/Conditioning/Image.yaml \
45
+ --ckpt PATH/TO/CHECKPOINT \
46
+ --out_dir outputs_image \
47
+ --num_samples 1 \
48
+ --image_path data_sprite.png \
49
+ --image_views 12 \
50
+ --image_pick random
51
+ ```
52
+
53
+ ## Citation
54
+ ```bibtex
55
+ @article{pu2025memorization,
56
+ title={Memorization in 3D Shape Generation: An Empirical Study},
57
+ author={Pu, Shu and Zeng, Boya and Zhou, Kaichen and Wang, Mengyu and Liu, Zhuang},
58
+ journal={arXiv preprint arXiv:2512.23628},
59
+ year={2025}
60
+ }
61
+ ```