Improving Art Style Classification Through Data Augmentation Using Diffusion Models.

Moyano, Miguel Ángel Martín; García-Aguilar, Iván; López-Rubio, Ezequiel; Luque-Baena, Rafael M.

Classifying pictorial styles in artworks is a complex challenge due to the diversity and lack of available datasets, which often limit the performance of machine learning models. To address this issue, we propose a novel data augmentation approach using Diffusion models in contrast to traditional augmentation techniques. Our method generates new samples based on the existing data, expanding the available dataset and enhancing the generalization capability of classification models. We evaluate the effectiveness of this data augmentation technique by training deep learning models with varying proportions of augmented and real data and assessing their performance in pictorial style classification. Our results demonstrate that the proposed Diffusion model-based augmentation significantly improves classification accuracy, suggesting that it can be a viable solution for overcoming data limitations in similar applications.

MACHINE learning; DATA augmentation; MACHINE performance; DEEP learning; ARTISTIC style; CLASSIFICATION

Electronics (2079-9292), 2024, Vol 13, Issue 24, p5038

2079-9292

Academic Journal

10.3390/electronics13245038