Улучшение GAN с помощью оптимального транспорта15.03.2018 07:00

Аннотация

Мы представляем Optimal Transport GAN (OT-GAN) — вариант генеративно-состязательных сетей, который минимизирует новую метрику, измеряющую расстояние между распределением генератора и распределением данных. Эту метрику, которую мы называем энергетическим расстоянием на мини-батчах (mini-batch energy distance), мы получаем путем объединения оптимального транспорта в первоначальной форме с энергетическим расстоянием, определяемым в пространстве признаков, изученном с помощью состязательного обучения. В результате получается высокоразделимая функция расстояния с несмещенными градиентами по мини-батчам. Экспериментально мы показываем, что OT-GAN демонстрирует высокую стабильность при обучении с большими мини-батчами, и представляем передовые результаты на нескольких популярных задачах-бенчмарках для генерации изображений.