Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,3 +1,68 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
tags:
|
| 5 |
+
- image-generation
|
| 6 |
+
- diffusion
|
| 7 |
+
- class-conditional
|
| 8 |
+
- imagenet
|
| 9 |
+
- pytorch
|
| 10 |
---
|
| 11 |
+
|
| 12 |
+
# LDM-is-AE: Latent Diffusion is an Intrinsic Auto-Encoder
|
| 13 |
+
|
| 14 |
+
Released **inference checkpoints** for *LDM-is-AE* (NeurIPS 2026).
|
| 15 |
+
Code: https://github.com/PolyU-VCLab/LDMisAE
|
| 16 |
+
|
| 17 |
+
## TL;DR
|
| 18 |
+
|
| 19 |
+
Latent diffusion is usually a two-stage pipeline -- train a VAE, then train a diffusion model in its
|
| 20 |
+
latent space. **LDM-is-AE removes the pipeline**: the DiT backbone is split into **DiT-E** / **DiT-D**
|
| 21 |
+
and the intermediate feature is supervised in the **image domain at every timestep**, so the backbone
|
| 22 |
+
itself becomes an intrinsic auto-encoder trained end-to-end in a single stage.
|
| 23 |
+
|
| 24 |
+
## Checkpoints
|
| 25 |
+
|
| 26 |
+
| File | Resolution | Keys | Size |
|
| 27 |
+
| --- | --- | --- | --- |
|
| 28 |
+
| `LDMisAE.256.ckpt` | 256x256 | `model_ema1`, `args`, `epoch` | 3.90 GB |
|
| 29 |
+
| `LDMisAE.512.ckpt` | 512x512 | `model_ema1`, `args`, `epoch` | 3.93 GB |
|
| 30 |
+
|
| 31 |
+
Class-conditional ImageNet models. The files are **ema1-only** (inference); resuming training needs a
|
| 32 |
+
full checkpoint.
|
| 33 |
+
|
| 34 |
+
## Results
|
| 35 |
+
|
| 36 |
+
| Setting | FID | IS |
|
| 37 |
+
| --- | --- | --- |
|
| 38 |
+
| ImageNet 256x256, class-conditional | **1.80** | **314** |
|
| 39 |
+
| ImageNet 512x512, class-conditional | **1.90** | **320** |
|
| 40 |
+
| Reconstruction (gFID) | **1.82** | -- |
|
| 41 |
+
|
| 42 |
+
## Usage
|
| 43 |
+
|
| 44 |
+
```bash
|
| 45 |
+
git clone https://github.com/PolyU-VCLab/LDMisAE.git && cd LDMisAE
|
| 46 |
+
pip install -r requirements.txt
|
| 47 |
+
huggingface-cli download xtudbxk/LDMisAE LDMisAE.256.ckpt --local-dir weights
|
| 48 |
+
CKPT=weights/LDMisAE.256.ckpt IMG_SIZE=256 CFG=2.25 NUM_IMAGES=50000 bash scripts/inference.sh
|
| 49 |
+
```
|
| 50 |
+
|
| 51 |
+
Keep `--ema_mode ema1` (the default); `none` / `ema2` fail fast with a clear error. The LPIPS `vgg.pth`
|
| 52 |
+
must be reachable before sampling -- see the repository README (*Quick Start*).
|
| 53 |
+
|
| 54 |
+
## Citation
|
| 55 |
+
|
| 56 |
+
```bibtex
|
| 57 |
+
@inproceedings{ldm_is_ae_2026,
|
| 58 |
+
title = {LDM-is-AE: Latent Diffusion is an Intrinsic Auto-Encoder for End-to-End Image Generation},
|
| 59 |
+
author = {Zhang, Zhengqiang and Sun, Lingchen and Wu, Rongyuan and Yi, Qiaosi and
|
| 60 |
+
Kong, Xiangtao and Xiao, Chaodong and Zhang, Lei},
|
| 61 |
+
booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
|
| 62 |
+
year = {2026}
|
| 63 |
+
}
|
| 64 |
+
```
|
| 65 |
+
|
| 66 |
+
## License
|
| 67 |
+
|
| 68 |
+
Code is **Apache-2.0** (https://github.com/PolyU-VCLab/LDMisAE). Weights are released for research use.
|