P ATCH -L EVEL C ONTRASTING WITHOUT P ATCH C ORRESPONDENCE FOR A CCURATE AND D ENSE C ONTRASTIVE R EPRESENTATION L EARNING
Anonymous authors
Paper under double-blind review
A BSTRACT
We propose ADCLR
Accurate and Dense Contrastive Representation Learning
, a novel self-supervised learning framework for learning accurate and dense vision representation.
To extract spatial-sensitive information, ADCLR introduces query patches for contrasting in addition with global contrasting.
Compared with previous dense contrasting methods
ADCLR mainly enjoys three merits i) achieving both global-discriminative and spatial-sensitive representation ii) modelefficient (no extra parameters in addition to the global contrasting baseline), and iii) correspondence-free and thus simpler to implement.
Our approach achieves new state-of-the-art performance for contrastive methods.
On classification tasks for
ViT-S
ADCLR achieves 78.1% top-1 accuracy on ImageNet with linear probing outperforming our baseline (DINO) without our devised techniques as plug-in by
For ViT-B
ADCLR achieves 79.8%,
0% accuracy on ImageNet by linear probing and finetune, outperforming DINO by 0.6%
4% accuracy
For dense tasks on MS
COCO
ADCLR achieves significant improvements of
AP on object detection
AP on instance segmentation outperforming previous
SOTA method SelfPatch by 2.2% and 1.2%, respectively.
On ADE20K
ADCLR outperforms SelfPatch by 1.0% mIoU
1.2% mAcc on the segmentation task
I NTRODUCTION
Self-supervised representation learning (SSL) has been attracting increasing attention for deep learning, whereby a prediction problem is often formulated by a pretext task for pre-training with unlabeled data.
SSL methods can mainly be divided into three categories: 1) Generative approaches (Goodfellow et al., 2014) learn to generate samples in the input space.
However, generation can be computationally expensive and may not be necessary for representation learning.
Contextual methods (Gidaris et al., 2018) design pretext tasks (denoising auto-encoders (Vincent et al., 2008), context auto encoders (Zhang et al., 2016), etc).
3) Contrastive methods (Chen et al., 2020a; Zhang et al., 2022; Chen et al., 2021; Caron et al., 2021) take augmented views of the same image as positive pairs and others as negative pairs.
Contrastive-based methods have shown great promise e.g. in image classification/detection, video classification (Caron et al., 2021), and others (Chen et al., 2021).
It has been recently shown (Chen et al., 2020a;
Wang et al., 2021) that existing contrastive learning in general aims to learn global-discriminative features, which may lack spatial sensitivity (Yi et al., 2022), and it limits their ability on downstream fine-tuning tasks, especially for dense vision tasks like detection and segmentation.
Consequently, object-level (Wei et al., 2021; Hénaff et al., 2022) and pixel-level (Xie et al., 2021c; Wang et al., 2021) contrastive objectives and frameworks are proposed.
Meanwhile, with the success of recent ViT-based visual backbones (Dosovitskiy et al., 2020; Liu et al., 2021), patch-level contrastive approaches (Yun et al., 2022) are devised, which achieve state-of-the-art performance on downstream dense tasks.
However, there are mainly three disadvantages in these dense contrasting methods.
i)
It is hard to balance the global and patchlevel losses in the dense contrasting methods (Xie et al., 2021c; Wang et al., 2021), causing their less competitive linear/fine-tune accuracy on the global classification task.
ii) Establishing the correspondence among pixels/patches usually requires bilinear interpolation, which is complex and heavily sensitive to random crop augmentation (in an extreme case, if two views have no intersection parts, there’s no correspondence relation).
iii) each corresponding pixel/patch need be involved in the final contrastive loss, which is time-consuming.
In this paper, we propose Accurate and Dense Contrastive Representation Learning (ADCLR), which is more global-discriminative, spatial-sensitive, correspondence-free and efficient.
The main contributions of ADCLR are: 1) Cross-view Query-based Patch-level Contrasting Paradigm:
For patch-level contrasting as recently used for dense tasks in negative-free Transformer-based SSL methods, we propose to augment two views, and perform contrasting crops from different views, for more effective learning with increased contrasting difficulty.
The motivation for our cross-view design instead of the commonlyused single-view in existing patch-level contrasting is: it is non-triv and even impossible to establish patch correspondence within a single view (especially adding random resized crop augmentation).
While introducing two views and replacing the correspondence establishing with more feasible query could meanwhile increase the patch appearance variance for more difficult contrasting.
The above module can be introduced to existing global contrasting only SSL baselines.
As shown in Fig.
1, the [ CLS ] tokens are used to extract global information, and the designed query patches are used to extract local information, making ADCLR both global-discriminative and spatial-sensitive.
2) Robust Unidirectional Cross-Attention Scheme under the above Paradigm: The above patchlevel contrasting paradigm can technically be prone to collapse to a trivial solution 1 (see more explanation in our theoretical analysis in Sec. 3.3) if we directly resort to the unidirectional selfattention scheme as used in the vanilla vision Transformers.
Instead, we design unidirectional cross-attention, which takes both query patches and raw patches from raw images as input.
For attention block, the data flow are: RP → RP , { RP, QP i } → QP i , QP i ↛
QP
j
( i ̸ = j ) where RP and QP i means raw patches (including [ CLS ] token) and the i -th query patch, respectively.
3) Boosting baseline DINO to new SOTA accuracy on classification and dense tasks: The proposed ADCLR can serve as a plugin based on existing Transformer-based and negative-free SSL baselines e.g. DINO (Caron et al., 2021).
Our experimental results on both linear probing, finetune classification as well as other downstream tasks show the effectiveness of ADCLR.
Specifically, for ViT-S, ADCLR outperforms DINO by 1.1% accuracy under linear probing protocols.
On
MSCOCO, ADCLR outperforms SelfPatch (Yun et al., 2022) by 2.2% and 1.2% mAP on detection and segmentation tasks.
For ViT-B, ADCLR achieves 79.8%, 84.0% accuracy on ImageNet by linear probing and finetuning, outperforming DINO by 0.6%, 0.4% accuracy, respectively.
R ELATED W ORKS
In this section, we review previous literature on general contrastive learning (for global tasks such as image classification) and dense ones (for downstream tasks like detection and segmentation).
General contrastive learning aims at learning the global information of images through enforcing contrastive objectives (e.g., InfoNCE loss, Hjelm et al.
on the final global image representations ( [ CLS ] token or from average pooling).
The pivotal idea is to align the embeddings of two augmented views from the same image while preventing trivial solutions (a.k.a. degenerated representations).
To reach this target, MoCo (He et al., 2020; Chen et al., 2020b) employs a memory bank to store and update negative examples, from which the negative examples could be randomly sampled, whereas SimCLR (Chen et al., 2020a) treats all other data samples within the same training batch as negative examples.
Due to the inconvenience and huge cost of using negative examples, later research turned to explore negative-example-free methods.
BYOL (Grill et al., 2020) and SimSiam (Chen &
He, 2021) design asymmetric architectures with additional predictors and stop-gradient to prevent using negative examples.
Barlow Twins (Zbontar et al., 2021), ZeroCL (Zhang et al., 2021) and
VICReg (Bardes et al., 2022) resort to feature-level decorrelation to mitigate trivial solutions.
Inspired by the success of Vision Transformers (ViT), CNN backbones are gradually replaced with ViTs Chen et al.
; Caron et al.
iBOT
(Zhou et al., 2022) further incorporates contrastive learning with masked image modeling and has achieved State-of-the-Art performance on various tasks.
Dense contrastive learning , on the contrary, targets learning local information through regularizing different local regions of images.
One common dense contrastive learning way is to mine the correspondence of each pixel (CNNs) or patch (ViTs) in a feature map (Wang et al., 2021;
Yun et al., 2022).
DenseCL (Wang et al., 2021) exploits the correspondence through sorting the similarities of pixels, while PixPro (Xie et al., 2021c) utilizes the augmentation wrapper to get the spatial correspondence of the pixel intersection between two views in the feature map.
Furthermore,
[Figure]
Detco (Xie et al., 2021a) tries to improve the performance of general contrastive learning approaches by augmenting multiple global and local views simultaneously.
Inspired by PixPro, Resim (Xiao et al., 2021)
uses Precise RoI Pooling (Jiang et al., 2018) to extract a feature vector from the associated feature map region for both views.
On the basis of DenseCL, SetSim (Wang et al., 2022) employs a threshold selection to filter out noisy backgrounds.
With the development of ViT in SSL (Chen et al., 2021; Caron et al., 2021), SelfPatch (Yun et al., 2022) treats the spatial neighbors of the patch as positive examples for learning more semantically meaningful relations among patches.
M ETHODOLOGY
We first review recent SSL approaches with ViTs as backbones (Caron et al., 2021; Yun et al., 2022) in Sec. 3.1.
Then we present ADCLR detailedly in Sec.
3.2 (we provide a sketch in Fig. 1).
3. P RELIMINARIES
Vision Transformers.
Denote an image by x ∈ R C × H × W , where H × W is the resolution of the image and C is the number of channels.
Plain ViT (Dosovitskiy et al., 2020) treats the image x as a sequence composed of non-overlapping patches { x ( i ) ∈ R CP 2 } Ni =1 , where each patch has
a fixed
P × P resolution.
Then, the patches are linearly transformed to D -dimensional patch embeddings z ( i ) =
Ex ( i ) + W ipos ∈ R D , where E ∈ R D × CP 2 is the linear projection and W pos ∈ R D is the positional embedding for the i -th patch.
A [ CLS ] token z [ CLS ] ∈ R D is subsequently prepended to the patch sequence to extract global information, so the resulting input sequence is represented as z
=
[ z [ CLS ] , z (1) , z (2) , · · · , z ( N ) ] .
Then, ViT uses a Transformer encoder (Vaswani et al., 2017) to generate both image-level ( [ CLS ] token) and patch-level (other tokens) .
In line with SelfPatch (Yun et al., 2022), we use f θ to denote the whole process of a ViT parameterized by θ :
[Equation]
where f [ CLS ] θ ( x ) and f ( i ) θ ( x ) are the representations of the whole image and i -th patch, respectively.
Self-supervised learning with ViTs.
Since our ADCLR is built on top of DINO (Caron et al., 2021), we shortly review its framework and objective function.
Given the image x , DINO constructs a positive pair ( x A , x B ) through random augmentation.
Then, DINO learns by maximizing the similarity between their representations.
The generic form of image-level self-supervised loss is:
[Equation]
where H ( a, b )
= − a log b is the cross entropy loss.
sg ( · ) means stop-gradient operation.
g γ is the
MLP projector, which is commonly used in previous SSL methods (Chen et al., 2020a; Grill et al., 2020).
γ ′ and θ ′ denote the exponential moving averages updated parameters in the teacher branch.
3. ADCLR: A CCURATE AND D ENSE C ONTRASTIVE R EPRESENTATION L EARNING
Query crops.
Although DINO is able to handle most cases (e.g., two augmented views contain the same objects), we find that the [ CLS ] token may extract mismatched information (see the attention map of [ CLS ] in Fig. 1).
To this end, we randomly crop each random image to generate Q query crops, where Q is a pre-defined hyper-parameter.
Then, we resize each query crop to the same resolution as raw patches (e.g., 16 × 16 or 8 × 8 ): X q ∈ R Q × CP 2 .
Denote the i -th query crop as x q i , we feed each query patch into a linear projector to get its embedding z ( q i ) .
Different from raw patches, it’s unnecessary to add positional embeddings to query patches as the region of the query crop may not in the augmented views.
Then, the input embeddings can be formulated as:
[Equation]
Note that z [ CLS ] and raw patches [ z (1) , z (2) , · · · , z ( N ) ] will be equipped with positional embeddings.
Unidirectional cross attention mechanism.
Instead of directly feeding the whole sequence (namely [ CLS ] token, raw patches, query patches) to the Transformer encoder, which may hurt the performance of the downstream task (as only [ CLS ] token and raw patches are fed to the Transformer for downstream tasks), we slightly modify the attention mechanism.
Given one z [ CLS ] token, raw patches sequence [ z (1) , z (2) , · · · , z ( N ) ] and query patches sequence [ z ( q 1 ) , z ( q 2 ) , · · · , z ( q Q ) ] .
Let z ′ =
[ z [ CLS ] , z (1) , z (2) , · · · , z ( N ) ] and z q =
[ z ( q 1 ) , z ( q 2 ) , · · · , z ( q Q ) ] .
For z [ CLS ] and raw patches, the propagation rule in the attention block is in line with plain Transformer, i.e.,
[Equation]
where Q z ′ =
z ′ W Q , K = z ′ W K , V = z ′ W V , and W Q , W K , W V are learnable weights.
For query patches z ( q i ) , the attention propagation rule is similar to [ CLS ] token, i.e.,
[Equation]
where Q z ( qi ) =
z ( q i ) W Q .
By Eq.
4 and Eq.
5, we can complete the attention block in modified
Transformer.
In a nutshell, we can summarize the propagation rule as RP → RP , { RP, QP i } →
QP i , QP i ↛
QP j ( i ̸ = j ) , where RP and QP i means raw patches (including [ CLS ] token) and i -th query patch, respectively.
We can find only raw patches and [ CLS ] token linearly express the raw patches and [ CLS ] in the attention mechanism.
This design can make the pretrained Transformer more suitable for downstream tasks without introducing extra parameters.
Objective function of ADCLR.
By feeding the whole sequence of two views x A and x B to the modified Transformer f θ and MLP projection head g γ , we can obtain the final representation of the image h
[ CLS ] A ; B , raw patches h A ; B and query patches h qA ; B .
Then, the overall objectives are:
[Equation]
where λ is the hyper-parameter to balance global and local loss.
The impact of λ is studied in Sec. 4.3.
N ECESSITY OF O UR U NIDIRECTIONAL C ROSS -A TTENTION M ECHANISM
Theorem 1 Bidirectional self-attention in standard vision Transformer leads to collapse.
Given a set of query patches { x } Qi =1 and a set of raw patches { x } Ni =1 , contrasting with the standard vision
Transformer-like backbones that use bidirectional self-attention mechanisms can cause ADCLR easily to collapse if Q is large enough ( Q >> the total number of raw patches of the image).
Remarks.
In fact, applying the standard bidirectional self-attention as used in vision Transformers not only incurs the solution collapse issue as shown in Theorem 1, but also may hurt the performance, as will be shown in the ablation studies in Sec. 4.3.
We guess this is because ADCLR introduces additional query patches during the pretraining stage, while the downstream task finetuning only
[Table]
involves raw patches and the bidiretional scheme would cause the mutual influence between query and raw patches, leading to the input inconsistency between pretraining and downstream tasks.
Therefore, we elaborately replace the bidirectional self-attention scheme by our devised unidirectional crossattention one as such the above two issues are addressed directly.
E XPERIMENTS 4. E XPERIMENT S ETUP
Dataset.
We conduct self-supervised pre-training on the ImageNet-1K (Deng et al., 2009) training set, as commonly used in SSL methods for both MIM (He et al., 2021) and contrastive learning (Chen et al., 2020a).
ImageNet-1k includes 1,000 classes which are even in distribution.
We also transfer the encoder pretrained by ADCLR on MS-COCO (Lin et al., 2014), ADE20K
(Zhou et al., 2017) and other fine-grained classfication datasets (Van Horn et al., 2018) for downstream evaluation.
Baselines.
We consider a variety of existing SSL methods based on the ResNet (He et al., 2016) and ViT (Dosovitskiy et al., 2020) architectures: (a) self-supervised ResNets: MoCo-v2 (Chen et al., 2020b), SwAV (Caron et al., 2020), Barlow Twins (Zbontar et al., 2021), ZeroCL (Zhang et al., 2021),
ARB (Zhang et al., 2022), DenseCL (Wang et al., 2021), ReSim (Xiao et al., 2021), and DetCo (Xie et al., 2021a); and (b) self-supervised ViTs: DINO (Caron et al., 2021), MoCo-v3 (Chen et al., 2021),
MoBY (Xie et al., 2021b), iBOT
(Zhou et al., 2022) and SelfPatch (Yun et al., 2022).
Backbone.
We use the standard ViT architecture (Dosovitskiy et al., 2020).
It has several blocks composed of multi-head self-attention (MHSA) (Vaswani et al., 2017), MLP, and LayerNorm (Ba et al., 2016).
In line with DINO and iBOT, we mainly adopt ViT-S/16 and ViT-B/16 as backbones.
Pre-training hyper-parameters.
In line with DINO, we train with Adamw (Loshchilov & Hutter, 2018) and a batch size of 1024, distributed over 32 GPUs using ViT-S/16.
The learning rate is linearly ramped up during the first 10 epochs to its base value determined with the following linear scaling rule (Chen et al., 2020a): lr = 0.0005, batchsize=256.
After warmup, we decay the learning rate with a cosine schedule (Loshchilov & Hutter, 2016).
The weight decay also follows a cosine scheduled from 0.04 to 0.4.
The temperature τ is set to 0.1 while we use a linear warm-up for τ t from 0.
to 0.07 during the first 30 epochs.
We follow the data augmentations of BYOL (Grill et al., 2020) (color jittering, Gaussian blur, and solarization) and multi-crop (Caron et al., 2020) with a bicubic interpolation to adapt the position embeddings to the scales.
We set λ = 0 .
5 and query crop ratio from 0.05 to 0.15.
The code with configuration details for reproducing will be publicly available.
Evaluation protocols.
Standard SSL protocols is to either learn a linear classifier on frozen features (Chen et al., 2020a;
He et al., 2020) or to finetune on downstream tasks (He et al., 2021; Chen et al., 2022).
For linear evaluations, we apply random resize crops and horizontal flips augmentation for training, and report accuracy on a central crop.
For finetuning evaluations (detection and segmentation on MS-COCO (Lin et al., 2014), segmentation on ADE20K
(Zhou et al., 2017)), we initialize networks with the pretrained weights to adapt with further training.
In line with (Zbontar et al., 2021), we also evaluate our method’s transfer ability on small-scale and fine-grained classification dataset.
[Table] [Table]
4. M AIN RESULTS
We first validate the ADCLR framework used in this study with the standard self-supervised benchmark on ImageNet.
Then, we study the performance of detection, segmentation
[Table]
and transfer learning.
Finetune on ImageNet-1k.
Table 1 shows the results with four different learning schemes,
i
[Table]
e., finetune from scratch, supervised pre-training, MIM-based approaches
[Table]
and contrastive learning.
We report linear probing and finetune accuracy, respectively.
Our finetune protocol is in line with iBOT
(Zhou et al., 2022), using strong regularization for training 100 epochs.
The compared baselines are MIM-based methods MAE (He et al., 2021), BEiT (Bao et al., 2021), CIM (Fang et al., 2022) as well as contrastive methods (Caron et al., 2021;
Zhou et al., 2022; Chen et al., 2021) and the combination of the two techniques: CAE (Chen et al., 2022) which is emerging.
Linear probing on ImageNet-1k.
For linear probing evaluation, we compare ADCLR with both
CNNs-based and ViTs-based contrastive learning methods.
Following iBOT
(Zhou et al., 2022), we use effective training epochs to measure the images actually seen by ADCLR.
Specifically, DINO and iBOT are by default trained with 2 global crops with the crop size 224*224 and 10 local crops with the size 96*96.
The calculation formula is r = 2 + ( 96224 ) 2 × 10 = 3 .
84 ≈ 4 .
Moreover, ADCLR introduces 4 query crops, and the total seen image is r = 2 + ( 96224 )
× 10 + ( 16224 ) 2 × 4 = 3 .
≈ 4 .
The overall seen image brought by query crops is only 0.02, which can be negligible.
We report k -NN and linear probing accuracy in Table 2.
ADCLR achieves 78.1% top-1 accuracy on ImageNet-1k with
ViT-S/16, outperforming baseline DINO +1.1% in 3200 effective epochs.
For ViT-B/16, ADCLR outperforms DINO and iBOT
+0.6% and +0.3% accuracy in 800 effective epochs, respectively.
4.2. T RANSFER LEARNING ACROSS DIFFERENT TASKS
MS-COCO Setup.
We evaluate pre-trained models on the MS-COCO object detection and instance segmentation tasks.
Here, all models are fine-tuned with Mask R-CNN (He et al., 2017) and FPN
(Lin et al., 2017) under the standard 1x schedule.
Following (Chen et al., 2022), we utilize multi-scale training and resize the image with the size of the short side between 480 and 800 and the long side no larger than 1333.
In pretraining stage, we set λ = 0 .
5 and Q = 10 for extract better dense information.
For fair comparisons with DINO and SelfPatch, we only use two local (96*96 resolution) and two global (224*224 resolution) views for pretraining.
The pretrining is in line with SelfPatch (Yun et al., 2022).
In finetune stage, we distributed training on 8 Tesla V100 32G GPUs set batch size 32 and set the learning rate as 3e-4 with the layer decay rate 0.75, which also follows SelfPatch.
Results.
Table 3 shows the detection and segmentation performance on MS-COCO.
We give the results of both 200 and 300 epochs’ pretraining to better demonstrate the advance of ADCLR.
We find that Although both SelfPatch and ADCLR perform patch-level contrastive loss, ADCLR breaks the limitation of contrasting between the same view.
The learning paradigm of query tokens seems to bring more variance than single view contrasting, increasing the difficulty of patch-level pretext tasks, and resulting in higher performance on detection and segmentation tasks.
Specifically, under 200 epochs pretraining, ADCLR outperforms SelfPatch with +1.7% and +0.7% points on detection (AP bb ) and segmentation (AP mk ) tasks, respectively.
Compared with baseline DINO (Caron et al., 2021), ADCLR outperforms DINO with a large range, e.g., +2.2% bounding box average precision, +1.2% mask average precision higher than DINO, which benefits from contrasting on query patches.
semantic segmentation performances of the recent self-supervised approaches pre-trained on ImageNet.
The metrics mIoU, aAcc
[Table]
and mAcc denote the mean intersection of union, all pixel accuracy, and mean class accuracy, respectively.
[Table]
segmentation on
ADE20K.
Setups.
We also evaluate semantic segmentation performances of pre-trained models on
ADE20K
(Zhou et al., 2017), which includes 150 finegrained semantic categories and 25k training data.
In line with SelfPatch (Yun et al., 2022), we report three metrics: the mean intersection of union (mIoU) averaged over all semantic categories, all pixel accuracy (aAcc), and mean class accuracy (mAcc).
We pretrain ADCLR in 200 epochs with two local and two global views and finetune the pretrained backbone in different architectures (FPN (Lin et al., 2017) and
UperNet (Xiao et al., 2018)) for fair comparisons.
Results.
The results are given in Table 4.
When finetuning the FPN with 40k iterations, ADCLR outperforms SelfPatch and DINO +1.2% and +3.9% mIoU points, respectively.
In UperNet with 160k iterations, ADCLR outperforms SelfPatch and DINO +1.0% and +3.0% mIoU points.
We guess the gain mainly comes from two aspects, 1) patch-level contrasting enhances the ability to capture local information (see the gap between global method DINO and local method SelfPatch and ADCLR); 2)
The learning paradigm of query tokens brings more accurate information (the gap between SelfPatch and ADCLR).
Transfer learning on the small-scale dataset.
We study transfer learning where we pre-train on
ImageNet-1K and fine-tune on several smaller datasets.
We follow the training recipe and protocol used in DINO and iBOT.
For pretraining, ViT-S and ViT-B are pretrained on 16 GPUs with 1024 batch size and 10 local views in 800 and 400 epochs, respectively.
The results are reported in Table 5.
While the results on several datasets (e.g., CIFAR10, CIFAR100, Flowers, and Cars) have almost plateaued,
ADCLR consistently performs favorably against other SSL frameworks, achieving state-of-the-art transfer results.
Specifically, on small-scale datasets, ADCLR outperforms baseline DINO (Caron et al., 2021) over 2.2% and 1.1% accuracy on iNaturalist18 and iNaturalist19 datasets.
We observe
ADCLR achieve greater performance gain over DINO in fine-grained classification datasets like iNaturalist18, iNaturalist19, and Flowers than CIFAR10 and CIFAR100, indicating ADCLR (query token) is more adept at capturing local information.
Compared with the state-of-the-art method iBOT
(Zhou et al., 2022), although ADCLR achieves similar results, iBOT spends more memory due to the MIM branch (under the same setting, 19.5G for iBOT and 16.7G for ADCLR).
[Table]
[Figure]
[Table]
[Figure]
4. A BLATION STUDY
To verify the robustness of ADCLR, we comprehensively study the sensitiveness of balance ratio λ , number of query tokens Q , the effect of cross attention mechanism, and crop ratio.
Effect of λ .
We conduct ablation experiments on different balance ratios λ to explore the effect of local objective L Local .
Specifically, we set batch size as 1024, and distribute training on 32 GPUs with 30 epochs as warm up out of the total 100 epochs.
For efficiency, we only use two global views (224 * 224), and 10 query views (16*16) without local view (96*96).
We report the results in
Fig. 2(a).
Compared with baseline DINO, ADCLR outperforms +1.3% top-1 accuracy when λ = 0 .
With too large λ , the accuracy will drop a little, which we guess ADCLR would overemphasize the local information, but ignore the global information and leading to the decreasing accuracy of global task (classification).
With smaller λ e.g. between 0.1 and 0.2, ADCLR is mainly determined by [ CLS ] token, which may bring unwanted contrasting loss (see Fig. 1) to hurt the final performance.
Effect of the number of query crops Q .
We set batch size as 1024 and pretrain ADCLR withepochs.
In line with ablation on λ , we only use two global views without local views.
Besides, we choose the best value of λ = 0 .
5 for this experiment.
We illustrate the results in Fig. 2(b).
The performance increases with the increase of Q , which is similar to the ensemble mechanism (with more base models, the ensemble model are more accurate).
We find ADCLR saturates when Q =and Q = 10 , which may be enough to capture local information.
We also try large Q , but when we set Q = 15 , we find there’s only little improvement, and thus we set Q = 10 by default.
[Table]
Effect of unidirectional cross attention.
As discussed in Sec. 3.2, we find unidirectional cross attention makes
ADCLR more effective to downstream tasks (with only the raw patches and [ CLS ] tokens as input).
We conduct experiments by using unidirectional and bidirectional cross attention mechanisms and the results are given in Fig. 2(c).
When Q = 1 , ADCLR (bidir) drops about 3% ∼ 4% accuracy.
When increasing the number of query view Q , the accuracy of the downstream classification task drops notably.
When we set Q = 196 (same with query token), we find ADCLR (bidir) only gets about 50% accuracy, while ADCLR (unidir) gets 68.8% accuracy.
Effect of query crops ratio s .
Since ADCLR introduces the query crops, we further study the performance with different local and global crop ratios.
Following iBOT
(Zhou et al., 2022), we conduct the experiments by tweaking s , where s is the scale dividing the local and global crops.
The query crops and local crops are sampled from (0.05, s ).
The global crops are sampled from (0.4, 1).
We fix the other hyper-parameter and pretrain ADCLR with 100 epochs.
We report the linear classification accuracy in Table 6.
We evaluate crop ratios smaller than 0.2 because of the low resolution of query crop With a large ratio, low-resolution crops may lose lots of local information.
[Table]
Effect of local crops.
We further conduct ablation on two 8-GPU machines on the number of local views as reported in Table 7.
We fix the batch size 1024 and evaluate it pretrained with 100 andepochs.
The multi-crop strategy follows
SwAV (Caron et al., 2020), i.e., 2 global views (224*224) and 2, 6, 10 local views (96*96).
Under the two-view setting, ADCLR achieves 68.9% top-1 accuracy with 100 epochs pretraining.
With the same setting, we reproduce the result of DINO, resulting 67.7% top-1 accuracy with 100 epochs pretraining, where our ADCLR outperforms DINO +1.2% accuracy.
For multi-crops setting, we run DINO with 300 epochs, using 2 × 224 2 + 10 × 96 2 crop strategy, resulting 76.2% (0.1% higher than reported in DINO) top-1 accuracy, ADCLR outperforms DINO by +0.8% topaccuracy.
We find ADCLR gets higher accuracy under the two-crops strategy, which we guess using two global views for DINO is difficult to extract local information and the object in global views cannot be matched exactly.
For ADCLR, the query tokens are cropped from the raw images with 16*16 pixels, which help capture accurate and dense information (see Fig. 1).
F URTHER D ISCUSSION TO R ELATED W ORKS
We discuss the relation of ADCLR against similar approaches to better position our work.
Relation to DINO (Caron et al., 2021).
Connection.
ADCLR adopts the learning framework of DINO (EMA update, sharpening and centering).
The global branch of ADCLR is the same as
DINO.
Differences.
On the basis of DINO, ADCLR proposes a local branch to learn spatial-sensitive information.
Besides, technically, ADCLR proposes the unidirectional cross attention mechanism.
Relation to iBOT
(Zhou et al., 2022).
Connection. .
Both iBOT and ADCLR learn global and local information.
Differences.
iBOT learns the local information by adding MIM objective, which is now widely developed in MAE (He et al., 2021), SimMIM (Xie et al., 2021d).
ADCLR learns the local information by maximizing the consistency of the same query crops in latent space (the variance is caused by the attention mechanism of two views).
Compared with iBOT (masks patches), query tokens in ADCLR may be out of the views, which increases the difficulty of local contrasting.
Relation to SelfPatch (Yun et al., 2022).
Connection.
Both ADCLR and SelfPatch adopt DINO’s framework on the patch level to learn dense information.
Differences.
The idea of SelfPatch is very straightforward, i.e., regard top- k neighbors of query patch as positives, which may bring the noise (the selection of k ).
Besides, the local branch of SelfPatch requires input views has the same location, which decreases the learning difficulty.
In contrast, ADCLR is simple yet also carefully designed.
The local branch of ADCLR reserves the variance caused by augmentation (ADCLR can solve the positional shift caused by random resize and crop).
Technically, SelfPatch designs the matching algorithm.
For each query crop, it calculates the similarities and sorts them, which may bring a lot more complexity.
ADCLR only introduces Q query patches, where 1 ≤ Q ≤ 10 .
The complexity brought by ADCLR can be ignored (ViT-B/16 and ViT-S/16 divide the image into 196 raw patches.)
C ONCLUSION
This paper proposes ADCLR, a new contrastive learning method aiming to capture accurate and local information for vision representation.
By introducing “query crop”, ADCLR could easily match spatial location without complex operation, e.g., bi-linear interpolation.
To make ADCLR more adaptable to downstream tasks and prevent collapse, we design unidirectional cross attention.
Experiments on classification, detection, and segmentation show the effectiveness.
Limitations.
Currently, the paradigm of “query crops” only supports low resolution (mapping the cropped region into small patches), which may lose semantic information.
Hence, we think the multi-scale paradigm could help ADCLR a lot, which may be a good extension in future works.
A P ROOF FOR T HEOREM
Proof For simplicity, we consider in one layer single-head self-attention block, and remove the g , b re-scale parameters in layer normalization (Ba et al., 2016).
Recall the self attention mechanism in
Transformer, i.e., Attn ( x , W Q , W K , W V ) =
Softmax ( ( xW Q )( xW K ) ⊤ √ d ) xW V , where W Q , W K and W V are learnable weights, and √ d is the hidden dimension.
Note that x ∈ R (1+ N + Q ) ×
d includes [ CLS ] token, raw patch sequence and query patch sequence.
Let s i ; j = ( x i W Q )( x j W K ) ⊤ √ d , where
[Equation]
e
[Equation]
N
[Equation]
Q
[Equation]
e − 1 + e , since the normalized ∥ x i W Q ∥ 22 = 1 .
Then, the input of two branches in contrastive learning becomes [ x 0 A , x 1 A , x 2 A , · · · , x QA , · · · , x N + Q A ] for view A and [ x 0 B , x 1 B , x 2 B , · · · , x QB , · · · , x N + Q B ] for view B. Note that we have x iA = x iB for 1 ≤ i ≤ Q .
Then, the embedding of i -th patch z iA and z iB through self-attention block can be written as:
[Equation]
Recall the objective of negative-free contrastive learning, i.e., aligning the representations max sim ( z iA , z iB ) for 0 ≤ i ≤ Q , where z 0 is the embedding of [ CLS ] token.
For simplicity, we use the mean square error (MSE) objective for analysis.
Take the gradient into Eq.
7, we get: ∂ L ( z iA , z iB )
[Equation]
where s i ; jA means attention value of i -th patch to j -th patch in view A branch.
Since x is standardized and Q >> N , Then, the numerator of Eq.
8 can be written as ∥ ( (cid:80) Qj
=
s i ;
jA x jA − (cid:80) Qj =
s i ; jA x jB ) W V ∥ 22 .
Then, the network could easily get collapse solution, i.e., making s i ; jA =
s i ; jB for 0 ≤ i, j ≤ Q , leading the L ( z iA , z iB )
= 0 , since x iA = x iB for 0 ≤ i ≤ Q .
Then, we complete the proof.
B I MPLEMENTATIONS
Pytorch-liked code of unidirectional cross attention.
class Attention(nn.
Module):defforward(self,cls, x, query): """:param cls: B, 1, C :param x: B, N, C :param query: B, Q, C """ qkv =
self.qkv(torch.cat([cls, x, query], dim=1)).reshape(B, 1+N+Q, 3, self.num_heads, C // self.num_heads).permute(2, 0, 3, 1, 4) q, k, v = qkv[0], qkv[1], qkv[2] # cls and x forward attn =
(q[:, :, :1+N] @
k[:, :, :1+N].
[Table]