U NCERTAINTY - GUIDED C ONTINUAL L EARNING WITH B AYESIAN N EURAL N ETWORKS
Anonymous authors
Paper under double-blind review
A BSTRACT
Continual learning aims to learn new tasks without forgetting previously learned ones.
This is especially challenging when one cannot access data from previous tasks and when the model has a ﬁxed capacity.
Current regularization-based continual learning algorithms need an external representation and extra computation to measure the parameters’ importance .
In contrast, we propose Uncertaintyguided Continual Bayesian Neural Networks (UCB), where the learning rate adapts according to the uncertainty deﬁned in the probability distribution of the weights in networks.
Uncertainty is a natural way to identify what to remember and what to change as we continually learn, and thus mitigate catastrophic forgetting.
We also show a variant of our model, which uses uncertainty for weight pruning and retains task performance after pruning by saving binary masks per tasks.
We evaluate our
UCB approach extensively on diverse object classiﬁcation datasets with short and long sequences of tasks and report superior or on-par performance compared to existing approaches.
Additionally, we show that our model does not necessarily need task information at test time, i e. it does not presume knowledge of which task a sample belongs to.
I NTRODUCTION
Humans can easily accumulate and maintain knowledge gained from previously observed tasks, and continuously learn to solve new problems or tasks.
Artiﬁcial learning systems typically forget prior tasks when they cannot access all training data at once but are presented with task data in sequence.
Overcoming these challenges is the focus of continual learning , sometimes also referred to as lifelong learning or sequential learning .
Catastrophic forgetting (McCloskey & Cohen, 1989; McClelland et al., 1995) refers to the signiﬁcant drop in the performance of a learner when switching from a trained task to a new one.
This phenomenon occurs because trained parameters on the initial task change in favor of learning new objectives.
This is the reason that naive ﬁnetuning intuitively suffers from catastrophic forgetting .
Given a network of limited capacity, one way to address this problem is to identify the importance of each parameter and penalize further changes to those parameters that were deemed to be important for the previous tasks (Kirkpatrick et al., 2017; Aljundi et al., 2018a; Zenke et al., 2017).
An alternative is to freeze the most important parameters and allow future tasks to only adapt the remaining parameters to new tasks (Mallya & Lazebnik, 2018).
Such models rely on the explicit parametrization of importance.
We propose here implicit uncertainty-guided importance representation.
Bayesian approaches to neural networks (MacKay, 1992b) can potentially avoid some of the pitfalls of explicit parameterization of importance in regular neural networks.
Bayesian techniques, naturally account for uncertainty in parameters estimates.
These networks represent each parameter with a distribution deﬁned by a mean and variance over possible values drawn from a shared latent probability distribution (Blundell et al., 2015).
Variational inference can approximate posterior distributions using Monte Carlo sampling for gradient estimation.
These networks act like ensemble methods in that they reduce the prediction variance but only use twice the number of parameters present in a regular neural network.
We propose the use of the predicted mean and variance of the latent distributions to characterize the importance of each parameter.
We perform continual learning with Bayesian neural networks by controlling the learning rate of each parameter as a function of its uncertainty.
Figure 1 illustrates how posterior distributions evolve for certain and uncertain weight
[Figure] [Figure]
gradient update
distributions while learning two consecutive tasks.
Intuitively, the more uncertain a parameter is, the more learnable it can be and therefore, larger gradient steps can be taken for it to learn the current task.
As a hard version of this regularization technique, we also show that pruning, i.e., preventing the most important model parameters from any change and learning new tasks with the remaining parameters, can be also integrated into UCB.
We refer to this method as UCB-P.
Contributions: We propose to perform continual learning with Bayesian neural networks and develop a new method which exploits the inherent measure of uncertainty therein to adapt the learning rate of individual parameters (Sec. 4).
Second, we introduce a hard-threshold variant of our method that decides which parameters to freeze (Sec. 4.2).
Third, in Sec. 5, we extensively validate our approach experimentally, comparing it to prior art both on single datasets split into different tasks, as well as for the more difﬁcult scenario of learning a sequence of different datasets.
Forth, in contrast to most prior work, our approach does not rely on knowledge about task boundaries at inference time, which humans do not need and might not be always available.
We show in Sec. 6 that our approach naturally supports this scenario and does not require task information at test time, sometimes also referred to as a “single head” scenario for all tasks.
We refer to evaluation metric of a “single head” model without task information at test time as “generalized accuracy”.
Our code will be released upon acceptance.
R ELATED W ORK
Conceptually, approaches to continual learning can be divided into the following categories: dynamic architectural methods, memory-based methods, and regularization methods.
Dynamic architectural methods : In this setting, the architecture grows while keeping past knowledge ﬁxed and storing new knowledge in different forms such as additional layers, nodes, or modules.
In this approach, the objective function remains ﬁxed whereas the model capacity grows –often exponentially– with the number of tasks.
Progressive networks (Rusu et al., 2016; Schwarz et al., 2018) was one of the earliest works in this direction and was successfully applied to reinforcement learning problems; the base architecture was duplicated and lateral connections added in response to new tasks.
Dynamically Expandable Network (DEN) (Yoon et al., 2018) also expands its network by selecting drifting units and retraining them on new tasks.
In contrast to our method, these approaches require the architecture grow with each new task.
Memory-based methods: In this regime, previous information is partially stored to be used later as a form of rehearsal (Robins, 1995).
Gradient episodic memory (GEM) (Lopez-Paz et al., 2017) uses this idea to store the data at the end of each episode to be used later to prevent gradient updates from deviating from their previous values.
GEM also allows for positive backward knowledge transfer, i.e, an improvement on previously learned tasks, and it was the ﬁrst method capable of learning using a single training example.
Recent approaches in this category have mitigated forgetting by using external data combined with distillation loss and/or conﬁdence-based sampling strategies to select the most representative samples.
(Castro et al., 2018; Wu et al., 2019; Lee et al., 2019)
Regularization methods : In these approaches, signiﬁcant changes to the representation learned for previous tasks are prevented.
This can be performed through regularizing the objective function or directly enforced on weight parameters.
Typically, this importance measure is engineered to represent the importance of each parameter.
Inspired by Bayesian learning, in elastic weight consolidation (EWC) method (Kirkpatrick et al., 2017) important parameters are those to have the highest in terms of the Fisher information matrix.
In Synaptic Intelligence (SI) (Zenke et al., 2017)
this parameter importance notion is engineered to correlate with the loss function: parameters that contribute more to the loss are more important.
Similar to SI, Memory-aware Synapses (MAS) (Aljundi et al., 2018a) proposed an online way of computing importance adaptive to the test set using the change in the model outputs w.r.t the inputs.
While all the above algorithms are task-dependent, in parallel development to this work, (Aljundi et al., 2018b) has recently investigated task-free continual learning by building upon MAS and using a protocol to update the weights instead of waiting until the tasks are ﬁnished.
PackNet (Mallya & Lazebnik, 2018) used iterative pruning to fully restrict gradient updates on important weights via binary masks.
This method requires knowing which task is being tested to use the appropriate mask.
PackNet also ranks the weight importance by their magnitude which is not guaranteed to be a proper importance indicative.
HAT (Serra et al., 2018) identiﬁes important neurons by learning an attention vector to the task embedding to control the gradient propagation.
It maintains the information learned on previous tasks using an almost-binary mask per previous tasks.
Bayesian approaches: Using Bayesian approach in learning neural networks has been studied for few decades (MacKay, 1992b;a).
Several approaches have been proposed for Bayesian neural networks, based on, e.g., the Laplace approximation (MacKay, 1992a), Hamiltonian Monte Carlo (Neal, 2012), variational inference (Hinton & Van Camp, 1993; Graves, 2011), and probabilistic backpropagation (Hern ´ andez-Lobato & Adams, 2015).
Variational continual learning (Nguyen et al., 2018) uses Bayesian inference to perform continual learning where new posterior distribution is simply obtained by multiplying the previous posterior by the likelihood of the dataset belonging to the new task.
They also showed that by using a core-set, a small representative set of data from previous tasks, VCL can experience less forgetting.
In contrast, we rely on Bayesian neural networks to use their predictive uncertainty to perform continual learning.
Moreover, we do not use episodic memory or any other way to access or store previous data in our approach.
Natural gradient descent methods: A fast natural gradient descent method for variational inference was introduced in (Khan & Nielsen, 2018) in which, the Fisher Information matrix is approximated using the generalized Gauss-Newton method.
In contrast, in our work, we use classic gradient descent.
Although second order optimization algorithms are proven to be more accurate than the ﬁrst order methods, they add considerable computational cost.
Tseran et al.
; Chen et al.
both investigate the effect of natural gradient descent methods as an alternative to classic gradient descent used in VCL and EWC methods.
GNG (Chen et al., 2019) uses Gaussian natural gradients in the
Adam optimizer (Kingma & Ba, 2014) in the framework of VCL because as opposed to conventional gradient methods which perform in Euclidian space, natural gradients cause a small difference in terms of distributions following the changes in parameters in the Riemannian space.
Similar to VCL, they obtained their best performance by adding a coreset of previous examples.
Tseran et al.
(2018) introduce two modiﬁcations to VCL called Natural-VCL (N-VCL) and VCL-Vadam.
N-VCL (Tseran et al., 2018) uses a Gauss-Newton approximation introduced by (Schraudolph, 2002;
Graves, 2011) to estimate the VCL objective function and used natural gradient method proposed in (Khan et al, 2018) to exploit the Riemannian geometry of the variational posterior by scaling the gradient with an adaptive learning rate equal to σ − 2 obtained by approximating the Fisher Information matrix in an online fashion.
VCL-Vadam (Tseran et al., 2018) is a simpler version of N-VCL to trade-off accuracy for simplicity which uses Vadam (Khan et al, 2018) to update the gradients by perturbing the weights with a Gaussian noise using a reparameterization trick and scaling by σ − 1 instead of its squared.
N-VCL/VCL-Vadam both use variational inference to adapt the learning rate within Adam optimizer at every time step, whereas in our method below, gradient decent is used with constant learning rate during each task where learning rate scales with uncertainty only after ﬁnishing a task.
We show extensive comparison with state-of-the-art results on short and relatively long sequence of vision datasets with Bayesian convolutional neural networks, whereas VCL-Vadam only rely on multi-layer perceptron networks.
We also like to highlight that this is the ﬁrst work which evaluates and shows the working of convolutional Bayesian Neural Networks rather than only fully connected
MLP models for continual learning.
B ACKGROUND : V ARIATIONAL B AYES - BY -B ACKPROP
In this section, we review the Bayes-by-Backprop (BBB) framework which was introduced by (Blundell et al., 2015); to learn a probability distribution over network parameters.
(Blundell et al., 2015) showed a back-propagation-compatible algorithm which acts as a regularizer and yields comparable performance to dropout on the MNIST dataset.
In Bayesian models, latent variables are drawn from a prior density p ( w ) which are related to the observations through the likelihood p ( x | w ) .
During inference, the posterior distribution p ( w | x ) is computed conditioned on the given input data.
However, in practice, this probability distribution is intractable and is often estimated through approximate inference.
Markov Chain Monte Carlo (MCMC) sampling (Hastings, 1970) has been widely used and explored for this purpose, see (Robert & Casella, 2013) for different methods under this category.
However, MCMC algorithms, despite providing guarantees for ﬁnding asymptotically exact samples from the target distribution, are not suitable for large datasets and/or large models as they are bounded by speed and scalability issues.
Alternatively, variational inference provides a faster solution to the same problem in which the posterior is approximated using optimization rather than being sampled from a chain (Hinton & Van Camp, 1993).Variational inference methods always take advantage of fast optimization techniques such as stochastic methods or distributed methods, which allow them to explore data models quickly.
See (Blei et al., 2017) for a complete review of the theory and (Shridhar et al., 2018) for more discussion on how to use Bayes by Backprop (BBB) in convolutioal neural networks.
3. B AYES BY B ACKPROP (BBB)
Let x ∈ IR n be a set of observed variables and w be a set of latent variables.
A neural network, as a probabilistic model P ( y | x , w ) , given a set of training examples D = ( x , y ) can output y which belongs to a set of classes by using the set of weight parameters w .
Variational inference aims to calculate this conditional probability distribution over the latent variables by ﬁnding the closest proxy to the exact posterior by solving an optimization problem.
We ﬁrst assume a family of probability densities over the latent variables w parametrized by θ , i.e., q ( w | θ ) .
We then ﬁnd the closest member of this family to the true conditional probability of interest
P ( w |D ) by minimizing the Kullback-Leibler (KL) divergence between q and P which is equivalent to minimizing variational free energy or maximizing the expected lower bound:
[Equation]
The objective function can be written as:
[Equation]
Eq. 2 can be approximated using N Monte Carlo samples w i from the variational posterior (Blundell et al., 2015):
[Equation]
We assume q ( w | θ ) to have a Gaussian pdf with diagonal covariance and parametrized by θ =
( µ, ρ ) .
A sample weight of the variational posterior can be obtained by sampling from a unit Gaussian and reparametrized by w = µ + σ ◦ (cid:15) where (cid:15) is the noise drawn from unit Gaussian, and ◦ is a pointwise multipliation.
Standard deviation is parametrized as σ = log(1 + exp( ρ )) and thus is always positive.
For the prior, as suggested by Blundell et al.
(2015), a scale mixture of two
Gaussian pdfs are chosen which are zero-centered while having different variances of σ 21 and σ 22 .
The uncertainty obtained for every parameter has been successfully used in model compression (Han et al., 2015) and uncertainty-based exploration in reinforcement learning (Blundell et al., 2015).
In this work we propose to use this framework to learn sequential tasks without forgetting using per-weight uncertainties.
U NCERTAINTY - GUIDED C ONTINUAL L EARNING IN B AYESIAN N EURAL N ETWORKS
In this section, we introduce Uncertainty-guided Continual learning approach with Bayesian neural networks (UCB), which exploits the estimated uncertainty of the parameters’ posterior distribution to regulate the change in “important” parameters both in a soft way (Section 4.1) or setting a hard threshold (Section 4.2).
4. UCB WITH LEARNING RATE REGULARIZATION
A common strategy to perform continual learning is to reduce forgetting by regularizing further changes in the model representation based on parameters’ importance .
In UCB the regularizing is performed with the learning rate such that the learning rate of each parameter and hence its gradient update becomes a function of its importance .
As shown in the following equations, in particular, we scale the learning rate of µ and ρ for each parameter distribution inversely proportional to its importance Ω to reduce changes in important parameters while allowing less important parameters to alter more in favor of learning new tasks.
[Equation]
The core idea of this work is to base the deﬁnition of importance on the well-deﬁned uncertainty in parameters distribution of Bayesian neural networks, i.e., setting the importance to be inversely proportional to the standard deviation σ which represents the parameter uncertainty in the Baysian neural network:
[Equation]
We explore different options to set Ω in our ablation study presented in Section A.2 of the appendix,
Table 1.
We empirically found that Ω µ = 1 /σ and not adapting the learning rate for ρ (i.e. Ω ρ = 1 ) yields the highest accuracy and the least forgetting.
The key beneﬁt of UCB with learning rate as the regularizer is that it neither requires additional memory, as opposed to pruning technique nor tracking the change in parameters with respect to the previously learned task, as needed in common weight regularization methods.
More importantly, this method does not need to be aware of task switching as it only needs to adjust the learning rates of the means in the posterior distribution based on their current uncertainty.
The complete algorithm for UCB is shown in Algorithm 1 with parameter update function given in
Algorithm 2.
4. UCB USING WEIGHT PRUNING (UCB-P)
In this section, we introduce a variant of our method, UCB- P , is related to recent efforts in weight pruning in the context of reducing inference computation and network compression (Liu et al., 2017;
Molchanov et al., 2016).
More speciﬁcally, weight pruning has been recently used in continual learning (Mallya & Lazebnik, 2018), where the goal is to continue learning multiple tasks using a single network’s capacity.
(Mallya & Lazebnik, 2018) accomplished this by freeing up parameters deemed to be unimportant to the current task according to their magnitude.
Forgetting is prevented in pruning by saving a task-speciﬁc binary mask of important vs. unimportant parameters.
Here, we adapt pruning to Bayesian neural networks.
Speciﬁcally, we propose a different criterion for measuring importance: the statistically-grounded uncertainty deﬁned in Bayesian neural networks.
Unlike regular deep neural networks, in a BBB model weight parameters are represented by probability distributions parametrized by their mean and standard deviation.
Similar to (Blundell et al., 2015), in order to take into account both mean and standard deviation, we use the signal-to-noise ratio (SNR) for each parameter deﬁned as
[Equation]
SNR is a commonly used measure in signal processing to distinguish between “useful” information from unwanted noise contained in a signal.
In the context of neural models, the SNR can be thought as an indicative of parameter importance; the higher the SNR, the more effective or important the parameter is to the model predictions for a given task.
[Table]
UCB- P , as shown in Algorithms 1 and 3, is performed as follows: for every layer, convolutional or fully-connected, the parameters are ordered by their SNR value and those with the lowest importance are pruned (set to zero).
The pruned parameters are marked using a binary mask so that they can be used later in learning new tasks whereas the important parameters remain ﬁxed throughout training on future tasks.
Once a task is learned, an associated binary mask is saved which will be used during inference to recover key parameters and hence the exact performance to the desired task.
The overhead memory per parameter in encoding the mask as well as saving it on the disk is as follows.
Assuming we have n tasks to learn using a single network, the total number of required bits to encode an accumulated mask for a parameter is at max log 2 n bits assuming a parameter deemed to be important from task 1 and kept being encoded in the mask.
R ESULTS 5. E XPERIMENTAL S ETUP
Datasets: We evaluate our approach in two common scenarios for continual learning: 1) classincremental learning of a single or two randomly alternating datasets, where each task covers only a subset of the classes in a dataset, and 2) continual learning of multiple datasets, where each task is a dataset.
We use Split MNIST with 5 tasks (5-Split MNIST) similar to (Nguyen et al., 2018;
Chen et al., 2019; Tseran et al., 2018) and permuted MNIST (Srivastava et al., 2013) for class incremental learning with similar experimental settings as used in (Serra et al., 2018; Tseran et al., 2018).
Furthermore, to have a better understanding of our method, we evaluate our approach on continually learning a sequence of 8 datasets with different distributions using the identical sequence as in (Serra et al., 2018), which includes FaceScrub (Ng & Winkler, 2014), MNIST, CIFAR100,
NotMNIST (Bulatov, 2011), SVHN (Netzer et al., 2011), CIFAR10, TrafﬁcSigns (Stallkamp et al.,
[Table]
2011), and FashionMNIST (Xiao et al., 2017).
Details of each are summarized in Table 4 in appendix.
No data augmentation of any kind has been used in our analysis.
Baselines: Within the Bayesian framework, we compare to three models which do not incorporate the importance of parameters, namely ﬁne-tuning, feature extraction, and joint training.
In ﬁne-tuning ( BBB - FT ), training continues upon arrival of new tasks without any forgetting avoidance strategy.
Feature extraction, denoted as ( BBB - FE ), refers to freezing all layers in the network after training the ﬁrst task and training only the last layer for the remaining tasks.
In joint training ( BBB - JT ) we learn all the tasks jointly in a multitask learning fashion which serves as the upper bound for average accuracy on all tasks, as it does not adhere to the continual learning scenario.
We also perform the counterparts for FT, FE, and JT using ordinary neural networks and denote them as ORD-FT, ORDFE, and ORD-JT.
From the prior work, we compare with state-of-the-art approaches including Elastic
Weight Consolidation (EWC) (Kirkpatrick et al., 2017), Incremental Moment Matching (IMM) (Lee et al., 2017), Learning Without Forgetting (LWF) (Li & Hoiem, 2016), Less-Forgetting Learning (LFL) (Jung et al., 2016), PathNet (Fernando et al., 2017), Progressive neural networks (PNNs) (Rusu et al., 2016), and Hard Attention Mask (HAT) (Serra et al., 2018) using implementations provided by (Serra et al., 2018).
On Permuted MNIST results for SI (Zenke et al., 2017) are reported from (Serra et al., 2018).
On Split and Permuted MNIST, results for VCL (Nguyen et al., 2018) are obtained using their original provided code whereas for VCL-GNG
(Chen et al., 2019) and VCL-Vadam (Tseran et al., 2018) results are reported from the original work without re-implementation.
Because our method lies into the regularization-based regime, we only compare against baselines which do not beneﬁt from episodic or coreset memory.
Hyperparameter tuning: Unlike commonly used tuning techniques which use a validation set composed of all classes in the dataset, we only rely on the ﬁrst two task and their validations set, similar to the setup in (Chaudhry et al., 2019).
In all our experiments we consider a 0 .
15 split for the validation set on the ﬁrst two tasks.
After tuning, training starts from the beginning of the sequence.
Our scheme is different from (Chaudhry et al., 2019), where the models are trained on the ﬁrst (e.g. three) tasks for validation and then training is restarted for the remaining ones and the reported performance is only on the remaining tasks.
Training details: It is important to note that in all our experiments, no pre-trained model is used .
We used stochastic gradient descent with a batch size of 64 and a learning rate of 0 .
01 , decaying it by a factor of 0 .
3 once the loss plateaued.
Dataset splits and batch shufﬂe are identically in all UCB experiments and all baselines.
Pruning procedure and mask size : Once a task is learned, we compute the performance drop for a set of arbitrary pruning percentages from the maximum training accuracy achieved when no pruning is applied.
The pruning portion is then chosen using a threshold beyond which the performance drop is not accepted.
Mask size is chosen without having the knowledge of how many tasks to learn in the future.
Upon learning each task we used a uniform distribution of pruning ratios ( 50 - 100% )
and picked the ratio resulted in at most 1% , 2% , and 3% forgetting for MNIST, CIFAR, and 8tasks experiments, respectively.
We did not tune this parameter because in our hyperparameter tuning, we only assume we have validation sets of the ﬁrst two tasks.
Parameter regularization and importance measurement: Table 1 ablates different ways to compute the importance Ω of an parameter in Eq.
4 and 5.
As shown in Table 1 the conﬁguration that yields the highest accuracy and the least forgetting (maximum BWT) occurs when the learning rate regularization is performed only on µ of the posteriors using Ω µ = 1 /σ as the importance and Ω ρ = 1 .
[Table]
Performance measurement: Let n be the total number of tasks.
Once all are learned, we evaluate our model on all n tasks.
ACC is the average test classiﬁcation accuracy across all tasks.
To measure forgetting we report backward transfer, BWT, which indicates how much learning new tasks has inﬂuenced the performance on previous tasks.
While BWT < 0 directly reports catastrophic forgetting , BWT > 0 indicates that learning new tasks has helped with the preceding tasks.
Formally,
BWT and ACC are as follows:
[Equation]
i =
R i,n (8) where R i,n is the test classiﬁcation accuracy on task i after sequentially ﬁnishing learning the n th task.
Note that in UCB- P , R i,i refers the test accuracy on task i before pruning and R i,n after pruning which is equivalent to the end of sequence performance.
In Section 6, we show that our UCB model can be used when tasks labels are not available at inference time by training it with a “single head” architecture with a sum of number of classes for all tasks.
We refer to the ACC measured for this scenario as “Generalized Accuracy”.
5. 5-S PLIT MNIST
We ﬁrst present our results for class incremental learning of MNIST (5-Split MNIST) in which we learn the digits 0 − 9 in ﬁve tasks with 2 classes at a time in 5 pairs of 0 / 1 , 2 / 3 , 4 / 5 , 6 / 7 , and 8 / 9 .
Table 2a shows the results for reference baselines in Bayesian and non-Bayesian neural networks including ﬁne-tuning ( BBB - FT , ORD - FT ), feature extraction ( BBB - FE , ORD - FE ) and, joint training ( BBB - JT , ORD - JT ) averaged over 3 runs and standard deviations are given in Table ??
in the appendix.
Although the MNIST dataset is an “easy” dataset, we observe throughout all experiments that Bayesian ﬁne-tuning and joint training perform signiﬁcantly better than their counterparts, ORD - FT and ORD - JT .
For Bayesian methods, we compare against VCL and its variations named as VCL with Variational Adam (VCL-Vadam), VCL with Adam and Gaussian natural gradients (VCL-GNG).
For non-Bayesian methods, we compare against HAT, IMM, and
EWC (EWC can be regarded as Bayesian-inspired).
VCL-Vadam (ACC= 99 . 17% ) appears to be outperforming VCL (ACC= 98 .
and VCL-GNG (ACC= 96 .
50% ) in average accuracy.
However, full comparison is not possible because forgetting was not reported for Vadam and GNG.
Nevertheless,
UCB (ACC= 99 .
63% ) is able to surpass all the baselines including VCL-Vadam in average accuracy while in zero forgetting it is on par with HAT (ACC= 99 .
We also report results on incrementally learning MNIST in two tasks (2-Split MNIST) in Table 8 in the appendix, where we compare it against PackNet, HAT, and LWF where PackNet, HAT, UCB- P , and UCB have zero forgetting while
UCB has marginally higher accuracy than all others.
5. P ERMUTED MNIST
Permuted MNIST is a popular variant of the MNIST dataset to evaluate continual learning approaches in which each task is considered as a random permutation of the original MNIST pixels.
Following the literature, we learn a sequence of 10 random permutations and report average accuracy at the end.
Table 2b shows ACC and BWT of UCB and UCB- P in comparison to state-of-the-art models using a small and a large network with 0 .
1M and 1 .
9M parameters, respectively (architecture details are given in Section A.2 of the appendix).
The accuracy achieved by UCB (ACC= 91 .
44 ± 0 .
using the small network outperforms the ACC reported by Serra et al.
(2018) for SI (ACC= 86 . 0% ),
EWC (ACC= 88 .
2% ), while HAT attains a slightly better performance (ACC= 91 .
Comparing the average accuracy reported in VCL-Vadam (ACC= 86 .
34% ) and VCL-GNG (
ACC= 90 .
50% ) as well as obtained results for VCL (ACC= 88 .
80% ) shows UCB with BWT=( 0 .
03% ± 0 . 00% ) is able to outperform other Bayesian approaches in accuracy while forgetting signiﬁcantly less compared to
VCL with BWT= − 7 . 9% .
While we do not experiment with memory in this work, not surprisingly adding memory to most approaches will improve their performance signiﬁcantly as it allows looking into past tasks.
E.g. Chen et al.
(2019) report ACC= 94 .
37% for VCL-GNC when adding a memory of size 200 .
Next, we compare the results for the larger network (1.9M).
While HAT and UCB have zero forgetting,
UCB, reaching ACC= 97 .
42 ± 0 .
01% , performs better than all baselines including HAT which obtains
ACC= 97 .
34 ± 0 .
05% using 1 . 9M parameters.
We also observe again that BBB - FT , despite being not speciﬁcally penalized to prevent forgetting, exhibits reasonable negative BWT values, performing better than IMM and LWF baselines.
It is close to joint training, BBB - JT , with ACC= 98 .
1% , which can be seen as an upper bound.
5. A LTERNATING CIFAR10 AND CIFAR
In this experiment, we randomly alternate between class incremental learning of CIFAR10 and
CIFAR100.
Both datasets are divided into 5 tasks each with 2 and 20 classes per task, respectively.
Table 2c presents ACC and BWT obtained with UCB- P , UCB, and three BBB reference methods compared against various continual learning baselines.
Among the baselines presented in Table 2c, PNN and PathNet are the only zero-forgetting-guaranteed approaches.
It is interesting to note that in this setup, some baselines (PathNet, LWF, and LFL) do not perform better than the naive accuracy achieved by feature extraction.
PathNet suffers from bad pre-assignment of the network’s capacity per task which causes poor performance on the initial task from which it never recovers.
IMM performs almost similar to ﬁne-tuning in ACC, yet forgets more.
PNN, EWC, and HAT are the only baselines that perform better than BBB - FE and BBB - FT .
EWC and HAT are both allowed to forget by construction, however, HAT shows zero forgetting behavior.
While EWC is outperformed by both of our UCB variants, HAT exhibits 1% better ACC over UCB- P .
Despite having a slightly higher forgetting, the overall accuracy of UCB is higher, reaching 79 .
5. M ULTIPLE DATASETS LEARNING
Finally, we present our results for continual learning of 8 tasks using UCB- P and UCB in Table 2d.
Similar to the previous experiments we look at both ACC and BWT obtained for UCB- P , UCB, BBB references (FT, FE, JT) as well as various baselines.
Considering the ACC achieved by BBB - FE or
BBB - FT ( 58 . 1% ) as a lower bound we observe again that some baselines are not able to do better than BBB - FT including LFL, PathNet, LWF, IMM, and EWC while PNN and HAT remain the only strong baselines for our UCB- P and UCB approaches.
UCB- P again outperforms PNN by 3 . 6% in
ACC.
HAT exhibits only − 0 .
1% BWT, but our UCB achieves 2 .
4% higher ACC.
S INGLE H EAD AND G ENERALIZED A CCURACY OF UCB
UCB can be used even if the task information is not given at test time.
For this purpose, at training time, instead of using a separate fully connected classiﬁcation head for each task, we use a single head with the total number of outputs for all tasks.
For example in the 8 -dataset experiment we only use one head with 293 number of output classes, rather than using 8 separate heads, during training and inference time.
Table 3 presents our results for UCB and BBB - FT trained with a single head against having a multi-head architecture, in columns 4 - 7 .
Interestingly, we see only a small performance degrade
[Table]
for UCB from training with multi-head to a single head.
The ACC reduction is 0 .
6% , 5 . 1% , and
1% for 2-Split MNIST, Permuted MNIST, Alternating CIFAR10/100, and sequence of 8 tasks experiments, respectively.
We evaluated UCB and BBB - FT with a more challenging metric where the prediction space covers the classes across all the tasks.
Hence, confusion of similar class labels across tasks can be measured.
Performance for this condition is reported as Generalized ACC in Table 3 in columns 2 - 3 .
We observe a small performance reduction in going from ACC to Generalized ACC, suggesting non-signiﬁcant confusion caused by the presence of more number of classes at test time.
The performance degradation from ACC to Generalized ACC is 0 .
6% , 3 . 1% , and 3 .
1% for 2-Split MNIST, Permuted MNIST,
Alternating CIFAR10/100, and sequence of 8 tasks, respectively.
This shows that UCB can perform competitively in more realistic conditions such as unavailability of task information at test time.
We believe the main insight of our approach is that instead of computing additional measurements of importance, which are often task, input or output dependent, we directly use predicted weight uncertainty to ﬁnd important parameters.
We can freeze them using a binary mask, as in UCB- P , or regularize changes conditioned on current uncertainty, as in UCB.
C ONCLUSION
In this work, we propose a continual learning formulation with Bayesian neural networks, called
UCB, that uses uncertainty predictions to perform continual learning: important parameters can be either fully preserved through a saved binary mask (UCB- P ) or allowed to change conditioned on their uncertainty for learning new tasks (UCB).
We demonstrated how the probabilistic uncertainty distributions per weight are helpful to continually learning short and long sequences of benchmark datasets compared against baselines and prior work.
We show that UCB performs superior or on par with state-of-the-art models such as HAT (Serra et al., 2018) across all the experiments.
Choosing between the two UCB variants depends on the application scenario: While UCB- P enforces no forgetting after the initial pruning stage by saving a small binary mask per task, UCB does not require additional memory and allows for more learning ﬂexibility in the network by allowing small forgetting to occur.
UCB can also be used in a single head setting where the right subset of classes belonging to the task is not known during inference leading to a competitive model that can be deployed where it is not possible to distinguish tasks in a continuous stream of the data at test time.
UCB can also be deployed in a single head scenario and where tasks information is not available at test time.
A A PPENDIX A. D ATASETS
Table 4 shows a summary of the datasets utilized in our work along with their size and number of classes.
In all the experiments we resized images to 32 × 32 × 3 if necessary.
For datasets with monochromatic images, we replicate the image across all RGB channels.
[Table]
A. I MPLEMENTATION D ETAILS
In this section we take a closer look at elements of our UCB model on MNIST and evaluate variants of parameter regularization, importance measurement, as well as the effect of the number of samples drawn from the posited posterior.
Bayes-by-backprop (BBB) Hyperparamters:
Table 5 shows the search space for hyperparamters in the BBB algorithm Blundell et al.
(2015) which we used for tuning on the validation set of the ﬁrst two tasks.
[Table]
Network architecture:
For Split MNIST and Permuted MNIST experiments, we have used a twolayer perceptron which has 1200 units.
Because there is more number of parameters in our Bayesian neural network compared to its equivalent regular neural net, we ensured fair comparison by matching the total number of parameters between the two to be 1 .
9M unless otherwise is stated.
For the multiple datasets learning scenario, as well as alternating incremental CIFAR10/100 datasets, we have used a ResNet18 Bayesian neural network with 7 . 1 - 11 .
3M parameters depending on the experiment.
However, the majority of the baselines provided in this work are originally developed using some variants of AlexNet structure and altering that, e.g. to ResNet18, resulted in degrading in their reported and experimented performance as shown in Table 6.
Therefore, we kept the architecture for baselines as AlexNet and ours as ResNet18 and only matched their number of parameters to ensure having equal capacity across different approaches.
[Table]
Number of Monte Carlo samples: UCB is ensured to be robust to random noise using multiple samples drawn from posteriors.
Here we explore different number of samples and the effect on ﬁnal performance for ACC and BWT.
We have used Ω µ = 1 /σ as importance and regularization has been performed on mean values only.
Following the result in Table 7 we chose the number of samples to be 10 for all experiments.
[Table]
A. A DDITIONAL RESULTS
Here we include some additional results such as Table 8 for 2-split MNIST and some complementary results for tables in the main text as follows: ?? , 9, and 10 include standard deviation for results shown in Table 2a, 2b, 2c, respectively.
[Table]
[Table] [Table]