🌙

面向图像翻译的多目标进化生成对抗网络压缩

Multiobjective Evolutionary Generative Adversarial Network Compression for Image Translation

IEEE Transactions on Evolutionary Computation · 2023
被引 24
ABS 4

中文导读

提出多目标进化算法MEGC,将GAN模型的计算成本与生成图像质量建模为两目标优化问题,通过帕累托集引导超网训练,在减少计算开销的同时提升图像质量,适用于资源受限平台。

Abstract

Generative Adversarial Networks (GANs) have achieved remarkable success in image translation tasks. However, its prohibitive computational overhead has been a major hurdle for deployment on resource constrained platforms. Due to the training instability and the complicated network architecture, existing model acceleration techniques cannot appropriately handle the GAN compression problem. To cope with these difficulties, we propose a multi-objective evolutionary algorithm to compress GAN models in the context of image translation tasks, which is termed as MEGC. Particularly, the conflict between the computational cost of the GAN model and the quality of generated image is explicitly modeled as a two-objective optimization problem, and the evolved Pareto set is utilized to guide the sampling process during supernet training, which can in turn divert the focus of the supernet training to well-performing compact subnets. Besides, an evaluation-free strategy is introduced to facilitate exploration in the search space while incurring no extra computational cost. Based on the above design, the proposed MEGC eliminates the requirement of subnet searching in the post-processing procedure. Experiments on image translation tasks under paired and unparied settings demonstrate the effectiveness of the proposed MEGC on reducing the computational cost of GANs while improving the quality of generated images compared to those of the full models.

计算机科学图像翻译生成对抗网络多目标优化模型压缩