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
檔案包含計畫實現的功能清單。非常感謝幫忙!也就是說,支持該專案的最簡單方法就是給這個倉庫一個!
謝謝你!