Neural Image Beauty Predictor Based on Bradley-Terry Model

TL;DR

This paper introduces a CNN-based image beauty assessment method using the Bradley-Terry model for pairwise evaluation, achieving around 70% accuracy and providing a novel approach to mimic human visual system judgments.

Contribution

It presents a new image beauty prediction framework combining the Bradley-Terry model with CNNs and compares multiple CNN architectures for improved accuracy.

Findings

Achieved about 70% accuracy in pairwise image beauty assessment.

Pretraining on AVA dataset improves model performance.

CNN architectures vary in effectiveness for beauty prediction.

Abstract

Image beauty assessment is an important subject of computer vision. Therefore, building a model to mimic the image beauty assessment becomes an important task. To better imitate the behaviours of the human visual system (HVS), a complete survey about images of different categories should be implemented. This work focuses on image beauty assessment. In this study, the pairwise evaluation method was used, which is based on the Bradley-Terry model. We believe that this method is more accurate than other image rating methods within an image group. Additionally, Convolution neural network (CNN), which is fit for image quality assessment, is used in this work. The first part of this study is a survey about the image beauty comparison of different images. The Bradley-Terry model is used for the calculated scores, which are the target of CNN model. The second part of this work focuses on the results of the image beauty prediction, including landscape images, architecture images and portrait images. The models are pretrained by the AVA dataset to improve the performance later. Then, the CNN model is trained with the surveyed images and corresponding scores. Furthermore, this work compares the results of four CNN base networks, i.e., Alex net, VGG net, Squeeze net and LSiM net, as discussed in literature. In the end, the model is evaluated by the accuracy in pairs, correlation coefficient and relative error calculated by survey results. Satisfactory results are achieved by our proposed methods with about 70 percent accuracy in pairs. Our work sheds more light on the novel image beauty assessment method. While more studies should be conducted, this method is a promising step.