Electronic Science and Technology ›› 2023, Vol. 36 ›› Issue (7): 32-38.doi: 10.16180/j.cnki.issn1007-7820.2023.07.005

Previous Articles     Next Articles

Lightweight Generative Adversarial Networks Based on Multi-Scale Gradient

SUN Hong,ZHAO Yingzhi   

  1. School of Optical-Electrical and Computer Engineering,University of Shanghai for Science and Technology,Shanghai 200093,China
  • Received:2022-01-10 Online:2023-07-15 Published:2023-06-21
  • Supported by:
    National Natural Science Foundation of China(61472256);National Natural Science Foundation of China(61170277);National Natural Science Foundation of China(61703277)

Abstract:

With the advancement of generative adversarial network research, the computational amount of the network model increases sharply, its own training instability still exists, and the quality of the generated image also needs to be improved. To solve the problems, a lightweight generative adversarial network is proposed, which introduces multi-scale gradient structure to solve the problem of unstable training. By combining the ideas of self-attention mechanism and dynamic convolution, the cyclic module and image enhancement module are used to improve the learning ability of the model under the premise of keeping fewer parameters. The verification experimental results show that the inception score is 2.75 and the FID is 70.1 on CelebA data set, the inception score is 2.61 and the FID is 73.2 on LUSN data set, which is better than that of the classical models such as SAGAN and DCGAN, and verifies the feasibility and performance of the proposed algorithm.

Key words: multi-scale gradient, dynamic convolution, cyclic block, half-attention mechanism, sparse attention, convolutional neural networks, deep learning, image generation, generative adversarial net

CLC Number: 

  • TP391