S ELF - TRAINING F OR F EW - SHOT T RANSFER A CROSS E XTREME T ASK D IFFERENCES
Anonymous authors
Paper under double-blind review
A BSTRACT
Most few-shot learning techniques are pre-trained on a large, labeled “base dataset”.
In problem domains where such large labeled datasets are not available for pre-training (e.g., X-ray, satellite images), one must resort to pre-training in a different “source” problem domain (e.g., ImageNet), which can be very different from the desired target task.
Traditional few-shot and transfer learning techniques fail in the presence of such extreme differences between the source and target tasks.
In this paper, we present a simple and effective solution to tackle this extreme domain gap: self-training a source domain representation on unlabeled data from the target domain.
We show that this improves one-shot performance on the target domain by 2.9 points on average on the challenging BSCD-FSL benchmark consisting of datasets from multiple domains.
I NTRODUCTION
Despite progress in visual recognition, training recognition systems for new classes in novel domains requires thousands of labeled training images per class.
For example, to train a recognition system for identifying crop types in satellite images, one would have to hire someone to go to the different locations on earth to get the labels of thousands of satellite images.
The high cost of collecting annotations precludes many downstream applications.
This issue has motivated research on few-shot learners : systems that can rapidly learn novel classes from a few examples .
However, most few-shot learners are trained on a large base dataset of classes from the same domain.
This is a problem in many domains (such as medical imagery, satellite images), where no large labeled dataset of base classes exists.
The only alternative is to train the fewshot learner on a different domain (a common choice is to use ImageNet).
Unfortunately, few-shot learning techniques often assume that novel and base classes share modes of variation (Wang et al., 2018), class-distinctive features (Snell et al., 2017), or other inductive biases.
These assumptions are broken when the difference between base and novel is as extreme as the difference between object classiﬁcation in internet photos and pneumonia detection in X-ray images.
As such, recent work has found that all few-shot learners fail in the face of such extreme task/domain differences, underperforming even naive transfer learning from ImageNet (Guo et al., 2020).
Another alternative comes to light when one considers that many of these problem domains have unlabeled data (e.g., undiagnosed X-ray images, or unlabeled satellite imagery).
This suggests the possibility of using self-supervised techniques on this unlabeled data to produce a good feature representation, which can then be used to train linear classiﬁers for the target classiﬁcation task using just a few labeled examples.
Indeed, recent work has explored self-supervised learning on a variety of domains (Wallace & Hariharan, 2020).
However, self-supervised learning starts tabula rasa , and as such requires extremely large amounts of unlabeled data (on the order of millions of images).
With more practical unlabeled datasets, self supervised techniques still struggle to outcompete
naive
ImageNet transfer (Wallace & Hariharan, 2020).
We are thus faced with a conundrum: on the one hand, few-shot learning techniques fail to bridge the extreme differences between ImageNet and domains such as X-rays.
On the other hand, self-supervised techniques fail when they ignore inductive biases from ImageNet.
A sweet spot in the middle, if it exists, is elusive.
In this paper, we solve this conundrum by presenting a strategy that adapts feature representations trained on source tasks to extremely different target domains, so that target task classiﬁers can then be trained on the adapted representation with very little labeled data.
Our key insight is that a pre-trained base classiﬁer from the source domain, when applied to the target domain, induces a grouping of images on the target domain.
This grouping captures what the pre-trained classiﬁer thinks are similar or dissimilar in the target domain.
Even though the classes of the pre-trained classiﬁer are themselves irrelevant in the target domain, the induced notions of similarity and dissimilarity might still be relevant and informative.
This induced notion of similarity is in contrast to current self-supervised techniques which often function by considering each image as its own class and dissimilar from every other image in the dataset (Wu et al., 2018; Chen et al., 2020).
We propose to train feature representations on the novel target domain to replicate this induced grouping .
This approach produces a feature representation that is (a) adapted to the target domain, while (b) maintaining prior knowledge from the source task to the extent that it is relevant.
A discerning reader might observe the similarity of this approach to self-training , except that our goal is to adapt the feature representation to the target domain, rather than improve the base classiﬁer itself.
We call our approach “ S elf T raining to A dapt R epresentations T
o U nseen P roblems”, or STARTUP .
In a recently released BSCD-FSL benchmark consisting of datasets from extremely different domains (Guo et al., 2020), we show that STARTUP provides signiﬁcant gains (up to 2.9 points on average) over few-shot learning, transfer learning and self-supervision state-of-the-art.
To the best of our knowledge, ours is the ﬁrst attempt to bridge such large task/domain gaps and successfully and consistently outperform naive transfer in cross-domain few-shot learning.
P ROBLEM S ETUP
Our goal is to build learners for novel domains that can be quickly trained to recognize new classes when presented with very few labeled data points ( “few-shot” ).
Formally, the target domain is deﬁned by a set of data points (e.g. images) X N , an unknown set of classes (or label space) Y N , and a distribution D N over X N × Y N .
A “few-shot learning task” in this domain will consist of a set of classes Y ⊂
Y N , a very small training set (“support”)
[Equation]
and a small test set (“query”)
[Equation]
When presented with such a few-shot learning task, the learner must rapidly learn the classes presented and accurately classify the query images.
As with prior few-shot learning work, we will assume that before being presented with few-shot learning tasks in the target domain, the learner has access to a large annotated dataset D B known as the base dataset.
However, crucially unlike prior work on few-shot learning , we assume that this base dataset is drawn from a very different distribution.
In fact, we assume that the base dataset is drawn from a completely disjoint image space X B and a disjoint set of classes Y B :
[Equation]
where X B is the set of data (or the source domain) and Y B is the set of base classes.
Because the base dataset is so different from the target domain, we introduce another difference vis-a-vis the conventional few-shot learning setup: the learner is given access to an additional unlabeled dataset from the target domain:
[Equation]
Put together, the learner will undergo two phases.
In the representation learning phase, the learner will pre-train its representation on D B and D u ; then it goes into the evaluation phase where it will be presented few-shot tasks from the target domain where it learns the novel classes (Figure 1).
R ELATED W ORK
Few-shot Learning (FSL).
This paper explores few-shot transfer, and as such the closest related work is on few-shot learning.
Few-shot learning techniques are typically predicated on some degree of similarity between classes in the base dataset and novel classes.
For example, they may assume that features that are discriminative for the base classes are also discriminative for the novel classes, suggesting a metric learning-based approach (Gidaris & Komodakis, 2018;
Qi et al., 2018;
Snell et al., 2017; Vinyals et al., 2016; Sung et al., 2018;
Hou et al., 2019) or transfer learning-based approach (Chen et al., 2019a; Wang et al., 2019; Kolesnikov et al., 2020; Tian et al., 2020).
Alternatively, they may assume that model initializations that lead to rapid convergence on the base classes are also good initializations for the novel classes (Finn et al., 2017; 2018; Ravi & Larochelle, 2017;
Nichol & Schulman; Rusu et al., 2019; Sun et al., 2019; Lee et al., 2019).
Other methods assume that modes of intra-class variation are shared, suggesting the possibility of learned, class-agnostic augmentation policies (Hariharan & Girshick, 2017; Wang et al., 2018;
Chen et al., 2019b).
Somewhat related is the use of a class-agnostic parametric model that can “denoise” few-shot models, be they from the base or novel classes (Gidaris & Komodakis, 2018; 2019).
In contrast to such strong assumptions of similarity between base and novel classes, this paper tackles few-shot learning problems where base and novel classes come from very different domains, also called cross-domain few-shot learning.
Cross-domain Few-shot Classiﬁcation (CD-FSL).
When the domain gap between the base and novel dataset is large, recent work (Guo et al., 2020; Chen et al., 2019a) has shown that existing stateof-the-art few-shot learners fail to generalize.
Tseng et al.
(2020) attempt to address this problem by simulating cross-domain transfer during training.
However, their approach assumes access to an equally diverse array of domains during training, and a much smaller domain gap at test time: for example, both base and novel datasets are from internet images.
Another relevant work (Ngiam et al., 2018) seeks to build domain-speciﬁc feature extractor by reweighting different classes of examples in the base dataset based on the target novel dataset but their work only investigates transfer between similar domains (both source and target are internet images).
Our paper tackles a more extreme domain gap.
Another relevant benchmark for this problem is (Zhai et al., 2019) but they assume access to more annotated examples (1k annotations) during test time than the usual FSL setup.
Few-shot learning with unlabeled data.
This paper uses unlabeled data from the target domain to bridge the domain gap.
Semi-supervised few-shot learning (SS-FSL) (Ren et al., 2018; Li et al., 2019; Yu et al., 2020; Rodr´ıguez et al., 2020; Wang et al., 2020) and transductive few-shot learning (T-FSL) (Liu et al., 2019; Dhillon et al., 2020; Hou et al., 2019; Wang et al., 2020; Rodr´ıguez et al., 2020) do use such unlabeled data, but only during evaluation, assuming that representations trained on the base dataset are good enough.
In contrast our approach leverages the unlabeled data during representation learning.
The two are orthogonal innovations and can be combined.
Self-Training.
Our approach is closely related to self-training, which has been shown to be effective for semi-supervised training and knowledge distillation.
In self-training, a teacher model trained on the labeled data is used to label the unlabeled data and another student model is trained on both the original labeled data and the unlabeled data labeled by the teacher.
Xie et al.
(2020) and Yalniz et al.
(2019) have shown that using self-training can improve ImageNet classiﬁcation performance.
Knowledge distillation (Hinton et al., 2015) is similar but aims to compress a large teacher network by training a student network to mimic the prediction of the teacher network.
A key difference between these and our work is that self-training / knowledge distillation focus on a single task of interest, i.e, there is no change in label space.
Our approach is similar, but we are interested in transferring to novel domains with a wholly different label space : an unexplored scenario.
Domain Adaptation.
Transfer to new domains is also in the purview of domain adaptation (Tzeng et al., 2017; Hoffman et al., 2018;
Long et al., 2018; Xu et al., 2019; Laradji & Babanezhad, 2020;
Wang & Deng, 2018; Wilson & Cook, 2020) where the goal is to transfer knowledge from the labelabundant source domain to a target domain where only unlabeled data is available.
However, a key assumption in domain adaptation is that the source domain and target domain share the same label space which does not hold for few-shot learning.
Self-supervised Learning.
Learning from unlabeled data has seen a resurgence of interest with advances in self-supervised learning.
Early self-supervised approaches were based on handcrafted “pretext tasks” such as solving jigsaw puzzles (Noroozi & Favaro, 2016), colorization (Zhang et al., 2016) or predicting rotation (Gidaris et al., 2018).
A more recent (and better performing) line of self-supervised learning is contrastive learning (Wu et al., 2018; Misra & Maaten, 2020;
He et al., 2020; Chen et al., 2020) which aims to learn representations by considering each image together with its augmentations as a separate class.
While self supervision has been shown to boost few-shot learning (Gidaris et al., 2019; Su et al., 2020), its utility in cases of large domain gaps between base and novel datasets have not been evaluated.
Our work focuses on this challenging scenario.
A PPROACH
Consider a classiﬁcation model f θ =
C ◦ φ where φ embeds input x into R d and C is a (typically linear) classiﬁer head that maps φ ( x ) to predicted probabilities P ( y | x ) .
θ is a vector of parameters.
During representation learning, STARTUP performs the following three steps:
Learn a teacher model θ 0 on the base dataset D B by minimizing the cross entropy loss 2.
Use the teacher model to construct a softly-labeled set D ∗ u
[Equation]
[Equation] [Equation]
Note
that ¯
y i
is
a
probability
distribution
as described above
Learn a new student model θ ∗ on D B and D ∗ u by optimizing:
[Equation]
where l CE is the cross entropy loss, l KL is the KL divergence and l unlabeled is any unsupervised/self-supervised loss function (See below).
The third term, l unlabeled , is intended to help the learner extract additional useful knowledge speciﬁc to the target domain.
We use a state-of-the-art self-supervised loss function based on contrastive learning: SimCLR (Chen et al., 2020).
The SimCLR loss encourages two augmentations of the same image to be closer in feature space to each other than to other images in the batch.
We refer the reader to the paper for the detailed loss formulation.
The ﬁrst two terms are similar to those in prior self-training literature (Xie et al., 2020).
However, while in prior self-training work, the second term ( l KL ) is thought to mainly introduce noise during training, we posit that l KL has a more substantial role to play here: it encourages the model to learn feature representations that emphasize the groupings induced by the pseudo-labels ¯
y i on the target domain.
We analyze this intuition in section 5.2.2.
4. E VALUATION
STARTUP is agnostic to inference methods during evaluation; any inference methods that rely on a representation (Snell et al., 2017; Gidaris & Komodakis, 2018) can be used with STARTUP .
For simplicity and based on results reported by Guo et al.
(2020), we freeze the representation φ after performing STARTUP and train a linear classiﬁer on the support set and evaluate the classiﬁer on the query set.
4. I NITIALIZATION S TRATEGIES
Xie et al.
(2020) found that training the student from scratch sometimes yields better results for ImageNet classiﬁcation.
To investigate, we focused on a variant of STARTUP where the SimCLR loss is omitted and experimented with three different initialization strategies - from scratch ( STARTUPRand (no SS)), from teacher ( STARTUP - T (no SS)) and using the teacher’s embedding with randomly initialized classiﬁer ( STARTUP (no SS)).
We found no conclusive evidence that one single initialization strategy is superior to the others across different datasets (See Appendix A.4) but we observe that ( STARTUP (no SS)) is either the best or the second best in all scenarios.
As such, we opt to use teacher’s embedding with a randomly initialized classiﬁer as the default student initialization.
E XPERIMENTS
We defer the implementation details to Appendix A.1.
5. F EW - SHOT T RANSFER A CROSS D RASTICALLY D IFFERENT D OMAINS
Benchmark.
We experiment with the challenging (BSCD-FSL) benchmark introduced in Guo et al.
The base dataset in this benchmark is miniImageNet (Vinyals et al., 2016), which is an object recognition task on internet images.
There are 4 novel datasets in the benchmark, none of which involve objects, and all of which come from a very different domain than internet images:
CropDiseases (recognizing plant diseases in leaf images), EuroSAT (predicting land-use from satellite images), ISIC2018 (identifying melanoma from images of skin lesions) and ChestX (diagnosing chest X-rays).
Guo et al.
found that state-of-the-art few-shot learners fail on this benchmark.
To construct our setup, we randomly sample 20% of data from each novel datasets to form the respective unlabeled datasets D u .
We use the rest for sampling tasks for evaluation.
Following Guo et al.
(2020), we evaluate 5-way k-shot classiﬁcation tasks (the support set consists of 5 classes and k examples per class) for k ∈ { 1 , 5 } and report the mean and 95% conﬁdence interval overfew-shot tasks (conclusions generalize to k ∈ { 20 , 50 } .
See Appendix A.2).
Baselines.
We compare to the techniques reported in Guo et al.
(2020), which includes most stateof-the-art approaches as well as a cross-domain few-shot technique Tseng et al.
The top performing among these is naive Transfer which simply trains a convolutional network to classify the base dataset, and uses the resulting representation to learn a linear classiﬁer when faced with novel few-shot tasks.
These techniques do not use the novel domain unlabeled data.
We also compare to another baseline, SimCLR that uses the novel domain unlabeled data D u to train a representation using SimCLR(Chen et al., 2020), and then uses the resulting representation to learn linear classiﬁers for few-shot tasks.
This builds upon state-of-the-art self-supervised techniques.
To compare to a baseline that uses both sources of data, we establish Transfer + SimCLR .
This baseline is similar to the SimCLR baseline except the embedding is initialized to Transfer’s embedding before SimCLR training.
Following the benchmark, all methods use a ResNet-10 (He et al., 2016) unless otherwise stated.
5.1. R ESULTS
We present our main results on miniImageNet → BSCD-FSL in Table 1.
STARTUP vs Few-shot learning techniques.
STARTUP performs signiﬁcantly better than all fewshot techniques in most datasets (except ChestX, where all methods are similar).
Compared to previous state-of-the-art, Transfer , we observe an average of 2.9 points improvement on the 1-shot case.
The improvement is particularly large on CropDisease, where STARTUP provides almost apoint increase for 1-shot classiﬁcation.
This improvement is signiﬁcant given the simplicity of our approach, and given that all meta-learning techniques underperform this baseline.
. Methods
[Table]
EuroSAT
CropDisease
[Table]
MAML
[Table]
[Table] [Table]
[Table] [Table]
ProtoNet
[Table] [Table]
[Table] [Table]
[Table]
ProtoNet + FWT
[Table] [Table]
[Table] [Table]
MetaOpt
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
Transfer 60.
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table]
+ SimCLR 57.
[Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
(no
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table] [Table]
[Table]
STARTUP vs SimCLR.
The SimCLR baseline in general tends to underperform naive transfer from miniImageNet, and consequently, STARTUP performs signiﬁcantly better than SimCLR on ISIC and EuroSAT.
The exception to this is CropDisease, where SimCLR produces a surprisingly good representation.
We conjecture that the base embedding is not a good starting point for this dataset.
However, we ﬁnd that using SimCLR as an auxilliary loss to train the student ( STARTUP vs STARTUP (no SS)) is beneﬁcial.
STARTUP vs Transfer + SimCLR.
STARTUP outperforms Transfer + SimCLR in most cases (except 5-shot in ChestX and 1-shot in ISIC).
We stress that the strength of STARTUP is not solely from
SimCLR but rather from both self-training and SIMCLR.
This is especially evident in EuroSAT since the STARTUP (no SS) variant outperforms Transfer and Transfer + SimCLR .
Larger and stronger teachers.
To unpack the impact of teacher quality, we experiment with a larger network and transfer from the full ILSVRC 2012 dataset (Deng et al., 2009) to BSCD-FSL.
In particular, we used the publicly available pre-trained ResNet-18 (He et al., 2016) as a teacher and train a student via STARTUP .
We compare this to a transfer baseline that uses the same network and ImageNet as the training set.
The result can be found in table 2.
Surprisingly, larger, richer embeddings do not always transfer better , in contrast to in-domain results reported by Hariharan &
Girshick (2017).
However, STARTUP is still useful in improving performance: the absolute improvement in performance for STARTUP compared to Transfer remains about the same in most datasets except EuroSAT and CropDisease where larger improvements are observed.
5. W HY SHOULD STARTUP WORK ?
While it is clear that STARTUP helps improve few shot transfer across extreme domain differences, it is not clear why or how it achieves this improvement.
Below, we look at a few possible hypotheses.
5.2. H YPOTHESIS 1: STARTUP ADDS NOISE WHICH INCREASES ROBUSTNESS .
Xie et al.
(2020) posit that self-training introduces noise when training the student and thus yielding a more robust student.
More robust students may be learning more generalizable representations, and this may be allowing STARTUP to bridge the domain gap.
Under this hypothesis, the function of the unlabeled data is only to add noise during training.
This in turn suggests that STARTUP should yield improvements on the target tasks even if trained on unlabeled data from a different domain .
To test this, we train a STARTUP ResNet-18 student on EuroSAT and ImageNet and evaluate it on CropDisease.
This model yields a 5-way 1-shot performance of 70.40 ± 0.86 (88.78 ± 0.
for 5-shot), signiﬁcantly underperforming the naive Transfer baseline (Table 2 .
See Appendix A. for different combinations of unlabeled dataset and target dataset).
This suggests that while the hypothesis is valid in conventional semi-supervised learning, it is incorrect in the cross-domain few-shot learning setup: unlabeled data are not merely functioning as noise .
Rather, STARTUP [Table] is learning inherent structure in the target domain useful for downstream classiﬁcation.
The question now becomes what inherent structure STARTUP is learning, which leads us to the next hypothesis.
5.2. H YPOTHESIS 2: STARTUP ENHANCES TEACHER - INDUCED GROUPINGS
The teacher produces a meaningful grouping of the data from the target domain.
The predictions made by the teacher essentially induce a grouping on the target domain.
Even though the base label space and novel label space are disjoint, the groupings produced by the teacher might not be entirely irrelevant for the downstream classiﬁcation task.
To test this, we ﬁrst assign each example in the novel datasets to its most probable prediction by the teacher (ResNet 18 trained on ImageNet).
We then compute the adjusted mutual information (AMI) (Vinh et al., 2010) between the resulting grouping and ground truth label.
AMI ranges from 0 for unrelated groupings to 1 for identical groupings.
From Table 3, we see that on EuroSAT and CropDisease, there is quite a bit of agreement between the induced grouping and ground truth label.
Interestingly, these are the two datasets where we observe the best transfer performance and most improvement from STARTUP (Table 2), suggesting correlations between the agreement and the downstream classiﬁcation task performance.
STARTUP enhances the grouping induced by the teacher.
Even though the induced groupings by the teacher can be meaningful, one could argue that those groupings are captured in the teacher model already, and no further action to update the representation is necessary.
However, we posit that STARTUP encourages the feature representations to emphasize the grouping.
To verify, we plot the t-SNE (Maaten & Hinton, 2008) of the data prior to STARTUP and after STARTUP for the two datasets in ﬁgure 2.
From the t-SNE plot, we observe more separation after doing STARTUP , signifying a representation with stronger discriminability.
Put together, this suggests that STARTUP works by (a) inducing a potentially meaningful grouping on the target domain data, and (b) training a representation that emphasizes this grouping.
5. F EW - SHOT T RANSFER A CROSS S IMILAR D OMAINS
Is STARTUP still useful when the gap between the base and target is smaller?
To answer this, we tested STARTUP on two popular within-domain few-shot learning benchmark: miniImageNet (Vinyals et al., 2016) and tieredImageNet (Ren et al., 2018).
For miniImageNet, we use 20% of the
[Table]
novel set as the unlabeled dataset and use the same teacher as in section 5.1. For tieredImageNet, we use ResNet-12 (Oreshkin et al., 2018) as our model architecture and evaluate two different setupstieredImageNet-less that uses 10% of the novel set as unlabeled data (following Ren et al.
and tieredImageNet-more that uses 50% of the novel set as unlabeled data.
We follow the same evaluation protocols in section 5.1.
We report the results in table 4.
We found that on miniImageNet, STARTUP and its variants neither helps nor hurts in most cases (compared to Transfer ), indicating that the representation is already well-matched.
On both variants of tieredImageNet, we found that STARTUP, with the right initialization, can in fact outperform Transfer .
In particular, in the less data case, it is beneﬁcial to initialize the student with the teacher model whereas in the more data case, training the students from scratch is superior.
In sum, these results show the potential of STARTUP variants to boost few-shot transfer even when the base and target domains are close.
Additional Ablation Studies: We conducted two additional ablation studies: (a) training the student with various amount of unlabeled data and (b) training the student without the base dataset.
We show that STARTUP beneﬁts from more unlabeled data (Appendix A.5) and training student without the base dataset can hurt performance in certain datasets but not all datasets (Appendix A.6).
C ONCLUSION
We investigate the use of unlabeled data from novel target domains to mitigate the performance degradation of few-shot learners due to large domain/task differences.
We introduce STARTUP - a simple yet effective approach that allows few-shot learners to adapt feature representations to the target domain while retaining class grouping induced by the base classiﬁer.
We show that STARTUP outperforms prior art on extreme cross-domain few-shot transfer.
A A PPENDIX A. I MPLEMENTATION D ETAILS
We implemented STARTUP by modifying the the publicly-available implementation 1 of BSCD-FSL by Guo et al.
A.1. T RAINING THE T EACHER
MiniImageNet: We train the teacher model using the code provided in the BSCD-FSL benchmark.
We keep everything the same except setting the batch size from 16 to 256. 2.
TieredImageNet:
We used the same setup as miniImageNet except we reduce the number of epochs to 90.
We do not use any image augmentation for tieredImageNet.
ImageNet: We used the pretrained ResNet18 available on PyTorch (Paszke et al., 2019)
A.1. T RAINING THE S TUDENT
Optimization Details.
Regardless of the base and novel datasets, the student model is trained for 1000 epochs where an epoch is deﬁned to be a complete pass on the unlabeled data.
We use a batch size of 256 on the unlabeled dataset and a batch size of 256 for the base dataset if applicable.
We use the SGD with momentum optimizer with momentum 0.9 and weight decay 1e-4.
To pick the suitable starting learning rate, 10% of the unlabeled data and 5% of the labeled data (1% when using ImageNet as the base dataset) are set aside as our internal validation set.
We pick the starting learning rate by training the student with starting learning rate lr ∈ { 1e-1, 5e-2, 3e-2, 1e-2, 5e-3, 3e-3, 1e-3 } for k epochs where k is the smallest epoch that guarantees at least 50 updates to the model and pick the learning rate that yields lowest loss on the validation set as the starting learning rate.
We reduce the learning rate by a factor of 2 when the training loss has not decreased byepochs.
The model that achieves the lowest loss on the internal validation set throughout theepochs of training is picked as the ﬁnal model.
SimCLR.
Our implementation of SimCLR’s loss function is based on a publicly available implementation of SimCLR 2 .
We added the two-layer projection head on top of the embedding function φ .
The temperature of NT-Xent is set to 1 since there is no validation set for BSCD-FSL for hyperparameter selection and we use a temperature of 1 when inferring the soft label of the unlabeled set.
For the stochastic image augmentations for SimCLR, we use the augmentations deﬁned for each novel dataset in Guo et al.
These augmentations include the commonly used “randomly resized crop”, color jittering, random horizontal ﬂipping.
For tieredImageNet and miniImageNet, we use the stochastic transformation implemented for the BSCD-FSL benchmark.
We refer readers to the BSCD-FSL implementation for more details.
When training the student on the base dataset, we use the augmentation used for training the teacher for fair comparison.
The batchsize for SIMCLR is set to 256.
A.1. T RAINING L INEAR C LASSIFIER .
We use the implementation by BSCD-FSL, i.e training the linear classiﬁer with standard cross entropy and SGD optimizer.
The linear classiﬁer is trained for 100 epochs with learning rate 0.01, momentum 0.9 and weight decay 1e-4.
A.1. B ASELINES
We use the same evaluation methods - linear classiﬁer.
Please see A.1.3 for classiﬁer training.
Transfer.
This is implemented using the teacher model as feature extractor.
Please see A.1.1 for details.
SimCLR.
This is implemented similarly to the SimCLR loss described in A.
. A.1. T -SNE
We use the publicly available scikit-learn implementation of t-SNE (Buitinck et al., 2013).
We used the default parameters except for the perplexity where we set to 50.
To speed up the experiment, we randomly sampled 25% of the data used for sampling few-shot tasks (80 % of the full dataset) and run t-SNE on this subset.
A. F ULL R ESULTS ON BSCD-FSL
We present the result on miniImageNet → BSCD-FSL for shot = 1, 5, 20, 50 in Table 5 and 6.
In addition to STARTUP, we also reported results on using teacher model as student initialization ( STARTUP -T and STARTUP - T (no SS)) in these tables for reference.
Results on ImageNet → BSCDFSL can be found in Table 7 and 8.
The conclusions we found in 5.1 still hold for higher shots in general.
A. U SING R OTATION FOR S ELF - SUPERVISION .
We use rotation (Gidaris et al.
(2018)) instead of SimCLR in STARTUP and report the results in in Table 5 and 6.
We observe that STARTUP (w/ Rotation) is able to outperform Transfer in CropDisease and EuroSAT but generally underperforms its SimCLR counterparts.
A. I NITIALIZATION S TRATEGIES FOR THE S TUDENT
We investigate the impact of different initialization strategies for the student on STARTUP.
For this experiment, we remove SimCLR from STARTUP and consider three initialization strategies for the student - from scratch ( STARTUP -Rand (no SS)), from teacher embedding with a randomly
Methods
[Table]
STARTUP - T 22.75 ± 0.
26.47 ± 0.
32.16 ± 0.
45.75 ± 0.
STARTUP 23.09 ± 0.
26.94 ± 0.
32.66 ± 0.
47.22 ± 0.
STARTUP (w/ Rotation) 22.83 ± 0.
26.38 ± 0.
31.54 ± 0.
45.68 ± 0.
Methods
EuroSAT
CropDisease
[Table]
MAML
[Table]
- 71.70 ± 0.
- 78.05 ± 0.
ProtoNet ∗ - 73.29 ± 0.
- 79.72 ± 0.
ProtoNet + FWT ∗ - 67.34 ± 0.
- 72.72 ± 0.
MetaOpt ∗ - 64.44 ± 0.
- 68.41 ± 0.
Transfer ∗ - 75.69 ± 0.
- 87.48 ± 0.
Transfer 60.73 ± 0.
80.30 ± 0.
69.97 ± 0.
90.16 ± 0.
SimCLR 43.52 ± 0.
59.05 ± 0.
78.23 ± 0.
92.57 ± 0.
Transfer + SimCLR 57.18 ± 0.
77.61 ± 0.
76.90 ± 0.
92.64 ± 0.
STARTUP - T (no SS) 63.00 ± 0.
81.25 ± 0.
71.11 ± 0.
90.79 ± 0.
STARTUP (no SS) 62.90 ± 0.
81.81 ± 0.
73.30 ± 0.
91.69 ± 0.
STARTUP - T 63.49 ± 0.
81.54 ± 0.
72.85 ± 0.
91.49 ± 0.
STARTUP 63.88 ± 0.
82.29 ± 0.
75.93 ± 0.
[Table] [Table]
± 0.
57.41 ± 0.
62.71 ± 0.
STARTUP - T 32.79 ± 0.
36.66 ± 0.
56.43 ± 0.
61.76 ± 0.
STARTUP 33.19 ± 0.
36.91 ± 0.
58.63 ± 0.
64.16 ± 0.
STARTUP (w/ Rotation) 31.94 ± 0.
36.33 ± 0.
57.97 ± 0.
63.44 ± 0.
Methods
EuroSAT
CropDisease k= k= k= k=
MAML ∗ 81.95 ± 0.
- 89.75 ± 0.
ProtoNet ∗ 82.27 ± 0.
80.48 ± 0.
88.15 ± 0.
90.81 ± 0.
ProtoNet + FWT ∗ 75.74 ± 0.
78.64 ± 0.
85.82 ± 0.
87.17 ± 0.
MetaOpt ∗ 79.19 ± 0.
83.62 ± 0.
82.89 ± 0.
91.76 ± 0.
Transfer ∗ 84.13 ± 0.
86.62 ± 0.
94.45 ± 0.
96.62 ± 0.
Transfer 88.31 ± 0.
91.09 ± 0.
96.10 ± 0.
97.69 ± 0.
SimCLR 72.25 ± 0.
78.64 ± 0.
97.26 ± 0.
98.44 ± 0.
Transfer + SimCLR 87.48 ± 0.
91.43 ± 0.
97.41 ± 0.
98.44 ± 0.
STARTUP - T (no SS) 88.44 ± 0.
91.13 ± 0.
96.31 ± 0.
97.80 ± 0.
STARTUP (no SS) 89.29 ± 0.
91.94 ± 0.
96.91 ± 0.
98.20 ± 0.
[Table]
initialized classiﬁer ( STARTUP (no SS)), from teacher model ( STARTUP - T (no SS)). We repeated the experiment in section 5.1 on miniImageNet → BSCD-FSL and report the results in table 9. We found that not a single initialization is superior to the others (for instance random initialization is the best on CropDisease but the worst on ISIC) however we did ﬁnd that initializing the student with the teacher’s embedding with a randomly initialized classiﬁer for STARTUP is either the best or second best in all scenarios so we set that as our default initialization. [Table]
[Table]
A. I MPACT OF D IFFERENT A MOUNT OF U NLABELED E XAMPLES
STARTUP uses unlabeled data to adapt feature representations to novel domains.
As with all learning techniques, it should perform better with more unlabeled data.
To investigate how the amount of unlabeled examples impacts STARTUP , we repeated the miniImageNet → ISIC experiments in 5.1 with various amount of unlabeled data (20% of the dataset (2003 examples) is set aside for evaluation).
The verdict is clear - STARTUP beneﬁts from more unlabeled data (Figure 3).
A. T RAINING THE S TUDENT WITHOUT THE B ASE D ATASET
STARTUP requires joint training on both the base dataset as well as the target domain.
But in many cases, the base dataset may not be available.
Removing the cross entropy loss on the base dataset when training the student essentially boils down to a ﬁne-tuning paradigm.
For miniImageNet →
BSCD-FSL (Table 10), we found no discernible difference between all datasets except on the ISIC where we observe signiﬁcant degradation in 5-shot performance. . Methods
[Table]
Methods [Table] [Figure]
[Table] [Figure] [Figure] D IFFERENT U NLABELED D
[Figure]
We consider the ImageNet → CD-FSL experiment.
We perform STARTUP on unlabeled data different from the target domain and present the result in Table 11.
We found that it is crucial that the unlabeled data to perform STARTUP on should be from the target domain of interest.
[Table] [Table]