Image Classification Improvement: Text-to-Image AI for Synthetic Dataset Approach

Olger Zambrano, Benaoumeur Senouci

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review

Abstract

In this paper, we share our experience in using synthetic image generation through text-to-image AI models with the aim of improving the performance of image classifiers models. The procedure for generating high-quality synthetic images with text-to-image AI models is described, utilizing generic prompts based on proximity in hypernymys to generate 200 images of samples (lions, cats, and monkeys). Two image classifiers were built using the synthetic images alongside with classical image augmentation techniques. Each classifier was evaluated using three different datasets. Stable Diffusion 2.1 was employed as the image generator, and the results demonstrate that our proposed technique of generating synthetic images can indeed enhance image classifier accuracy. The results rely on the proficiency of the AI and the appropriateness of utilized prompts. We think our results can be improved by advances on the Latent Diffusion models.
Original languageEnglish
Title of host publicationProceedings - 2023 49th Euromicro Conference on Software Engineering and Advanced Applications, SEAA 2023
PublisherIEEE
Publication date2023
Pages74-77
ISBN (Electronic)979-8-3503-4235-2
DOIs
Publication statusPublished - 2023
Event49th Euromicro Conference on Software Engineering and Advanced Applications (SEAA) - Durres, Albania
Duration: 6. Sept 20238. Sept 2023

Conference

Conference49th Euromicro Conference on Software Engineering and Advanced Applications (SEAA)
Country/TerritoryAlbania
CityDurres
Period06/09/202308/09/2023

Keywords

  • Synthetic dataset
  • image augmentation techniques
  • image classifiers
  • text-to-image AI models

Cite this