DermoCC-GAN: A new approach for standardizing dermatological images using generative adversarial networks.

ABSTRACT

BACKGROUND AND OBJECTIVE: Dermatological images are typically diagnosed based on visual analysis of the skin lesion acquired using a dermoscope. However, the final quality of the acquired image is highly dependent on the illumination conditions during the acquisition phase. This variability in the light source can affect the dermatologist's diagnosis and decrease the accuracy of computer-aided diagnosis systems. Color constancy algorithms have proven to be a powerful tool to address this issue by allowing the standardization of the image illumination source, but the most commonly used algorithms still present some inherent limitations due to assumptions made on the original image. In this work, we propose a novel Dermatological Color Constancy Generative Adversarial Network (DermoCC-GAN) algorithm to overcome the current limitations by formulating the color constancy task as an image-to-image translation problem. METHODS: A generative adversarial network was trained with a custom heuristic algorithm that performs well on the training set. The model hence learns the domain transfer task (from original to color standardized image) and is then able to accurately apply the color constancy on test images characterized by different illumination conditions. RESULTS: The proposed algorithm outperforms state-of-the-art color constancy algorithms for dermatological images in terms of normalized median intensity and when using the color-normalized images in a deep learning framework for lesion classification (accuracy of the seven-class classifier 79.2%) and segmentation (dice score 90.9%). In addition, we validated the proposed approach on two different external datasets with highly satisfactory results. CONCLUSIONS: The novel strategy presented here shows how it is possible to generalize a heuristic method for color constancy for dermatological image analysis by training a GAN. The overall approach presented here can be easily extended to numerous other applications.