Source-linked AI summary
Self Supervision to Distillation for Long-Tailed Visual Recognition
Tianhao Li, Limin Wang, Gangshan Wu
TL;DR
Long-tailed recognition remains difficult because class imbalance degrades performance, while re-balancing risks overfitting and decoupling leaves label imbalance in feature learning. SSD combines self-distillation with self-supervision-guided soft-label generation; it achieves state-of-the-art results across three benchmarks.
Problem
Long-tailed data imbalance degrades recognition, and prior re-balancing and decoupling approaches leave challenges in overfitting or representation-stage label imbalance.
Source: §1 · p. 1Method
SSD uses a multi-stage self-distillation framework with self-supervision-guided soft labels that model relations between classes.
Source: §1 · p. 2Results
SSD achieves state-of-the-art performance on ImageNet-LT, CIFAR100-LT, and iNaturalist 2018, and exceeds the stronger longer-scheduler LWS baseline by 2.7% on iNaturalist 2018.
Source: §4.2.3 · p. 7, §5 · p. 8Takeaways & Limitations
The authors conclude that distilled supervision can address re-balanced methods’ overfitting issue while providing relatively balanced information for feature learning.
Source: §5 · p. 8
Abstract
from arXiv · showhide
Deep learning has achieved remarkable progress for visual recognition on large-scale balanced datasets but still performs poorly on real-world long-tailed data. Previous methods often adopt class re-balanced training strategies to effectively alleviate the imbalance issue, but might be a risk of over-fitting tail classes. The recent decoupling method overcomes over-fitting issues by using a multi-stage training scheme, yet, it is still incapable of capturing tail class information in the feature learning stage. In this paper, we show that soft label can serve as a powerful solution to incorporate label correlation into a multi-stage training scheme for long-tailed recognition. The intrinsic relation between classes embodied by soft labels turns out to be helpful for long-tailed recognition by transferring knowledge from head to tail classes. Specifically, we propose a conceptually simple yet particularly effective multi-stage training scheme, termed as Self Supervised to Distillation (SSD). This scheme is composed of two parts. First, we introduce a self-distillation framework for long-tailed recognition, which can mine the label relation automatically. Second, we present a new distillation label generation module guided by self-supervision. The distilled labels integrate information from both label and data domains that can model long-tailed distribution effectively. We conduct extensive experiments and our method achieves the state-of-the-art results on three long-tailed recognition benchmarks: ImageNet-LT, CIFAR100-LT and iNaturalist 2018. Our SSD outperforms the strong LWS baseline by from $2.7\%$ to $4.5\%$ on various datasets. The code is available at https://github.com/MCG-NJU/SSD-LT.
1. Introduction
Long-tailed data makes balanced datasets difficult to collect, and direct training or re-balancing can leave recognition and representation learning vulnerable to imbalance. SSD incorporates soft-label class relations into multi-stage training and achieves state-of-the-art results on three long-tailed benchmarks.
- Class re-balancing can overfit tail classes, while decoupled training addresses overfitting but still handles label imbalance poorly during representation learning.Re-balancing methods include resampling and cost-sensitive reweighting.Source: §1 · p. 1
- Soft labels capture relations between classes and can transfer knowledge from head classes to tail classes within multi-stage training.The paper presents soft labels as a way to incorporate label correlation into the training strategy.Source: §1 · p. 1
- SSD combines a self-distillation framework that mines label relations with a self-supervision-guided module that generates less biased, more informative soft labels.Self-supervision uses the long-tailed training set and treats images equally when generating soft labels.Source: §1 · p. 2
- The training pipeline learns an initial teacher with label and self-supervision, adjusts a separate classifier to produce soft labels, then trains a self-distillation network with soft and hard labels.Two classification heads handle the two supervision types because of their semantic gap and possible head-class bias.Source: §1 · p. 2
- SSD leverages soft-label modeling in feature learning to combine re-balanced sampling with decoupled training.Source: §1 · p. 2
- SSD achieves state-of-the-art performance on ImageNet-LT, CIFAR100-LT, and iNaturalist 2018.Source: §1 · p. 2
2. Related Work
Related work covers class re-balancing, distilled-label training, and self-supervised learning. SSD differs by applying distilled labels in a multi-stage scheme for long-tailed data and using self-supervision to improve label generation.
2.1. Long-tailed Classification
- Re-balancing methods resample data or alter losses, but may overfit data-scarce classes or underfit data-abundant classes; decoupling transfers frozen head-class features during classifier training.Other approaches augment tail classes, adjust class weights or margins, group classes, or modify negative gradients.Source: §2.1 · pp. 2–3
2.2. Learning with Distilled Labels
- Unlike BAN and self-training, SSD produces distilled labels through an additional classifier-adjustment stage for long-tailed data.LFME distills experts trained on category groups, while RIDE distills between models with different numbers of experts; SSD uses distilled labels in multi-stage training to transfer knowledge from head to tail classes.Source: §2.2 · p. 3
2.3. Self-supervised Learning
- Self-supervised learning uses proxy tasks without manual annotations; SSD uses an auxiliary self-supervised task to improve distilled-label quality, unlike self-supervised pretraining in.Examples of proxy tasks include predicting image context or rotation, colorization, jigsaw puzzles, and instance discrimination.Source: §2.3 · p. 3
3. Methodology
SSD uses a multi-stage pipeline to learn features, generate soft labels, and train a self-distillation network with both soft and hard supervision. Self-supervision guides feature learning, while a class-balanced classifier generates the labels used in the final stage.
3.1. Overall framework
- SSD first learns features with supervised and self-supervised tasks, then generates intermediate soft labels, and finally trains with self-distillation.The stages use instance-balanced sampling, class-balanced sampling with the feature backbone fixed, and hybrid supervision, respectively.Source: §3.1 · p. 3, Figure 2 · p. 4
3.2. Feature learning enhanced by self-supervision
- Balanced self-supervised tasks treat images equally without label influence, complementing semantic classification under the original long-tailed distribution.The paper uses rotation prediction and instance discrimination as proxy tasks.Source: §3.1 · p. 3, §3.2 · p. 4
3.1. Overall framework
- A classifier tuned with class-balanced sampling on frozen features produces soft labels that integrate label-driven and data-driven information.The method uses Learnable Weight Scaling (LWS) to rescale classifier weights and reduce the tendency toward head classes.Source: §3.1 · pp. 3–4
3.3. Long-tailed recognition via self-distillation
- The final network is trained from scratch with separate hard-label and soft-label classifiers, combining cross-entropy and distillation losses.The paper states that this self-distilled model can outperform the teacher; classifier fine-tuning is optional but recommended for further improvement.Source: §3.1 · p. 4, §3.3 · p. 5
4.1. Experimental settings
The experiments use ImageNet-LT, CIFAR100-LT, and iNaturalist 2018, with evaluation on balanced validation or test sets. The protocol reports top-1 accuracy and, for ImageNet-LT, accuracy across three training-frequency groups.
4.1. Experimental settings
- ImageNet-LT is sampled from ImageNet-2012 with 1,000 classes and 115.8K training images, ranging from 1,280 to 5 images per class.Its validation and test sets are balanced and contain 20K and 50K images, respectively; hyperparameters are selected on validation data.Source: §4.1 · p. 5
- CIFAR100-LT uses 100 categories and imbalance factors of 10, 50, and 100, defined as the ratio of the most- to least-frequent class counts.The validation set contains 100 images per class.Source: §4.1 · p. 5
- iNaturalist 2018 contains 8,142 fine-grained species, 437.5K training images, and an imbalance factor of 500.The official validation set, with 3 images per class, is used for testing.Source: §4.1 · p. 5
- Evaluation reports top-1 accuracy on the corresponding balanced validation or test set, plus Many-shot, Medium-shot, and Few-shot accuracy for ImageNet-LT.The ImageNet-LT groups are defined by at least 100, 20–100, and at most 20 training samples per class, respectively.Source: §4.1 · p. 5
4.2. Comparisons with the state-of-the-art methods
SSD is compared with state-of-the-art methods on ImageNet-LT, CIFAR100-LT, and iNaturalist 2018. It reports gains over LWS on all three benchmarks, with state-of-the-art performance across CIFAR100-LT imbalance factors.
4.2.1 Experimental results on ImageNet-LT
- 56.0% overall accuracy sets a new ImageNet-LT state of the art; SSD beats CIFAR100-LT LWS by 3.7%–4.5% and iNaturalist LWS by 2.7% even with its longer scheduler.The CIFAR100-LT gains span imbalance factors 100, 50, and 10; the iNaturalist comparison is against the 69.5% longer-scheduler LWS baseline.Source: §4.2.1 · p. 6, §4.2.2 · p. 6, §4.2.3 · p. 7
- ImageNet-LT comparisons use ResNeXt-50, and the authors report consistent gains over re-implemented cRT and LWS baselines on every split.They also report that smaller batch size and learning rate improve performance (50.7% v.s. 49.9%), while the 1.5× scheduler adds +1.4% for LWS.Source: §4.2.1 · pp. 5–6, Table 1 · p. 6
4.2.2 Experimental results on CIFAR100-LT
- SSD achieves state-of-the-art performance on CIFAR100-LT across imbalance factors 100, 50, and 10, beating LWS by 3.7%–4.5%.The experiments use ResNet-32; phase I uses rotation prediction as its self-supervised task.Source: §4.2.2 · p. 6, Table 2 · p. 6
4.2.3 Experimental results on iNaturalist 2018
- On iNaturalist 2018, SSD improves over the 69.5% longer-scheduler LWS baseline by 2.7%; it also gains 2.7% over LWS with the 1× scheduler.The comparison uses ResNet-50 and evaluates 1× and 2× training schedulers.Source: Table 3 · p. 6, §4.2.3 · p. 7
4.3. Ablation studies
The ablations examine self-supervised feature learning, self-distillation strategies, and training accuracy across shot groups. They report gains from self-supervision and hybrid supervision, with instance discrimination more stable across class splits.
4.3.1 Effectiveness of each training stage
- Instance discrimination as self-supervision improves overall performance by 2.6 points over the 46.3% CE baseline and raises accuracy across all three shot splits.The reported gains are +1.9% for many-shot, +3.3% for medium-shot, and +1.6% for few-shot classes; the feature visualization shows better separation, especially for tail classes.Source: §4.3.1 · p. 7, Figure 3 · p. 7
- III-hard (test) beats Stage-I by 2.7% and the CE baseline by 5.3%; IV-LWS adds a further 0.3%.III-hard uses the hard-label classifier; IV-LWS applies LWS to that classifier.Source: §4.3.1 · p. 7
4.3.2 Study on different self-distillation strategies
- The hybrid soft classifier outperforms single self-distillation, while coupled self-distillation does not reach the 54.1% teacher result.The authors attribute limited medium- and few-shot improvement for the coupled approach to interference between hard and soft labels; they suggest hard labels complement soft supervision.Source: §4.3.2 · p. 8, Table 5 · p. 8
4.3.3 Evaluation on self-supervised task
- On ImageNet-LT, instance-discrimination accuracy is more stable across shot splits than supervised classification, whose accuracy remains low for tail categories.The authors interpret this pattern as evidence that self-supervised learning treats images equally during training, relieving the effect of imbalanced labels.Source: §4.3.3 · p. 8, Figure 4 · p. 8
4.4. Visualization
The visualization examines how temperature shapes the training-sample distribution of distilled labels.
4.4. Visualization
- A larger temperature makes the distilled-label distribution flatter; the experiments use T = 2, selected by cross-validation.The authors divide logits by temperature before applying softmax and sum the outputs across training samples to calculate the distribution.Source: §4.4 · p. 8
5. Conclusion
The conclusion presents SSD as a multi-stage long-tailed recognition framework that uses distilled labels and self-supervision-guided label generation.
5. Conclusion
- SSD combines distilled supervision with self-supervision-guided soft labels and reports state-of-the-art performance on ImageNet-LT, CIFAR100-LT, and iNaturalist 2018.The proposed labels leverage both top-down semantics and bottom-up data structure.Source: §5 · p. 8
Related papers
This paper cites
- A Simple Framework for Contrastive Learning of Visual Representations2020
- Colorful Image Colorization2016
- Class-Balanced Loss Based on Effective Number of Samples2019
- Born Again Neural Networks2018
- Learning Imbalanced Datasets with Label-Distribution-Aware Margin Loss2019
- Decoupling Representation and Classifier for Long-Tailed Recognition2019