概要
RTDMD（Reward-Tilted Distribution Matching Distillation）は、少ないステップ数で高品質な画像を生成する「few-step generator」を強化するための二段階フレームワークです。
報酬勾配強化学習と分布マッチング蒸留を組み合わせることで、SD3-M、SD3.5-M、FLUX.2 4Bといった既存の画像生成モデルの性能を飛躍的に向上させます。
特に、4NFEという少ない評価回数で、蒸留されたFLUX.2 4Bモデルが、より大きなオリジナルモデルFLUX.2 9B（50NFE）を上回る性能を示す点が大きな特徴です。
本リポジトリはPyTorchで実装されており、画像生成モデルの研究開発者や、少ない計算コストで最先端の画像生成技術を利用したいユーザーに適しています。
互換性・特徴
- PyTorch
- Python
- CLI
- GPU必須
- 画像生成
- 研究用途
基本情報
| ライセンス | Apache-2.0 |
| Stars | 29 |
| Forks | 1 |
| カテゴリ | 画像生成 |
| アクティビティ | low |
GitHub: https://github.com/Harahan/RTDMD