W ARPING THE S PACE : W EIGHT S PACE R OTATION FOR C LASS -I NCREMENTAL F EW -S HOT L EARNING
A BSTRACT
Class-incremental few-shot learning, where new sets of classes are provided sequentially with only a few training samples, presents a great challenge due to catastrophic forgetting of old knowledge and overfitting caused by lack of data.
During finetuning on new classes, the performance on previous classes deteriorates quickly even when only a small fraction of parameters are updated, since the previous knowledge is broadly associated with most of the model parameters in the original parameter space.
In this paper, we introduce WaRP, the weight space rotation process , which transforms the original parameter space into a new space so that we can push most of the previous knowledge compactly into only a few important parameters.
By properly identifying and freezing these key parameters in the new weight space, we can finetune the remaining parameters without affecting the knowledge of previous classes.
As a result, WaRP provides an additional room for the model to effectively learn new classes in future incremental sessions.
Experimental results confirm the effectiveness of our solution and show the improved performance over the state-of-the-art methods.
I NTRODUCTION
Humans can easily acquire new concepts while preserving old experiences continually over the course of their life span.
With a growing desire to imitate such ability, incremental or continual learning has been brought into the spotlight in the AI community recently (Hung et al., 2019; Wortsman et al., 2020; Saha et al., 2021).
Here, due to storage and privacy constraints, it is impractical to save all the training samples of previous tasks during the training process (Desai et al., 2021).
The most challenging issue in this setup is to preserve the knowledge of previous tasks against catastrophic forgetting (Serra et al., 2018).
More recently, an increasing need for such learning capability when dealing with rare data (e.g., military image, medical data, photos of rare animals) has encouraged many researchers to focus on a more challenging setup, known as class-incremental few-shot learning (CIFSL).
In CIFSL, each task consists of only a few training samples, making the problem much harder as we must additionally handle the severe overfitting issue caused by lack of training data.
Prior works on CIFSL typically take the following two steps: (i) pretraining the model on the first task (base classes) and (ii) adapting the pretrained model to new classes (novel classes) in each training session (e.g., via finetuning), assuming that the first task contains a sufficiently large number of training samples.
One recent work, F2M (Shi et al., 2021), tries to find the flat local minima during the pretraining stage and then finetunes the model within this flat area for learning novel classes so that the forgetting issue could be resolved.
Another line of work, named FSLL (Mazumder et al., 2021), mitigates the forgetting issue by keeping some important parameters frozen and finetunes only the remaining trainable parameters during the incremental learning sessions.
Several other works adopt different strategies to learn new classes well during incremental sessions without forgetting (Tao et al., 2020;
Cheraghian et al., 2021a; Zhang et al., 2021; Kukleva et al., 2021; Chen & Lee, 2021; Aky¨urek et al., 2022).
Motivation.
One key strategy shared among some of these prior works is: model update is performed in the original parameter space , which is defined here with the standard basis .
To illustrate the concept of standard basis in this context, consider an example with 2 × 2 weight matrix W for a specific layer.
By defining w ij as the ( i, j ) -th parameter of W , we can unroll W as follows:
[Equation]
Here, we denote E = { e 11 , e 12 , e 21 , e 22 } as standard basis, upon which the weight space is parameterized by w = { w 11 , w 12 , w 21 , w 22 } .
Neural networks are typically implemented in the weight space spanned by E , and thus the learning processes (e.g. gradient computation, parameter update) are performed with respect to the parameters in w as shown in Figure 1.
However, our empirical observation in Figure 2, which showcases the effect of simple finetuning in two different spaces, suggests that the accuracies of previous classes are extremely vulnerable to model finetuning in the
[Figure]
original space, even when we finetune only a small fraction of parameters; finetuning only 3% of model parameters (freeze 97%) throughout the incremental sessions significantly degrades the performance on base classes.
This implies that the previous knowledge is more broadly associated with, to a certain extent, most of the model parameters in the original space.
Motivated by this, we pose the following question: Can we find another basis, i.e., a new weight parameter space, such that we can push most of the previous knowledge compactly into only a few key parameters?
Contributions.
In this paper, we introduce the concept of W eight sp a ce R otation P rocess (WaRP) that provides a solution to the above question.
By viewing the weight matrix of a neural network from a different perspective, WaRP transforms the original parameter space into a new space.
For any given orthonormal set of matrices, we can reparameterize the weight in (1) as
[Equation]
update
[Equation]
update
[Equation]
update
[Equation]
where B = {K 11 , K 12 , K 21 , K 22 } is a newly constructed basis that consists of orthonormal matrices and (cid:101) w ij is the weight parameter that is reparameterized according to K ij .
As shown in Figure 1, we can always rotate the axes by which the weight is represented for an arbitrary basis B .
Here, if (cid:101) w 12 in (2) plays a key role in determining the output of the layer, then {K 11 , K 21 , K 22 } can be viewed as flat directions.
Thus, finetuning can be performed along the flat directions by freezing the important parameter (cid:101) w 12 in the new space, which effectively preserves the previous knowledge (Figure 2).
Figure 1 provides high-level descriptions of WaRP and our finetuning process along the flat direction.
By introducing the concept of WaRP, we propose a new strategy to construct an appropriate basis B using the low-rankness property of activation, and redefine the model parameters in this new space so that we can push most of the knowledge of base classes compactly into only a few parameters.
Once we construct a new weight space, we identify and freeze the important parameters for keeping the knowledge of previous classes at the end of each training session, and finetune only the remaining parameters in the next session without performance degradation on previous knowledge.
With a careful construction of new basis and a proper criterion for identifying important parameters, we can preserve the knowledge of base classes after finetuning the model on novel classes at each incremental session.
The knowledge of novel classes can be also preserved by accumulating the important parameters and keeping them frozen.
It turns out that WaRP is easy to implement and the proposed score criterion is efficient to compute.
Experimental results on various benchmarks show the improved performance of our method over the state-of-the-art baselines.
W EIGHT S PACE R OTATION
Before describing our solution for CIFSL, we first introduce the general concept of WaRP proposed in this paper.
To begin with, we view the weight matrix W ∈ R m × n as a linear combination of orthonormal matrices { e ij } i,j , where e ij ∈ R m × n is defined as a matrix having one for the ( i, j ) -th element and zeros for others.
We can rewrite W as
[Equation]
where w ij denotes the ( i, j ) -th element of W which is the coefficient of e ij .
Since neural networks are parameterized by coefficients w := { w ij } i,j in practical deep learning implementations, typical learning operations such as gradient descent are conducted on the weight space spanned by the standard basis E := { e ij } i,j .
However, the learned knowledge of previous tasks is likely to be associated with most of the parameters in w (i.e., most directions in standard basis E ); updating a small fraction of selected parameters could give rise to significant performance degradation on previous classes as shown in Figure 2, suggesting that the previous performance is highly vulnerable to finetuning in the original space.
In this paper, we extend the above linear algebraic interpretation of the weight matrix (based on standard basis E ) using an arbitrary basis B = {K
ij ∈ R m × n | i ∈
[ m ] , j ∈ [ n ] } that spans R m × n .
Specifically, let B be an arbitrary set of orthonormal matrices satisfying: (i) ⟨K , K ′ ⟩ = 0 if K ̸ = K ′ for K , K ′ ∈ B , and (ii) ∥K∥ = (cid:112) ⟨K , K⟩ = 1 for all K ∈ B , where ∥ · ∥ and ⟨· , ·⟩ are the norm and inner product properly defined in R m × n , respectively.
Since a vector space H :
= R m × n equipped with the inner product operation ⟨ A, B ⟩ := trace ( A ⊤ B ) (for A, B ∈ H ) is Hilbert space, one can always rewrite W ∈ H as a linear combination of orthonormal matrices in B as follows:
[Equation]
Note that this equation generalizes (3) to the case with an arbitrary basis B ; the special case with
K ij = e ij (i.e., B = E ) reduces to the result in equation (3).
Thus for any orthonormal set of matrices
B spanning R m × n , we can always reparameterize the layer of the model with (cid:101) w := {⟨ W, K⟩} K∈B in place of w .
In the next section, we describe how WaRP can address forgetting in CIFSL.
P ROPOSED A LGORITHM FOR CIFSL
Problem Setup and Notations.
The main goal of CIFSL is to incrementally train the model on a new set of classes at each training session, while avoiding forgetting of old classes.
Let T k be the task given in k -th session with tuples of training sample and its class label, i.e., T k = { ( x kt , y kt ) } N k t =1 , where N k denotes the number of samples in T k .
For simplicity, we omit super/subscripts of x kt and y kt if these can be clearly referred from the context.
By defining n s as the total number of training sessions, we assume that the corresponding sets of categories in each task, denoted by C 1 , . . .
, C n s , are completely disjoint.
As in previous works (Tao et al., 2020; Shi et al., 2021), we consider a setup where a sufficiently large amount training data are available in the first task T 1 .
In contrast, the subsequent tasks T 2 . .
, T n s consist of only a small number of samples per class, e.g., 5 samples per class.
We consider a practical setup where all training samples of previous tasks cannot be accessed during training due to privacy and storage constraints.
We use the term ‘task’ and ‘session’ interchangeably throughout the paper.
We let W l ∈ R m l × n l be the weight matrix of l -th layer in the neural network.
We discard the layer index l and use W ∈ R m × n for weight whenever the context obviates it.
I n denotes the n × n identity matrix, [ n ] denotes the set { 1 , . . .
n } , and { a i } i or { a i } i ∈I represents { a i | i ∈ I} for some index set I .
Overview of Approach.
Given the model trained on the first task T 1 , our first step is to construct a new basis via WaRP on which most directions are flat.
Once new basis is constructed in the first session, this new basis is fixed throughout the whole remaining (incremental) sessions.
At the end of each session, we identify the important parameters based on the scoring criterion and freeze them.
In the next session, we finetune only the remaining parameters, namely, the directions of flat axes, and the same process is repeated for the future tasks.
Throughout the learning process, the important parameters are accumulated and kept frozen to preserve the knowledge of both novel and base classes.
In the following, we start by describing how to construct an appropriate basis via WaRP for CIFSL and how to implement it in practice (Sec. 3.1).
Then we describe our criterion for identifying the important parameters in each incremental session (Sec. 3.2).
Finally, we describe the overall procedure of our algorithm for CIFSL with details (Sec. 3.3).
3. C ONSTRUCTING N EW B ASIS FOR W EIGHT S PACE R OTATION
To find an appropriate basis to reparameterize W ∈ R m × n , we start by expressing the basis by using a pair of unitary matrices.
Given any two unitary matrices V ∈ R m × m , U ∈ R n × n , it can be seen that the set B = {K ij :=
v i u ⊤ j ∈ R m × n | i ∈ [ m ] , j ∈ [ n ] } is orthonormal, where v i and u j denote the i -th and j -th column vectors of V and U , respectively.
Thus from (4), we can rewrite W as
[Equation]
where (cid:101) w ij := ⟨ W, v i u ⊤ j ⟩ .
In other words, W is reparameterized by { (cid:101) w ij } i,j using the basis { v i u ⊤ j } i,j constructed with arbitrary two unitary matrices V and U .
Now the question is, how to design appropriate V and U to make most of the elements in the basis to be flat directions?
For a specific layer l , we define a matrix Φ l =
[ ϕ l ( x 1 ) , ϕ l ( x 2 ) , . . . ,
ϕ l ( x N 1 )] where ϕ l ( x t ) denotes the activation of the l -th layer for the t -th sample in the first task T 1 .
Then the singular value decomposition of covariance of activations is given by
[Equation]
where Σ l is a diagonal matrix having singular values on its diagonal and U l is the corresponding unitary matrix.
Here, note that the activations of deep neural networks are known to have low-rankness property as shown in (Suzuki et al., 2020b; Wang et al., 2021), i.e., singular values decrease rapidly and most of the singular values are close to 0 .
Thanks to this low-rankness property, by adopting
U =
U l in (5) for the l -th layer W l and defining (cid:101) w lij =
⟨ W l , v i u ⊤ j ⟩ , we have
[Equation]
for any unitary matrix V and any sample x in T 1 .
Namely, the effect of activation ϕ l ( x ) is highly likely to be negligible in most directions in B = { v i u ⊤ j } i,j during forward propagation.
In other words, most directions in B are flat and thus only a few directions and their corresponding parameters are meaningful.
So if we can properly identify the important parameters among { (cid:101) w ij } i,j and freeze them, finetuning along the remaining directions (for which v i u ⊤ j ϕ l ( x ) ≈ 0 ) does rarely change the output W l ϕ l ( x ) for x in T 1 .
This implies that the feature embedding of previous classes can be preserved by freezing only a small fraction of parameters in this new weight space.
Implementation.
Note that the actual implementation of WaRP in (5) turns out to be simple in practice.
By using the fact that V V ⊤ =
I m and UU ⊤ = I n , we have W = V V ⊤ WUU ⊤ =
V (cid:102) WU ⊤ where (cid:102) W :=
V ⊤ WU .
From the definition above, (cid:101) w ij is ( i, j ) -th element of (cid:102) W .
Thus to apply WaRP, we just multiply unitary matrices to W and reparameterize the layer with (cid:102) W while V , U remain fixed.
Throughout the algorithm, we set V =
I m which we found to be a valid choice.
Convolutional Case.
Implementing WaRP for convolutional layers is almost the same with the case of fully-connected layers except for flattening the feature maps and reshaping the weight/unitary matrices, to make the pipelines workable as in fully-connected layer.
The detailed implementation for convolutional layer can be found in Appendix A.2.
3. I DENTIFYING I MPORTANT P ARAMETERS
In the new weight space constructed in Sec. 3.1, our scheme identifies the important parameters at the end of each training session and keep freezing them throughout the training process.
The remaining trainable parameters (except the accumulated important parameters) are finetuned in the next session.
In this subsection, we describe our strategy for selecting the important parameters in each session.
To identify the trainable parameters (i.e., flat directions), at the end of each k -th session, we take the score criterion that is well compatible with our new space constructed by WaRP.
By the chain rule, taking the derivative of the loss function, denoted by L , with respect to the parameter (cid:101) w lij yields:
[Equation]
where dz l denotes the gradient of the loss with respect to the output of the layer z l = W l ϕ l ( x ) for any x in T k .
Interestingly, it turns out that the term ( a ) in equation (8) also appears in (7).
In addition, as shown in (8), the trend of the magnitude of dL/d (cid:101) w lij follows the tendency of ( a ) .
From these observations, it can be seen that the gradient can indirectly capture the influence of finetuning within each direction v i u ⊤ j on T k .
Motivated by this, we consider the following score criterion:
[Equation]
where L k ( b ) is the loss computed with batch b in T k , and D n b k is a set that consists of N k /n b batches with size n b in T k .
In words, the score is computed by summing up the magnitudes of the gradients using the samples in T k .
Here, if s lij is small, then ∆ (cid:101) w ij v
i
u ⊤ j ϕ l ( x ) is likely to be small where ∆ (cid:101) w ij simulates the gradient with respect to (cid:101) w ij .
Thus, finetuning with respect to this (cid:101) w ij (following the direction of v i u ⊤ j ) on new classes may not distort the feature embedding of all previous classes .
Remarks.
We wish to highlight two points here.
First, WaRP in Sec.
3.1 and the score criterion in
Sec. 3.2 work in a highly complementary manner; if we use the standard basis, the flat directions are limited and thus the performance could be restricted regardless of the scoring criterion.
If we use a different score criterion (e.g. weight magnitude as done in (Mazumder et al., 2021))
other than (9) in the new weight space, we cannot fully exploit the low-rankness as in (8) and thus cannot properly identify flat directions.
These results are confirmed via experiments in Sec.
Secondly, due to the simplicity of implementing WaRP, we can rely on auto-grad computation of gradient with respect to the weight matrix; we can take the derivative with respect to (cid:102) W rather than with respect to (cid:101) w ij one by one, to obtain the scores at once.
Thus it is efficient to compute.
3. O VERALL P ROCEDURE FOR CIFSL
Based on WaRP and the proposed score criterion, we describe the overall procedure of our algorithm.
First Session ( k = 1 ).
We pretrain the embedding network and the linear classifier W cls by minimizing the cross-entropy loss L CE ( T k ) =
− 1 N k (cid:80) N k t =1 (cid:80) c ∈C k 1 { y kt = c } log (cid:16) exp( γ c,t ) (cid:80) c ′∈∪ ki =1 C i exp( γ c ′ ,t ) (cid:17) with k = 1 , where 1 {·} is an indicator function and γ c,t denotes the output of classifier corresponding to class c .
When pretraining is done, we replace the weight vector of linear classifier W cls corresponding to class c with the prototype (average of feature embeddings corresponding to class c ), which we found is a good option for balancing the model performance on base classes and future novel classes.
Now we construct a new basis for each layer according to Sec. 3 . 1 and apply WaRP.
Here, to alleviate the overfitting issue on novel classes during the remaining incremental sessions, we restrict the learnable space by applying WaRP only to the last few layers while keeping other layers fixed throughout the whole training process.
Note that once we construct a new basis in the first session, this basis is fixed throughout the remaining sessions and thus model updates are performed in the new/fixed space.
See Appendix A.4 for discussions on validity of the new space for novel classes.
At the end of the first session, we identify the important/trainable parameters based on the proposed importance score (9) in Sec. 3.2.
We select the parameters with the scores up to top α × 100% as important parameters and let the remaining ones as trainable parameters, where α is the parameter keeping ratio, a hyperparameter defined in range [0 , 1] .
To indicate whether each element of (cid:102) W l is important or not, we introduce a 0-1 mask M l ∈ R m l × n l as follows:
[Equation]
where s α is the score value corresponding to top α × 100% of the scores of all parameters in all selected layers, i.e., in { s lij |
i ∈ [ m l ] , j ∈ [ n l ] , selected layers l } .
If the ( i, j ) -th element of M l is 1 , the ( i, j ) -th element of (cid:102) W l is regarded as an important parameter.
Incremental Sessions ( k ≥ 2 ).
In the beginning of each incremental session k ≥ 2 , we compute the prototypes corresponding to new classes in C k and utilize them as a classifier along with the prototypes of the base classes.
To preserve the knowledge of previous classes during finetuning, we only update the unimportant parameters using the importance mask M l obtained from the previous session.
We block the gradient computation with respect to important parameters at each layer l as
[Equation]
and let W l ← V (cid:102) W l U ⊤ , where stop grad ( · ) is an operation that detaches the input from the computational graph during loss computation, and ⊙ denotes element-wise multiplication.
Thus, gradient descent with respect to (cid:102) W l in (11) leads to model update of the unimportant parameters only.
In each incremental session k , we finetune the model (cid:102) W l using the cross-entropy loss L CE ( T k ) for each selected layer l , while keeping the weights of the classifier fixed during finetuning.
After finetuning is finished on task T k , we also identify the important parameters for T k to be considered for the next session; we obtain the importance mask M l,k using criterion in (10) for a given α .
To preserve not only the knowledge of T k but also of classes in previous sessions, we accumulate important parameters identified so far:
[Equation]
All the knowledge encountered so far can be properly preserved by this accumulated mask.
Finally, when each k -th session is finished, the prediction is made on the test samples from all encountered classes, i.e., for a given input, the prediction ˆ y test is given by ˆ y test =
arg max c ′ ∈∪ ki =1 C i γ c ′ ,t .
The whole process described above is repeated until the last incremental session, i.e., for k = 2 , 3 , . . .
n s .
R ELATED W ORKS
Class-Incremental Few-shot Learning.
There have been many attempts to solve class-incremental few-shot learning where only a few labeled samples are available in each task (Cheraghian et al., 2021b; Zhu et al., 2021;
Zhou et al., 2022b; Cheraghian et al., 2021a).
To address both catastrophic forgetting and overfitting issues in CIFSL, some prior works take meta-learning approaches (Gidaris & Komodakis, 2018;
Ren et al., 2019; Yoon et al., 2020).
They typically mimic the inference stage during training so that the model can rapidly adapt to new classes.
However, these works assume that only one set of classes is given during the incremental session and do not consider a scenario in which “more than one” sets of new classes are consecutively provided.
TOPIC (Tao et al., 2020) is the first work that considered more than one incoming sets of classes, providing a cornerstone for future emerging studies (Aky¨urek et al., 2022; Chen & Lee, 2021;
Zhang et al., 2021; Mazumder et al., 2021; Shi et al., 2021;
Zhou et al., 2022a).
The authors of TOPIC adopted neural gas network to maintain the topology of feature space, preventing forgetting issue.
FSLL (Mazumder et al., 2021) addressed the overfitting issue by choosing a few trainable parameters to be updated for finetuning and prevented forgetting issue by imposing additional regularization to minimize the deviation from the pretrained model parameters.
However, this additional regularization may limit the learnability of the model on new classes.
Unlike many existing works, (Shi et al., 2021; Zhou et al., 2022a) turned their focus on preparing a desired pretrained model in advance suitable for future incremental sessions.
Specifically, the authors of F2M (Shi et al., 2021) inject noise to model parameters during pretraining to find the flat local minima, and finetune the model within this flat area during incremental sessions.
However, pretraining with noise injection can lead to a longer training time for the model to converge and induce a little compromise on the performance of the pretrained model in return for the flatness of local minima.
Compared to existing works where the proposed algorithms are performed in the ‘original parameter space’ (e.g. freezing/updating the parameters in standard basis as done in FSLL),
[Table]
our method explores the ‘new parameter space’ where the axes are aligned with many flat directions along which the model is stably updated, by reparameterzing the weight as in equation (4).
With a careful construction of the new weight space, we can finetune along the flat directions by properly identifying and freezing the important parameters at each session, enjoying both advantages of F2M (Shi et al., 2021) and FSLL (Mazumder et al., 2021), without any dedicated process to find the flat region in advance or additional regularization on model parameters.
Although not directly targeting
CIFSL, there is another work (Liu et al., 2018) that reparameterizes the weight using a rotation matrix for tackling continual learning.
Its focus is mainly on a good diagonalization of the Fisher information matrix (FIM) that shows up in the objective function of (Kirkpatrick et al., 2017).
In our work, however, the role of rotation is different in that we rotate the axes of loss landscape for identifying/utilizing the flat directions aligned with new basis by further exploiting the low-rankness property.
Utilizing Low-Rankness Property of Activation.
There are a few works that exploit the lowrankness property of activation in deep neural network (Suzuki et al., 2020b;a; Saha et al., 2021;
Wang et al., 2021).
This suggests that the inputs are highly concentrated on the subspace spanned by only a few number of certain orthogonal vectors.
(Suzuki et al., 2020b; Wang et al., 2021) confirm this property by showing the empirical distribution of singular values at some layers.
The authors of (Suzuki et al., 2020a) proposed a network compression scheme by exploiting this property which brings on small number of degrees of freedom.
In (Suzuki et al., 2020b), the authors derived the tight generalization bound of deep neural network with the help of high compressibility of the model induced by this property.
Another line of works (Saha et al., 2021; Wang et al., 2021) try to solve continual learning problem by utilizing the low-rankness property.
At each learning session, they update the model parameters using the gradient orthogonal to the subspace which the previous tasks are concentrated on.
Note that our work turns out to generalize the update process of (Saha et al., 2021; Wang et al., 2021) in that if we freeze the parameters of the weight matrix in a column-wise manner, the gradient update of our method can be reduced to that of these works as special cases.
E XPERIMENTS 5. E XPERIMENTAL S ETUP & D ETAILS
Datasets.
We evaluate our method on three benchmark datasets, CIFAR100 (Krizhevsky et al., 2019), mini ImageNet (Vinyals et al., 2016) and CUB200 (Wah et al., 2011), in the CIFSL setting.
For both
CIFAR100 and mini ImageNet, we split the total of 100 classes into 60 classes (for base classes) and 40 classes (for novel classes).
The 40 novel classes are divided into 8 different sets havingcategories each, and allocated to each incremental session following the 5-way 5-shot setting.
For
CUB200, we split the total of 200 classes into 100 for the base classes and the remaining 100 for the novel classes, where the 100 novel classes are divided into 10 different sets having 10 categories each, following the 10-way 5-shot setting in each incremental session.
We follow the same split configuration proposed by TOPIC (Tao et al., 2020) in all datasets.
Model Architecture.
For the embedding network, following the standard setup in (Tao et al., 2020;
Mazumder et al., 2021; Shi et al., 2021), we use ResNet20 for CIFAR100 and use ResNet18 (He et al., 2016) for the others.
Additionally, to see how a wider model affects the ability of our method for making additional room for learning new classes, we also conduct experiments with ResNet
[Table] [Table]
on CIFAR100, which is wider than ResNet20.
We randomly initialize the network before training on base classes for both CIFAR100 and mini ImageNet.
For CUB200, we deploy the Pytorch’s built-in pretrained model for initialization as existing works have done
[Table]
Evaluation & Design Parameter.
At each session
[Table]
we evaluate our method by measuring the accuracy on test samples from all encountered classes so far.
We conduct 5 simulations under different random seeds and report average values.
Regarding the hyperparameters, the parameter keeping ratio is set to be α = 0 .
1 for all experiments.
We apply WaRP to the last 1 or 2 resnet blocks while the remaining parameters including batchnorm layers are fixed after the pretraining stage in the first session.
For computing the output of the classifier, we use the standard dot product in the beginning, and replace it with cosine similarity (Gidaris & Komodakis, 2018) after adopting the prototypes as a classifier.
More detailed hyperparameter settings can be found in Appendix A.1.
Baselines.
We consider the following recent methods as baselines: iCaRL (Rebuffi et al., 2017),
Rebalance (Hou et al., 2019), GPM (Saha et al., 2021), EEIL (Castro et al., 2018), TOPIC (Tao et al., 2020), FSLL (Mazumder et al., 2021), CEC (Zhang et al., 2021) and F2M (Shi et al., 2021).
The performance of these baselines are mostly taken from (Zhang et al., 2021; Shi et al., 2021) for fair comparisons.
We also consider simple finetuning, which is equivalent to our method with α =using a smaller learning rate.
5. M AIN E XPERIMENTAL R ESULTS
We first report the evaluated accuracy of each scheme on mini ImageNet, CUB200 and CIFAR for ResNet18, in Tables 1, 2 and 3, respectively.
Due to space limitation, the result on CIFAR for ResNet20 is left to Appendix A.3.
As can be seen from the results in Tables 1 and 2, WaRP consistently outperforms all the baselines in whole incremental sessions.
More specifically, iCaRL,
Rebalance and EEIL are far below WaRP, since all these methods are not originally developed for handling the lack of data in CIFSL.
The naive finetuning method shows poor performance in all results as the model is highly overfitted to new classes and forgets previous ones rapidly due to scarcity of data.
It is worth noting that the performance gain of WaRP against one of SOTA methods,
F2M, is 2.68% for mini ImageNet and 1.12% for CUB200.
For CIFAR100 dataset, since there are no prior results for ResNet18, we reproduced the results of the following SOTA methods: GPM, FSLL,
CEC and F2M.
In Table 3, the trend is consistent with the results on mini ImageNet and CUB200, where the performance gains over the F2M and CEC are 0.75% and 1.30%, respectively.
To see the effect of finetuning in WaRP clearly, we compare our method with a simple and strong baseline which takes the prototype of each class as classifier for both base and novel classes.
Since this baseline is equivalent to the case of our method with no finetuning, we can directly see the pure gains of finetuning in WaRP by comparing with this baseline.
As can seen from the results on CUB in Table 4, this prototype-based baseline already outperforms or is comparable to SOTA methods, which is consistent with the observation from F2M (Shi et al., 2021).
We can see performance gains
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
of WaRP over this strong baseline
coherently in all incremental sessions, where the gain increases as the session progresses.
5. F URTHER S TUDIES ON W A RP
Performance on Base and Novel Classes.
In Figure 3a, we report the accuracies on base classes and novel classes separately.
We compare our WaRP with FSLL (Mazumder et al., 2021), which also finetunes the model by selecting trainable parameters as ours
[Figure]
but in the original parameter space defined with the standard basis.
We also report the performance of the prototype-based baseline considered in Table 4.
We have the following key observations in Figure 3a.
First, for all sessions, the base class performance of WaRP sacrifices little compared to the prototype baseline (0.61% drop in the last session) while achieving better result than FSLL (1.40% gain in the last session).
This indicates that WaRP can effectively keep the knowledge of base classes throughout the learning process.
Regarding the novel classes, WaRP has considerable gains compared to both FSLL (1.2%- 2.8% gain) and the prototype-based baseline (3%-5% gain), indicating that WaRP effectively provides additional room for learning new classes in the new space, compared to FSLL which may limit the learnability on new classes due to additional regularization.
The overall results confirm the ability of
WaRP for preserving the knowledge of previous classes and learning new classes.
Ablation on Choices of Basis/Score.
To see how the proposed new basis and score criterion work in a highly complementary manner, in Figure 3b, we compare the performance of 4 different (basis, score) combinations: we consider standard basis and new basis for the basis candidates.
For the score criterion, we consider weight score (Ws) and gradient score (Gs), which are the magnitudes of weight and gradient, respectively.
As we can see, if we do not apply our new space and score criterion simultaneously, the performance extremely degrades as the session progresses while our method shows much better performance than other combinations.
Ablation on Varying α .
In Figure 3c, we consider ablation on varying α ranging from 0.01 to 0.
to see the robustness of WaRP against the perturbing hyperparameter.
For an extreme case with α = 0 .
01 , the accuracy is low since most of model parameters are updated and thus the previous knowledge cannot be preserved.
However, if α is larger than a specific small threshold (0.05 in this case), WaRP consistently provides additional gain against the baseline, suggesting that the performance gain is robust to perturbing α .
C ONCLUSION
In this paper, we introduced the concept of weight space rotation process (WaRP), a general framework for reparameterizing the weight matrix of the neural network.
By taking advantage of WaRP, we proposed a novel strategy for CIFSL that effectively learns new classes while preserving previous knowledge, alleviating forgetting/overfitting issues.
Experimental results on various benchmarks confirmed the advantage of our method over the state-of-the-arts.
Although we focused on CIFSL in this paper, the general concept of WaRP can be utilized in any deep learning applications where the knowledge of the model should be kept in a few parameters, which we leave for a future work.
R EPRODUCIBILITY S TATEMENT
The detailed experimental setups including datasets, model architectures, evaluation metrics and hyperparameters are described in Sec.
5.1 and Appendix A.1.
We also provide the source code in Supplementary Material.
We run the simulations 5 times with different random seeds using
NVIDIA GeForce RTX 3090 GPU machine and report average values.
The code is available at https://github.com/EdwinKim3069/WaRP-CIFSL .
A CKNOWLEDGMENTS
This work was supported by Samsung Electronics Co., Ltd. and by IITP funds from MSIT of Korea (No. 2020-0-00626).
A A PPENDIX A. H YPERPARAMETERS
Optimizer.
We use SGD optimizer with a momentum of 0 .
9 throughout the whole sessions for all our simulations.
First Session ( k = 1 ).
We pretrain the model with the batch size of 128 in the first session for all datasets.
The number of pretraining epochs is 210 for ResNet18 on both CIFAR100 and mini ImageNet and 300 for ResNet20 on CIFAR100.
Regarding ResNet18 on CUB200, as we adopt Pytorch built-in pretrained model as done in most of previous works (Tao et al., 2020; Zhang et al., 2021), we pretrain the model using a smaller learning rate of 0.01 following (Chen & Lee, 2021), and the total number of epochs is 120.
The learning rate decays by 1/10 at last 90, 60, 30 epoch to the end of training (for example, the learning rate decays at 210, 240, 270 if the total training epochs is 300).
Incremental Sessions ( k ≥ 2 ).
Following (Mazumder et al., 2021), we finetune the model with training epochs of 30 for all datasets/architectures in WaRP experiments.
We use the learning rate of 0.01 for ResNet18 and 0.001 for ResNet20.
When searching the learning rate, we randomly split trainset of base classes into train/validation so that the validation set has a ratio of about 10% in total.
We also randomly sample the data from trainset of novel classes (different from the 5-shots that we use for training) and use them as validation set for novel classes.
We search the learning rate within the grids of [1e-1, 1e-2, 1e-3] on CUB200 for ResNet18 and on CIFAR100 for ResNet20.
As for the mini ImageNet and CIFAR100 for ResNet18, we follow the setting of CUB200.
Overall Settings for WaRP Algorithm.
Throughout the whole datasets/architectures, we randomly sample 20 batches ( 20 × 128 = 2 , 560 samples) from base classes and use them for computing SVD in the first session.
For computing score criterion, we set the batch size in equation (9) to be n b =for base classes and n b = 1 for new classes.
As for the layers to be applied weight space rotation, we apply WaRP to the last two resnet blocks (out of four blocks in total) for ResNet18 model and apply to the last one block (out of three blocks in total) for ResNet20 model.
We do not apply WaRP to the batch-normalization layer.
Note that we freeze all the layers, to which WaRP is not applied, after completing the first session.
The hyperparameter of keeping ratio α at each session is set to be 0.1.
The temperature for cosine similarity (Gidaris & Komodakis, 2018) in classifier is set to be a constant value of 16, following (Zhang et al., 2021).
A. I MPLEMENTATION OF W A RP FOR C ONVOLUTIONAL L AYER C ASE
We first denote the filter (kernel) of convolutional layer by W ∈ R m × n × k × k where k is size of kernel, m is the target number of output channels and n is the number of input channels, and let the activation (input of the layer) be denoted by Φ ∈ R n × h × w where h and w are height and width of activation feature map respectively.
For computing SVD of activation map, all the patches in the feature map, that the kernel slides across, are flattened as a vector form and all these flattened vectors are stacked to be 2-dimensional matrix.
For example, if the kernel size k is 3 and the stride and padding arerespectively, the shape of stacked version of activation matrix is (3 × 3 × n, h × w ) and the resulting unitary matrix would be U 2-dim ∈ R 9 n × 9 n .
Then after we reshape the filter W into 2-dimensional weight matrix as W 2-dim ∈ R m × nk 2 , the remaining process is the same with fully-connected layer case; we reparameterize the layer with (cid:102) W 2-dim =
V ⊤ 2-dim W 2-dim U 2-dim where V 2-dim ∈ R m × m
and
U 2-dim ∈ R nk 2 × nk 2 are unitary matrices.
During computing the loss, there are two options we can choose to implement W ← V (cid:102) WU ⊤ .
The first one is just to compute W 2-dim = V 2-dim (cid:102) W 2-dim U ⊤ 2-dim and reshape W 2-dim to be the same with
W ∈ R m × n × k × k at every loss computation.
This approach is easy to implement but might be computationally inefficient.
To pursuit a slight computational efficiency, we take the second approach: configure this layer as three convolutions with filters of
V conv ∈ R m × m × 1 × 1 , (cid:102) W conv ∈ R m × nk 2 × 1 ×and U conv ∈ R nk 2 × n × k × k where V conv , (cid:102) W conv and U conv are reshaped version of V 2-dim , (cid:102) W 2-dim and
U 2-dim respectively.
A. A DDITIONAL E XPERIMENTAL R ESULTS
We report additional results on CIFAR100 for ResNet20 as shown in Table 5.
As ResNet20 is too narrow (e.g., the number of channels in the last layer is 64) to make additional room that can be reaped by WaRP for learning new classes, the gain is rather minimal compared to the results for wider network of ResNet18 (the number of channels in the last layer is 512) as shown in Table 1, 2 and 3.
However, as can be seen in Table 6, WaRP still consistently provides gains in all sessions against prototype-based baseline which we considered in Sec. 5.2.
Note that the prototype-based baseline is strong in that this baseline already outperforms or is comparable to state-of-the-art methods, which is also confirmed in F2M (Shi et al., 2021).
Moreover, the effectiveness of WaRP is not restricted by the dataset; result on the same dataset, CIFAR100, for ResNet18 achieves state-of-the-art as can be seen in Table 3 and reads appreciable gains compared to ResNet20 case as shown in Table 7.
Thus, the limited gain in small network can be easily handled by replacing the layers of the model with wider layers in practical deployment of our method.
Table 8 and Figure 4 refers to the results on repeated experiments of Table 4 and Figure 3c on different datasets respectively.
As shown in Figure 4, WaRP consistently provides the gains against strong baseline (prototype) once the α is larger than a certain threshold value, which suggests that the WaRP is fairly robust to perturbing hyperparameter α value.
[Table]
[Table] [Table]
Additionally
[Table]
we further compare our method with more recent works (Zhou et al., 2022b; Chi et al., 2022; Hersche et al., 2022; Liu et al., 2022) on mini ImageNet dataset.
Note that as (Hersche et al., 2022) adopted ResNet12 network for evaluating their method, we reproduced the results of (Hersche et al., 2022) by adopting ResNet18 to match the architecture.
As shown in the table
[Table]
we confirm our method outperforms all the baselines, except for (Zhou et al., 2022b)
[Table]
but still shows better performance in most sessions.
Moreover, we found that (Aky¨urek et al., 2022) provided the results with ResNet12 architecture as done in (Hersche et al., 2022), thus we also evaluate our
[Table]
[Figure] [Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
[Table]
[Figure]
method for ResNet12 architecture and compare with (Aky ¨ urek et al., 2022; Hersche et al., 2022) on mini ImageNet.
As can be seen in the Table 10, our method shows better performance than the baselines which is consistent with the case for ResNet18.
[Table]
Lastly
[Table]
we compare with (Hersche et al., 2022) on CIFAR100 dataset for ResNet12 (we further evaluate our method on CIFAR100 for ResNet12 instead of ResNet18)
[Table]
as (Hersche et al., 2022) also provided the results on CIFAR100 for ResNet12.
Table 11 indicates that our method also shows the improved performance compared to (Hersche et al., 2022) on CIFAR
[Table]
which is consistent with the results on mini ImageNet
[Table]
A. V ALIDITY OF THE N EW B ASIS ON N OVEL C LASSES
As aforementioned, once the new basis is constructed in the first session, the new basis is fixed during the whole remaining sessions.
The future incoming tasks are not used to reconstruct or modify the basis to avoid unexpected behavior or statistical unreliability due to their lack of training samples.
Here one may argue that the new basis is suitable only for preserving the knowledge of the base classes.
Recall that we apply WaRP only to the last few layers while remaining the other layers fixed to resolve the overfitting issue.
Here, since the activation of each layer depends not only on the task itself but also on preceding layers, the important parameters identified in the new basis are shared among the base and novel classes, i.e., the novel classes are also aligned with the new basis to some extent.
To figure out this, we measure the keep ratio of important parameters that is accumulated as the session progressed.
As shown in Figure 5, although we set keep ratio α as 0.1 for all sessions, the measured accumulated ratio does not grow by multiples of 0.1.
Rather, the
[Table] [Table]
accumulated ratio increases by some value smaller than α .
As can be seen from the observation, the important parameters at each session somewhat overlap since the activation also depends on the preceding layers.
This suggests that the new basis constructed in the first session would be aligned with the future tasks to a certain extent.
Hence the new basis is also eligible for preserving the knowledge of the novel classes with our score criterion in (9).
[Figure]
Along with this observation, we further verify the ability to preserve the novel tasks by showing how the mode accuracy on novel classes varies as session grows.
The table 12 and 13 show the accuracy on T 2 and T 3 respectively, i.e., the first and second incremental novel task.
We evaluated on mini ImageNet dataset.
Moreover, to single out the effectiveness of WaRP in preserving the previous knowledge, we further evaluate the accuracies on a certain novel task when the model knows task ID.
In other words, the prediction is made when the output logits corresponding to this novel task are only given.
By doing so, we can exclude the effect of forgetting induced by the interference from the output logits
[Table] [Table]
corresponding to other tasks.
As can be seen in the table 14 and 15, the new basis constructed using base classes is also suitable for preserving the knowledge of the novel classes.
[Table]
C OMPARISON WITH E XISTING N ETWORK P RUNING S CHEMES
As our method identifies the important parameters
[Table]
we can compare our method against the network pruning schemes.
We consider three network pruning methods: two for structured (channel/filter) pruning (Li et al., 2017;
Liu et al., 2021) and one for unstructured pruning, SNIP
(Lee et al., 2019).
We adopted the methods of identifying parameters, including the score criterion and the structure of identification (e.g. whether identifying channel-wise or element-wise manner) proposed in these works.
Although we set the ratio of important parameters α as 0 . 1 in WaRP experiments, during the reproducing the results on the methods from (Li et al., 2017; Liu et al., 2021; Lee et al., 2019), we tuned the parameter of this ratio α to get their best performances.
We found that α = 0 .
9 ∼ 0 .
95 is suitable for (Li et al., 2017; Liu et al., 2021; Lee et al., 2019) as all of them still consider the original parameter space which exhibits the fundamental limitation as shown in the Figure 2.
The results are evaluated on mini ImageNet dataset.
As can be seen in the table 16, our method outperforms all of score criterion suggested in pruning schemes we considered, which indicates that the new space and its compatible score criterion are very effective in solving CIFSL.
A. C OMPUTATIONAL C OST
We first separate the computational aspects of our scheme into two parts and discuss them separately:
(a) constructing new basis in the first session, (b) finetuning/inference in the incremental sessions.
(a) After pretraining the model in the first session, we construct new basis by using singular value decomposition (SVD).
The computational complexity of SVD is known as O ( n 3 ) where the covariance of activation is in the shape of n × n .
Although constructing the new basis appears to have somewhat large complexity due to the SVD, we only need to construct it just once in the first session.
Moreover, the real time taken to construct the basis is not that significant compared to that of pretraining in the first session in practice.
The table 17 shows the elapsed times for constructing the basis and pretraining with only 5 epochs on miniImageNet dataset.
We tested on NVIDIA GeForce RTXGPU.
(b) During finetuning in the new basis, the complexity induced in the layer, to which the WaRP is applied, increases as much as when training with 3 layers due to V and U multiplied on both sides of
[Table] [Table]
W in practical implementation.
However, this difference does not that significantly slow down the finetuning as the model is finetuned only with a small iterations using only a few labeled samples in the actual incremental session.
The table 18 shows the elapsed times for finetuning in both new and original space on miniImageNet dataset (we report averaged time per each incremental session).
We tested on NVIDIA GeForce RTX 3090 GPU.
Moreover, we would like to highlight that, after completing the finetuning, we recover this layer to its original one, i.e., W = V (cid:102) WU ⊤ so that the inference time complexity does not increase.
A. N OTES ON THE S PACE OF M ATRIX
In equation (4), we deal with the “space of matrix” and generalize the linear algebraic concepts (which we are familiar with in space R k , for example) to this matrix space.
Here, we define the inner product as ⟨ A, B ⟩ = trace ( A ⊤ B ) in Sec. 2.
In fact, this inner product actually can be seen as a typical dot product in the vector space R k for any k (i.e. a · b = (cid:80) ki
=1 a i b i for any vectors a , b ∈ R k where a i and b i are i -th element of a and b respectively), if A, B ∈ R m × n are flattened as a vector form.
In other words, it can be easily seen that ⟨ A, B ⟩ =
A flatten · B flatten where A flatten , B flatten ∈ R nm is flattened version of A and B respectively.
Thus, once W and K ij are flattened, we can easily interpret the equation (4) as the flattened version of W and K ij are in the vector space R nm and the linear algebraic concepts we are familiar with in R k such as linear combination, orthonormal vectors are directly adopted to this equation.
[Table] [Table]