git clone https://github.com/QuentinWach/image-ranker.git
cd image-ranker
python -m venv venv
source venv/bin/activate # On Windows, use `venvScriptsactivate`
pip install flask trueskill
python app.py
。http://localhost:5000
。 每个图像由两个值表示:
新项目以默认 μ(通常为 25,但此处为 0)和高 σ(通常为 8.33)开始。当比较两个项目时,它们的 μ 和 σ 值用于计算预期结果。实际结果与此预期进行比较。赢家的 μ 增加,输家的 μ 减少。两个项目的 σ 通常都会减少(表示确定性增加)。变化的幅度取决于:
它使用高斯分布来建模技能水平,并采用因子图和消息传递来实现高效更新。项目通常按 μ - 3σ(保守估计)进行排名。
重要的是,该算法在每次比较时同时更新所有先前排名的项目,而不是仅更新新图像。这意味着该算法可以考虑比较中可用的所有信息,而不仅仅是成对比较。
因此,总的来说,该系统允许使用不完整的比较数据进行有效排名,使其非常适合无法进行详尽的成对比较的大型项目!
您可以选择启用顺序消除来排名
您可以随时通过单击“随机播放”按钮手动随机播放图像对,或者每三个比较自动随机播放一次。如果您想尽快最小化排名的不确定性,这非常有用。仅排名几次且具有较高不确定性 σ 的图像将被优先考虑。这样,您就不必花费更多时间对您已经确定的图像进行排名,而是可以更快地获得分数非常相似的图像的更准确排名。
Image Ranker是让任何人都能够根据自己的特定需求定制创建自己的基础模型的整体努力的一部分。
训练后的基础模型使它们真正有用。例如,如果没有经过后期训练,大型语言模型甚至可能不会与你聊天。对于图像也是如此。为此,一种常见的技术是 RLHF,它使用奖励模型根据用户偏好来奖励或惩罚生成基础模型的输出。为了创建这个奖励模型,我们需要知道用户的偏好,这需要一个数据集,这里是图像。因此,无论是对现有模型(如稳定扩散或通量)进行一些根本性的改变,还是训练自己的模型,重要的是能够以某种方式对图像进行排名,以了解哪些图像更好。这就是这个应用程序的用武之地。
如果您有任何疑问,请在 GitHub 上提出问题!请随意分叉这个项目,提出建议或贡献新功能。 OPEN_TODO.md
文件包含计划实现的功能列表。非常感谢帮助!也就是说,支持该项目的最简单方法就是给这个仓库一个!
谢谢你!