Population-Based Multi-Objective Training of Discriminators for Semi-Supervised GANs

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, quick

Summary

A new training strategy, Population-Based Multi-Objective Training, addresses the instability often found in semi-supervised generative adversarial networks (SSL-GANs). This method reframes discriminator learning as a multi-objective optimization problem, moving beyond aggregating supervised and unsupervised components into a single scalar loss. Instead, it maintains a population of discriminators, ranking them by Pareto dominance to explore various trade-offs between classification accuracy and the ability to discriminate between real and fake samples. This approach aims to enhance both the accuracy of the learned classifiers and the realism of samples produced by the generator. Experiments conducted on MNIST with limited labels demonstrated improved training robustness compared to existing SSL-GAN and CE-SSL-GAN baselines, with an elitist variant consistently achieving the highest classification accuracy.

Key takeaway

For Machine Learning Engineers developing semi-supervised GANs and facing training instability, you should consider adopting a population-based, multi-objective training strategy. This approach, which uses Pareto dominance to balance classification accuracy and real/fake discrimination, significantly improves robustness. Specifically, implementing the elitist variant can yield higher classification accuracy on limited-label datasets like MNIST, enhancing your model's overall performance.

Key insights

Multi-objective, population-based training stabilizes semi-supervised GANs, improving both classification accuracy and sample realism by managing trade-offs.

Principles

Method

Maintain a discriminator population, ranking members by Pareto dominance to explore classification accuracy vs. real/fake discrimination trade-offs, avoiding single scalar loss aggregation.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.