VibrantLeaves: A principled parametric image generator for training deep restoration models

Abstract

In this paper, we introduce a synthetic image generator relying on a few simple principles, specifically focusing on geometric modeling, textures, and a simple modeling of image acquisition. These principles, integrated into the classical Dead Leaves model, allow for the creation of high-quality training sets for image restoration tasks. Standard image denoising and super-resolution networks trained on these datasets achieve performance comparable to those trained on natural image datasets. The motivation behind this approach stems from the limitations of Deep Neural Networks in image restoration tasks. Despite their impressive performance, these networks are often poorly understood and prone to biases inherited from standard natural image training sets. To mitigate these issues, we emphasize the need for a better control over training sets, particularly through the use of synthetic and abstract datasets. Furthermore, our work includes a detailed analysis of the principles considered, identifying which image properties are necessary for maintaining high performance, thus taking a first step towards explainability. Besides, we show that neural networks trained with our synthetic sets naturally inherit robustness to the invariance properties of these sets, namely rotation and scale.