GAN Regularizer for Stable Training and Diversity

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Training Generative Adversarial Networks (GANs) is computationally intensive and often results in mode collapse, where generated data collapses to one or a few points, and requires significant computational resources, making it time-consuming and inefficient.

Innovation Solution

The implementation of a Regularizer that encourages the discriminator to properly use its capacity by promoting high entropy in hidden representations and diverse data distribution, allowing for adaptive training steps and improved allocation of model capacity, thereby reducing computational complexity and enhancing training efficiency.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Reliability

If training GANs using conventional methods, then the discriminator learns to distinguish real and generated data, but the training process becomes computationally intensive and time-consuming

Engineering Contradiction:
Improvetraining stabilityVSAvoidtraining time
Core Design Contradiction:
ReliabilityVSLoss of time

Solution Approach 1:

The patent applies a regularizer to the discriminator's loss function before training begins, pre-configuring the discriminator to allocate its capacity evenly across different regions of the data space. This preliminary action prevents mode collapse from occurring during training, eliminating the need for lengthy training periods to achieve stable results.

Inventive Principle:
Principle #10Preliminary action

Solution Approach 2:

The patent modifies the discriminator's objective function by adding a regularizer term that changes the parameter allocation strategy. Instead of allowing the discriminator to focus on easy distinctions, the regularizer enforces even capacity allocation across all regions, fundamentally changing how the discriminator learns and reducing training time required for stability.

Inventive Principle:
Principle #35Parameter changes

2Measurement precision

If training GANs using conventional methods, then the discriminator can identify fake data, but mode collapse occurs where generated data collapses to one or a few points

Engineering Contradiction:
Improvediscriminator accuracyVSAvoidsample diversity
Core Design Contradiction:
Measurement precisionVSAdaptability or versatility

Solution Approach 1:

The patent applies the regularizer to ensure that different regions of the data space receive equal attention from the discriminator. By making the capacity allocation local to each region rather than globally optimized, the discriminator maintains high accuracy in identifying fake data while simultaneously ensuring diverse generation across all regions, preventing mode collapse.

Inventive Principle:
Principle #3Local quality

Solution Approach 2:

The regularizer creates an equipotential landscape for the discriminator's capacity allocation, where all regions of the data space have equal potential for being distinguished. This prevents the discriminator from concentrating its capacity in a few easy-to-distinguish regions, thereby maintaining both accuracy and diversity simultaneously.

Inventive Principle:
Principle #12Equipotentiality

3Reliability

If increasing the discriminator's model capacity to improve generation quality, then the discriminative ability improves, but the computational complexity increases

Engineering Contradiction:
Improvegeneration qualityVSAvoidcomputational complexity
Core Design Contradiction:
ReliabilityVSDevice complexity

Solution Approach 1:

The patent changes the parameter allocation strategy within the existing discriminator architecture by introducing a regularizer. This modifies how the discriminator's existing capacity is utilized, achieving improved generation quality through better parameter distribution rather than increasing the overall model size, thus avoiding increased computational complexity.

Inventive Principle:
Principle #35Parameter changes

Data Source

PatentUS12056605B2System and method for improved neural network training
Publication Date: 2024.08.06 ROYAL BANK OF CANADA
  • US12056605B2 patent drawing
  • US12056605B2 patent drawing
  • US12056605B2 patent drawing

AI summary

A system, electronic device and method for improved neural network training are provided. The electronic device includes: a processor, a memory storing a Generative adversarial network (GAN) to learn from unlabeled data by engaging a generative model in an adversarial game with a discriminator; and one or more programs stored in the memory and configured to be executed by the one or more processors, the one or more programs including instructions for training the Generative adversarial network using a regularizer to encourage the discriminator to properly use its capacity and hidden representations of the discriminator to have high entropy.