Statistical Guarantees of Group-Invariant GANs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Ziyu, Katsoulakis, Markos A., Rey-Bellet, Luc, Zhu, Wei
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912268800753664
author Chen, Ziyu
Katsoulakis, Markos A.
Rey-Bellet, Luc
Zhu, Wei
author_facet Chen, Ziyu
Katsoulakis, Markos A.
Rey-Bellet, Luc
Zhu, Wei
contents This work presents the first statistical performance guarantees for group-invariant generative models. Many real data, such as images and molecules, are invariant to certain group symmetries, which can be taken advantage of to learn more efficiently as we rigorously demonstrate in this work. Here we specifically study generative adversarial networks (GANs), and quantify the gains when incorporating symmetries into the model. Group-invariant GANs are a type of GANs in which the generators and discriminators are hardwired with group symmetries. Empirical studies have shown that these networks are capable of learning group-invariant distributions with significantly improved data efficiency. In this study, we aim to rigorously quantify this improvement by analyzing the reduction in sample complexity and in the discriminator approximation error for group-invariant GANs. Our findings indicate that when learning group-invariant distributions, the number of samples required for group-invariant GANs decreases proportionally by a factor of the group size and the discriminator approximation error has a reduced lower bound. Importantly, the overall error reduction cannot be achieved merely through data augmentation on the training data. Numerical results substantiate our theory and highlight the stark contrast between learning with group-invariant GANs and using data augmentation. This work also sheds light on the study of other generative models with group symmetries, such as score-based generative models.
format Preprint
id arxiv_https___arxiv_org_abs_2305_13517
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Statistical Guarantees of Group-Invariant GANs
Chen, Ziyu
Katsoulakis, Markos A.
Rey-Bellet, Luc
Zhu, Wei
Machine Learning
62E10, 62E17, 60-08
This work presents the first statistical performance guarantees for group-invariant generative models. Many real data, such as images and molecules, are invariant to certain group symmetries, which can be taken advantage of to learn more efficiently as we rigorously demonstrate in this work. Here we specifically study generative adversarial networks (GANs), and quantify the gains when incorporating symmetries into the model. Group-invariant GANs are a type of GANs in which the generators and discriminators are hardwired with group symmetries. Empirical studies have shown that these networks are capable of learning group-invariant distributions with significantly improved data efficiency. In this study, we aim to rigorously quantify this improvement by analyzing the reduction in sample complexity and in the discriminator approximation error for group-invariant GANs. Our findings indicate that when learning group-invariant distributions, the number of samples required for group-invariant GANs decreases proportionally by a factor of the group size and the discriminator approximation error has a reduced lower bound. Importantly, the overall error reduction cannot be achieved merely through data augmentation on the training data. Numerical results substantiate our theory and highlight the stark contrast between learning with group-invariant GANs and using data augmentation. This work also sheds light on the study of other generative models with group symmetries, such as score-based generative models.
title Statistical Guarantees of Group-Invariant GANs
topic Machine Learning
62E10, 62E17, 60-08
url https://arxiv.org/abs/2305.13517