Add model card for E-GRPO
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,3 +1,30 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: apache-2.0
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
pipeline_tag: text-to-image
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
+
# E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models
|
| 7 |
+
|
| 8 |
+
This repository contains the weights for E-GRPO, as presented in the paper [E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models](https://huggingface.co/papers/2601.00423).
|
| 9 |
+
|
| 10 |
+
## Introduction
|
| 11 |
+
|
| 12 |
+
E-GRPO (Entropy-Guided Group Relative Policy Optimization) is a reinforcement learning approach designed to enhance flow-matching models for human preference alignment. The key insight is that high-entropy denoising steps are more critical for policy optimization. The authors propose a merging-step strategy that focuses training on these important steps, leading to more efficient and effective exploration compared to standard SDE or ODE sampling methods.
|
| 13 |
+
|
| 14 |
+
## Resources
|
| 15 |
+
|
| 16 |
+
- **Paper:** [E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models](https://huggingface.co/papers/2601.00423)
|
| 17 |
+
- **Code:** [GitHub - shengjun-zhang/VisualGRPO](https://github.com/shengjun-zhang/VisualGRPO)
|
| 18 |
+
|
| 19 |
+
## Citation
|
| 20 |
+
|
| 21 |
+
If you find this work helpful for your research, please consider citing:
|
| 22 |
+
|
| 23 |
+
```bibtex
|
| 24 |
+
@article{zhang2025egrpo,
|
| 25 |
+
title={E-GRPO: High Entropy Steps Drive Effective Reinforcement Learning for Flow Models},
|
| 26 |
+
author={Zhang, Shengjun and Zhang, Zhang and Dai, Chensheng and Duan, Yueqi},
|
| 27 |
+
journal={arXiv preprint arXiv:2601.00423},
|
| 28 |
+
year={2025}
|
| 29 |
+
}
|
| 30 |
+
```
|