Can Synthetic Images Improve CNN Performance in Wound Image Classification?

Stud Health Technol Inform. 2023 May 18:302:927-931. doi: 10.3233/SHTI230311.

Abstract

For artificial intelligence (AI) based systems to become clinically relevant, they must perform well. Machine Learning (ML) based AI systems require a large amount of labelled training data to achieve this level. In cases of a shortage of such large amounts, Generative Adversarial Networks (GAN) are a standard tool for synthesising artificial training images that can be used to augment the data set. We investigated the quality of synthetic wound images regarding two aspects: (i) improvement of wound-type classification by a Convolutional Neural Network (CNN) and (ii) how realistic such images look to clinical experts (n = 217). Concerning (i), results show a slight classification improvement. However, the connection between classification performance and the size of the artificial data set is still unclear. Regarding (ii), although the GAN could produce highly realistic images, the clinical experts took them for real in only 31% of the cases. It can be concluded that image quality may play a more significant role than data size in improving the CNN-based classification result.

Keywords: artificial intelligence; classification; convolutional neural network; data augmentation; generative adversarial networks; synthetic images; wound imaging.

MeSH terms

  • Artificial Intelligence*
  • Image Processing, Computer-Assisted
  • Machine Learning
  • Neural Networks, Computer*