Improve model card: Add pipeline tag, library name, paper & code links
Browse filesThis PR improves the model card by:
- Adding the `pipeline_tag: unconditional-image-generation` to ensure the model is discoverable under this task.
- Adding `library_name: diffusers` as the model is compatible with the Diffusers library, which enables the "How to use" widget.
- Adding a direct link to the paper: https://huggingface.co/papers/2505.14502.
- Adding a link to the GitHub repository: https://github.com/poppuppy/secant_expectation.
Please review and merge this PR if everything looks good.
README.md
CHANGED
|
@@ -1,3 +1,29 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
library_name: diffusers
|
| 4 |
+
pipeline_tag: unconditional-image-generation
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
This repository contains the official implementation of the paper: [Learning to Integrate Diffusion ODEs by Averaging the Derivatives](https://huggingface.co/papers/2505.14502).
|
| 8 |
+
|
| 9 |
+
Code: https://github.com/poppuppy/secant_expectation
|
| 10 |
+
|
| 11 |
+
## About
|
| 12 |
+
This work introduces an intermediate strategy for accelerating diffusion model inference by learning ODE integration. It uses loss functions derived from the derivative-integral relationship, inspired by Monte Carlo integration and Picard iteration. Referred to as "secant losses," this approach balances performance and cost, leading to great training stability. The secant version of EDM achieves a 10-step FID of 2.14 on CIFAR-10, while the secant version of SiT-XL/2 attains a 4-step FID of 2.27 and an 8-step FID of 1.96 on ImageNet-256x256.
|
| 13 |
+
|
| 14 |
+
## Installation
|
| 15 |
+
For installation instructions and how to train and evaluate the models, please refer to the [GitHub repository](https://github.com/poppuppy/secant_expectation).
|
| 16 |
+
|
| 17 |
+
## Checkpoints
|
| 18 |
+
Trained checkpoints for different settings (SDEI, STEE, SDEE, STEE, various step counts) are provided on [Hugging Face](https://huggingface.co/poppuppy/secant_expectation).
|
| 19 |
+
|
| 20 |
+
## Citation
|
| 21 |
+
If you find this work useful in your research, please consider citing:
|
| 22 |
+
```bibtex
|
| 23 |
+
@article{liu2025learning,
|
| 24 |
+
title={Learning to Integrate Diffusion ODEs by Averaging the Derivatives},
|
| 25 |
+
author={Liu, Wenze and Yue, Xiangyu},
|
| 26 |
+
journal={arXiv preprint arXiv:2505.14502},
|
| 27 |
+
year={2025}
|
| 28 |
+
}
|
| 29 |
+
```
|