HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Enhancing GANs with MMD Neural Architecture Search, PMish Activation Function, and Adaptive Rank Decomposition

{Majid Rabbani Raghuveer Rao Jamison Heard Sohail Dianat Mahsa Mozaffari Prasanna Reddy Pulakurthi}

Abstract

Generative Adversarial Networks (GANs) have gained considerable attention owing to their impressive ability to generate high-quality, realistic images from a desired data distribution. This research introduces advancements in GANs by developing an improved activation function, a novel training strategy, and an adaptive rank decomposition method to compress the network. The proposed activation function, called Parametric Mish (PMish), automatically adjusts a trainable parameter to control the smoothness and shape of the activation function. Our method employs a Neural Architecture Search (NAS) to discover the optimal architecture for image generation while using the Maximum Mean Discrepancy (MMD) repulsive loss for adversarial training. The proposed novel training strategy improves performance by progressively increasing the upper bound of the bounded MMD-GAN repulsive loss. Finally, the proposed Adaptive Rank Decomposition (ARD) method reduces the complexity of the network with minimal impact on its generative performance, thus enabling efficient deployment on resource-limited platforms. The effectiveness of these advancements is rigorously tested on standard benchmark datasets such as CIFAR-10, CIFAR-100, STL-10, and CelebA, where significant improvements over existing techniques are demonstrated. The implementation code is available at: https://github.com/PrasannaPulakurthi/MMD-PMish-NAS

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-celeba-64x64MMD-PMish-NAS
FID: 1.92
Model Size (MB): 6.67
image-generation-on-celeba-64x64MMD-PMish-NAS (Compressed)
FID: 2.03
Model Size (MB): 3.04
image-generation-on-cifar-100MMD-PMish-NAS
FID: 7.68
Inception Score: 10.68
Model Size (MB): 17.9
image-generation-on-cifar-100MMD-PMish-NAS (Compressed)
FID: 8.37
Inception Score: 10.34
Model Size (MB): 2.3
image-generation-on-stl-10MMD-PMish-NAS
FID: 11.61
Inception score: 11.79
Model Size (MB): 19.47
image-generation-on-stl-10MMD-PMish-NAS (Compressed)
FID: 13.07
Inception score: 11.85
Model Size (MB): 2.78

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