P RINCIPLED W EIGHT I NITIALIZATION FOR H YPERNETWORKS
Anonymous authors
Paper under double-blind review
A BSTRACT
Hypernetworks are meta neural networks that generate weights for a main neural network in an end-to-end differentiable manner.
Despite extensive applications ranging from multi-task learning to Bayesian deep learning, the problem of optimizing hypernetworks has not been studied to date.
We observe that classical weight initialization methods like Glorot & Bengio (2010) and He et al.
(2015), when applied directly on a hypernet, fail to produce weights for the mainnet in the correct scale.
We develop principled techniques for weight initialization in hypernets, and show that they lead to more stable mainnet weights, lower training loss, and faster convergence.
I NTRODUCTION
Meta-learning describes a broad family of techniques in machine learning that deals with the problem of learning to learn.
An emerging branch of meta-learning involves the use of hypernetworks , which are meta neural networks that generate the weights of a main neural network to solve a given task in an end-to-end differentiable manner.
Hypernetworks were originally introduced by Ha et al.
as a way to induce weight-sharing and achieve model compression by training the same meta network to learn the weights belonging to different layers in the main network.
Since then, hypernetworks have found numerous applications including but not limited to: weight pruning (Liu et al., 2019), neural architecture search (Brock et al., 2017; Zhang et al., 2018), Bayesian neural networks (Krueger et al., 2017; Ukai et al., 2018; Pawlowski et al., 2017; Henning et al., 2018;
Deutsch et al., 2019), multi-task learning (Pan et al., 2018;
Shen et al., 2017; Klocek et al., 2019;
Serr ` a et al., 2019;
Meyerson & Miikkulainen, 2019), continual learning (von Oswald et al., 2019), generative models (Suarez, 2017; Ratzlaff & Fuxin, 2019), ensemble learning (Kristiadi & Fischer, 2019), hyperparameter optimization (Lorraine & Duvenaud, 2018), and adversarial defense (Sun et al., 2017).
Despite the intensiﬁed study of applications of hypernetworks, the problem of optimizing them to this day remains signiﬁcantly understudied.
Given the lack of principled approaches to training hypernetworks, prior work in the area is mostly limited to ad-hoc approaches based on trial and error (c.f. Section 3).
For example, it is common to initialize the weights of a hypernetwork by sampling a “small” random number.
Nonetheless, these ad-hoc methods do lead to successful hypernetwork training primarily due to the use of the Adam optimizer (Kingma & Ba, 2014), which has the desirable property of being invariant to the scale of the gradients.
However, even Adam will not work if the loss diverges (i.e. integer overﬂow) at initialization, which will happen in sufﬁciently big models.
The normalization of badly scaled gradients also results in noisy training dynamics where the loss function suffers from bigger ﬂuctuations during training compared to vanilla stochastic gradient descent (SGD).
Wilson et al.
(2017) showed that while adaptive optimizers like Adam may exhibit lower training error, they fail to generalize as well to the test set as non-adaptive gradient methods.
Moreover, Adam incurs a computational overhead and requires 3X the amount of memory for the gradients compared to vanilla SGD.
Small random number sampling is reminiscent of early neural network research (Rumelhart et al., 1986) before the advent of classical weight initialization methods like Xavier init (Glorot & Bengio, 2010) and Kaiming init (He et al., 2015).
Since then, a big lesson learned by the neural network optimization community is that architecture speciﬁc initialization schemes are important to the robust training of deep networks, as shown recently in the case of residual networks (Zhang et al., 2019).
In fact, weight initialization for hypernetworks was recognized as an outstanding open prob- lem by prior work (Deutsch et al., 2019) that had questioned the suitability of classical initialization methods for hypernetworks.
Our results We show that when classical methods are used to initialize the weights of hypernetworks, they fail to produce mainnet weights in the correct scale, leading to exploding activations and losses.
This is because classical network weights transform one layer’s activations into another, while hypernet weights have the added function of transforming the hypernet’s activations into the mainnet’s weights.
Our solution is to develop principled techniques for weight initialization in hypernetworks based on variance analysis.
The hypernet case poses unique challenges.
For example, in contrast to variance analysis for classical networks, the case for hypernetworks can be asymmetrical between the forward and backward pass.
The asymmetry arises when the gradient ﬂow from the mainnet into the hypernet is affected by the biases, whereas in general, this does not occur for gradient ﬂow in the mainnet.
This underscores again why architecture speciﬁc initialization schemes are essential.
We show both theoretically and experimentally that our methods produce hypernet weights in the correct scale.
Proper initialization mitigates exploding activations and gradients or the need to depend on Adam.
Our experiments reveal that it leads to more stable mainnet weights, lower training loss, and faster convergence.
Section 2 brieﬂy covers the relevant technical preliminaries and Section 3 reviews problems with the ad-hoc methods currently deployed by hypernetwork practitioners.
We derive novel weight initialization formulae for hypernetworks in Section 4, empirically evaluate our proposed methods in Section 5, and ﬁnally conclude in Section 6.
P RELIMINARIES
Deﬁnition.
A hypernetwork is a meta neural network H with its own parameters φ that generates the weights of a main network θ from some embedding e in a differentiable manner: θ =
H φ ( e ) .
Unlike a classical network, in a hypernetwork, the weights of the main network are not model parameters.
Thus the gradients ∆ θ have to be further backpropagated to the weights of the hypernetwork ∆ φ , and then trained via gradient descent φ t +1 =
φ t − λ ∆ φ t .
This fundamental difference suggests that conventional knowledge about neural networks may not apply directly to hypernetworks and novel ways of thinking about weight initialization, optimization dynamics and architecture design for hypernetworks are sorely needed.
2. R ICCI C ALCULUS
We propose the use of Ricci calculus, as opposed to the more commonly used matrix calculus, as a suitable mathematical language for thinking about hypernetworks.
Ricci calculus is useful because it allows us to reason about the derivatives of higher-order tensors with notational ease.
For readers not familiar with the index-based notation of Ricci calculus, please refer to Laue et al.
(2018) for a good introduction to the topic written from a machine learning perspective.
For a general nth-order tensor T i 1 ,...,i n , we use d i k to refer to the dimension of the index set that i k is drawn from.
We include explicit summations where the relevant expressions might be ambiguous, and use Einstein summation convention otherwise.
We use square brackets to denote different layers for added clarity, so for example W [ t ] denotes the t -th weight layer.
2. X AVIER I NITIALIZATION
Glorot & Bengio (2010) derived weight initialization formulae for a feedforward neural network by conducting a variance analysis over activations and gradients.
For a linear layer y i
=
W ij x j
+ b i , suppose we make the following Xavier Assumptions at initialization: (1) The W ij , x j , and b i are all independent of each other.
(2) ∀ i, j : E
[ W ij ] = 0 .
(3) ∀ j : E [ x j ] = 0 .
(4) ∀ i : b i = 0 .
Then, E [ y i ] = 0 and Var ( y i ) =
d j Var ( W ij ) Var ( x j ) .
To keep the variance of the output and input activations the same, i.e. Var ( y i )
=
Var ( x j ) , we have to sample W ij from a distribution whose variance is equal to the reciprocal of the fan-in: Var ( W i j ) = 1 d j .
If analogous assumptions hold for the backward pass, then to keep the variance of the output and input gradients the same, we have to sample W ij from a distribution whose variance is equal to the reciprocal of the fan-out: Var ( W ij )
= 1 d i .
Thus, the forward pass and backward pass result in symmetrical formulae.
Glorot & Bengio (2010) proposed an initialization based on their harmonic mean: Var ( W ij )
= 2 d j + d i .
In general, a feedforward network is non-linear, so these assumptions are strictly invalid.
But symmetric activation functions with unit derivative at 0 results in a roughly linear regime at initialization.
2. K AIMING I NITIALIZATION
He et al.
(2015) extended Glorot & Bengio (2010)’s analysis by looking at the case of ReLU activation functions, i.e. y i
= W ij ReLU ( x j ) + b i .
We can write z j = ReLU ( x j ) to get
Var ( y i ) =
(cid:88)
[Equation]
This results in an extra factor of 2 in the variance formula.
W i j have to be symmetric around 0 to enforce Xavier Assumption 3 as the activations and gradients propagate through the layers.
He et al.
(2015) argued that both the forward or backward version of the formula can be adopted, since the activations or gradients will only be scaled by a depth-independent factor.
For convolutional layers, we have to further divide the variance by the size of the receptive ﬁeld.
‘Xavier init’ and ‘Kaiming init’ are terms that are sometimes used interchangeably.
Where there might be confusion, we will refer to the forward version as fan-in init, the backward version as fan-out init, and the harmonic mean version as harmonic init.
R EVIEW OF C URRENT M ETHODS
In the seminal Ha et al.
(2016) paper, the authors identiﬁed two distinct classes of hypernetworks: dynamic (for recurrent networks) and static (for convolutional networks).
They proposed Orthogonal init (Saxe et al., 2013) for the dynamic class, but omitted discussion of initialization for the static class.
The static class has since proven to be the dominant variant, covering all kinds of non-recurrent networks (not just convolutional), and thus will be the central object of our investigation.
Through an extensive literature and code review, we found that hypernet practitioners mostly depend on the Adam optimizer, which is invariant to and normalizes the scale of gradients, for training and resort to one of four weight initialization methods:
M1 Xavier or Kaiming init (as found in Pawlowski et al.
Serr ` a et al.
von Oswald et al.
; Balazevic et al.
M2 Small random values (as found in Krueger et al.
; Lorraine & Duvenaud (2018)).
M3 Kaiming init, but with the output layer scaled by 110 (as found in Ukai et al.
M4 Kaiming init, but with the hypernet embedding set to be a suitably scaled constant (as found in Meyerson & Miikkulainen (2019)).
M1 uses classical neural network initialization methods to initialize hypernetworks.
This fails to produce weights for the main network in the correct scale.
Consider the following illustrative example of a one-layer linear hypernet generating a linear mainnet with T + 1 layers, given embeddings sampled from a standard normal distribution and weights sampled entry-wise from a zero-mean distribution.
We leave the biases out for now.
[Equation]
In this case, if the variance of the weights in the hypernet Var ( H [ t ] i t +1 i t k t ) is equal to the reciprocal of the fan-in d k t , then the variance of the activations Var ( x [ T + 1] i t +1 )
= (cid:81)
Tt =1 d i t explodes.
If it is equal to the reciprocal of the fan-out d i t +1 , then the activation variance Var ( x [ T + 1] i t +1 )
= (cid:81)
Tt =1 d kt d it +1 is likely to vanish, since the size of the embedding vector is typically small relatively to the width of the mainnet weight layer being generated.
Where the fan-in is of a different scale than the fan-out, the harmonic mean has a scale close to that of the smaller number.
Therefore, the fan-in, fan-out, and harmonic variants of Xavier and Kaiming init will all result in activations and gradients that scale exponentially with the depth of the mainnet.
M2 and M3 introduce additional hyperparameters into the model, and the ad-hoc manner in which they work is reminiscent of pre deep learning neural network research, before the introduction of classical initialization methods like Xavier and Kaiming init.
This ad-hoc manner is not only inelegant and consumes more compute, but will likely fail for deeper and more complex hypernetworks.
M4 proposes to set the embeddings e [ t ] k t to a suitable constant (d − 1 / 2 i t in this case), such that both
W [ t ] i t
+1 i t and H [ t ] i
t +1 i t k t can seem to be initialized with the same variance as Kaiming init.
This ensures that the variance of the activations in the mainnet are preserved through the layers, but the restrictions on the embeddings might not be desirable in many applications.
Luckily, the ﬁx appears simple — set Var ( H [ t ] i t +1 i t k t )
=
1 d it d kt .
This results in the variance of the generated weights in the mainnet Var ( W [ t ] i t
+1 i t )
= 1 d it resembling conventional neural networks initialized with fan-in init.
This suggests a general hypernet weight initialization strategy: initialize the weights of the hypernet such that they approximate weights coming from classical neural network initialization.
We elaborate on and generalize this intuition in Section 4.
H YPERFAN I NITIALIZATION
Most hypernetwork architectures use a linear output layer so that gradients can pass from the mainnet into the hypernet directly without any non-linearities.
We make use of this fact in developing methods called hyperfan-in init and hyperfan-out init for hypernetwork weight initialization based on the principle of variance analysis.
4. H YPERFAN - IN
Proposition.
Suppose a hypernetwork comprises a linear output layer.
Then, the variance between the input and output activations of a linear layer in the mainnet y i
=
W ij x j + b i can be preserved using fan-in init in the hypernetwork with appropriately scaled output layers.
Case 1.
The hypernet generates the weights but not the biases of the mainnet.
The bias in the mainnet is initialized to zero.
We can write the weight generation in the form W ij =
H ijk h ( e ) k + β ij where h computes all but the last layer of the hypernet and ( H, β ) form the output layer.
We make the following Hyperfan Assumptions at initialization: (1) Xavier assumptions hold for all the layers in the hypernet.
The H ijk , h ( e ) k , β ij , x j , and b i are all independent of each other.
(3) ∀ i, j, k : E
[ H ijk ] = 0 .
(4) E [ x j ] = 0 .
(5) ∀ i : b i = 0 .
Use fan-in init to initialize the weights for h .
Then, Var ( h ( e ) k ) =
Var ( e l ) .
If we initialize H with the formula Var ( H ijk ) = 1 d j d k Var ( e l ) and β with zeros, we arrive at Var ( W ij )
= 1 d j , which is the formula for fan-in init in the mainnet.
The Hyperfan assumptions imply the Xavier assumptions hold in the mainnet, thus preserving the input and output activations.
[Equation]
Case 2.
The hypernet generates both the weights and biases of the mainnet.
We can write the weight and bias generation in the form W ij =
H ijk h ( e [1]) k
+ β ij and b i =
G il g ( e [2]) l + γ i respectively, where h and g compute all but the last layer of the hypernet, and ( H, β ) and ( G, γ ) form the output layers.
We modify Hyperfan Assumption 2 so it includes G il , g ( e [2]) l , and γ i , and further assume Var ( x j ) = 1 , which holds at initialization with the common practice of data standardization.
Use fan-in init to initialize the weights for h and g .
Then, Var ( h ( e [1]) k ) =
Var ( e [1] m ) and
Var ( g ( e [2]) l ) =
Var ( e [2] n ) .
If we initialize H with the formula Var ( H ijk ) = 1 2 d j d k Var ( e [1] m ) , G with the formula Var ( G il )
=
1 2 d l Var ( e [2] n ) , and β, γ with zeros, then the input and output activations in the mainnet can be preserved.
[Equation]
If we initialize G ij to zeros, then its contribution to the variance will increase during training, causing exploding activations in the mainnet.
Hence, we prefer to introduce a factor of 1 / 2 to divide the variance between the weight and bias generation, where the variance of each component is allowed to either decrease or increase during training.
This becomes a problem if the variance of the activations in the mainnet deviates too far away from 1 , but we found that it works well in practice.
4. H YPERFAN - OUT
A similar derivation can be done for the backward pass using analogous assumptions on gradients ﬂowing in the mainnet:
[Equation]
through mainnet weights:
[Equation]
and through mainnet biases:
[Equation]
If we initialize the output layer H with the analogous hyperfan-out formula Var ( H [ t ] i t
+1 i t k t )
= 1 d it +1 d kt Var ( e kt ) and the rest of the hypernet with fan-in init, then we can preserve input and output gradients on the mainnet:
Var ( ∂L ∂x [ t ] i t ) =
Var ( ∂L ∂x [ t +1] i t +1 ) .
However, note that the gradients will shrink when ﬂowing from the mainnet to the hypernet:
Var (
[Equation] [Equation]
( e kt )
Var ( ∂L ∂W [ t ] it +1 it )
[Equation]
and scaled by a depth-independent factor due to the use of fan-in rather than fan-out init.
In the classical case, the forward version (fan-in init) and the backward version (fan-out init) are symmetrical.
This remains true for hypernets if they only generated the weights of the mainnet.
However, if they were to also generate the biases, then the symmetry no longer holds, since the biases do not affect the gradient ﬂow in the mainnet but they do so for the hypernet (c.f. Equation 4).
Nevertheless, we can initialize G so that it helps hyperfan-out init preserve activation variance on the forward pass as much as possible (keeping the assumption that Var ( x j ) = 1 as before):
[Equation] [Equation]
We summarize the variance formulae for hyperfan-in and hyperfan-out init in Table 1.
It is not uncommon to re-use the same hypernet to generate different parts of the mainnet, as was originally done in Ha et al.
We discuss this case in more detail in Appendix Section A
[Table]
E XPERIMENTS
We evaluated our proposed methods on four sets of experiments involving different use cases of hypernetworks: feedforward networks, continual learning, convolutional networks and Bayesian neural networks.
In all cases, we optimize with vanilla SGD and sample from the uniform distribution according to the variance formula given by the init method.
More experimental details can be found in Appendix Section B.
5. F EEDFORWARD N ETWORKS ON MNIST
As an illustrative ﬁrst experiment, we train a feedforward network with ﬁve hidden layers (hidden units), a hyperbolic tangent activation function, and a softmax output layer, on MNIST across four different settings: (1) a classical network with Xavier init, (2) a hypernet with Xavier init that generates the weights of the mainnet, (3) a hypernet with hyperfan-in init that generates the weights of the mainnet, (4) and a hypernet with hyperfan-out init that generates the weights of the mainnet.
The use of hyperfan init methods on a hypernetwork reproduces mainnet weights similar to those that have been trained from Xavier init on a classical network, while the use of Xavier init on a hypernetwork causes exploding activations right at the beginning of training (see Figure 1).
Observe in Figure 2 that when the hypernetwork is initialized in the proper scale, the magnitude of generated weights stabilizes quickly.
This in turn leads to a more stable training regime, as seen in Figure 3.
More visualizations of the activations and gradients of both the mainnet and hypernet can be viewed in Appendix Section B.1.
Qualitatively similar observations were made when we replaced the activation function with ReLU and Xavier with Kaiming init, with Kaiming init leading to even bigger activations at initialization.
Suppose now the hypernet generates both the weights and biases of the mainnet instead of just the weights.
We found that this architectural change leads the hyperfan init methods to take more time (but still less than Xavier init), to generate stable mainnet weights (c.f. Figure 25 in the Appendix).
[Figure]
[Figure] [Figure]
5. C ONTINUAL L
[Figure]
Continual learning solves the problem of learning tasks in sequence without forgetting prior tasks.
von Oswald et al.
(2019) used a hypernetwork to learn embeddings for each task as a way to efﬁciently regularize the training process to prevent catastrophic forgetting.
We compare different initialization schemes on their hypernetwork implementation, which generates the weights and biases of a ReLU mainnet with two hidden layers to solve a sequence of three regression tasks.
In Figure 4, we plot the training loss averaged over 15 different runs, with the shaded area showing the standard error.
We observe that the hyperfan methods produce smaller training losses at initialization and during training, eventually converging to a smaller loss for each task.
[Figure]
5. C ONVOLUTIONAL N ETWORKS ON CIFAR-
Ha et al.
(2016) applied a hypernetwork on a convolutional network for image classiﬁcation on
CIFAR-10.
We note that our initialization methods do not handle residual connections, which were in their chosen mainnet architecture and are important topics for future study.
Instead, we implemented their hypernetwork architecture on a mainnet with the All Convolutional Net architecture (Springenberg et al., 2014) that is composed of convolutional layers and ReLU activation functions.
After searching through a dense grid of learning rates
[Figure]
we failed to enable the fan-in version of
Kaiming init to train even with very small learning rates.
The fan-out version managed to begin delayed training, starting from around epoch 270 (see Figure 5).
By contrast, both hyperfan-in and hyperfan-out init led to successful training immediately.
This shows a good init can make it possible to successfully train models that would have otherwise been unamenable to training on a bad init
[Figure]
5. B AYESIAN N EURAL N ETWORKS ON I MAGE N ET
Bayesian neural networks improve model calibration and provide uncertainty estimation, which guard against the pitfalls of overconﬁdent networks.
Ukai et al.
(2018) developed a Bayesian neural network by using a hypernetwork to simulate an expressive prior distribution.
We trained a similar hypernetwork by applying Ukai et al.
(2018)’s methods on ImageNet, but differed in our choice of
MobileNet (Howard et al., 2017) as a mainnet architecture that does not have residual connections.
In the work of Ukai et al.
(2018), it was noticed that even with the use of batch normalization in the mainnet, classical initialization approaches still led to diverging losses (due to exploding activations, c.f. Section 3).
We observe similar results in our experiment (see Figure 6) — the fan-in version of
Kaiming init, which is the default initialization in popular deep learning libraries like PyTorch and
Chainer, resulted in substantially higher initial losses and led to slower training than the hyperfan methods.
We found that this ﬁnding still holds even when the hypernet does not generate the last layer of the mainnet.
This shows that while batch normalization helps, it is not the solution for a bad init that causes exploding activations.
Our approach solves this problem in a principled way, and is preferable to trial-and-error based heuristics that Ukai et al.
(2018) had to resort to in order to train their model.
Surprisingly, the fan-out version of Kaiming init led to similar results as the hyperfan methods, suggesting that batch normalization might be sufﬁcient to correct the bad initializations that result in vanishing activations.
That being said, hypernet practitioners should not expect batch normalization to be the panacea for problems caused by bad initialization, especially in memory-constrained scenarios.
In a Bayesian neural network application (especially in hypernet architectures without relaxed weight-sharing), the blowup in the number of parameters limits the use of big batch sizes, which is essential to the performance of batch normalization (Wu & He, 2018).
For example, in this experiment, our hypernet model requires 32 times as many parameters as a classical MobileNet.
To the best of our knowledge, the interaction between batch normalization and initialization is not well-understood, even in the classical case, and thus, our ﬁndings prompt an interesting direction for future research.
[Figure]
C ONCLUSION
In all our experiments, hyperfan-in and hyperfan-out both lead to successful hypernetwork training with SGD.
We did not ﬁnd a good reason to prefer one over the other (similar to He et al.
(2015)’s observation in the classical case for fan-in and fan-out init).
For a long time, the promise of deep nets to learn rich representations of the world was left unfulﬁlled due to the inability to train these models.
The discovery of greedy layer-wise pre-training (Hinton et al., 2006; Bengio et al., 2007) and later, Xavier and Kaiming init, as weight initialization strategies to enable such training was a pivotal achievement that kickstarted the deep learning revolution.
This underscores the importance of model initialization as a fundamental step in learning complex representations.
In this work, we developed the ﬁrst principled weight initialization methods for hypernetworks, a rapidly growing branch of meta-learning.
We hope our work will spur momentum towards the development of principled techniques for building and training hypernetworks, and eventually lead to signiﬁcant progress in learning meta representations.
Other non-hypernetwork methods of neural network generation (Stanley et al., 2009; Koutnik et al., 2010) would also do well to consider if their generated weights result in exploding activations and how to avoid that if so.
A PPENDIX A R E - USING H YPERNET W EIGHTS A. F OR M AINNET W EIGHTS OF THE S AME S IZE
For model compression or weight-sharing purposes, different parts of the mainnet might be generated by the same hypernet function.
This will cause some assumptions of independence in our analysis to be invalid.
Consider the example of the same hypernet being used to generate multiple different mainnet weight layers of the same size, i.e. H [ t ] i t
+1 i t k
[Equation]
[ t
[Equation]
1] i t
[Equation]
2 i t
[Equation]
1 k
[Equation]
d i t
[Equation] [Equation]
i t
[Equation] [Equation]
i t .
Then, x [ t + 1] i t +
= H [ t ] i
t +1 i t k t e
[ t ] k t x
[ t ] i t (cid:54)⊥⊥ W [
[Equation]
i
[Equation]
2 i
[Equation]
=
[Equation] [Equation]
i
[Equation]
2 i t +1 k e
[Equation]
] k t +1 .
The relaxation of some of these independence assumptions does not always prove to be a big problem in practice, because the correlations introduced by repeated use of H can be minimized with the use of ﬂat distributions like the uniform distribution.
It can even be helpful, since the re-use of the same hypernet for different layers causes the gradient ﬂowing through the hypernet output layer to be the sum of the gradients from the weights of these layers:
[Equation]
[Equation] [Equation]
it
H
[Equation]
+
[Equation]
t k , thus combating the shrinking effect.
A. F OR M AINNET W EIGHTS OF D IFFERENT S IZES
Similar reasoning applies if the same hypernet was used to generate differently sized subsets of weights in the mainnet.
However, we encourage avoiding this kind of hypernet architecture design if not otherwise essential, since it will complicate the initialization formulae listed in Table 1.
Consider Ha et al.
(2016)’s
hypernetwork architecture.
Their two-layer hypernet generated weight chunks of size ( K, n, n ) for a main convolutional network where K = 16 was found to be the highest common factor among the size of mainnet layers, and n 2 = 9 was the size of the receptive ﬁeld.
We simplify the presentation by writing i for i t , j for j t , k for k t,m , and l for l t,m .
[Equation]
Because the output layer ( H, β ) in the hypernet was re-used to generate mainnet weight matrices of different sizes (i.e. in general, i t (cid:54) =
i t +1 , j t (cid:54) =
j t +1 ), G effectively becomes the output layer that we want to be considering for hyperfan-in and hyperfan-out initialization.
Hence, to achieve fan-in in the mainnet Var ( W [ t ] ij ) = 1 d j , we have to use fan-in init for H (i.e.
Var ( H i ( mod K ) k ) = 1 d k (cid:54) = 1 d j d k Var ( e [ t ][ m t ] l ) ), and hyperfan-in init for G (i.e. Var ( G [ t ][ m t ] kl ) =
1 d j d l Var ( e [ t ][ m t ] l ) ).
Analogously, to achieve fan-out in the mainnet Var ( W [ t ] ij ) = 1 d i , we have to use fan-in init
B M ORE E XPERIMENTAL D ETAILS B. F EEDFORWARD N ETWORKS ON MNIST
The networks were trained on MNIST for 30 epochs with batch size 10 using a learning rate of 0 . 0005 for the hypernets and 0 .
01 for the classical network.
The hypernets had one linear layer with embeddings of size 50 and different hidden layers in the mainnet were all generated by the same hypernet output layer with a different embedding, which was randomly sampled from U ( −√ 3 , √ 3) and ﬁxed.
We use the mean cross entropy loss for training, but the summed cross entropy loss for testing.
We show activation and gradient plots for two cases: (i) the hypernet generates only the weights of the mainnet, and (ii) the hypernet generates both the weights and biases of the mainnet.
(i) covers
Figures 3, 1, 7, 8, 9, 10, 11, 12, 2, 13, 14, 15, and 16.
(ii) covers Figures 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, and 29.
The activations and gradients in our plots were calculated by averaging across a ﬁxed held-out set of 300 examples drawn randomly from the test set.
In Figures 1, 8, 9, 11, 12, 13, 14, 16, 18, 20, 21, 23, 24, 26, 27, and 29, the y axis shows the number of activations/gradients, while the x axis shows the value of the activations/gradients.
The value of activations/gradients from the hypernet output layer correspond to the value of mainnet weights.
In Figures 2, 7, 10, 15, 19, 22, 25, and 28, the y axis shows the mean value of the activations/gradients, while each increment on the x axis corresponds to a measurement that was taken every 1000 training batches, with the bars denoting one standard deviation away from the mean.
[Figure] [Figure]
[Figure] [Figure]
[Figure] [Figure]
N u m b e r
o f A c t i v a t i o n s
[Figure]
[Figure]
u m b e r o f
A c t i v a t i o n s
[Figure] [Figure]
[Figure] [Figure]
B.1.
H YPERNET G ENERATES B OTH M AINNET W EIGHTS AND B IASES
[Figure] [Figure]
[Figure]
e a n A c t i v a t i o n V a l u e
[Figure]
[Figure] [Figure]
[Figure]
d i
[Figure]
n t V a
[Figure] [Figure]
[Figure] [Figure]
Output Layer Activations during Training on MNIST
[Figure]
[Figure] [Figure]
[Figure]
B.1. R EMARK ON THE C OMBINATION OF F AN - IN AND F AN - OUT I NIT
Glorot & Bengio (2010) proposed to use the harmonic mean of the two different initialization formulae derived from the forward and backward pass.
He et al.
(2015) commented that either version sufﬁces for convergence, and that it does not really matter given that the difference between the two will be a depth-independent factor.
We experimented with the harmonic, geometric, and arithmetic means of the two different formulae in both the classical and the hypernet case.
There was no indication of any signiﬁcant beneﬁt from taking any of the three different means in both cases.
Thus, we conﬁrm and concur with He et al.
(2015)’s original observation that either the fan-in or the fan-out version sufﬁces.
B. C ONTINUAL L EARNING
The mainnet is a feedforward network with two hidden layers ( 10 hidden units) and the ReLU activation function.
The weights and biases of the mainnet are generated from a hypernet with two hidden layers ( 10 hidden units) and trainable embeddings of size 2 sampled from U ( −√ 3 , √ 3) .
We keep the same continual learning hyperparameter β output value of 0 .
005 and pick the best learning rate for each initialization method from { 10 − 2 , 10 − 3 , 10 − 4 , 10 − 5 } .
Notably, Kaiming (fan-in) could only be trained from learning rate 10 − 5 , with losses diverging soon after initialization using the other learning rates.
Each task was trained for 6000 training iterations using batch size 32 , with Figureplotted from losses measured at every 100 iterations.
B. C ONVOLUTIONAL N ETWORKS ON CIFAR-
The networks were trained on CIFAR-10 for 500 epochs starting with an initial learning rate of 0 .
using batch size 100 , and decaying with γ = 0 .
1 at epochs 350 and 450 .
The hypernet is composed of two layers ( 50 hidden units) with separate embeddings and separate input layers but shared output layers.
The weight generation happens in blocks of (96 , 3 , 3) where K = 96 is the highest common factor between the different sizes of the convolutional layers in the mainnet and n = 3 is the size of the convolutional ﬁlters (see Appendix Section A.2 for a more detailed explanation on the hypernet architecture).
The embeddings are size 50 and ﬁxed after random sampling from
U ( −√ 3 , √ 3) .
We use the mean cross entropy loss for training, but the summed cross entropy loss for testing.
B. B AYESIAN N EURAL N ETWORK ON I MAGE N ET
Ukai et al.
(2018) showed that a Bayesian neural network can be developed by using a hypernetwork to express a prior distribution without substantial changes to the vanilla hypernetwork setting.
Their methods simply require putting L 2 -regularization on the model parameters and sampling from stochastic embeddings.
We trained a linear hypernet to generate the weights of a MobileNet mainnet architecture (excluding the batch normalization layers), using the block-wise sampling strategy described in Ukai et al.
(2018), with a factor of 0 .
0005 for the L 2 -regularization.
We initialize ﬁxed embeddings of size 32 sampled from U ( −√ 3 , √ 3) , and sample additive stochastic noise coming from U ( − 0 .
at the beginning of every mini-batch training.
Due to expensive computational requirements involved in training on ImageNet, we trained it with batch size 256 and learning rate 0 .
1 for 5 epochs, or equivalently, 25025 iterations.