HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SAN: Inducing Metrizability of GAN with Discriminative Normalized Linear Layer

Yuhta Takida Masaaki Imaizumi Takashi Shibuya Chieh-Hsin Lai Toshimitsu Uesaka Naoki Murata Yuki Mitsufuji

SAN: Inducing Metrizability of GAN with Discriminative Normalized Linear Layer

Abstract

Generative adversarial networks (GANs) learn a target probability distribution by optimizing a generator and a discriminator with minimax objectives. This paper addresses the question of whether such optimization actually provides the generator with gradients that make its distribution close to the target distribution. We derive metrizable conditions, sufficient conditions for the discriminator to serve as the distance between the distributions by connecting the GAN formulation with the concept of sliced optimal transport. Furthermore, by leveraging these theoretical results, we propose a novel GAN training scheme, called slicing adversarial network (SAN). With only simple modifications, a broad class of existing GANs can be converted to SANs. Experiments on synthetic and image datasets support our theoretical results and the SAN's effectiveness as compared to usual GANs. Furthermore, we also apply SAN to StyleGAN-XL, which leads to state-of-the-art FID score amongst GANs for class conditional generation on ImageNet 256$\times$256. Our implementation is available on https://ytakida.github.io/san.

Code Repositories

sony/san
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-cifar-10StyleSAN-XL
FID: 1.36
image-generation-on-ffhq-1024-x-1024StyleSAN-XL
FID: 1.61
image-generation-on-ffhq-256-x-256StyleSAN-XL
FID: 1.68
image-generation-on-ffhq-512-x-512StyleSAN-XL
FID: 1.77
image-generation-on-imagenet-256x256StyleSAN-XL
FID: 2.14

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
SAN: Inducing Metrizability of GAN with Discriminative Normalized Linear Layer | Papers | HyperAI