CoolFace
Modelpublic

baidu/ERNIE-Image-Aes

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
18likes330downloads
Model Card

ERNIE-Image-Aes: Robust Image Aesthetics Scoring with Balanced Category Generalization

<!-- tier_preview --> <img src="https://cdn-uploads.huggingface.co/production/uploads/5f8d780e5d083370c711f575/EVsGxYPd7kVWWIKlBj-d9.png" width="100%">

[๐Ÿ“„ Paper]

๐ŸŒŸ Highlights

ERNIE-Image-Aes is a 8B vision-language model for image aesthetic scoring, initialized from ArtiMuse and fine-tuned on a diverse, professionally annotated dataset. It substantially outperforms existing aesthetic predictors (LAION-AES, ArtiMuse, UniPercept) in generalization across diverse image categories.

Key advantages:

  • โ€”Balanced predictions across photography, anime, design, everyday snapshots, and film photography
  • โ€”No systematic bias toward specific image types (e.g., AI-generated content or black-and-white photos)
  • โ€”Swiss-tournament based pairwise annotation for high-quality training labels
  • โ€”Achieves 0.7445 SRCC and 0.7598 PLCC on ERIA-1K benchmark

๐Ÿ” Motivation

Off-the-shelf aesthetic predictors exhibit systematic biases:

ModelBias
LAION-AestheticDisproportionately high scores for AI-generated/anime content
ArtiMuseOverscores black-and-white photography and casual everyday snapshots
UniPerceptStrong preference for monochrome images; overscores casual snapshots

ERNIE-Image-Aes addresses these failure modes through a purpose-built annotation pipeline with explicit category balance.

๐Ÿ“Š Results on ERIA-1K Benchmark

ModelSRCCPLCC
LAION AES0.29440.3138
ArtiMuse0.42770.4704
UniPercept0.45330.4748
ERNIE-Image-Aes0.74450.7598

Annotation Protocol:

  • โ€”Pairwise Swiss-system tournament for stable and reproducible rankings
  • โ€”Tier labels from 1 to 10
  • โ€”Annotators recruited from professional backgrounds (Central Academy of Fine Arts, Sichuan Fine Arts Institute, Communication University of China, etc.)
  • โ€”All annotators passed aesthetic calibration screening prior to participation

โš™๏ธ Setup

Please follow the setup instructions in the ArtiMuse repository.

๐Ÿ™ Acknowledgements

Our work builds upon ArtiMuse and InternVL-3. We sincerely thank the authors for their excellent contributions to the community.

โœ’๏ธ Citation

If you find this work useful, please consider citing:

bibtex