Add model card
#1
by
nielsr
HF Staff
- opened
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 |
+
```
|