F EATURE T RANSFORMERS : A U NIFIED R EPRESENTATION L EARNING F RAMEWORK FOR L IFELONG L EARNING
Anonymous authors
Paper under double-blind review
A BSTRACT
Despite the recent advances in representation learning, lifelong learning continues to be one of the most challenging and unconquered problems.
Catastrophic forgetting and data privacy constitute two of the important challenges for a successful lifelong learner.
Further, existing techniques are designed to handle only speciﬁc manifestations of lifelong learning, whereas a practical lifelong learner is expected to switch and adapt seamlessly to different scenarios.
In this paper, we present a single, uniﬁed mathematical framework for handling the myriad variants of lifelong learning, while alleviating these two challenges.
We utilize an external memory to store only the features representing past data and learn richer and newer representations incrementally through transformation neural networks - feature transformers .
We deﬁne, simulate and demonstrate exemplary performance on a realistic lifelong experimental setting using the MNIST rotations dataset, paving the way for practical lifelong learners.
To illustrate the applicability of our method in data sensitive domains like healthcare, we study the pneumothorax classiﬁcation problem from X-ray images, achieving near gold standard performance.
We also benchmark our approach with a number of state-of-the art methods on
MNIST rotations and iCIFAR100 datasets demonstrating superior performance.
I NTRODUCTION
Deep learning algorithms have achieved tremendous success on various challenging tasks like object detection, language translation, medical image segmentation, etc.
Lifelong learning - the ability to adapt, beneﬁt and sustain performance post deployment with more data and feedback, is an important goal for artiﬁcial intelligence (AI) and extremely crucial for sustained utility of these algorithms in domains like healthcare (learning from rare cases), self driving (learning new object detectors), etc.
While there is no single standard deﬁnition of lifelong learning, most of the research in this ﬁeld can be classiﬁed into one of the following sub-categories:
Incremental learning - encountering new data, with no change in distribution, for same task 2.
Domain adaptation - data from modiﬁed target distributions but for the same task 3.
New-task learning - data from tasks that were not presented before
Majority of the successful techniques study these variants in isolation.
However, a realistic lifelong learning scenario would involve a mixture of these manifestations over time.
Another major impediment to a successful lifelong learner is data privacy .
In domains like healthcare, it is near impossible to have data access beyond scope (both time and geography) which leads to catastophic forgetting (McCloskey & Cohen, 1989)-
the inability of machine learning models to retain past knowledge while learning from new data.
In this paper, we provide a uniﬁed framework - feature transformers - for practical lifelong learning to handle data privacy and catastrophic forgetting.
In summary, the major contributions of our work are as follows:
• Deﬁne a realistic and a challenging lifelong learning scenario and formulate a unique and generic mathematical framework - feature transformer, to work in this setting successfully.
• Ensure data privacy by storing only features from previous episodes while successfully combating catastrophic forgetting.
• Principled way of utilizing additional neural-compute to tackle complex incremental tasks.
• Demonstrate state-of-the-art results on common continuous learning benchmarks for new task learning like MNIST rotations, iCIFAR dataset and a challenging healthcare problem.
• Demonstrate exemplary performance even under severe constraints on memory and compute, thus operating under the assumptions of practical lifelong learning.
Following review of related work in section 2, we describe the mathematical formulation of feature transformers and their practical realization in section 3.
Section 4 contains experiments and results, followed by discussion in section 5.
R ELATED WORK
A conventional deep learning classiﬁer can be viewed as an automatic feature extractor followed by a classiﬁer, trained jointly.
For lifelong learning, the methods in literature can be broadly classiﬁed into: 1) learning with ﬁxed feature representations and 2) incrementally evolving representations.
Fixed Represenations : In this class of approaches, a representation for the data is learnt from the initial task and remains frozen for ensuing tasks, while only the classiﬁcation layers are modiﬁed.
The simplest baseline approaches include ﬁne-tuning (Oquab et al., 2014) and Mensink et al.
(2012) where the idea of using ﬁxed representations was extended further by using the nearest class mean classiﬁer.
In spite of their simplicity, these algorithms do not perform well in practice due to the limiting constraint of a ﬁxed representation throughout the incremental learning phase.
Incrementally evolving representations : The last few years have witnessed renewed interest in developing methods that allow changing data representation with the addition of new tasks.
Naive methods of retraining with only new data suffer from catastrophic forgetting (McCloskey & Cohen, 1989).
To overcome this problem, most methods have attempted different manifestations of rehearsal (Robins, 1995) - replaying data from the previous tasks.
However, access to all the previous data is usually not feasible due to size, compute and privacy concerns, researchers have attempted to reproduce past through proxy information, known as pseudorehearsal .
Pseudorehearsal techniques can be further classiﬁed depending on how the previous information is stored /used:
• Knowledge distillation based approaches : An example method for lifelong learning using knowledge distillation (Hinton et al., 2015) is learning without forgetting (LwF)
(Li & Hoiem, 2017) - where distillation loss was added to match output of the updated network to that of the old network on the old task output variables.
LwF and its extensions does not scale to a large number of new tasks, suffers from catastrophic forgetting and importantly do not address incremental learning or domain adapatation settings.
• Rehearsal using exemplar sets : Rebufﬁ et al.
(2017) propose iCaRL, an incremental learning approach by storing an exemplar set of data from the previous tasks and augmenting it with the new data.
In more recent works, Javed & Paracha (2018) and Castro et al.
(2018) have argued that a decoupled nearest mean classiﬁer from Rebufﬁ et al.
(2017) is not essential and have proposed joint learning of feature extraction and classiﬁcation.
Lopez-Paz et al.
(2017) propose Gradient Episodic Memory, a technique which stores data from previous classes and constrains the gradient update while learning new tasks.
• Rehearsal using generative models : Triki Rannen et al.
(2017) propose to reproduce past knowledge by using task-speciﬁc under-complete autoencoders.
When a new task is presented, optimization is constrained by preventing the reconstructions of these autoencoders from changing, thereby ensuring that the features necessary for the previous tasks are not destroyed.
Shin et al.
; He et al.
; Venkatesan et al.
(2017) employ generative adversarial networks for recreating the history from previous tasks.
• Network regularization strategies : Methods belonging to this class aim to identify and selectively modify parts of the neural network which are critical to remember past knowledge or by explicitly penalizing loss of performance on old tasks.
Kirkpatrick et al.
; Liu et al.
(2018) use the Fischer information matrix to identify the most crucial weights responsible for prediction of a given task and lower the learning rate for these tasks.
Lee et al.
(2017) propose a dynamically expanding network to increase the capacity for
new tasks if the previous architecture is insufﬁcient to represent the data.
Our framework lies at the intersection of pseudorehearsal methods and progressive neural networks, with scope for judiciously utilizing extra capacity, while resisting catastrophic forgetting.
L IFE - LONG LEARNING VIA FEATURE TRANSFORMATIONS
Before we present the feature transformer method, we introduce the terminologies and notations.
We consider training a deep neural network which classiﬁes an input to one of the classes c ∈
[ C ] (cid:44) { 1 , 2 , · · · , C } .
We refer to the operation of classifying an input to a particular class as a task .
To this end, the classiﬁer is trained with training dataset ( X, Y ) , drawn from a joint distribution (
X , Y ) .
We view the deep neural network, deﬁned by the parameters ( θ , κ ) , as the composition of a feature extractor Φ θ : X → F , and a classiﬁer Ψ κ
[Equation]
where X is the space of input data, and F is a space of low-dimensional feature vectors.
We concisely denote the training of the neural network by TRAIN ( θ , κ ; D ) , which minimizes a loss function on training data D =
( X, Y )
and produces the network parameters ( θ , κ ) .
Let us also deﬁne the set of all computed features on input F =
Φ θ ( X ) .
When the loss function only penalizes misclassiﬁcation, the network is expected to learn only the class separation boundaries in the feature space.
However, as we demonstrate experimentally, good separation of class speciﬁc features enables stable learning of representations, which directly has a bearing on the performance of life-long learning.
Therefore, in all our training procedures, we also use a feature loss which promotes feature separation
[Equation]
In the lifelong learning context, we denote the time varying aspect of the network, training data and the classes, by using the time symbol t ∈ N (cid:44) { 0 , 1 , 2 , · · · } in superscript on these objects.
Realistically, at any time t > 0 , the classiﬁer encounters any of the following canonical situations:
The number of classes remain the same for the classiﬁer as at t − 1 .
The model encounters new training data for a subset of classes, without change in their distribution: ∀ t > 0 , C ( t ) =
C ( t − 1) , T ( t )⊆
[ C ( t ) ] and ∀ τ ∈ T ( t ) 1 , ( X ( t ) τ , Y ( t ) τ ) ∼ ( X ( t ) τ , Y ( t ) τ ) =
( X ( t − 1) τ , Y ( t − 1) τ ) .
The number of classes remain the same for the classiﬁer as at t − 1 .
The model encounters new training data from modiﬁed input distribution(s), for a subset of classes: ∀ t > 0 , C ( t ) =
C ( t − 1) , T ( t )⊆
[ C ( t ) ] and ∀ τ ∈ T ( t ) 2 , ( X ( t ) τ , Y ( t ) τ ) ∼ ( X ( t ) τ , Y ( t ) τ ) (cid:54) =
( X ( t − 1) τ , Y ( t − 1) τ ) .
The model encounters new class(es) and corresponding new data: ∀ t > 0 , C ( t ) > C ( t − 1) , T ( t )=
[ C ( t ) ]
\ [ C ( t − 1) ] and ∀ τ ∈ T ( t ) 3 , ( X ( t ) τ , Y ( t ) τ ) ∼ ( X ( t ) τ , Y ( t ) τ ) .
In the most generic scenario a combination of all the three situations can occur at any time index t , with training data available for the classes in the set T ( t ) (cid:44) T ( t )∪ T ( t )∪ T ( t ) 3 .
However, it is important to note that at every index t , the classiﬁer is trained to classify all the C ( t ) classes.
3. F EATURE TRANSFORMATION BY AUGMENTING NETWORK CAPACITY
At any time t − 1 , the classiﬁer is optimized to classify all the classes [ C ( t − 1) ] and also the set of features F ( t − 1) are well separated according to classes.
At t , when new training data D ( t ) = ∪ τ ∈T ( t ) ( X ( t ) τ , Y ( t ) τ ) is encountered, the features extracted using the previous feature extractor
[Equation]
are not guaranteed to be optimized for classifying the new data and new classes.
In order to achieve good performance on new data and classes, we propose to change the feature representation at time t , just before the classiﬁcation stage.
We achieve this by deﬁning a feature transformer
[Equation]
parameterized by ∆ θ ( t ) , which maps any feature extracted by Φ θ ( t − 1) to a new representation.
The new feature extractor is now given by Φ θ ( t ) (cid:44) Φ ∆ θ ( t ) ◦ Φ θ ( t − 1) , where θ ( t ) (cid:44) θ ( t − 1) ∪ ∆ θ ( t ) .
Practically, this is realized by augmenting the capacity of the feature extractor at each time t by using one or more fully connected layers 2 .
It is however possible that Φ ∆ θ ( t ) could be simply an identity transform and feature transformers learnt in previous episodes could be adapted for new data.
This helps in controlling the growth of network capacity over time and this aspect of our work is described in section 4.5.
The feature transformer is trained, along with a new classiﬁer layer, using the composite loss function of the form in equation 2, by invoking TRAIN (∆ θ ( t ) , κ ( t ) ; D ( t ) ) , with D ( t ) =
( ∂ F ( t ) , Y ( t ) ) 3 .
This ensures that the classiﬁer performs well on the new data.
However, strikingly, training a feature transformer at t does not involve changing the feature extractor Φ θ ( t − 1) at all, and this helps us in alleviating catastrophic forgetting by efﬁciently making use of already computed features F ( t − 1) through a memory module.
3. R EMEMBERING HISTORY VIA MEMORY
The set of all extracted features F ( t − 1) serves as a good abstraction of the model, for all the tasks and data encountered till t − 1 .
Therefore, if F ( t − 1) is available to the model when it encounters new tasks and data, then the feature transformer at t can take advantage of this knowledge to retain the classiﬁcation performance on previous tasks and data as well.
To this end, we assume the availability of an un-ending memory module M , equipped with READ () , WRITE () and ERASE () procedures, that can store F ( t − 1) and can retrieve the same at t .
In situations where memory is scarce, only a relevant subset of F ( t − 1) can be stored and retrieved.
We train the feature transformer at any t > 0 by invoking TRAIN (∆ θ ( t ) , κ ( t ) ; D ( t ) ) , with the combined set of features
[Equation]
We then obtain the new set of features
[Equation]
and replace F ( t − 1) in memory by (a subset of) F ( t ) .
With the assumption of inﬁnite memory and capacity augmentation at every episode, our feature transformers framework is presented in algorithm 1.
3. F EATURE T RANSFORMERS IN ACTION
Before we present experimental results to demonstrate the ability of feature transformers for lifelong learning of new tasks, we ﬁrst show how feature transformers operate when simply new episodes of data are presented.
As described in section 3, we use a composite loss comprising of classiﬁcation loss and feature loss to train the classiﬁer.
To promote feature clustering/separation, we propose to use center-loss as described in Wen et al.
Dropping the time index t for the brevity, with the convention that the ground truth class for each x is encoded using one-hot vector ¯ y
=
[ y 1 , y 2 , · · · , y C ] T , and ( Ψ κ ◦ Φ θ ( x ))
c is
the c th component of the classiﬁer output, we use the following loss function for all the classiﬁer training procedures:
[Equation]
where D = ∪ τ ∈T ( X τ , Y τ ) is the given training data set, and µ c is the centroid of all features corresponding to input data labelled as c .
Figure 1 provides snapshot of feature transformation algorithm when a new episode of data is encountered.
We consider X-Ray lung images (from Wang et al.
(2017a))
consisting of two classes: (i) normal and (ii) pneumothorax.
At a time index ( t − 1) , the classiﬁer model is trained on
Input Task set T ( t ) , and training data ∪ τ ∈T ( t ) ( X ( t )
τ , Y ( t ) τ ) , ∀ t ≥
Output ( θ ( t ) , κ ( t ) ) , ∀ t t ← 0 , ERASE( M ) /
* Set initial time, erase memory */
D (0) ← ∪ τ ∈T (0) ( X (0) τ , Y (0) τ ) /
* Obtain initial tasks and training data */
TRAIN ( θ (0) , κ (0) ; D (0) ) /
* Train initial network */
F (0) ← ∪ τ ∈T (0) ( Φ θ (0) ( X (0) τ )) /
* Compute features */
WRITE ( M , ( F (0) , Y (0) ) ) /
* Write features to memory */ while TRUE do t ← t + 1 , obtain T ( t ) , ∪ τ ∈T ( t ) ( X ( t ) τ , Y ( t ) τ ) /
* Obtain current tasks and training data */
Compute ∂ F ( t ) using equation/* Compute old model features on new data */ ( F ( t − 1) , Y ( t − 1) ) ← READ ( M ) /
* Read previously computed features from memory */
Form D ( t ) using equation/* Form composite training data */
TRAIN (∆ θ ( t ) , κ ( t ) ; D ( t ) ) /
* Train feature transformer */ Φ θ ( t ) ← Φ ∆ θ ( t ) ◦ Φ θ ( t − 1) /
* Obtain new feature extractor */
Compute F ( t ) using equation/* Compute new features */
ERASE( M ) , WRITE ( M , ( F ( t ) , ∪ t (cid:48) ∈ [1 , 2 , ··· ,t ] Y ( t (cid:48) ) ) )
/* Erase & write new features*/ end
 ( t ) <latexit sha1_base64="PqpZ0URy/9EQtKjFJ4bKwR/diS0=
[Table]
/6Cpw1hxaPBQhqrtMQ1SBNBAgRLakQLmexJa3vgq81sPoLQIgzucRND12TAQA8EZGqlXrLrXIJG5Xij7euKbL3HrI5H2kl8SdXEEyO6TMp6maa9Ysiv2FHSZOHNSInPUe8VPtx/y2IcAuWRadxw7wm7CFAouIS24sYaI8TEbQsfQgPmgu8n0tpSeGKVPB6EyL0A6VX93JMzX2Zam0mc40oteJv7ndWIcXHQTEUQxQsBngwaxpBjSLCjaFwo4yokhjCthdqV8xBTjaOIsmBCcxZOXSbNaceyKc3tWql3O48iTI3JMysQh56RGbkidNAgnj+SZvJI368l6sd6tj1lpzpr3HJI/sL5+ANnkooQ=</latexit>
M <latexit sha1_base64="phZHaf37MWpQea5oXK/jTdgtkH0=">AAAB8nicbVDLSsNAFL2pr1pfVZdugkVwVRIRdFl040aoYB/QhjKZTtqhk5kwcyOU0M9w40IRt36NO//GSZuFth4YOJxzL3PuCRPBDXret1NaW9/
[Table]
/YPqodHbaNSTVmLKqF0NySGCS5ZCzkK1k00I3EoWCec3OZ
[Table]
<latexit sha1_base64="cyY4ZrPM6zr/1y4CZqRy4tJeApk=">AAAB63icbVA9SwNBEJ2LXzF+RS1tFoNgFe5E0DJoYxnBxEByhL3NJlmyu3fszgnhyF+wsVDE1j9k579xL7lCEx8MPN6bYWZelEhh0fe/vdLa+sbmVnm7srO7t39QPTxq2zg1jLdYLGPTiajlUmjeQoGSdxLDqYokf4wmt7n/+MSNFbF+wGnCQ0VHWgwFo5hLPZYm/WrNr/tzkFUSFKQGBZr96ldvELNUcY1MUmu7gZ9gmFGDgkk+q/RSyxPKJnTEu45qqrgNs/mtM3LmlAEZxsaVRjJXf09kVFk7VZHrVBTHdtnLxf+8borD6zATOkmRa7ZYNEwlwZjkj5OBMJyhnDpCmRHuVsLG1FCGLp6KCyFYfnmVtC/qgV8P7i9rjZsijjKcwCmcQwBX0IA7aEILGIzhGV7hzVPei/fufSxaS14xcwx/4H3+AB4XjkY=</latexit>
[Figure]
images with the loss in equation 7.
As shown by the t-SNE plot (A), the feature extractor Φ θ ( t − 1) produces features which are well-separated, and these features get stored in memory M .
However, at time t , when a set of 2000 new images is encountered, Φ θ ( t − 1) produces features that are scattered (t-SNE plot (B)).
To improve the separation on the new data, the feature transformer is trained using the (well-separated) features in M as well as poorly separated features (from new data), with the loss function promoting good separation in new representation.
This ensures that all theimages seen until time t is well separated (t-SNE plots (C) and (D)).
This is repeated for all time indices t .
Thus, the feature transformer, along with appropriate loss function, continuously changes the representation to ensure good classiﬁcation performance.
E XPERIMENTAL RESULTS
In this section, we benchmark our algorithm’s performance in various scenarios on relevant datasets.
• Realistic Lifelong scenario along with traditional multi-task(MT) on MNIST rotations • Incremental learning on Pneumothorax identiﬁcation (data-sensitive domain) • New-task learning on iCIFAR
4. P RACTICAL L IFELONG LEARNER - MNIST R OTATIONS DATASET
To simulate a realistic lifelong learning scenario as described in Section 2, we use MNIST rotations dataset - each task contains digits rotated by a ﬁxed angle between 0 and 180 degrees.
Firstly, we randomly permute the rotation angles (to test domain adaptation).
For every rotation angle, we randomly permute the class labels (0-10) and divide them into two subsets of 5 classes each (learning new tasks).
We divide the number of training samples into two different subsets (incremental learning).
Table 1 details the episodes of a lifelong learning experiment from the described procedure
[Table]
E XPERIMENT DETAILS AND R ESULTS
We used a basic CNN architecture composed of 3 conv-layers of 32 ﬁlters and kernel-size - (3x3) followed by 2 dense layers “fc1” and “fc2” of feature length 256 and a softmax.
Our feature transformer networks at every episode aims to transform “fc1” features using 2 additional dense layers of feature length 256.
Feature transformers from previous episode serve as initialization for current episode and these models were optimized for the cumulative loss (equation 7), with λ = 0 .
All the feature transformers were trained for only 3 epochs, with batch size of 32.
We compare our results to the two obvious life-long learners - naive and cumulative training, which serve as the lower and upper bounds of performance respectively in the absence of other competitive algorithms for this setting.
While the naive learner ﬁnetunes the entire network on the latest episode data, cumulative learner accumulates data from all the episodes seen so far and retrains the model.
Fig.
2a highlights the remarkable performance of the proposed approach.
We show the performance evolution over the ﬁrst 25 episodes in Fig.
2a while, performance comparisons over 80 episodes across multiple experiment runs are averaged and shown in Table.
2b.
We demonstrate that just by storing features in memory and learning feature transformers at every episode, we achieve almost similar performance to the gold-standard result.
Further, it also underlines the applicability of our approach as single framework to combat different requirements of lifelong learning.
Finally, we also compare our results (ﬁgure 3) in the conventional MT setting described in LopezPaz et al.
(2017), whose open source implementation we used for our experiments.
For details on the terminology and compared methods, the readers are encouraged to refer to Lopez-Paz et al.
Fig.
3a demonstrates clear superiority of our approach over multiple methods compared.
Additionally, backward transfer (BWT) - quantitative metric that models the deterioration of performance on older tasks while learning new tasks, is also negligible for our method, which shows resistance to catastrophic forgetting.
Fig.
3b highlights this further as shown by accuracy on ﬁrst task after learning subsequent tasks.
4. P NEUMOTHORAX IDENTIFICATION FROM X- RAY I MAGES
We simulate a practical manifestation of lifelong learning where a model trained to detect pneumothorax is deployed in a hospital with data arriving incrementally.
We utilize a subset of ChestXRay (Wang et al., 2017b) dataset, which consists of chest X-rays labeled with corresponding diagnoses.
We simulated incremental learning by providing the 8k training images in incremental batches of 2k and measured the performance on held-out validation set of 2k images.
Fig.
4a establishes the baselines for the experiment.
As in previous experiment, naive and cumulative training deﬁne the performance bounds.
To clearly highlight the value of our feature transfom, we also add another strong baseline - naive Learner with center loss, which learns on the recent batch but with an augmented loss function (equation 7).
In spite of a gain of 5% due to center loss, there is still a loss of 4% performance in the incremental learning set-up in the fourth batch of 2k.
Experimental Details and Results : We used a pre-trained VGG-network (Simonyan & Zisserman, 2014) as the base network and explored the use of features from different layers of the VGG-network
[Figure]
A pp r o p r i a t e V a li d a t i o n A cc u r a c y MNIST Rotations - Performance on
Simulator
[Figure]
[Figure]
namely - post the two pooling layers and fully connected layers. Feature transformer network essentially had one additional dense layer per step and was optimized for (equation 7).
[Figure]
t i o n
[Figure]
u r a c y
[Figure]
Fig.
4a captures the performance of feature transformer with the base features being extracted from ﬁrst pooling layer - block3 pool.
After fourth batch of data, feature transformer result almost matches the performance of cumulative training.
This performance is achieved despite not having access to the full images but only the stored features.
Table.
4b also presents the performance of feature transformer depending upon the base features used.
It can be noted that performance is lowest for the layer that is closer to the classiﬁcation layer - fc 2. This is intuitively satisfying because, the further layers in a deep neural network will be more ﬁnely tuned towards the speciﬁc task and deprives the feature transform of any general features.
ACC
BWT 0.
c l a ss i f
i c a t i o n a cc u r a c y
CIFARsingle independent iCARL EWC
GEM Feature Transformer (
[Figure]
setting    9 11 13 15 17 19 21 Batches 0.
V a li d a t i o n
A cc u r a c y iCaRl
AR
CIFAR100 -Single
Incremental Setting
[Figure]
Feature Transformer Cumulative Learner
[Figure]
4. I CIFAR100 D ATASET
We present the 100 classes from CIFAR100 dataset in a sequence of 20 tasks comprising of 5 classes each.
Similar to Lomonaco & Maltoni (2017), we use the deﬁnitions for MT and single-incremental task (SIT). In an MT setting, evaluation is performed only on the new tasks exposed to the learner in the current episode.
We start with a VGG type architecture, pretrained on iCIFAR10 dataset as suggested by Lopez-Paz et al.
(2017) and Lomonaco & Maltoni (2017).
Base features are extracted from ﬂatten layer (before the fully connected layers) and our feature transformers included two dense layers with feature length = 256.
Similar to our earlier experiments, the feature transformer module from the previous episode initializes the current episode transformer and is optimized for the cumulative loss, with NoEpochs = 30 and batch size = 32.
4.3. M ULTI - TASK SETTING
Fig 5a demonstrates the superiority of our approach by a signiﬁcant margin of > 10% .
Further, we have negligible Backward Transfer
. 4.3. S INGLE I NCREMENTAL T ASK SETTING
Fig 5b captures the performance over 20 batches of data for our method along with cumulative and naive learner.
Unsurprisingly, naive learner performs very poorly, while feature transformer displays exemplary performance numbers of 40% validation acccuracy after encountering allepisodes, while compared to cumulative learner at 50% .
Our method signiﬁcantly outperforms AR (Lomonaco & Maltoni, 2017), while iCaRL (Rebufﬁ et al., 2017) achieves best-in-class performance close to gold-standard cumulative learner.
This is not surprising because iCaRL is an explicit rehearsal technique, where exemplar images from previous episodes are stored and replayed while learning new tasks, while we only store low dimensional features.
4. E FFECT OF L IMITED M EMORY
One of the drawbacks of the proposed approach is the assumption of inﬁnite memory and need to store features computed on all samples observed so far.
To understand the extent of this limitation, we performed ablation experiments limiting the amount of history replayed as well as computing the storage requirements involved.
Storing all features is not necessary
We studied the effect of size of memory by limiting the number of samples stored to a smaller percentage.
We observed that performance dropped from 97% to 94% when we reduced the memory size to only 25% the original size on MNIST rotations (Table. 2).
We performed similar experiments on Pneumothorax classiﬁcation problem and achieved similar trends as shown Table.
3, clearly demonstrating the resistance of the proposed method.
Storing all features is not prohibitive Additionally, calculations of size-on-disk suggests that storing features of the entire history is not prohibitive.
A typical natural/medical image is 256*256* integers or more, whereas our representation is only 4096 ﬂoats (16kb).
Even the largest available medical image repository of 100k X-ray images takes 1.6GB which is not huge.
These are con- [Table] [Table] servative estimates.
A standard medical image can be of much larger size (1024*1024) and in 3-D (minimum 10 slices).
Any exemplar-based method (iCARL) will have severe storage limitations than our method.
Additionally, storing 50 low-dimensional features occupies same memory as storing one exemplar image.
This directly leads to storing more history compactly while addressing catastrophic forgetting and privacy.
4. C ONTROLLING THE GROWTH OF NETWORK CAPACITY
In the description of the feature transformer framework in section 3.1 and section 3.2, we provided a generic treatment of the method, where, at the end of the each episode, the features are transformed up to date and then stored in memory for the next episode.
With this scheme, it becomes imperative to always augment the capacity of the network in order to learn new representations, resulting in ever-growing network capacity.
However, this problem can be easily alleviated by partitioning the entire network into a base feature extractor and feature transformer layers.
The base feature extractor remains always ﬁxed, and it is only the output of base feature extractor that is always stored in memory.
With this scheme, the feature transformer layers need not grow in capacity and only a few already existing layers can be adapted for the new data.
When the capacity of the feature transformer layers is not sufﬁcient, then it can be augmented by adding one or more extra layers.
In either case, the stored base features are sufﬁcient to train.
Effect of varying additional capacity
We varied the size of feature transformers and observed the difference in performance.
Tableshows that by halving the additional capacity does not change the performance on MNIST rotations dataset at all.
In addition, we froze the capacity of feature transformers after 5 th episode and adapted them till end of 80 episodes.
It is striking that performance is still high.
Similarly, for Pneumothorax classiﬁcation, Table 5 shows the performance comparisons with varying capacity of 2 , 1 and zero fc layers post third episode.
These experiments (along with Sec 4.4) clearly demonstrate that power of the proposed approach comes from learning separable representations continually and not necessarily from storing all features or additional capacity
[Table]
[Table]
D ISCUSSION
In the ﬁnal section [Table] we discuss various points concerning our proposed approach [Table]
Bayesian interpretation of Feature transformers
Our feature transformers framework - learning a new representation with every new episode of data, can be interpreted as a maximum-a-posteriori representation learning, with the previous representation acting as prior.
The implementation described in this paper using the combination loss (equation 7) is one instantiation of a general incremental representation learning possible in our framework.
We have cast the MAP estimate problem to a tractable optimization problem constrained by a center-loss.
In future, we plan to explore other manifestations of our approach with different loss functions that can ensure better separability .
Information Loss, Incremental Capacity, Data Privacy
As shown in Table.
4b, feature transformer becomes less effective if the base features do not contain enough relevant information.
This also means that additional capacity that every feature transformer adds may not help or in-fact be counter-productive.
If the base features are extracted from layers close to the input image, there will be problem of traceability which violates the data privacy requirement we want to accomplish.
We feel this a potential trade-off between performance and data privacy which we will investigate in future.
Model compaction for cascade of feature transformers
Another approach to control the growth of network capacity is model compaction, which will be our future work.
At any point in time, the entire set of feature transformer layers can be replaced by a smaller and simpler network (possibly using distillation techniques) and then again allowed to grow subsequently.
This cycle of grow-and-purge can be used to effectively manage the overall capacity of the network.