nielsr HF Staff commited on
Commit
7cb370e
·
verified ·
1 Parent(s): 8b51db8

Improve model card: Add pipeline tag, library name, paper & code links

Browse files

This 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.

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