Multi-OctConv: Reducing Memory Requirements in Image Generative Adversarial Networks

Tobar F.M.; Torres C.E.

Keywords: GANs; image generation; memory usage

Abstract

Generative Adversarial Networks (GANs) for image generation of human faces have provided excellent results in recent years. However, we were able to identify a common problem among them: high memory usage in their training phase due to the convolutional encoder architecture used in these models. We address this issue by replacing the traditional convolutional layers in a model by what we call a Multi-Octave Convolution (M-OctConv) without modifying its architecture an advantage of this method is that it can be easily combined with traditional memory reduction techniques, such as pruning. We evaluate our proposition on StarGAN model achieving up to 40% of memory usage reduction without affecting the quality of the generated images.

Más información

Título según SCOPUS: Multi-OctConv: Reducing Memory Requirements in Image Generative Adversarial Networks
Título de la Revista: Proceedings - International Conference of the Chilean Computer Science Society, SCCC
Volumen: 2020-
Editorial: IEEE Computer Society
Fecha de publicación: 2020
Idioma: English
DOI:

10.1109/SCCC51225.2020.9281213

Notas: SCOPUS