File size: 1,602 Bytes
a2312f2 039d62f baa1359 f5b7b44 baa1359 f5b7b44 2206c77 f5b7b44 baa1359 f5b7b44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
---
license: mit
---
# ConvNeXtV2-IllustrationScorer
**Q0: What does this model do?**
A: π This model scores your anime-style illustrations based on 4 metrics. π
**Q1: What does the 4 metrics mean?**
A: π The 4 metrics measures the "Liking Rate", "Collection Rate", "AI-generated Probability", and "View Number / Uploaded Interval (i.e. Popularity)". π
**Q2: Why the "Rate" seems not being a rate?**
A: β¨ This is because the author did not train this model by regressing these "Rates". Instead, these values are obtained in a contrastive learning manner (i.e., ranking the top-k images for each "Rate"). This is because the author has observed that almost no gradient can be significantly observed by backwarding on these "Rates" if the model is trained by regressing these values. And simply, the author assumed that the model tried to minimize the Absolute Error Loss by "remembering the average value", which is not an expected result. β¨
**Q3: What are the training data?**
A: π€ All training data (~55K) are obtained from PIXIV. π€
**Q4: Why this model is trained.**
A: πΎ The author initially hoped to finetune the [Anything-V5](https://civitai.com/models/9409?modelVersionId=90854) model by RLHF based on [D3PO (arxiv.2311.13231)](https://github.com/yk7333/d3po), and this model is designed to play the role of a multi-objective reward model. And for fun :)πΎ
![result](./assests/result.png)
## Acknowledgement
π¨ Thanks to [SUSTech CCSE](https://hpc.sustech.edu.cn/), this model is trained on A100-80G x 1. π¨
π€ Any suggestion is welcome :) π€
|