Representation Distribution Matching

关键功能

通过分布匹配训练一步式图像生成器。
无需在线教师、对抗损失或轨迹匹配。
结合批内精确排斥和 Nyström 吸引项。
使用基于训练数据计算的冻结参考分布。
平衡一组 14 个预训练编码器。
支持用于文生图的联合图像-文本分布匹配。
将 FLUX.2 [klein] 从四步后训练为一步。
发布 MIT 许可证代码和公开模型检查点。

该方法将批内精确排斥与对冻结参考分布的 Nyström 吸引项结合起来,并结合梯度缓存使用大批量新生成样本。它平衡了 14 种编码器表示,包括用于条件生成的图像和文本特征,因此模型无法只优化单一表示而在其他空间中留下可见伪影。


RDM 适用于从事快速文生图生成的研究人员和基础设施团队。该发布展示了一步式 ImageNet 生成在 SW_r14 距离上达到 1.30,并将四步 FLUX.2 [klein] 后训练为一步模型,在 GenEval 和 PickScore 上超过教师模型,同时提供公开代码和检查点以便复现。

在您的网站添加此徽章,以获得更多点赞并登上搜索结果顶部!

嵌入按钮预览 - 浅色主题
嵌入按钮预览 - 深色主题

Subscribe to the AI Search Newsletter

Get top updates in AI to your inbox every weekend. It's free!