Original: D O GAN S LEARN THE DISTRIBUTION ? S OME T HEORY AND E MPIRICS
Revised: D O GAN S LEARN THE DISTRIBUTION ? S OME T HEORY AND E MPIRICS

Original: Anonymous authors
Revised: Anonymous authors

Original: Paper under double-blind review
Revised: Paper under double-blind review

Original: A BSTRACT
Revised: A BSTRACT

Original: Do GANS (Generative Adversarial Nets) actually learn the target distribution?
Revised: Do GANS (Generative Adversarial Nets) actually learn the target distribution?

Original: The foundational paper of Goodfellow et al.
Revised: The foundational paper of Goodfellow et al.

Original: (2014) suggested they do, if they were given “sufﬁciently large” deep nets, sample size, and computation time.
Revised: (2014) suggested they do, if they were given “sufﬁciently large” deep nets, sample size, and computation time.

Original: A recent theoretical analysis in Arora et al.
Revised: A recent theoretical analysis in Arora et al.

Original: (2017) raised doubts whether the same holds when discriminator has bounded size.
Revised: (2017) raised doubts whether the same holds when discriminator has bounded size.

Original: It showed that the training objective can approach its optimum value even if the generated distribution has very low support —in other words, the training objective is unable to prevent mode collapse .
Revised: It showed that the training objective can approach its optimum value even if the generated distribution has very low support —in other words, the training objective is unable to prevent mode collapse .

Original: The current paper makes two contributions.
Revised: The current paper makes two contributions.

Original: (1) It proposes a novel test for estimating support size using the birthday paradox of discrete probability.
Revised: (1) It proposes a novel test for estimating support size using the birthday paradox of discrete probability.

Original: Using this evidence is presented that well-known GANs approaches do learn distributions of fairly low support.
Revised: Using this evidence is presented that well-known GANs approaches do learn distributions of fairly low support.

Original: (2) It theoretically studies encoder-decoder GANs architectures (e.g., BiGAN/ALI), which were proposed to learn more meaningful features via GANs and (consequently) to also solve the mode-collapse issue.
Revised: (2) It theoretically studies encoder-decoder GANs architectures (e.g., BiGAN/ALI), which were proposed to learn more meaningful features via GANs and (consequently) to also solve the mode-collapse issue.

Original: Our result shows that such encoder-decoder training objectives also cannot guarantee learning of the full distribution because they cannot prevent serious mode collapse.
Revised: Our result shows that such encoder-decoder training objectives also cannot guarantee learning of the full distribution because they cannot prevent serious mode collapse.

Original: More seriously, they cannot prevent learning meaningless codes for data, contrary to usual intuition.
Revised: More seriously, they cannot prevent learning meaningless codes for data, contrary to usual intuition.

Original:  I NTRODUCTION
Revised:  I NTRODUCTION

Original: From the earliest papers on Generative Adversarial Networks (GANs) the question has been raised whether or not they actually come close to learning the distribution they are trained with (henceforth refered to as the target distribution)?
Revised: From the earliest papers on Generative Adversarial Networks (GANs) the question has been raised whether or not they actually come close to learning the distribution they are trained with (henceforth refered to as the target distribution)?

Original: These methods train a generator deep net that converts a random seed into a realistic-looking image.
Revised: These methods train a generator deep net that converts a random seed into a realistic-looking image.

Original: Concurrently they train a discriminator deep net to discriminate between its output and real images, which in turn is used to produce gradient feedback to improve the generator net.
Revised: Concurrently they train a discriminator deep net to discriminate between its output and real images, which in turn is used to produce gradient feedback to improve the generator net.

Original: In practice the generator deep net starts producing realistic outputs by the end, and the objective approaches its optimal value.
Revised: In practice the generator deep net starts producing realistic outputs by the end, and the objective approaches its optimal value.

Original: But does this mean the deep net has learnt the target distribution of real images?
Revised: But does this mean the deep net has learnt the target distribution of real images?

Original: Standard analysis introduced in (Goodfellow et al., 2014) shows that given “ sufﬁciently large” generator and discriminator nets, sample size, and computation time the training does succeed in learning the underlying distribution arbitrarily closely (measured in Jensen-Shannon divergence).
Revised: Standard analysis introduced in (Goodfellow et al., 2014) shows that given “ sufﬁciently large” generator and discriminator nets, sample size, and computation time the training does succeed in learning the underlying distribution arbitrarily closely (measured in Jensen-Shannon divergence).

Original: But this does not settle the question of what happens with realistic sample and net sizes.
Revised: But this does not settle the question of what happens with realistic sample and net sizes.

Original: Note that GANs differ from many previous methods for learning distributions in that they do not provide an estimate of a measure of distributional ﬁt —e.g., perplexity score.
Revised: Note that GANs differ from many previous methods for learning distributions in that they do not provide an estimate of a measure of distributional ﬁt —e.g., perplexity score.

Original: Therefore researchers have probed their performance using surrogate qualitative tests, which were usually designed to rule out the most obvious failure mode of the training, namely, that the GAN has simply memorized the training data.
Revised: Therefore researchers have probed their performance using surrogate qualitative tests, which were usually designed to rule out the most obvious failure mode of the training, namely, that the GAN has simply memorized the training data.

Original: One test checks the similarity of each generated image to the nearest images in the training set.
Revised: One test checks the similarity of each generated image to the nearest images in the training set.

Original: Another takes two random seeds s 1 , s 2 that produced realistic images and checks the images produced using seeds lying on the line joining s 1 , s 2 .
Revised: Another takes two random seeds s 1 , s 2 that produced realistic images and checks the images produced using seeds lying on the line joining s 1 , s 2 .

Original: If such “interpolating” images are reasonable and original as well, then this may be taken as evidence that the generated distribution has many novel images.
Revised: If such “interpolating” images are reasonable and original as well, then this may be taken as evidence that the generated distribution has many novel images.

Original: Yet other tests check for existence of semantically meaningful directions in the latent space, meaning that varying the seed along these directions leads to predictable changes e.g., (in case of images of human faces) changes in facial hair, or pose.
Revised: Yet other tests check for existence of semantically meaningful directions in the latent space, meaning that varying the seed along these directions leads to predictable changes e.g., (in case of images of human faces) changes in facial hair, or pose.

Original: A recent test proposed by Wu et al.
Revised: A recent test proposed by Wu et al.

Original: (2016) checks the log-likelihoods of GANs using Annealed Importance
Revised: (2016) checks the log-likelihoods of GANs using Annealed Importance

Original: Sampling, whose results indicate the mismatch between generator’s distribution and the target distribution.
Revised: Sampling, whose results indicate the mismatch between generator’s distribution and the target distribution.

Original: Poole et al.
Revised: Poole et al.

Original: (2016) proposed a method to trade-off between sample quality and sample diversity but they don’t provide a clear deﬁnition or a quantitative metric of sample diversity.
Revised: (2016) proposed a method to trade-off between sample quality and sample diversity but they don’t provide a clear deﬁnition or a quantitative metric of sample diversity.

Original: Recently a new theoretical analysis of GANs with ﬁnite sample sizes and ﬁnite discriminator size Arora et al.
Revised: Recently a new theoretical analysis of GANs with ﬁnite sample sizes and ﬁnite discriminator size Arora et al.

Original: (2017) revealed the possibility that training objective can sometimes approach optimality even if the generator is far from having actually learnt the distribution.
Revised: (2017) revealed the possibility that training objective can sometimes approach optimality even if the generator is far from having actually learnt the distribution.

Original: Speciﬁcally, if the discriminator has size p , then the training objective could be (cid:15) close to optimal even though the output distribution is supported on only O ( p log p/(cid:15) 2 ) images.
Revised: Speciﬁcally, if the discriminator has size p , then the training objective could be (cid:15) close to optimal even though the output distribution is supported on only O ( p log p/(cid:15) 2 ) images.

Original: By contrast one imagines that the target distribution usually must have very large support.
Revised: By contrast one imagines that the target distribution usually must have very large support.

Original: For example, the set of all possible images of human faces (a frequent setting in GANs work) must involve all combinations of hair color/style, facial features, complexion, expression, pose, lighting, race, etc., and thus the possible set of images of faces approaches inﬁnity.
Revised: For example, the set of all possible images of human faces (a frequent setting in GANs work) must involve all combinations of hair color/style, facial features, complexion, expression, pose, lighting, race, etc., and thus the possible set of images of faces approaches inﬁnity.

Original: The above paper raises the possibility that the discriminator may be unable to meaningfully distinguish such a diverse target distribution from a trained distribution with fairly small support.
Revised: The above paper raises the possibility that the discriminator may be unable to meaningfully distinguish such a diverse target distribution from a trained distribution with fairly small support.

Original: Furthermore, the paper notes that this failure mode is different from the one usually feared, namely the generator memorizing training samples.
Revised: Furthermore, the paper notes that this failure mode is different from the one usually feared, namely the generator memorizing training samples.

Original: The analysis of
Revised: The analysis of

Original: Arora et al.
Revised: Arora et al.

Original: (2017) raises the possibility that the trained distribution has small support, and yet all its samples could be completely disjoint from the training samples.
Revised: (2017) raises the possibility that the trained distribution has small support, and yet all its samples could be completely disjoint from the training samples.

Original: However, the above analysis was only a theoretical one, exhibiting a particular near-equilibrium solution that can happen from certain hyper-parameter combinations.
Revised: However, the above analysis was only a theoretical one, exhibiting a particular near-equilibrium solution that can happen from certain hyper-parameter combinations.

Original: It left open the possibility that real-life GANs training avoids such solutions thanks to some not-as-yet-understood property of back-propagation, SGD, or hyper-parameter choices.
Revised: It left open the possibility that real-life GANs training avoids such solutions thanks to some not-as-yet-understood property of back-propagation, SGD, or hyper-parameter choices.

Original: Thus further experimental investigation seems necessary.
Revised: Thus further experimental investigation seems necessary.

Original: And yet it seems difﬁcult at ﬁrst sight to do such an empirical evaluation of the support size of a distribution: it is not humanly possible to go through hundreds of thousands of images, whereas automated tests of image similarity can be thrown off by small changes in lighting, pose etc.
Revised: And yet it seems difﬁcult at ﬁrst sight to do such an empirical evaluation of the support size of a distribution: it is not humanly possible to go through hundreds of thousands of images, whereas automated tests of image similarity can be thrown off by small changes in lighting, pose etc.

Original: The current paper makes two important contributions.
Revised: The current paper makes two important contributions.

Original: On the empirical side it introduces a new test for the support size of the trained distribution, and uses it to ﬁnd that unfortunately these mode collapse problems do arise in many wellregarded GAN training methods.
Revised: On the empirical side it introduces a new test for the support size of the trained distribution, and uses it to ﬁnd that unfortunately these mode collapse problems do arise in many wellregarded GAN training methods.

Original: On the theoretical side we prove the limitations of encoder-decoder frameworks like
Revised: On the theoretical side we prove the limitations of encoder-decoder frameworks like

Original: BiGAN (Donahue et al., 2017) and Adversarially Learned Inference or ALI (Dumoulin et al., 2017), which, inspired by autoencoder models, require the setup to learn an inference mechanism as well as a generative mechanism.
Revised: BiGAN (Donahue et al., 2017) and Adversarially Learned Inference or ALI (Dumoulin et al., 2017), which, inspired by autoencoder models, require the setup to learn an inference mechanism as well as a generative mechanism.

Original: The result of Arora et al.
Revised: The result of Arora et al.

Original: (2017) applies only to standard GAN training objectives (including JS and Wasserstein), but not to encoder-decoder setups.
Revised: (2017) applies only to standard GAN training objectives (including JS and Wasserstein), but not to encoder-decoder setups.

Original: The clear hope in deﬁning encoder-decoder setups is that the encoding mechanism “inverts” the generator and thus forces the generator to learn meaningful featurizations of data that are useful in downstream applications.
Revised: The clear hope in deﬁning encoder-decoder setups is that the encoding mechanism “inverts” the generator and thus forces the generator to learn meaningful featurizations of data that are useful in downstream applications.

Original: In fact it has often been proposed that this need to learn meaningful featurizations will also solve the mode collapse problem: (Dumoulin et al., 2017) provide experiments on 2-dimensional mixtures of Gaussians suggesting this phenomenon.
Revised: In fact it has often been proposed that this need to learn meaningful featurizations will also solve the mode collapse problem: (Dumoulin et al., 2017) provide experiments on 2-dimensional mixtures of Gaussians suggesting this phenomenon.

Original: Our analysis shows not only that encoder-decoder training objectives cannot avoid mode collapse, but that they also cannot enforce learning of meaningful codes/features.
Revised: Our analysis shows not only that encoder-decoder training objectives cannot avoid mode collapse, but that they also cannot enforce learning of meaningful codes/features.

Original: (See Section 3.)
Revised: (See Section 3.)

Original: 1. B IRTHDAY PARADOX TEST FOR SUPPORT SIZE
Revised: 1. B IRTHDAY PARADOX TEST FOR SUPPORT SIZE

Original: Let’s consider a simple test that approximately tests the support size of a discrete distribution.
Revised: Let’s consider a simple test that approximately tests the support size of a discrete distribution.

Original: Suppose a distribution has support N .
Revised: Suppose a distribution has support N .

Original: The famous birthday paradox 1 says that a sample of size about √ N would be quite likely to have a duplicate.
Revised: The famous birthday paradox 1 says that a sample of size about √ N would be quite likely to have a duplicate.

Original: Thus our proposed birthday paradox test for GANs is as follows.
Revised: Thus our proposed birthday paradox test for GANs is as follows.

Original: Birthday Paradox Test
Revised: Birthday Paradox Test

Original: (a) Sample a batch ofsize s  from the generator.
Revised: (a) Sample a batch of s images from the generator.

Original: (b) Use an automated measure of image similarity to ﬂag the ∼ 20 most similar pairs in the sample.
Revised: (b) Use an automated measure of image similarity to ﬂag the ∼ 20 most similar pairs in the batch.

Original: (c) Visually inspect the ﬂagged pairs and check for duplicates.
Revised: (c) Visually inspect the ﬂagged pairs and check for duplicates.

Original: (d) Repeat.
Revised: (d) Repeat.

Original: If this test reveals that batches of size s have duplicate images with good probability, then suspect that the distribution has support size about s 2 .
Revised: If this test reveals that batches of size s have duplicate images with good probability, then suspect that the distribution has support size about s 2 .

Original: Note that the test is not deﬁnitive, because the distribution could assign a probability 10% to a single image, and be uniform on a huge number of other images.
Revised: Note that the test is not deﬁnitive, because the distribution could assign a probability 10% to a single image, and be uniform on a huge number of other images.

Original: Then the test would be quite likely to ﬁnd a duplicate even withsamples, even though the true support size is huge.
Revised: Then the test would be quite likely to ﬁnd a duplicate even with 20 samples, even though the true support size is huge.

Original: But such nonuniformity (a lot of probability being assigned to a few images) is the only failure mode of the birthday paradox test calculation, and such nonuniformity would itself be considered a failure mode of GANs training.
Revised: But such nonuniformity (a lot of probability being assigned to a few images) is the only failure mode of the birthday paradox test calculation, and such nonuniformity would itself be considered a failure mode of GANs training.

Original: This is captured in the following theorems:
Revised: This is captured in the following theorems:

Original: Theorem 1.
Revised: Theorem 1.

Original: Given a discrete probability distribution P on a set Ω , if there exists a subset S ⊆ Ω of size N such that (cid:80) s ∈ S P ( s ) ≥ ρ , then the probability of encountering at least one collision among M i.i.d.
Revised: Given a discrete probability distribution P on a set Ω , if there exists a subset S ⊆ Ω of size N such that (cid:80) s ∈ S P ( s ) ≥ ρ , then the probability of encountering at least one collision among M i.i.d.

Original: samples from P is ≥ 1 − exp( − ( M 2 − M ) ρ 2 N )
Revised: samples from P is ≥ 1 − exp( − ( M 2 − M ) ρ 2 N )

Original: Theorem 2.
Revised: Theorem 2.

Original: Given a discrete probability distribution P on a set Ω , if the probability of encountering at least one collision among M i.i.d.
Revised: Given a discrete probability distribution P on a set Ω , if the probability of encountering at least one collision among M i.i.d.

Original: samples from P is γ , then for ρ = 1 − o (1) , there exists
Revised: samples from P is γ , then for ρ = 1 − o (1) , there exists

Original: a subset S ⊆ Ω such that (cid:80) s ∈ S P ( s ) ≥ ρ with size ≤
Revised: a subset S ⊆ Ω such that (cid:80) s ∈ S P ( s ) ≥ ρ with size ≤

Original: [Equation]
Revised: [Equation]

Original: , under realistic assumptions on parameters.
Revised: , under realistic assumptions on parameters.

Original: The proofs of these theorems are included in Appendix A. Brieﬂy, the proof of theorem 1 is similar to one for the usual birthday paradox — calculate 1 − Pr
Revised: The proofs of these theorems are included in Appendix A. Brieﬂy, the proof of theorem 1 is similar to one for the usual birthday paradox — calculate 1 − Pr

Original: [ there is no collision within set S among M samples ] .
Revised: [ there is no collision within set S among M samples ] .

Original: For theorem 2, we use β = 1 (cid:80) X ∈ Ω P ( X ) 2 as a surrogate for the degree of uniformity of P .
Revised: For theorem 2, we use β = 1 (cid:80) X ∈ Ω P ( X ) 2 as a surrogate for the degree of uniformity of P .

Original: An upper bound on the probability of not seeing a collision within M samples can be shown in terms of β , using the Theorem 3 from Wiener (2005).
Revised: An upper bound on the probability of not seeing a collision within M samples can be shown in terms of β , using the Theorem 3 from Wiener (2005).

Original:  B IRTHDAY PARADOX TEST FOR GAN S : E XPERIMENTAL DETAILS
Revised:  B IRTHDAY PARADOX TEST FOR GAN S : E XPERIMENTAL DETAILS

Original: In the GAN setting, the distribution is continuous, not discrete.
Revised: In the GAN setting, the distribution is continuous, not discrete.

Original: When support size is inﬁnite then in a ﬁnite sample, we should not expect exact duplicate images where every pixel is identical.
Revised: When support size is inﬁnite then in a ﬁnite sample, we should not expect exact duplicate images where every pixel is identical.

Original: Thus a priori one imagines the birthday paradox test to completely not work.
Revised: Thus a priori one imagines the birthday paradox test to completely not work.

Original: But surprisingly, it still works if we look for near-duplicates.
Revised: But surprisingly, it still works if we look for near-duplicates.

Original: Given a ﬁnite sample, we select the 20 closest pairs according to some heuristic metric, thus obtaining a candidate pool of potential near-duplicates inspect.
Revised: Given a ﬁnite sample, we select the 20 closest pairs according to some heuristic metric, thus obtaining a candidate pool of potential near-duplicates inspect.

Original: Then we visually identify if any of them would be considered duplicates by humans.
Revised: Then we visually identify if any of them would be considered duplicates by humans.

Original: Our test were done using two datasets, CelebA (faces) (Liu et al., 2015) and CIFAR-10 (Krizhevsky, 2009) .
Revised: Our test were done using two datasets, CelebA (faces) (Liu et al., 2015) and CIFAR-10 (Krizhevsky, 2009) .

Original: We report in
Revised: We report in

Original: Appendix B results on the Bedroom dataset from the LSUN (Yu et al., 2015).
Revised: Appendix B results on the Bedroom dataset from the LSUN (Yu et al., 2015).

Original: CelebA is a large-scale face attributes dataset with more than 200K celebrity images (of 11k distinct individuals), each with 40 attribute annotations.
Revised: CelebA is a large-scale face attributes dataset with more than 200K celebrity images (of 11k distinct individuals), each with 40 attribute annotations.

Original: The images in this dataset cover large pose variations and background clutter.
Revised: The images in this dataset cover large pose variations and background clutter.

Original: Images of faces seem a good testing ground for our birthday paradox test since we humans are especially attuned to minor differences in faces.
Revised: Images of faces seem a good testing ground for our birthday paradox test since we humans are especially attuned to minor differences in faces.

Original: The CIFAR-10 datasetconsists of 60k 32x32 colour images in 10 classes (airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck), with 6k images per class.
Revised: The CIFAR-10 datasetconsists of 60k 32x32 colour images in 10 classes.

Original: For faces, we ﬁnd Euclidean distance in pixel space works well as a heuristic similarity measure, probably because the samples are centered and aligned.
Revised: For faces, we ﬁnd Euclidean distance in pixel space works well as a heuristic similarity measure, probably because the samples are centered and aligned.

Original: For CIFAR-10, we pre-train a discriminative Convolutional Neural Net for the full classiﬁcation problem, and use the top layer representation as an embedding.
Revised: For CIFAR-10, we pre-train a discriminative Convolutional Neural Net for the full classiﬁcation problem, and use the top layer representation as an embedding.

Original: Heuristic similarity is then measured as the Euclidean distance in the embedding space.
Revised: Heuristic similarity is then measured as the Euclidean distance in the embedding space.

Original: These metrics can be crude, but note that improving them can only lower our estimate of the support size, since a better similarity measure can only increase the number of duplicates found.
Revised: These metrics can be crude, but note that improving them can only lower our estimate of the support size, since a better similarity measure can only increase the number of duplicates found.

Original: Thus our reported estimates should be considered as upper bounds on the support size of the distribution.
Revised: Thus our reported estimates should be considered as upper bounds on the support size of the distribution.

Original: Note: Some GANs (and also older methods such as variational autoencoders) implicitly or explicitly apply noise to the training and generated images.
Revised: Note: Some GANs (and also older methods such as variational autoencoders) implicitly or explicitly apply noise to the training and generated images.

Original: This seems useful if the goal is to compute a perplexity score, which involves the model being able to assign a nonzero probability to every image.
Revised: This seems useful if the goal is to compute a perplexity score, which involves the model being able to assign a nonzero probability to every image.

Original: Such noised images are usually very blurry and the birthday paradox test does not work well for them, primarily because the automated measure of similarity no longer works well.
Revised: Such noised images are usually very blurry and the birthday paradox test does not work well for them, primarily because the automated measure of similarity no longer works well.

Original: Even visually judging similarity of noised images is difﬁcult.
Revised: Even visually judging similarity of noised images is difﬁcult.

Original: Thus our experiments work best with GANs that generate sharper, realistic images.
Revised: Thus our experiments work best with GANs that generate sharper, realistic images.

Original: 2. R ESULTS ON C ELEB A DATASET
Revised: 2. R ESULTS ON C ELEB A DATASET

Original: We test the following methods, doing the birthday paradox test with Euclidean distance in pixel space as the heuristic similarity measure 2 .
Revised: We test the following methods, doing the birthday paradox test with Euclidean distance in pixel space as the heuristic similarity measure 2 .

Original: For judging whether two images are the same, we set the criterion that the two faces are not exactly identical but look like doppelgangers.
Revised: For judging whether two images are the same, we set the criterion that the two faces are not exactly identical but look like doppelgangers.

Original: (Of course, in real life the only doppelgangers we know are usually twins.)
Revised: (Of course, in real life the only doppelgangers we know are usually twins.)

Original: • DCGAN —unconditional, with JSD objective as described in Goodfellow et al.
Revised: • DCGAN —unconditional, with JSD objective as described in Goodfellow et al.

Original: (2014) and Radford et al.
Revised: (2014) and Radford et al.

Original: • MIX+ GAN protocol introduced in Arora et al.
Revised: • MIX+ GAN protocol introduced in Arora et al.

Original: (2017), speciﬁcally, MIX+DCGAN with 3 mixture components.
Revised: (2017), speciﬁcally, MIX+DCGAN with 3 mixture components.

Original: • ALI (Dumoulin et al., 2017) (or equivalently BiGANs (Donahue et al., 2017)).
Revised: • ALI (Dumoulin et al., 2017) (or equivalently BiGANs (Donahue et al., 2017)).

Original: For fair comparison, we set the discriminator of ALI (or BiGANs) to be roughly the same in size as that of the DCGAN model, since the results of Section 2.1.1 below suggests that the discriminator size has a strong effect on diversity of the learnt distribution.
Revised: For fair comparison, we set the discriminator of ALI (or BiGANs) to be roughly the same in size as that of the DCGAN model, since the results of Section 2.1.1 below suggests that the discriminator size has a strong effect on diversity of the learnt distribution.

Original: We ﬁnd that with probability ≥ 50% , a batch of ∼ 800 samples contains at least one pair of duplicates for both
Revised: We ﬁnd that with probability ≥ 50% , a batch of ∼ 800 samples contains at least one pair of duplicates for both

Original: DCGAN and MIX+DCGAN.
Revised: DCGAN and MIX+DCGAN.

Original: Figure 1 displays duplicates and their nearest neighbors in training set.
Revised: Figure 1 displays duplicates and their nearest neighbors in training set.

Original: These results suggest that the support size of the distribution is less than 800 2 ≈ 640000 , being at the same order of the size of the training set, but this distribution is not just memorizing the training set (see the dashed boxes).
Revised: These results suggest that the support size of the distribution is less than 800 2 ≈ 640000 , being at the same order of the size of the training set, but this distribution is not just memorizing the training set (see the dashed boxes).

Original: ALI (or BiGANs) appears to be more diverse, in that collisions appear with 50% probability only with a batch size of 1200 , implying a support size of a million.
Revised: ALI (or BiGANs) appears to be more diverse, in that collisions appear with 50% probability only with a batch size of 1200 , implying a support size of a million.

Original: This is 6 x the training set, but still much smaller than the diversity one would expect among human faces 4 .
Revised: This is 6 x the training set, but still much smaller than the diversity one would expect among human faces 4 .

Original: Nevertheless, these tests do support the suggestion in Dumoulin et al.
Revised: Nevertheless, these tests do support the suggestion in Dumoulin et al.

Original: (2017) and Donahue et al.
Revised: (2017) and Donahue et al.

Original: (2017) that the bidirectional structure prevents some of the mode collapse observed in usual GANs.
Revised: (2017) that the bidirectional structure prevents some of the mode collapse observed in usual GANs.

Original: 2.1. D IVERSITY VS D ISCRIMINATOR S IZE
Revised: 2.1. D IVERSITY VS D ISCRIMINATOR S IZE

Original: The analysis of Arora et al Arora et al.
Revised: The analysis of Arora et al Arora et al.

Original: (2017) suggested that the support size could be as low as near-linear in the capacity of the discriminator; in other words, there is a near-equilibrium in which a distribution of such a small support could sufﬁce to fool the best discriminator.
Revised: (2017) suggested that the support size could be as low as near-linear in the capacity of the discriminator; in other words, there is a near-equilibrium in which a distribution of such a small support could sufﬁce to fool the best discriminator.

Original: So it is worth investigating whether training in real life allows generator nets to exploit this “loophole” in the training that we now know is in principle available to them.
Revised: So it is worth investigating whether training in real life allows generator nets to exploit this “loophole” in the training that we now know is in principle available to them.

Original: While a comprehensive test is beyond the scope of this paper, we do a ﬁrst test with a simplistic version of discriminator size (i.e., capacity).
Revised: While a comprehensive test is beyond the scope of this paper, we do a ﬁrst test with a simplistic version of discriminator size (i.e., capacity).

Original: We build DCGANs with increasingly larger discriminators while ﬁxing the other hyper-parameters.
Revised: We build DCGANs with increasingly larger discriminators while ﬁxing the other hyper-parameters.

Original: The discriminator used here is a 5-layer Convolutional Neural Network such that the number of output channels of each layer is 1 × , 2 × , 4 × , 8 × dim where dim is chosen to be 16 , 24 , 32 , . . .
Revised: The discriminator used here is a 5-layer Convolutional Neural Network such that the number of output channels of each layer is 1 × , 2 × , 4 × , 8 × dim where dim is chosen to be 16 , 24 , 32 , . . .

Original: Thus the discriminator size should be proportional to dim 2 .
Revised: Thus the discriminator size should be proportional to dim 2 .

Original: Figure 2 suggests that in this simple setup the diversity of the learnt distribution does indeed grow near-linearly with the discriminator size.
Revised: Figure 2 suggests that in this simple setup the diversity of the learnt distribution does indeed grow near-linearly with the discriminator size.

Original: 2. R ESULTS FOR CIFAR- 
Revised: 2. R ESULTS FOR CIFAR-

Original: On CIFAR-10, Euclidean distance in pixel space is not informative.
Revised: On CIFAR-10, Euclidean distance in pixel space is not informative.

Original: So we adopt a classifying CNN with 3 convolutional layers, 2 fully-connected layer and a 10-class soft-max output pretrained with a multi-class classiﬁcation objective, and use its top layer features as embeddings for similarity test using Euclidean distance.
Revised: So we adopt a classifying CNN with 3 convolutional layers, 2 fully-connected layer and a 10-class soft-max output pretrained with a multi-class classiﬁcation objective, and use its top layer features as embeddings for similarity test using Euclidean distance.

Original: Firstly, we ﬁnd that the result of the test is affected by the quality of samples.
Revised: Firstly, we ﬁnd that the result of the test is affected by the quality of samples.

Original: If the training uses noised samples (with noise being added either explicitly or implicitly in the objective) then the generated samples are also quite noisy.
Revised: If the training uses noised samples (with noise being added either explicitly or implicitly in the objective) then the generated samples are also quite noisy.

Original: Then the most similar samples in a batch tend to be blurry blobs of low quality.
Revised: Then the most similar samples in a batch tend to be blurry blobs of low quality.

Original: Indeed, when we test a DCGAN (even the best variant with 7.16 Inception Score reported in Huang et al.
Revised: Indeed, when we test a DCGAN (even the best variant with 7.16 Inception Score reported in Huang et al.

Original: , the pairs returned are mostly blobs.
Revised: , the pairs returned are mostly blobs.

Original: To get meaningful test results, we turn to a Stacked GAN which is the state-of-the-art generative model on CIFAR-
Revised: To get meaningful test results, we turn to a Stacked GAN which is the state-of-the-art generative model on CIFAR-

Original: (Inception Score 8.59 (Huang et al., 2016)).
Revised: (Inception Score 8.59 (Huang et al., 2016)).

Original: It also generates the most real-looking images.
Revised: It also generates the most real-looking images.

Original: Since this model is trained by conditioning on class label, we measure its diversity within each class separately.
Revised: Since this model is trained by conditioning on class label, we measure its diversity within each class separately.

Original: The batch sizes needed for duplicates are shown in Table 1 .
Revised: The batch sizes needed for duplicates are shown in Table 1 .

Original: Duplicate samples shown in Figure 3 .
Revised: Duplicate samples shown in Figure 3 .

Original: [Table]
Revised: [Table]

Original: We also check whether the duplicate images detected are close to any of the training images.
Revised: We also check whether the duplicate images detected are close to any of the training images.

Original: To do so we look for the nearest neighbor in the training set using our heuristic similarity measure and visually inspected the closest suspects.
Revised: To do so we look for the nearest neighbor in the training set using our heuristic similarity measure and visually inspected the closest suspects.

Original: We ﬁnd that the closest image is quite different from the duplicates, suggesting the issue with GANs is indeed lack of diversity (low support size) instead of memorizing training set.
Revised: We ﬁnd that the closest image is quite different from the duplicates, suggesting the issue with GANs is indeed lack of diversity (low support size) instead of memorizing training set.

Original:  L IMITATIONS OF E NCODER -D ECODER GAN ARCHITECTURES
Revised:  L IMITATIONS OF E NCODER -D ECODER GAN ARCHITECTURES

Original: We recall the Adversarial Feature Learning (BiGAN) setup from Donahue et al.
Revised: We recall the Adversarial Feature Learning (BiGAN) setup from Donahue et al.

Original: The “generative” player consists of two parts: a generator G and an encoder E .
Revised: The “generative” player consists of two parts: a generator G and an encoder E .

Original: The generator takes as input a latent variable z and produces a sample G ( z ) ; the encoder takes as input a data sample x and produces a guess for the latent variable E ( x ) .
Revised: The generator takes as input a latent variable z and produces a sample G ( z ) ; the encoder takes as input a data sample x and produces a guess for the latent variable E ( x ) .

Original: This produces two joint distributions over pairs of latent variables and data samples: ( z, G ( z )) and ( E ( x ) , x ) .
Revised: This produces two joint distributions over pairs of latent variables and data samples: ( z, G ( z )) and ( E ( x ) , x ) .

Original: The goal of the “generative” player is to convince the discriminator that these two distributions are the same, whereas the discriminator is being trained to distinguish between them.
Revised: The goal of the “generative” player is to convince the discriminator that these two distributions are the same, whereas the discriminator is being trained to distinguish between them.

Original: In the ideal case, the hope is that the “generative” player converges to ( z, G ( z )) and ( E ( x ) , x ) both being jointly distributed as p ( z, x ) where p is the joint distribution of the latent variables and data – i.e. G ( z ) is distributed as p ( x | z ) : the true generator distribution; and E ( x ) is distributed as p ( z | x ) : the true encoder distribution.
Revised: In the ideal case, the hope is that the “generative” player converges to ( z, G ( z )) and ( E ( x ) , x ) both being jointly distributed as p ( z, x ) where p is the joint distribution of the latent variables and data – i.e. G ( z ) is distributed as p ( x | z ) : the true generator distribution; and E ( x ) is distributed as p ( z | x ) : the true encoder distribution.

Original: Using usual min-max formalism for adversarial training, the BiGAN objective is written as:
Revised: Using usual min-max formalism for adversarial training, the BiGAN objective is written as:

Original: [Equation] [Equation]
Revised: [Equation] [Equation]

Original: where ˆ µ is the empirical distribution over data samples x ; ˆ ν is a distribution over random “seeds” for the latent variables: typically sampled from a simple distribution like a standard Gaussian; and φ is a concave “measuring” function.
Revised: where ˆ µ is the empirical distribution over data samples x ; ˆ ν is a distribution over random “seeds” for the latent variables: typically sampled from a simple distribution like a standard Gaussian; and φ is a concave “measuring” function.

Original: (The standard choice is log , though other options have been proposed in the literature.)
Revised: (The standard choice is log , though other options have been proposed in the literature.)

Original: For our purposes, we will assume that φ outputs values in the range [ − ∆ , ∆] , ∆ ≥ 1 , and is L φ -Lipschitz.
Revised: For our purposes, we will assume that φ outputs values in the range [ − ∆ , ∆] , ∆ ≥ 1 , and is L φ -Lipschitz.

Original: For ease of exposition we will refer to µ as the image distribution .
Revised: For ease of exposition we will refer to µ as the image distribution .

Original: The proof is more elegant if we assume that µ consists of images that have been noised —concretely, think of replacing every 100 th pixel by Gaussian noise.
Revised: The proof is more elegant if we assume that µ consists of images that have been noised —concretely, think of replacing every 100 th pixel by Gaussian noise.

Original: (Formally speaking, total number of pixels that have to be noised is the dimension of the latent space, which is typically extremely small compared to the number of pixels).

Original: Such noised images would of course look ﬁne to our eyes, and we would expect the learnt encoder/decoder to not be affected by this noise.
Revised: Such noised images would of course look ﬁne to our eyes, and we would expect the learnt encoder/decoder to not be affected by this noise.

Original: 5 For concreteness, we will take the seed/code distribution ν to be a spherical zero-mean Gaussian (in an arbitrary dimension and with an arbitrary variance).
Revised: For concreteness, we will take the seed/code distribution ν to be a spherical zero-mean Gaussian (in an arbitrary dimension and with an arbitrary variance).

Original: Furthermore, we will assume that Domain ( µ ) =
Revised: Furthermore, we will assume that Domain ( µ ) =

Original: R d , Domain ( ν ) =
Revised: R d , Domain ( ν ) =

Original: R ˜ d where we think of ˜ d (cid:28) d (which is certainly the case in practice – ˜ d is typically very small.)
Revised: R ˜ d with ˜ d < d (we think of ˜ d (cid:28) d , which is certainly the case in practice).

Original: As in Arora et al.
Revised: As in Arora et al.

Original: we assume that discriminators are L -lipschitz with respect to their trainable parameters, and the support size of the generator’s distribution will depend upon this L and the capacity p (= number of parameters) of the discriminator.
Revised: we assume that discriminators are L -lipschitz with respect to their trainable parameters, and the support size of the generator’s distribution will depend upon this L and the capacity p (= number of parameters) of the discriminator.

Original: Theorem 3 (Main) .
Revised: Theorem 3 (Main) .

Original: There exists a generator G of support p
Revised: There exists a generator G of support p

Original: [Equation]
Revised: [Equation]

Original: and an encoder E with at most ˜ d non-zero weights, s.t. for all discriminators D that are L -Lipschitz and have capacity less than p , it holds that
Revised: and an encoder E with at most ˜ d non-zero weights, s.t. for all discriminators D that are L -Lipschitz and have capacity less than p , it holds that

Original: [Equation]
Revised: [Equation]

Original: The interpretation of the above theorem isexactly as stated in the introductory sections: the encoder E has very small complexity (we will subsequently specify it precisely and show it simply extracts noise out of the input x ); the generator G is a small-support distribution (so presumably far from the true data distribution).
Revised: The interpretation of the above theorem is as stated before: the encoder E has very small complexity (we will subsequently specify it precisely and show it simply extracts noise from the input x ); the generator G is a small-support distribution (so presumably far from the true data distribution).

Original: Nevertheless, the value of the BiGAN objective is small.
Revised: Nevertheless, the value of the BiGAN objective is small.

Original: The argument of Arora et al.
Revised: The argument of Arora et al.

Original: (2017) seems unable to apply to this setting.
Revised: (2017) seems unable to apply to this setting.

Original: It is a simple concentration/epsilon-net argument showing that the discriminator of capacity p cannot distinguish between a generator that samples from µ versus one that memorizes a subset of p log p (cid:15)random images in µ and outputs one randomly from this subset.
Revised: It is a simple concentration/epsilonnet argument showing that the discriminator of capacity p cannot distinguish between a generator that samples from µ versus one that memorizes a subset of p log p (cid:15)random images in µ and outputs one randomly from this subset.

Original: By contrast, in the current setting we need to say what happens with the encoder.
Revised: By contrast, in the current setting we need to say what happens with the encoder.

Original: The precise noise model: Denoting by ˜ µ the distribution of unnnoised images, and ν the distribution of seeds/codes, we deﬁne the distribution of noised images µ as the following distribution: to produce a sample in µ take a sample ˜ x from ˜ µ and z from ν independently and output x = ˜ x
Revised: The precise noise model: Denoting by ˜ µ the distribution of unnnoised images, and ν the distribution of seeds/codes, we deﬁne the distribution of noised images µ as the following distribution: to produce a sample in µ take a sample ˜ x from ˜ µ and z from ν independently and output x = ˜ x

Original: (cid:126) z , which is deﬁned as x i =(cid:40)
Revised: (cid:126) z , which is deﬁned as x [Equation] =

Revised: [Equation]

Revised: (cid:98)

Revised: [Equation]

Original: z i (cid:98) d ˜ d (cid:99) , if i ≡ 0( mod (cid:98) d ˜ d (cid:99) )
Revised: , if i ≡ 0( mod (cid:98) [Equation] (cid:99) [Equation]

Original: ˜ x i , otherwise
Revised: ˜ x i , otherwise

Original: In other words, set every (cid:98) d ˜ d (cid:99) -th to one of the coordinates of z .
Revised: In other words, set every (cid:98) d ˜ d (cid:99) -th to one of the coordinates of z .

Original: In practical settings, ˜ d (cid:28) d is usually chosen, so noising ˜ d coordinates out of all d will have no visually noticeable effect.
Revised: In practical settings, ˜ d (cid:28) d is usually chosen, so noising ˜ d coordinates out of all d will have no visually noticeable effect.

Original: We refer to this new distribution µ as ˜ µ (cid:126) ν .

Original: 3. P ROOF S KETCH , T HEOREM 3; D ETAILED PROOF IN APPENDIX
Revised: 3. P ROOF S KETCH , T HEOREM 3; D ETAILED PROOF IN APPENDIX

Original: The main idea is to show the existence of the generator/encoder pair via a probabilistic construction that is shown to succeed with high probability.
Revised: The main idea is to show the existence of the generator/encoder pair via a probabilistic construction that is shown to succeed with high probability.

Original: • Encoder E : The encoder just extracts the noise from the noised image (by selecting the relevant ˜ d coordinates).
Revised: • Encoder E : The encoder just extracts the noise from the noised image (by selecting the relevant ˜ d coordinates).

Original: Namely, E (˜ x (cid:126) z ) =
Revised: Namely, E (˜ x (cid:126) z ) =

Original: z .
Revised: z .

Original: (So the code is just gaussian noise and has no meaningful content.)
Revised: (So the code is just gaussian noise and has no meaningful content.)

Original: It’s easy to see this can be captured using a ReLU network with ˜ d
Revised: It’s easy to see this can be captured using a ReLU network with ˜ d

Original: weights: we can simply connect the i -th output to the ( i (cid:98) d ˜ d (cid:99) )
Revised: weights: we can simply connect the i -th output to the ( i (cid:98) d ˜ d (cid:99) )

Original: -th input using an edge of weight 1.
Revised: -th input using an edge of weight 1.

Original: • Generator G : This is designed to produce a distribution of support size m :=
Revised: • Generator G : This is designed to produce a distribution of support size m :=

Original: [Equation] LL [Equation]
Revised: p ∆ 2 log 2 ( p ∆ LL φ /(cid:15) )

Original: [Table]
Revised: (cid:15)

Original: We ﬁrst deﬁne a partition of Domain ( ν ) =
Revised: We ﬁrst deﬁne a partition of Domain ( ν ) =

Original: R ˜ d into m equal-measure blocks under ν .
Revised: R ˜ d into m equal-measure blocks under ν .

Original: Next, we sample m samples x ∗ 1 , x ∗ 2 , . . .
Revised: Next, we sample m samples x ∗ 1 , x ∗ 2 , . . .

Original: , x ∗ m from the image distribution.
Revised: , x ∗ m from the image distribution.

Original: Finally, for a sample z , we deﬁne ind ( z ) to be the index of the block in the partition in which z lies, and deﬁne the generator as G ( z ) =
Revised: Finally, for a sample z , we deﬁne ind ( z ) to be the index of the block in the partition in which z lies, and deﬁne the generator as G ( z ) =

Original: x ∗ ind ( z ) (cid:126) z .
Revised: x ∗ ind ( z ) (cid:126) z .

Original: Since the set of samples x ∗ i : i ∈ [ m ] is random, this speciﬁes a distribution over generators.
Revised: Since the set of samples x ∗ i : i ∈ [ m ] is random, this speciﬁes a distribution over generators.

Original: We prove that with high probability, one of these generators satisﬁes the statement of Theorem 3.
Revised: We prove that with high probability, one of these generators satisﬁes the statement of Theorem 3.

Original: Moreover, we show that such a generator can be easily implemented using a ReLU network of complexity
Revised: Moreover, we show that such a generator can be easily implemented using a ReLU network of complexity

Original: O ( md ) in Theorem 5.
Revised: O ( md ) in Theorem 5.

Original: The basic intuition of the proof is as follows.
Revised: The basic intuition of the proof is as follows.

Original: As a useful piece of notation, we will call a set T of samples from ν non-colliding if no two lie in the same block.
Revised: We will call a set T of samples from ν non-colliding if no two lie in the same block.

Original: Let T nc be the distribution over non-colliding sets { z 1 , z 2 , . . .
Revised: Let T nc be the distribution over non-colliding sets { z 1 , z 2 , . . .

Original: z m } , s.t. each z
Revised: z m } , s.t. each z

Original: i is sampled independently from the conditional distribution of ν inside the i-th block of the partition.
Revised: i is sampled independently from the conditional distribution of ν inside the i-th block of the partition.

Original: First, we notice that under the distribution for G we deﬁned, it holds that
Revised: First, we notice that under the distribution for G we deﬁned, it holds that

Original: [Equation]
Revised: [Equation]

Original: In other words, the “expected” encoder correctly matches the expectation of φ ( D ( x, E ( x ))) , so that the discriminator is fooled.
Revised: In other words, the “expected” encoder correctly matches the expectation of φ ( D ( x, E ( x ))) , so that the discriminator is fooled.

Original: We want to show that E G E z ∼ ν φ ( G ( z ) , z ) concentrates enough around this expectation, as a function of the randomness in G , so that we can say with high probability over the choice of G , | E x ∼ µ φ ( D ( x, E ( x )))
Revised: We want to show that E G E z ∼ ν φ ( G ( z ) , z ) concentrates enough around this expectation, as a function of the randomness in G , so that we can say with high probability over the choice of G , | E x ∼ µ φ ( D ( x, E ( x )))

Original: − E z ∼ ν φ ( G ( z ) , z ) | is small.
Revised: − E z ∼ ν φ ( G ( z ) , z ) | is small.

Original: We handle the concentration argument in two steps:
Revised: We handle the concentration argument in two steps:

Original: • First, we note that we can calculate the expectation of φ ( D ( G ( z ) , z )) when z ∼ ν by calculating the empirical expectation over m -sized non-colliding sets T sampled according to T nc .
Revised: • First, we note that we can calculate the expectation of φ ( D ( G ( z ) , z )) when z ∼ ν by calculating the empirical expectation over m -sized non-colliding sets T sampled according to T nc .

Original: Namely, as we show in LemmaD. in
Revised: Namely, as we show in Lemma

Original: the Appendix:
Revised: D.1:

Original: [Equation]
Revised: [Equation]

Original: This follows easily from the fact that all blocks in the partition have equal measure under ν .
Revised: This follows easily from the fact that all blocks in the partition have equal measure under ν .

Original: Thus, we have reduced our task to arguing about the concentration of E T ∼T [Equation]
Revised: Thus, we have reduced our task to arguing about the concentration of E T ∼T nc E z ∼ T φ ( D ( G ( z ) , z ))

Original: ) , z )) (viewed as a random variable in G ).
Revised: (viewed as a random variable in G ).

Original: Towards this, we consider the random variable E z ∼ T φ ( D ( G ( z ) , z )) as a function of the randomness in G and T both.
Revised: Towards this, we consider the random variable E z ∼ T φ ( D ( G ( z ) , z )) as a function of the randomness in G and T both

Original: Since T is a non-colliding set of samples, we can write
Revised: Since T is a non-colliding set of samples, we can write

Original: [Equation]
Revised: [Equation]

Original: for some function f , where the random variables x ∗ i , z i are all mutually independent – thus use McDiarmid’s inequality to argue about the concentration of f in terms of both T and G .
Revised: for some function f , where the random variables x ∗ i , z i are all mutually independent – thus use McDiarmid’s inequality to argue about the concentration of f in terms of both T and G .

Original: From this, we can use Markov’s inequality to argue that all but an exponentially small (in p ) fraction of encoders G satisfy that: for all but an exponentially small (in p ) fraction of non-colliding sets T , | E z ∼ 
Revised: From this, we can use Markov’s inequality to argue that all but an exponentially small (in p ) fraction of encoders G satisfy that: for all but an exponentially small (in p ) fraction of non-colliding sets T , | E z ∼ T φ ( D ( G ( z ) , z )) − E G E T ∼T nc E z ∼ T φ

Original: [Equation] φ ( D ( G ( z ) , z )) − E G E T ∼T nc E z ∼ T φ ( D ( G ( z ) , z )) | is small.
Revised: ( D ( G ( z ) , z )) | is small.

Original: Note that this has to hold for all discriminators D – so we need to additionally build an epsilon-net, and union bound over all discriminators, similarly as in Arora et al.
Revised: Note that this has to hold for all discriminators D – so we need to additionally build an epsilon-net, and union bound over all discriminators, similarly as in Arora et al.

Original: From this fact, it’s easy to extrapolate that for such G ,
Revised: From this fact, it’s easy to extrapolate that for such G ,

Original: [Equation]
Revised: [Equation]

Original: is small, as we want.
Revised: is small, as we want.

Original: The details are in Lemma D.2 in the Appendix.
Revised: The details are in Lemma D.2 in the Appendix.

Original:  C ONCLUSIONS
Revised:  C ONCLUSIONS

Original: The paper reveals gaps in current thinking about GANs, and hopes to stimulate further theoretical and empirical study.
Revised: The paper reveals gaps in current thinking about GANs, and hopes to stimulate further theoretical and empirical study.

Original: GANs research has always struggled with the issue of mode collapse, and recent theoretical analysis of Arora et al.
Revised: GANs research has always struggled with the issue of mode collapse, and recent theoretical analysis of Arora et al.

Original: (2017) shows that the GANs training objective is not capable of preventing mode collapse.
Revised: (2017) shows that the GANs training objective is not capable of preventing mode collapse.

Original: This exhibits the existence of bad solutions in the optimization landscape.
Revised: This exhibits the existence of bad solutions in the optimization landscape.

Original: This in itself is not deﬁnitive, since existence of bad solutions is also known for the more traditional classiﬁcation tasks Zhang et al.
Revised: This in itself is not deﬁnitive, since existence of bad solutions is also known for the more traditional classiﬁcation tasks Zhang et al.

Original: (2017), where heldout sets can nevertheless prove that a good solution has been reached.
Revised: (2017), where heldout sets can nevertheless prove that a good solution has been reached.

Original: The difference in case of GANs is lack of an obvious way to establish that training succeeded.
Revised: The difference in case of GANs is lack of an obvious way to establish that training succeeded.

Original: Our new Birthday Paradox test gives a new benchmark for testing the support size (i.e., diversity of images) in a distribution.
Revised: Our new Birthday Paradox test gives a new benchmark for testing the support size (i.e., diversity of images) in a distribution.

Original: Though it may appear weak, experiments using this test suggest that current GANs approaches, speciﬁcally, the ones that produce images of higher visual quality, do suffer mode collapse.
Revised: Though it may appear weak, experiments using this test suggest that current GANs approaches, speciﬁcally, the ones that produce images of higher visual quality, do suffer mode collapse.

Original: Our rough experiments also suggest —again in line with the previous theoretical analysis—that the size of the distribution’s support scales near-linearly with discriminator capacity; though this needs more exhaustive testing.
Revised: Our rough experiments also suggest —again in line with the previous theoretical analysis—that the size of the distribution’s support scales near-linearly with discriminator capacity; though this needs more exhaustive testing.

Original: Researchers have raised the possibility that the best use of GANs is not distribution learning but feature learning.
Revised: Researchers have raised the possibility that the best use of GANs is not distribution learning but feature learning.

Original: Encoder-decoder GAN architectures seem promising since they try to force the generator to use “meaningful” encodings of the image.
Revised: Encoder-decoder GAN architectures seem promising since they try to force the generator to use “meaningful” encodings of the image.

Original: While such architectures do exhibit slightly better diversity in our experiments, our theoretical result suggest that the the encoder-decoder objective is also unable to avoid mode collapse, furthermore, also fails to guarantee meaningful codes.
Revised: While such architectures do exhibit slightly better diversity in our experiments, our theoretical result suggest that the the encoder-decoder objective is also unable to avoid mode collapse, furthermore, also fails to guarantee meaningful codes.

Original: R
Revised: R

Original: A B IRTHDAY PARADOX CALCULATION
Revised: A B IRTHDAY PARADOX CALCULATION

Original: In this section, we prove the statemts of Theorems 1 and 2.
Revised: In this section, we prove the statemts of Theorems 1 and 2.

Original: Proof of Theorem 1.
Revised: Proof of Theorem 1.

Original: Pr[ there is at least a collision among M samples ]
Revised: Pr[ there is at least a collision among M samples ]

Original: [Equation]
Revised: [Equation]

Original: there is no collision within set S among M samples
Revised: there is no collision within set S among M samples

Original: [Equation]
Revised: [Equation]

Original: We use the fact that the worst case is when the ρ probability mass is uniformly distributed on S .
Revised: We use the fact that the worst case is when the ρ probability mass is uniformly distributed on S .

Original: Proof of Theorem 2.
Revised: Proof of Theorem 2.

Original: Suppose X 1 , X 2 , . . .
Revised: Suppose X 1 , X 2 , . . .

Original: are i.i.d.
Revised: are i.i.d.

Original: samples from the discrete distribution P .
Revised: samples from the discrete distribution P .

Original: We deﬁne T = inf { t ≥ 2 , X t ∈ { X 1 , X 2 , . . .
Revised: We deﬁne T = inf { t ≥ 2 , X t ∈ { X 1 , X 2 , . . .

Original: , X t − 1 }} to be the collision time and also we use β = 1 Pr[ T =2] = 1 (cid:80) X ∈ Ω P ( X ) 2 as a surrogate for the uniformity of P .
Revised: , X t − 1 }} to be the collision time and also we use β = 1 Pr[ T =2] = 1 (cid:80) X ∈ Ω P ( X ) 2 as a surrogate for the uniformity of P .

Original: According Theorem 3 in Wiener (2005), Pr [ T ≥ M ] can be upper-bounded using β .
Revised: According Theorem 3 in Wiener (2005), Pr [ T ≥ M ] can be upper-bounded using β .

Original: Speciﬁcally, with β > 1000 and M ≤ 2 √ β ln β , which is usually true when P is the distribution of a generative model of images,
Revised: Speciﬁcally, with β > 1000 and M ≤ 2 √ β ln β , which is usually true when P is the distribution of a generative model of images,

Original: [Equation]
Revised: [Equation]

Original: To estimate β , we notice that
Revised: To estimate β , we notice that

Original: [Equation]
Revised: [Equation]

Original: [ there is at least a collision among
Revised: [ there is at least a collision among

Original: [Equation] [Equation]
Revised: [Equation] [Equation]

Original: ) which immediately implies
Revised: ) which immediately implies

Original: [Equation]
Revised: [Equation]

Original: This gives us a upper-bound of the uniformity of distribution P , which we can utilize.
Revised: This gives us a upper-bound of the uniformity of distribution P , which we can utilize.

Original: Let S ⊆
Revised: Let S ⊆

Original: Ω be the smallest set with probability mass ≥ ρ and suppose it size is N .
Revised: Ω be the smallest set with probability mass ≥ ρ and suppose it size is N .

Original: To estimate the largest possible N such that the previous inequality holds, we let
Revised: To estimate the largest possible N such that the previous inequality holds, we let

Original: [Equation]
Revised: [Equation]

Original: from which we obtain
Revised: from which we obtain

Original: [Equation]
Revised: [Equation]

Original: B E XPLORATORY RESULTS ON B EDROOM DATASET
Revised: B E XPLORATORY RESULTS ON B EDROOM DATASET

Original: We test DCGANs trained on the trained on 64 × 64 center cropped Bedroom dataset(LSUN) using Euclidean distance to extract collision candidates since it is impossible to train a CNN classiﬁer on such single-category (bedroom) dataset.
Revised: We test DCGANs trained on the trained on 64 × 64 center cropped Bedroom dataset(LSUN) using Euclidean distance to extract collision candidates since it is impossible to train a CNN classiﬁer on such single-category (bedroom) dataset.

Original: We notice that the most similar pairs are likely to be the corrupted samples with the same noise pattern (topcollision candidates all contain such patterns).
Revised: We notice that the most similar pairs are likely to be the corrupted samples with the same noise pattern (topcollision candidates all contain such patterns).

Original: When ignoring the noisy pairs, the most similar ”clean” pairs are not even similar according to human eyes.
Revised: When ignoring the noisy pairs, the most similar ”clean” pairs are not even similar according to human eyes.

Original: This implies that the distribution puts signiﬁcant probability on noise patterns, which can be seen as a form of under-ﬁtting (also reported in the DCGAN paper).
Revised: This implies that the distribution puts signiﬁcant probability on noise patterns, which can be seen as a form of under-ﬁtting (also reported in the DCGAN paper).

Original: We manually counted the number of samples with a ﬁxed noise pattern from a batch of 900 i.i.d samples.
Revised: We manually counted the number of samples with a ﬁxed noise pattern from a batch of 900 i.i.d samples.

Original: We ﬁnd 43 such corrupted samples among the 900 generated images, which implies 43 / 900 ≈ 5% probability.
Revised: We ﬁnd 43 such corrupted samples among the 900 generated images, which implies 43 / 900 ≈ 5% probability.

Original: C B IRTHDAY PARADOX TEST FOR VAE S
Revised: C B IRTHDAY PARADOX TEST FOR VAE S

Original: Given these ﬁndings, it is natural to wonder about the diversity of distributions learned using earlier methods such as Variational Auto-Encoders Kingma & Welling (2014) (VAEs).
Revised: Given these ﬁndings, it is natural to wonder about the diversity of distributions learned using earlier methods such as Variational Auto-Encoders Kingma & Welling (2014) (VAEs).

Original: Instead of using feedback from the discriminator, these methods train the generator net using feedback from an approximate perplexity calculation.
Revised: Instead of using feedback from the discriminator, these methods train the generator net using feedback from an approximate perplexity calculation.

Original: Thus the analysis of Arora et al.
Revised: Thus the analysis of Arora et al.

Original: (2017) does not apply as is to such methods and it is conceivable they exhibit higher diversity.
Revised: (2017) does not apply as is to such methods and it is conceivable they exhibit higher diversity.

Original: However, we ﬁnd the birthday paradox test difﬁcult to run since samples from a VAE trained on CelebA are not realistic or sharp enough for a human to deﬁnitively conclude whether or not two images are almost the same.
Revised: However, we ﬁnd the birthday paradox test difﬁcult to run since samples from a VAE trained on CelebA are not realistic or sharp enough for a human to deﬁnitively conclude whether or not two images are almost the same.

Original: Fig 5 shows examples of some indicative parts (hair, eyes, mouth, etc.) are
Revised: Fig 5 shows examples of some indicative parts (hair, eyes, mouth, etc.) are

Original: D B I GAN’ S DON ’ T EVADE THE CURSE OF LOW DIVERSITY
Revised: D B I GAN’ S DON ’ T EVADE THE CURSE OF LOW DIVERSITY

Original: We recall the basic notation from the main part: the image distribution will be denoted as µ , and the code/seed distribution as ν , which we assume is a spherical Gaussian.
Revised: We recall the basic notation from the main part: the image distribution will be denoted as µ , and the code/seed distribution as ν , which we assume is a spherical Gaussian.

Original: For concreteness, we assumed the domain of ˜ µ is R d and the domain of ν is R ˜ d with ˜ d ≤ d .
Revised: For concreteness, we assumed the domain of ˜ µ is R d and the domain of ν is R ˜ d with ˜ d ≤ d .

Original: (As we said, we are thinking of ˜ d (cid:28) d .)
Revised: (As we said, we are thinking of ˜ d (cid:28) d .)

Original: We also introduced the quantity m :=
Revised: We also introduced the quantity m :=

Original: p ∆ 2 log 2 ( p ∆ LL φ /(cid:15) )
Revised: p ∆ 2 log 2 ( p ∆ LL φ /(cid:15) )

Original: (cid:15).
Revised: (cid:15).

Original: Before proving Theorem 3, let’s note that the claim can easily be made into a ﬁnite-sample version.
Revised: Before proving Theorem 3, let’s note that the claim can easily be made into a ﬁnite-sample version.

Original: Namely:
Revised: Namely:

Original: Corollary D.1 (Main, ﬁnite sample version) .
Revised: Corollary D.1 (Main, ﬁnite sample version) .

Original: There exists a generator G of support m , s.t.
Revised: There exists a generator G of support m , s.t.

Original: if ˆ µ is the uniform distribution over a training set S of size at least m , and
Revised: if ˆ µ is the uniform distribution over a training set S of size at least m , and

Original: ˆ ν is the uniform distribution over a sample T from ν of size at least m , for all discriminators D that are L -Lipschitz and have less than p parameters, with probability 1 − exp( − Ω( p )) over the choice of training set S , T we have:
Revised: ˆ ν is the uniform distribution over a sample T from ν of size at least m , for all discriminators D that are L -Lipschitz and have less than p parameters, with probability 1 − exp( − Ω( p )) over the choice of training set S , T we have:

Original: [Equation]
Revised: [Equation]

Original: Proof.
Revised: Proof.

Original: As is noted in Theorem B.2 in Arora et al.
Revised: As is noted in Theorem B.2 in Arora et al.

Original: (2017), we can build a (cid:15)LL φ -net for the discriminators with a size bounded by e p log( LL φ p/(cid:15) ) .
Revised: (2017), we can build a (cid:15)LL φ -net for the discriminators with a size bounded by e p log( LL φ p/(cid:15) ) .

Original: By Chernoff and union bounding over the points in the (cid:15)LL φ -net, with probability at least 1 − exp( − Ω( p )) over the choice of a training set S , we have
Revised: By Chernoff and union bounding over the points in the (cid:15)LL φ -net, with probability at least 1 − exp( − Ω( p )) over the choice of a training set S , we have

Original: [Equation]
Revised: [Equation]

Original: for all discriminators D with capacity at most p .
Revised: for all discriminators D with capacity at most p .

Original: Similarly, with probability at least 1 − exp( − Ω( p )) over the choice of a noise set T ,
Revised: Similarly, with probability at least 1 − exp( − Ω( p )) over the choice of a noise set T ,

Original: [Equation]
Revised: [Equation]

Original: Union bounding over these two events, we get the statement of the Corollary.
Revised: Union bounding over these two events, we get the statement of the Corollary.

Original: Spelling out the distribution over generators more explicitly, we will in fact show:
Revised: Spelling out the distribution over generators more explicitly, we will in fact show:

Original: Theorem 4 (Main, more detailed) .
Revised: Theorem 4 (Main, more detailed) .

Original: Let G follow the distribution over generators deﬁned in Section 3.
Revised: Let G follow the distribution over generators deﬁned in Section 3.

Original: With probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G , for all discriminators D that are L-Lipschitz and have capacity bounded by p :
Revised: With probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G , for all discriminators D that are L-Lipschitz and have capacity bounded by p :

Original: [Equation]
Revised: [Equation]

Original: D. P ROOF OF THE MAIN CLAIM
Revised: D. P ROOF OF THE MAIN CLAIM

Original: Let us recall we call a set T of samples from ν non-colliding if no two lie in the same block
Revised: Let us recall we call a set T of samples from ν non-colliding if no two lie in the same block

Original: and we denoted T nc to be the distribution over non-colliding sets { z 1 , z 2 , . . .
Revised: and we denoted T nc to be the distribution over non-colliding sets { z 1 , z 2 , . . .

Original: z m } , s.t. each z
Revised: z m } , s.t. each z

Original: i is sampled independently from the conditional distribution of ν inside the i-th block of the partition.
Revised: i is sampled independently from the conditional distribution of ν inside the i-th block of the partition.

Original: First, notice the following Lemma:
Revised: First, notice the following Lemma:

Original: Lemma D.1 (Reducing to expectations over non-colliding sets) .
Revised: Lemma D.1 (Reducing to expectations over non-colliding sets) .

Original: Let G be a ﬁxed generator, and D a ﬁxed discriminator.
Revised: Let G be a ﬁxed generator, and D a ﬁxed discriminator.

Original: Then,
Revised: Then,

Original: [Equation]
Revised: [Equation]

Original: Proof.
Revised: Proof.

Original: By deﬁnition,
Revised: By deﬁnition,

Original: [Equation]
Revised: [Equation]

Original: where ( T nc ) i is the conditional distribution of ν in the i -th block of the partition.
Revised: where ( T nc ) i is the conditional distribution of ν in the i -th block of the partition.

Original: However, since the blocks form an equipartitioning, we have
Revised: However, since the blocks form an equipartitioning, we have

Original: [Equation]
Revised: [Equation]

Original: Lemma D.2 (Concentration of good generators) .
Revised: Lemma D.2 (Concentration of good generators) .

Original: With probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G ,
Revised: With probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G ,

Original: [Equation]
Revised: [Equation]

Original: for all discriminators D of capacity at most p .
Revised: for all discriminators D of capacity at most p .

Original: Proof.
Revised: Proof.

Original: Consider E z ∼ T φ
Revised: Consider E z ∼ T φ

Original: ( D ( G ( z ) , z )) as a random variable in T ∼ T nc and G for a ﬁxed D .
Revised: ( D ( G ( z ) , z )) as a random variable in T ∼ T nc and G for a ﬁxed D .

Original: We can write
Revised: We can write

Original: [Equation]
Revised: [Equation]

Original: where the random variables x ∗ i ,
Revised: where the random variables x ∗ i ,

Original: z i are all mutually independent.
Revised: z i are all mutually independent.

Original: Note that the arguments that f is a function of are all independent variables, so we can apply McDiarmid’s inequality.
Revised: Note that the arguments that f is a function of are all independent variables, so we can apply McDiarmid’s inequality.

Original: Towards that, let’s denote by z − i the vector of all inputs to f , except for z i .
Revised: Towards that, let’s denote by z − i the vector of all inputs to f , except for z i .

Original: Notice that
Revised: Notice that

Original: [Equation]
Revised: [Equation]

Original: (as changing z i to ˜ z i only affect one out of the m terms in E z ∼ T φ ( D ( G ( z ) , z )) ).
Revised: (as changing z i to ˜ z i only affect one out of the m terms in E z ∼ T φ ( D ( G ( z ) , z )) ).

Original: Analogously we have
Revised: Analogously we have

Original: [Equation]
Revised: [Equation]

Original: Denoting R D,T,G = f
Revised: Denoting R D,T,G = f

Original: ( x ∗ i , z
Revised: ( x ∗ i , z

Original: i , i ∈ [ m ]) − E T,G [ f ( x ∗ i , z i , i ∈ [ m ])] , by McDiarmid we get
Revised: i , i ∈ [ m ]) − E T,G [ f ( x ∗ i , z i , i ∈ [ m ])] , by McDiarmid we get

Original: [Equation]
Revised: [Equation]

Original: Building a (cid:15)LL φ -net for the discriminators and union bounding over the points in the net, we get that
Revised: Building a (cid:15)LL φ -net for the discriminators and union bounding over the points in the net, we get that

Original: Pr T,G ( ∃ D, | R D,T,G | ≥ (cid:15)/ 2) ≤ exp( − Ω( p log(∆ /(cid:15) ))) .
Revised: Pr T,G ( ∃ D, | R D,T,G | ≥ (cid:15)/ 2) ≤ exp( − Ω( p log(∆ /(cid:15) ))) .

Original: On the other hand, we also have
Revised: On the other hand, we also have

Original: [Equation]
Revised: [Equation]

Original: so by Markov’s inequality, with probability 1 − exp( − Ω( p log(∆ /(cid:15) )) over the choice of G , ,
Revised: so by Markov’s inequality, with probability 1 − exp( − Ω( p log(∆ /(cid:15) )) over the choice of G , ,

Original: [Equation]
Revised: [Equation]

Original: Let us denote by ˜ T ( G ) the sets T , s.t. ∀ D, | R D,T,G | ≤ (cid:15)/ 2 .
Revised: Let us denote by ˜ T ( G ) the sets T , s.t. ∀ D, | R D,T,G | ≤ (cid:15)/ 2 .

Original: Then, with probability 1 − exp( − Ω( p log(∆ /(cid:15) )))) over the choice of G , we have that for all D of capacity at most p ,
Revised: Then, with probability 1 − exp( − Ω( p log(∆ /(cid:15) )))) over the choice of G , we have that for all D of capacity at most p ,

Original: [Equation]
Revised: [Equation]

Original: which is what we want
Revised: which is what we want

Original: [Equation]
Revised: [Equation]

Original: With these in mind, we can prove the main theorem:
Revised: With these in mind, we can prove the main theorem:

Original: Proof of Theorem 3.
Revised: Proof of Theorem 3.

Original: From Lemma D.1 and Lemma D.2 we get that with probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G ,
Revised: From Lemma D.1 and Lemma D.2 we get that with probability 1 − exp( − Ω( p log(∆ /(cid:15) ))) over the choice of G ,

Original: [Equation]
Revised: [Equation]

Original: for all discriminators D of capacity at most p .
Revised: for all discriminators D of capacity at most p .

Original: On the other hand, it’s easy to see that
Revised: On the other hand, it’s easy to see that

Original: [Equation]
Revised: [Equation]

Original: Furthermore,
Revised: Furthermore,

Original: [Equation]
Revised: [Equation]

Original: by the deﬁnition of E ( x ) , which proves the statement of the theorem.
Revised: by the deﬁnition of E ( x ) , which proves the statement of the theorem.

Original: D. R EPRESENTABILITY RESULTS
Revised: D. R EPRESENTABILITY RESULTS

Original: In this section, we show that a generator G of the type we described in the previous section can be implemented easily using a ReLU network.
Revised: In this section, we show that a generator G of the type we described in the previous section can be implemented easily using a ReLU network.

Original: The encoder can be parametrized by the set x ∗ 1 , x ∗ 2 , . . .
Revised: The encoder can be parametrized by the set x ∗ 1 , x ∗ 2 , . . .

Original: x ∗ m of “memorized” training samples.
Revised: x ∗ m of “memorized” training samples.

Original: The high-level overview of the network is rather obvious: we partition R ˜
Revised: The high-level overview of the network is rather obvious: we partition R ˜

Original: d into equal-measure blocks; subsequently, for a noise sample z , we determine which block it belongs to, and output the corresponding sample x ∗ i , which is memorized in the weights of the network.
Revised: d into equal-measure blocks; subsequently, for a noise sample z , we determine which block it belongs to, and output the corresponding sample x ∗ i , which is memorized in the weights of the network.

Original: Theorem 5.
Revised: Theorem 5.

Original: Let G be the generator determined by the samples x ∗ 1 , x ∗ 2 , . . .
Revised: Let G be the generator determined by the samples x ∗ 1 , x ∗ 2 , . . .

Original: , x ∗ m ∈ R d , i.e. G ( z ) =
Revised: , x ∗ m ∈ R d , i.e. G ( z ) =

Original: x ∗ ind ( z ) (cid:126) z .
Revised: x ∗ ind ( z ) (cid:126) z .

Original: For any δ > 0 , there exists a ReLU network with O ( md ) non-zero weights, which implements a function ˜ G , s.t.
Revised: For any δ > 0 , there exists a ReLU network with O ( md ) non-zero weights, which implements a function ˜ G , s.t.

Original: (cid:107) ˜
Revised: (cid:107) ˜

Original: G ( ν ) − G ( ν ) (cid:107) TV ≤ δ , where (cid:107) · (cid:107)
Revised: G ( ν ) − G ( ν ) (cid:107) TV ≤ δ , where (cid:107) · (cid:107)

Original: [Table]
Revised: [Table]

Original: Proof.
Revised: Proof.

Original: The construction of the network is pictorially depicted on Figure D.2.
Revised: The construction of the network is pictorially depicted on Figure D.2.

Original: We expand on each of the individual parts of the network.
Revised: We expand on each of the individual parts of the network.

Original: First, we show how to implement the partitioning into sectors.
Revised: First, we show how to implement the partitioning into sectors.

Original: The easiest way to do this is to use the fact that the coordinates of a spherical Gaussian are independent, and simply partition each dimension separately into equimeasure blocks, depending on the value of | z i | : the absolute value of the i -th coordinate.
Revised: The easiest way to do this is to use the fact that the coordinates of a spherical Gaussian are independent, and simply partition each dimension separately into equimeasure blocks, depending on the value of | z i | : the absolute value of the i -th coordinate.

Original: Concretely, without loss of generality, let’s assume m = k ˜ d , for some k ∈ N .
Revised: Concretely, without loss of generality, let’s assume m = k ˜ d , for some k ∈ N .

Original: Let us denote by τ i ∈ R :
Revised: Let us denote by τ i ∈ R :

Original: i ∈
Revised: i ∈

Original: [ k ] , τ 0 = 

Revised: [ k ] , τ 0 =

Original: the real-numbers, s.t.
Revised: the real-numbers, s.t.

Original: Pr z i ∼ ν
Revised: Pr z i ∼ ν

Original: i [ | z i | ∈ ( τ i − 1 , τ i )]
Revised: i [ | z i | ∈ ( τ i − 1 , τ i )]

Original: = 1 k . .
Revised: = 1 k . .

Original: We will associate to each d -tuple
Revised: We will associate to each d -tuple

Original: ( i 1 , i 2 , . .
Revised: ( i 1 , i 2 , . .

Original: , i d ) ∈ [ k ] d the cell { z ∈ R ˜
Revised: , i d ) ∈ [ k ] d the cell { z ∈ R ˜

Original: d : | z j | ∈ ( τ i j − 1 , τ i j ) , ∀ j ∈
Revised: d : | z j | ∈ ( τ i j − 1 , τ i j ) , ∀ j ∈

Original: [ ˜ d ] } .
Revised: [ ˜ d ] } .

Original: These blocks clearly equipartition R ˜
Revised: These blocks clearly equipartition R ˜

Original: d with respect to the Gaussian measure.
Revised: d with respect to the Gaussian measure.

Original: Inferring the ˜ d -tuple after calculating the absolute values | z j | (which is trivially representable as a ReLU network as max(0 , z j ) + max(0 , − z j ) ) can be done using the “selector” circuit introduced in ‘Arora et al.
Revised: Inferring the ˜ d -tuple after calculating the absolute values | z j | (which is trivially representable as a ReLU network as max(0 , z j ) + max(0 , − z j ) ) can be done using the “selector” circuit introduced in ‘Arora et al.

Original: Namely, by Lemma 3 there, there exists a ReLU network with O ( k ) non-zero weights that takes | z j | as input and outputs k numbers ( b j, 1 , b j, 2 , . . .
Revised: Namely, by Lemma 3 there, there exists a ReLU network with O ( k ) non-zero weights that takes | z j | as input and outputs k numbers ( b j, 1 , b j, 2 , . . .

Original: , b j,k ) , s.t.
Revised: , b j,k ) , s.t.

Original: b j,i j = 1 and b j,l = 0 , j (cid:54) =
Revised: b j,i j = 1 and b j,l = 0 , j (cid:54) =

Original: i j with probability 1 − δ over the randomness of z j , j ∈ [ ˜ d ] .
Revised: i j with probability 1 − δ over the randomness of z j , j ∈ [ ˜ d ] .

Original: Since we care about G ( ν ) and ˜ G ( ν ) being close in total variation distance only, we can focus on the case where all z j are such that b j,i j = 1 and b j,l = 0 , j (cid:54) =
Revised: Since we care about G ( ν ) and ˜ G ( ν ) being close in total variation distance only, we can focus on the case where all z j are such that b j,i j = 1 and b j,l = 0 , j (cid:54) =

Original: i j for some indices
Revised: i j for some indices

Original: i j .
Revised: i j .

Original: We wish to now “turn on” the memorized weights for the corresponding block in the partition.
Revised: We wish to now “turn on” the memorized weights for the corresponding block in the partition.

Original: To do this, we ﬁrst pass the calculated ˜ d -tuple through a network which interprets it as a number in k -ary and calculates it’s equivalent decimal representation.
Revised: To do this, we ﬁrst pass the calculated ˜ d -tuple through a network which interprets it as a number in k -ary and calculates it’s equivalent decimal representation.

Original: This is easily implementable as a ReLU network with O ( ˜ dk ) weights calculating
Revised: This is easily implementable as a ReLU network with O ( ˜ dk ) weights calculating

Original: L = (cid:80) ˜ dj
Revised: L = (cid:80) ˜ dj

Original: =1 k j (cid:80) kl
Revised: =1 k j (cid:80) kl

Original: =1 b j,l i j .
Revised: =1 b j,l i j .

Original: Then, we use use a simple circuit with O ( m ) non-zero weights to output m numbers ( B 1 , B 2 , . . .
Revised: Then, we use use a simple circuit with O ( m ) non-zero weights to output m numbers ( B 1 , B 2 , . . .

Original: , B m ) , s.t.
Revised: , B m ) , s.t.

Original: B L = 1 and B i = 0 , i (cid:54) =
Revised: B L = 1 and B i = 0 , i (cid:54) =

Original: L (implemented in the obvious way).
Revised: L (implemented in the obvious way).

Original: The subnetwork of B i will be responsiple for the i -th memorized sample.
Revised: The subnetwork of B i will be responsiple for the i -th memorized sample.

Original: Namely, we attach to each coordinate B i a curcuit with fan-out of degree d , s.t.
Revised: Namely, we attach to each coordinate B i a curcuit with fan-out of degree d , s.t.

Original: the weight of edge j ∈
Revised: the weight of edge j ∈

Original: [ d ] is x ∗ i,j .
Revised: [ d ] is x ∗ i,j .

Original: Let’s denote these outputs as F i,j , i ∈ [ m ] , j ∈ [ d ] and let ˜ F
Revised: Let’s denote these outputs as F i,j , i ∈ [ m ] , j ∈ [ d ] and let ˜ F

Original: i : i ∈ [ d ] be deﬁned as ˜ F i = (cid:80) mj
Revised: i : i ∈ [ d ] be deﬁned as ˜ F i = (cid:80) mj

Original: =1 F i,j .
Revised: =1 F i,j .

Original: It’s easy to see since B i = 0 , i (cid:54) =
Revised: It’s easy to see since B i = 0 , i (cid:54) =

Original: L that ˜ F i
Revised: L that ˜ F i

Original: =
Revised: =

Original: x
Revised: x

Original: ∗ L,i .
Revised: ∗ L,i .

Original: Finally, the operation (cid:126) can be trivially implemented using additional d weights: we simply connect each output i either to z i (cid:98) d ˜ d (cid:99) if i ≡ 

Revised: Finally, the operation (cid:126) can be trivially implemented using additional d weights: we simply connect each output i either to z i (cid:98) d ˜ d (cid:99) if i ≡

Original: ( mod (cid:98) d ˜ d (cid:99) ) or to ˜ F i otherwise.
Revised: ( mod (cid:98) d ˜ d (cid:99) ) or to ˜ F i otherwise.

Original: Adding up the sizes of the individual components, we see the total number of non-zero weights is O ( md ) , as we wanted.
Revised: Adding up the sizes of the individual components, we see the total number of non-zero weights is O ( md ) , as we wanted.

Original: ~
Revised: ~