计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (9): 89-95.DOI: 10.3778/j.issn.1002-8331.2004-0387

• 理论与研发 • 上一篇    下一篇

梯度策略的多目标GANs帕累托最优解算法

张波,徐黎明,黄志伟,要小鹏   

  1. 1.西南医科大学 医学信息与工程学院,四川 泸州 646000
    2.重庆邮电大学 计算机科学与技术学院,重庆 400065
  • 出版日期:2021-05-01 发布日期:2021-04-29

Multi-objective GANs Pareto Optimality Algorithm Using Gradient Strategy

ZHANG Bo, XU Liming, HUANG Zhiwei, YAO Xiaopeng   

  1. 1.School of Medical Information and Engineering, Southwest Medical University, Luzhou, Sichuan 646000, China
    2.School of Computer Science and Technology, Chongqing University of Posts and Telecommunications, Chongqing 400065, China
  • Online:2021-05-01 Published:2021-04-29

摘要:

针对基于梯度策略的多目标优化算法无法适用于多目标、高维度的生成对抗网络(Generative Adversarial Nets, GANs)及多目标GANs中利用交叉验证产生次优解,极难求得最优解等问题,提出一种基于梯度策略的多目标GANs帕累托最优解算法。该算法采用硬参数共享方式,将多目标优化分解为多个两目标优化,确定多目标权重参数后,沿着梯度方向进行线性搜索,最终确定帕累托最优解。理论上,在弱条件约束下,证明了所提算法能够确切地产生帕累托最优解。实验上,将所提算法应用到图像处理的常见领域,对比所提算法与原算法的性能。结果表明,当目标数量大于2时,所提算法能够产生明显的性能优势。

关键词: 梯度策略, 多目标生成对抗网络, 帕累托最优解, 图像处理

Abstract:

To address the problems that the gradient-based multi-objective algorithm cannot be applicable for multi-objective Generative Adversarial Nets(GANs) with high dimension and multiple tasks. Then, considering a fact that solutions with cross-of-validation can be sub-optimal and it is hard to search global solutions in optimizing multi-objective GANs, this paper presents a multi-objective GANs Pareto optimality algorithm based on gradient strategy. The proposed algorithm uses hardware parameter sharing method and decomposes multi-objective optimization into multiple binary-objective optimizations. Then, it computes and determines all the weighted parameters and searches Pareto optimality along the gradient direction, which can yield exactly Pareto optimality. Theoretically, it has been proved that the proposed method can result in one Pareto optimality with detailed demonstration. Practically, the proposed algorithm has been applied into common sub-fields of image processing to compare the source and proposed algorithm in the same setting. The experimental results show that the proposed algorithm has outperformed than source algorithms when the number of tasks is over 2.

Key words: gradient strategy, multi-objective generative adversarial nets, Pareto optimality, image processing