I MAGES AS W EIGHT M ATRICES : S EQUENTIAL I MAGE G ENERATION T HROUGH S YNAPTIC L EARNING R ULES
A BSTRACT
Work on fast weight programmers has demonstrated the effectiveness of key/value outer product-based learning rules for sequentially generating a weight matrix (WM) of a neural net (NN) by another NN or itself.
However, the weight generation steps are typically not visually interpretable by humans, because the contents stored in the WM of an NN are not.
Here we apply the same principle to generate natural images.
The resulting fast weight painters (FPAs) learn to execute sequences of delta learning rules to sequentially generate images as sums of outer products of selfinvented keys and values, one rank at a time, as if each image was a WM of an NN.
We train our FPAs in the generative adversarial networks framework, and evaluate on various image datasets.
We show how these generic learning rules can generate images with respectable visual quality without any explicit inductive bias for images.
While the performance largely lags behind the one of specialised state-ofthe-art image generators, our approach allows for visualising how synaptic learning rules iteratively produce complex connection patterns, yielding human-interpretable meaningful images.
Finally, we also show that an additional convolutional U-Net (now popular in diffusion models) at the output of an FPA can learn one-step “denoising” of FPA-generated images to enhance their quality. Our code is public.
I NTRODUCTION
A Fast Weight Programmer (Schmidhuber, 1991a; 1992) is a neural network (NN) that can learn to continually generate and rapidly modify the weight matrix (i e., the program) of another NN in response to a stream of observations to solve the task at hand (reviewed in Sec. 2.1).
At the heart of the weight generation process lies an expressive yet scalable parameterisation of update rules (or learning rules, or programming instructions) that iteratively modify the weight matrix to obtain any arbitrary weight patterns/programs suitable for solving the given task.
Several recent works (Schlag et al., 2021a; Irie et al., 2021; 2022c;b) have demonstrated outer products with the delta rule (Widrow & Hoff, 1960;
Schlag et al., 2021b) as an effective mechanism for weight generation.
In particular, this has been shown to outperform the purely additive Hebbian update rule (Hebb, 1949) used in the Linear Transformers (Katharopoulos et al., 2020; Choromanski et al., 2021) in various settings including language modelling (Schlag et al., 2021a), time series prediction (Irie et al., 2022b), and reinforcement learning for playing video games (Irie et al., 2021).
However, despite its intuitive equations—treating the fast weight matrix as a key/value associative memory—, the effective “actions” of these learning rules on the “contents” stored in the weight matrix still remain opaque, because in general the values stored in a weight matrix are not easily interpretable by humans.
Now what if we let a fast weight programmer generate a “weight matrix” that corresponds to some human-interpretable data?
While outer product-based pattern generation may have a good inductive bias for generating a weight matrix of a linear layer 2 , it can also be seen as a generic mechanism for iteratively generating any high dimensional data.
So let us apply the same principle to generate and incrementally refine natural images.
We treat a colour image as three weight matrices representing synaptic connection weights of a fictive NN, and generate them iteratively through sequences of delta learning rules whose key/value patterns and learning rates are produced by an actual NN that we train.
The resulting Fast Weight Painters (FPAs) learn to sequentially generate images, as sums of outer products, one rank at a time, through sequential applications of delta learning rules.
Intuitively, the delta rule allows a painter to look into the currently generated image in a computationally efficient way, and to apply a change to the image at each painting step.
We empirically observe that the delta rules largely improve the quality of the generated images compared to the purely additive outer product rules.
We train our FPAs in the framework of Generative Adversarial Networks (GAN; Goodfellow et al.
(2014); Niemitalo (2010); Schmidhuber (1990); reviewed in Sec. 2.2).
We evaluate our model on six standard image generation datasets (CelebA, LSUN-Church, Metfaces, AFHQ-Cat/Dog/Wild; all at the resolution of 64x64), and report both qualitative image quality as well as the commonly used
Fr´echet Inception Distance (FID) evaluation metric (Heusel et al., 2017).
Performance is compared to the one of the state-of-the-art StyleGAN2 (Karras et al. , 2020b;a) and the speed-optimised “lightweight” GAN (LightGAN; Liu et al.
While the performance still largely lags behind the one of StyleGAN2, we show that our generic models can generate images of respectable visual quality without any explicit inductive bias for image processing (e.g., no convolution is used in the generator).
This confirms and illustrates that generic learning rules can effectively produce complex weight patterns that, in our case, yield natural images in various domains.
Importantly, we can visualise each step of such weight generation in the human-interpretable image domain.
This is a unique feature of our work since learning rules are typically not visually meaningful to humans in the standard weight generation scenario—see the example shown in Figure 1 (weight generation for few-shot image classification).
Clearly, our goal is not to achieve the best possible image generator (for that, much better convolutional architectures exist).
Instead, we use natural images to visually illustrate the behaviour of an NN that learns to execute sequences of learning rules.
Nevertheless, it is also interesting to see how a convolutional NN can further improve the quality of FPA-generated images.
For this purpose, we conduct an additional study where we add to the FPA’s output, a now popular convolutional
U-Net (Ronneberger et al., 2015; Salimans et al., 2017) used as the standard architecture (Ho et al., 2020; Song et al., 2021; Dhariwal & Nichol, 2021) for denoising diffusion models (Sohl-Dickstein et al., 2015).
The image-to-image transforming U-Net learns (in this case) one-step “denoising” of
FPA-generated images and effectively improves their quality.
corresponding step.
B ACKGROUND
Here we review the two background concepts that are essential to describe our approach in Sec. 3:
Fast Weight Programmers (Sec. 2.1) and Generative Adversarial Networks (Sec. 2.2).
2. F AST W EIGHT P ROGRAMMERS (FWP S )
A Fast Weight Programmer (FWP; Schmidhuber (1991a; 1992)) is a general-purpose auto-regressive sequence-processing NN originally proposed as an alternative to the standard recurrent NN (RNN).
The system consists of two networks: the slow net, or the programmer 3 that learns (typically by gradient descent) to generate and rapidly modify a weight matrix (i.e., the program) of another net, the fast net, based on the input available at each time step.
The context-sensitive fast weight matrix serves as short-term memory of this sequence processor.
This concept has seen a recent revival due to its formal connection (Katharopoulos et al., 2020; Schlag et al., 2021a) to Transformers (Vaswani et al., 2017).
In fact, Transformers with linearised attention (Katharopoulos et al., 2020) have a dual form (Aizerman et al., 1964; Irie et al., 2022a) that is an FWP with an outer product-based weight generation mechanism (Schmidhuber, 1991a; 1992).
Let t , d key , d in , and d out denote positive integers.
A notable example of such a model is the DeltaNet (Schlag et al., 2021a)
that, at each time step t , transforms an input x t ∈ R d in into an output y t ∈ R d out while updating the fast weight matrix W [Equation] key (starting from W 0 = 0 ) as follows:
[Equation]
3) where the slow net (Eq. 1; with a learnable weight matrix W slow ∈ R (2 ∗ d key + d out +1) × d in ) generates query q t ∈ R d key , key k t ∈ R d key , value v t ∈ R d out vectors as well as a scalar β t ∈ R (to which we apply the sigmoid function σ ).
ϕ denotes an element-wise activation function whose output elements are positive and sum up to one (we use softmax) that is crucial for stability (Schlag et al., 2021a).
Eq.
2 corresponds to the rank-one update of the fast weight matrix, from W t − 1 to W t , through the delta learning rule (Widrow & Hoff, 1960) where the slow net-generated patterns, v t , ϕ ( k t ) , and σ ( β t ) , play the role of target , input , and learning rate of the learning rule respectively.
We note that if we replace this Eq. 2 with a purely additive Hebbian learning rule and set the learning rate to 1, i.e., W t
=
W t − 1 + v t ⊗ ϕ ( k t ) , we fall back to the standard Linear Transformer (Katharopoulos et al., 2020).
In fact, some previous works consider other learning rules in the context of linear
Transformers/FWP, e.g., a gated update rule (Peng et al., 2021) and the Oja learning rule (Oja, 1982;
Irie et al., 2022b).
Our main focus is on the delta rule above, but we’ll also support this choice with an ablation study comparing it to the purely additive rule.
An example evolution of the fast weight matrix following delta rules of Eq.
2 is shown in Figure 1.
As stated above, the visualisation itself does not provide useful information.
The core idea of this work is to apply Eq.
2 to image generation, and conduct similar visualisation.
2. G ENERATIVE A DVERSARIAL N ETWORKS
The framework of Generative Adversarial Networks (GANs; Goodfellow et al.
(2014); Niemitalo (2010)) trains two neural networks G (as in generator) and D (as in discriminator), given some dataset R , by making them compete against each other.
The GAN framework itself is a special instantiation (Schmidhuber, 2020) of the more general min-max game concept of Adversarial Artificial
Curiosity (Schmidhuber, 1990; 1991b), and has applications across various modalities, e.g., to speech (Binkowski et al., 2020) and with limited success also to text (de Masson d’Autume et al., 2019), but here we focus on the standard image generation setting (Goodfellow et al., 2014).
In what follows, let c , h , w , and d denote positive integers.
Given an input vector z ∈ R d whose elements are randomly sampled from the zero-mean unit-variance Gaussian distribution N (0 , 1) , G generates an image-like data G ( z ) ∈ R c × h × w with a height of h , a width of w , and c channels (typically c = 3 ; a non-transparent colour image).
D takes an image-like input X ∈ R c × h × w and outputs a scalar
D ( X ) ∈ R between 0 and 1.
The input X is either a sample from the real image dataset R or it is a “fake” sample generated by G .
The training objective of D is to correctly classify these inputs as real (label ‘1’) or fake (label ‘0’), while G is trained to fool D , i.e., the error of D is the gain of G .
The two networks are trained simultaneously with alternating parameter updates.
The goal of this process is to obtain a G capable of generating “fake” images indistinguishable from the real ones.
In practice, there are several ways of specifying the exact form of the objective function.
We refer to the corresponding description (Sec. 3.3) and the experimental section for further details.
Over the past decade, many papers have improved various aspects of GANs including the architecture (e.g., Karras et al.
(2019; 2021)), loss function (e.g., Lim & Ye (2017)), data augmentation strategies (e.g., Zhao et al.
Karras et al.
(2020a), and even the evaluation metric (e.g., Heusel et al.
(2017)) to achieve higher quality images at higher resolutions.
In this work, our baselines are the state-of-the-art StyleGAN
(Karras et al., 2020b;a) and a speed-optimised “light-weight” GAN (LightGAN; Liu et al.
F AST W EIGHT “P AINTERS ” (FPA S )
A Fast Weight Painter (FPA) is a generative model of images based on the weight generation process of outer product-based Fast Weight Programmers (FWP; Sec. 2.1; Eq. 2).
Conceptually, such a model can be trained as a generator in the GAN framework (Sec. 2.2) or as a decoder in a Variational
Auto-Encoder (VAE; Kingma & Welling (2014)).
Here, we train it in the former setting that offers a rich set of accessible baselines.
In the orginal FWPs, the slow net or the programmer’s goal is to generate useful programs/weights for the fast net to solve a given problem.
In the proposed FPAs under the GAN framework, the “slow net” or the painter ’s objective is to generate images that maximise the “fast net”/discriminator/critic’s prediction error.
Here we describe the main architecture of our FPA (Sec. 3.1), its extension through the U-Net (Sec. 3.2), and other GAN specifications (Sec. 3.3).
3. M AIN A RCHITECTURE
Like a typical generator in the GAN framework, an FPA is an NN that transforms a randomly-sampled latent noise vector into an image.
Its general idea is to use a sequence processing NN to decode the input vector for a fixed number of steps, where in each step, we generate a key/value vector pair and a learning rate that are used in a synaptic learning rule to generate a rank-one update to the currently generated image (starting from 0).
The final image thus corresponds to the sum of these update terms.
The number of decoding (or painting ) steps is a hyper-parameter of the model or the training setup.
In what follows, let T , c , d key , d value , d latent , d in , d hidden , denote positive integers.
Here we first provide an abstract overview of the building blocks of the FPA, followed by specific descriptions of each block.
Given a random input vector z ∈ R d latent , and a number of painting steps T , an FPA generates an image W ∈ R c × d value × d key with c channels, a height of d value , and a width of d key , through the following sequence of operations:
[Equation]
where for t ∈ { 1, ..., T } , x t ∈ R d in , h t ∈ R d hidden , and W t ∈ R c
× d value × d key with W 0 = 0 .
InputGenerator , SequenceProcessor , and UpdateNet denote the abstract blocks of operations described as follows:
Input Generator.
As its name indicates, the role of InputGenerator is to transform the latent vector z ∈ R d latent to a sequence of input vectors ( x 1 , ..., x T ) with x t ∈ R d in for t ∈ { 1, ..., T } , for the subsequent sequence processing NN, SequenceProcessor .
In practice, we consider two variants.
[Figure]
In the first variant (v1), the input generator does not do anything: it feeds the same latent vector to
SequenceProcessor at every step, i.e., x t = z for all t with d in = d latent .
In the second variant (v2),
InputGenerator is an NN that maps from an input vector of dimension d latent to an output vector of size T ∗ d in .
The latter is split into T vectors x t ( t ∈ { 1 , ..., T } ) of size d in each.
An obvious advantage of the first approach is that it scales independently of T , and it may also be rolled out for an arbitrary number of steps.
In most datasets, however, we found the second variant to perform better, yielding more stable GAN training.
Sequence Processor.
SequenceProcessor further processes the input vector sequence produced by
InputGenerator .
In our preliminary experiments, we found that auto-regressive processing and in particular RNNs are a good choice for this component (e.g., we did not manage to train any models successfully when the standard Transformer was used instead).
We use a multi-layer long short-term memory (LSTM; Hochreiter & Schmidhuber (1997))
RNN in all of our models.
We also allocate a separate NN that maps the latent vector z to the initial hidden states of each RNN layer (omitted for clarity in Figure 2).
Image Update Network.
The actual painting finally emerges in the image update network,
UpdateNet .
The role of UpdateNet is to apply modifications to the current image W t − 1 (starting from W 0 = 0 ) and obtain a new image W t , through an application of the delta learning rule.
At each time step t , the input h t ∈ R d hidden is projected to key k t ∈ R c ∗ d key , value v t ∈ R c ∗ d value , and learning rate β t ∈ R c vectors by using a learnable weight matrix W
[Equation]
Similarly to multi-head attention in Transformers (Vaswani et al., 2017), the generated vectors are split into c sub-vectors, one for each channel, i.e., for each painting step t ∈ { 1, ..., T } , we have [ k 1 t , ..., k ct ] =
k t for keys, [ v 1 t , ..., v ct ] = v t for values, and [ β 1 t , ..., β ct ] =
β t for learning rates, where k it ∈ R d key , v it ∈ R d value , and β it
∈ R for i ∈ { 1, ..., c } denoting the channel index.
Finally, for each step t ∈ { 1, ..., T } and for each channel index i ∈ { 1, ..., c } , the corresponding image channel W it ∈ R d value × d key is updated through the delta update rule (same as in Eq. 2):
[Equation]
Here each channel of the image is effectively treated as a “weight matrix,” and “painting” is based on synaptic learning rules.
The output image is finally obtained from the final step t =
T as W =
[ W 1 T , ..., W cT
] ∈ R c
× d value × d key (to which we apply tanh ).
We’ll visualise the iterative generation process of Eq.
9 in the experimental section.
An overview of this architecture is depicted in Figure 2.
3. O PTIONAL F INAL U-N ET R EFINEMENT S TEP
Experimentally, we find the process described above alone can generate reasonably fine looking images without any explicit inductive bias for images.
However, the resulting evaluation scores are worse than those of the baseline methods based on convolutional architectures.
This motivates us to further investigate how the quality of images generated by an FPA can be improved by an additional convolutional component.
For this purpose, we use the image-to-image transforming U-Net architecture (Ronneberger et al., 2015; Salimans et al., 2017), the core architecture (Ho et al., 2020; Song et al., 2021; Dhariwal & Nichol, 2021) of the now popular denoising diffusion models (SohlDickstein et al., 2015).
We apply the U-Net to the output of the FPA, i.e., after Eq.
7, it transforms
the image W = W T ∈ R c × d value × d key into another image of the same size W ′ ∈ R c × d [Figure]
[Equation]
[Figure]
key : [Equation]
From the U-Net’s perspective, the output of the FPA is a “noisy” image.
Its operation can be viewed as a one-step “denoising.”
This process is depicted in Figure 3.
In practice, we append this U-Net to the output of a pre-trained FPA, and train the resulting model in the GAN framework, while only updating parameters of the U-Net.
We also discuss end-to-end training in Appendix C.1.
However, in the main setting, all our U-Net models are trained with a pre-trained frozen FPA.
In the experimental section, we’ll show that such a U-Net can effectively improve the quality of FPA-generated images. [Figure]
3. D ISCRIMINATOR A RCHITECTURE AND O THER S PECIFICATIONS
We adopt the training configurations and the discriminator architecture of the LightGAN (Liu et al., 2021), i.e., we replace its generator by our FPA.
In our preliminary study, we also tried the StyleGAN setting, however, we did not observe any obvious benefit.
In the LightGAN framework, in addition to the main objective function of the hinge loss (Lim & Ye, 2017; Tran et al., 2017), the discriminator is regularised with an auxiliary image reconstruction loss at two different resolutions (for that two “simple” image decoders are added to the discriminator).
For details, we refer to Appendix B.3.
E XPERIMENTS 4. B ENCHMARKING ON THE S TANDARD D ATASETS
We start with evaluating how well the proposed FPA performs as an image generator without explicit inductive bias for images.
For this, we train our model on six datasets: CelebA (Liu et al., 2015),
LSUN Church (Yu et al., 2015), Animal Faces HQ (AFHQ) Cat/Dog/Wild (Choi et al., 2020), and
MetFaces (Karras et al., 2020a), all at the resolution of 64x64.
No data augmentation is used as we want to keep the comparison as simple as possible.
For details of the datasets, we refer to Appendix
B.1.
As a first set of experiments, we set the number of generation steps for FPAs to T =(we’ll vary this number in Sec. 4.3); such that the output images can be of full rank.
We provide all hyper-parameters in Appendix B.2 and discuss training/generation speed in Appendix C.2.
Following the standard practice, we compute the FID using 50 K sampled images and all real images (for further discussion on the FID computation, we refer to Appendix B.4).
Table 1 shows the FID scores.
We first observe that the state-of-the-art StyleGAN2 outperforms our FPA by a large margin.
The performance gap is the smallest in the case of the small dataset, MetFaces, while for larger datasets,
CelebA and LSUN-Church, the gap is large.
At the same time, the reasonable FID values show that the FPA is quite successful.
Qualitatively, we observe that the FPA can produce images with a respectable quality, even though we also observe that the StyleGAN2 tends to generate fine-looking
[Table]
images more consistently, and with a higher diversity.
Figure 4 displays the curated output images generated by various models.
By looking at these examples, it is hard to guess that they are generated as sums of outer products through sequences of synaptic learning rules.
The visualisation confirms the FPA’s respectable performance.
In Table 1, we also observe that the extra convolutional U-Net (Sec. 3.2) largely improves the quality of the images generated by the FPA.
Its performance is comparable to that of the LightGAN baseline across all datasets.
For further discussion of the UNet’s effect, we refer to Appendix C.1. [Figure]
4. V ISUALISING THE I TERATIVE G ENERATION P ROCESS
How are images like those shown in Figure 4 generated iteratively as sums of outer products by the
FPA?
Here we visualise this iterative generation process.
Generally speaking, we found almost all examples interesting to visualise.
We show a first set of examples in Figure 5.
As we noted above, all FPAs in Table 1/Figure 4 use T = 64 painting steps.
The first thing we observe is that for many steps, the “key” is almost one-hot (which is encouraged by the softmax), i.e., the part of the image is generated almost column-wise.
For other parts, such as generation/refinement of the background (e.g., steps 57-64 in AFHQ-Wild or steps 1-16 in MetFaces), rank-1 update covers a large region of the image.
Generally we can recognise the “intended action” of each learning rule step on the image (e.g., adding the extra colour in the background in steps 59-64 in LSUN Church, or drawing a draft of the animal face in steps 7-34 in AFHQ-Wild).
We discuss many more examples in Appendix A. [Figure]
{+MetFaces 16   
+}
4. A BLATION S TUDIES
[Table]
Varying Number of Painting Steps.
In all examples above, we train FPAs with a number of painting steps T =such that the 64x64 output can be of full rank.
Here we study FPAs with reduced numbers of steps.
The task should remain feasible, since natural images typically remain fine-looking (at least to human eyes) under low-rank approximations (Andrews & Patterson, 1976).
We select the model configuration that achieves the best FID with T = 64 , and train the same model with fewer steps { 8, 16, 32 } .
Table 2 shows the FIDs.
Globally, we find that fewer steps tend to greatly hurt performance.
An exception is MetFaces (and AFHQ-Cat to some extent) where the degradation is much smaller.
Figure 6 shows examples of 16-step generation of 64x64-resolution images for these two datasets, where we observe that FPAs find and exploit symmetries (for AFHQ-Cat) and other regularities (for MetFaces) as shortcuts for low rank/complexity image generation.
Choice of Learning Rules.
As mentioned in Sec. 2.1, the delta rule is not the only way of parameterising the weight update (Eq. 9).
However, we experimentally observe that both the purely additive rule and the Oja rule underperform the delta rule.
This is in line with previous works on FWPs (see introduction).
The best FID we obtain on the CelebA using the purely additive rules is above 80, much worse than 18.3 obtained by the delta rule (Table 1).
With the Oja rule, we did not manage to train any reasonable model on CelebA in our settings. [Figure]
D ISCUSSION
Limitations.
We have shown how the concept of FWPs can be applied to image generation, to visualise sequences of NN-controlled learning rules that produce natural images.
While this contributes to the study of learning rules in general, we note that our visualisations are tailored to the image generation task: mapping a random vector to a sequence of images, where only the final image (their sum) is “evaluated.”
In many FWP use cases, however, the generated WMs are queried/used at every time step to solve some sequence processing task.
What the learning rules actually do in such scenarios remains opaque.
Also, from the perspective of image generation methods, there are many aspects that we do not investigate here.
For example, we directly use the convolutional LightGAN discriminator.
However, since the FPA is non-convolutional, there may be alternative architectures with better feedback/gradient properties for training FPAs.
Also, our experiments are limited to images with a resolution of 64x64.
Increasing the resolution is typically not straightforward (Karras et al., 2018) but outside the scope of this work.
Diffusion models.
We use the GAN framework (Sec. 2.2) to train our FPA, and mention the alternative of using VAEs.
At first glance, it also seems attractive to use rank-1 noise as an efficient alternative to the expensive U-Net of diffusion models (Sohl-Dickstein et al., 2015).
Unfortunately, unlike in the standard Gaussian case (Feller, 1949), if the forward process is based on rank-1 noises (e.g., obtained as outer products of two random Gaussian noise vectors), we have no guarantee that the reversal process can be parameterised in the same way using an outer product.
Nevertheless, generally speaking, the exchange of ideas between the fields of image and NN weight generation/processing may stimulate both research domains.
An example is the use of discrete cosine transform (DCT) to parameterise a
WM of an NN (e.g., Koutn´ık et al.
(2010a;b); van Steenkiste et al.
; Irie & Schmidhuber (2021)).
Other Perspectives.
We saw that the same generic computational mechanism can be used for both fast weight programming and image generation (painting).
From a cognitive science perspective, it may be interesting to compare painting and programming as sequential processes.
C ONCLUSION
We apply the NN weight generation principles of Fast Weight Programmers (FWPs) to the task of image generation.
The resulting Fast Weight Painters (FPAs) effectively learn to generate weight matrices looking like natural images, through the execution of sequences of NN-controlled learning rules applied to self-invented learning patterns.
This allows us to visualise the iterative FWP process in six different image domains interpretable by humans.
While this method is certainly not the best approach for image generation, our results clearly demonstrate/illustrate/visualise how an NN can learn to control sequences of synaptic weight learning rules in a goal-directed way to generate complex and meaningful weight patterns. A CKNOWLEDGEMENTS We thank R´obert Csord´as for helpful comments on Figure 2. This research was partially funded by
ERC Advanced grant no: 742870, project AlgoRNN and by Swiss National Science Foundation grant no: 200021 192356, project NEUSYM.
We are thankful for hardware donations from NVIDIA and
IBM. The resources used for this work were partially provided by Swiss National Supercomputing Centre (CSCS) project s1145 and s1154. A M ORE V ISUALISATION E XAMPLES
As a continuation of Sec.
4.2, we provide further visualisations to illustrate the image generation process of FPAs.
Figure 7 shows some interesting effects we observe on CelebA.
In Example 1 (top), from step 1 to 16, the image is column-wise generated from the left.
Then from step 18, the generation of the right part of the face starts from the right.
Around step 40 and 46, we see that the separately generated left and right parts of the face do not fit each other.
But then the FPA fixes this in three steps from stepto 48 that “harmonise” the image.
Example 2 (bottom) shows another similar example.
Figure 8 illustrates another typical behaviour we observe on both CelebA and AFHQ-Dog.
In the
CelebA example, we can see that the face is already generated around step 48.
The rest of the steps are used to add extra hair.
Similarly in the AFHQ-Dog example, step-48 image is already a fine-looking dog.
An extra body is painted (rather unsuccessfully) from the right in the remaining steps. [Figure] [Figure] of Figure 5.
[Figure]
[Figure] [Figure] [Figure] B E XPERIMENTAL D ETAILS B. D ATASETS
We use six standard benchmark datasets for image generation: CelebA
(Liu et al., 2015), LSUN
Church (Yu et al., 2015), Animal Faces HQ (AFHQ) Cat/Dog/Wild (Choi et al., 2020), and MetFaces (Karras et al., 2020a).
The number of images are 203 K for CelebA, 126 K for LSUN Church, 5 K for each subsets of AFHQ, and 1336 for the smallest MetFaces.
For further information about the datasets, we refer to the original papers.
B. H YPER -P ARAMETERS
We conduct FPA tuning in terms of both hyper-parameters and model variations.
Regarding the latter, we use three specifications: InputGenerator version v1 vs. v2 (see their descriptions in Sec. 3.1), input generator with or without tanh , and with or without a linear layer that maps the latent vector to the RNN initial states (otherwise the initial states are zeros).
We also train models with/without applying tanh to the generated image at the FPA output.
Using the notations introduced in Sec.
3.1, the hyper-parameter search is conducted across the following ranges: d latent ∈ { 128 , 256 , 512 } , d hidden ∈ { 1024 , 2048 , 4096 } , and the number of RNN layers L ∈ { 1 , 2 } .
When InputGenerator is v2, we have d in ∈ { 8 , 16 , 32 } and furthermore, we add an extra linear layer between the input generator and the RNN with a dimension d ′ in ∈ { 128 , 256 , 512 } .
The batch size and learning rate are fixed to 20 and 2 e − 4 respectively.
We compute the FID score every 5 K training steps to monitor the performance.
We take the hyper-parameters achieving the best FID as the best configuration.
Tablesummarises the corresponding results.
For MetFaces, a more parameter-efficient v2 variant (with 20 M parameters) achieves a FID of 40.5 that is close to 36.3 (Table 1) achieved by the best model.
However, as the parameter count does not matter for our study, we strictly take the best model.
[Table]
B. GAN D ETAILS & I MPLEMENTATION
As stated in Sec. 3.3, our GAN setting is based on that of the LightGAN by Liu et al.
For architectural details of the discriminator, we refer to the original paper.
To be more specific, our code is based on the unofficial public LightGAN implementation https://github.com/ lucidrains/lightweight-gan .
which is slightly different from the official implementation.
In particular, the auxiliary reconstruction losses for the discriminator require 8x8 and 16x16 resolution images.
In the original implementation, these are obtained directly from the intermediate layers of the generator, while in this unofficial implementation, they are obtained by scaling down the final output image of the generator by interpolation.
The latter is the only compatible approach for FPAs since an FPA does not produce such intermediate images with small resolutions unlike the standard convolution based generators.
For any further details, we refer to our public code.
For the StyleGAN  baseline, we use the official implementation of StyleGAN3 (Karras et al., 2021) that also supports StyleGAN2: https://github.com/NVlabs/stylegan3 .
B. FID COMPUTATION
While the Fr´echet Inception Distance (FID; Heusel et al.
(2017)) is a widely used metric for evaluating machine-generated images, it is sensitive to many details (Parmar et al., 2022).
We also observe that it is crucial to use the same setting for all models consistently.
We use the following procedure for all datasets and models (including the StyleGAN2 baseline).
We store both the generated and resized real images in JPEG format, and use the pytorch-fid implementation of https: //github.com/mseitzer/pytorch-fid to compute the FID.
While the usage of PNG is generally recommended by Parmar et al.
(2022), since consistency is all we need for the purpose of our study, and JPEG is more convenient for frequently monitoring FID scores during training (as JPEG images are faster to store, and take less disk space), we opt for it here. Nevertheless, in Table 4, we also report FID scores computed using the generated and resized real images saved in PNG format.
These are computed using the models from Table 1, i.e., the best model checkpoint found based on JPEG-based FID scores during training. All numbers grow beyond those in Table 1 including those of the baselines, but the general trend does not change. The FID scores reported in all other tables are computed using the JPEG format. [Table]
C E XTRA R ESULTS & D ISCUSSION C. D ISCUSSION OF THE U-N ET EXTENSION
Visualising the U-Net’s Effect. In Sec. 4.1, we report the U-Net’s quality improvements of FPAgenerated images in terms of the FID metric.
Here we further look into its effect on the actual images.
Figure 9 displays some example images generated by FPA/U-Net models before and after the U-Net application.
We observe that in general, the U-Net is successful at improving the quality of the FPA-generated images without completely ignoring the original images.
However, we also observe examples where this is not the case, i.e., the U-Net generates almost completely new images as illustrated by the last row of MetFaces and AFHQ Dog examples.
End-to-End Training.
As described in Sec. 3.2, we train the FPA/U-Net model using pre-trained FPA parameters.
We also tried end-to-end training from scratch, but observed that the U-Net starts learning before the FPA can generate fine-looking images.
As a result, the resulting U-Net ignores the FPA output, hence the FPA does not receive any useful gradients for learning to generate meaningful images, and the FPA remains a noise generator.
We also tried to train it by providing both the FPA and U-Net outputs to the discriminator (and by stopping the gradient flow from the U-Net to the FPA).
This alleviates the problem of the FPA remaining a noise generator, but is still not good enough to make the U-Net properly act as a denoising/refinement component of the FPA. Choice of the Pre-Trained FPA. In general, we take the best performing standalone FPA as the pre-trained model to train a FPA/U-Net model (Sec. 3.2).
In some cases, however, we find that the best standalone FPA model does not yield the best FPA/U-Net model. This is the case for AFHQ-Cat and Dog as illustrated in Table 5. The FPA/U-Net model based on the best pre-trained FPA with an input generator of type v1 (Sec. 3.1) yields a slightly better FID than the one based on the best v variant, even though the trend is the opposite for the standalone FPA models.
In Table 1 of the main text, we report the overall best FIDs. Our U-Net implementation is based on the diffusion model implementation of https://github. com/lucidrains/denoising-diffusion-pytorch .
We use three down/up-sampling
ResNet blocks with a total of 8 M parameters.
[Figure]
[Table]
C. T RAINING AND G ENERATION S PEED
While it is difficult to compare speed across different implementations, the generation speed of
StyleGAN2 and the FPA are similar: about 55 images are generated every second for a batch size of one on a V100 GPU, while the LightGAN can generate 160 images per second in the same setting.
The training/convergence speed of the FPA is similar to or better than that of the LightGAN on the relatively large datasets.
For example, our FPA converges after about 150 K training steps on CelebA (vs. 285 K steps for the LightGAN), and 70 K steps on LSUN Church (similarly to the LightGAN) with a batch size of 20.
The FPA/U-Net variant converges more slowly.
For example on LSUN, it continues improving until 380 K steps in the same setting.
On the small datasets, e.g., on AFHQ-Cat, the LightGAN converges faster (about 30 K steps) than the FPA (which continues improving until about 280 K steps) in the same setting.
Any training run can be completed within one to three days on a single V100 GPU.