Papers › ScaleKD: Strong Vision Transformers Could Be Excellent Teachers

ScaleKD: Strong Vision Transformers Could Be Excellent Teachers

11 Nov 2024arXiv:2411.06786archive 2025-07-28

Jiawei Fan, Chao Li, Xiaolong Liu, Anbang Yao

In this paper, we question if well pre-trained vision transformer (ViT) models could be used as teachers that exhibit scalable properties to advance cross architecture knowledge distillation (KD) research, in the context of using large-scale datasets for evaluation. To make this possible, our analysis underlines the importance of seeking effective strategies to align (1) feature computing paradigm differences, (2) model scale differences, and (3) knowledge density differences. By combining three coupled components namely cross attention projector, dual-view feature mimicking and teacher parameter perception tailored to address the above problems, we present a simple and effective KD method, called ScaleKD. Our method can train student backbones that span across a variety of convolutional neural network (CNN), multi-layer perceptron (MLP), and ViT architectures on image classification datasets, achieving state-of-the-art distillation performance. For instance, taking a well pre-trained Swin-L as the teacher model, our method gets 75.15%|82.03%|84.16%|78.63%|81.96%|83.93%|83.80%|85.53% top-1 accuracies for MobileNet-V1|ResNet-50|ConvNeXt-T|Mixer-S/16|Mixer-B/16|ViT-S/16|Swin-T|ViT-B/16 models trained on ImageNet-1K dataset from scratch, showing 3.05%|3.39%|2.02%|4.61%|5.52%|4.03%|2.62%|3.73% absolute gains to the individually trained counterparts. Intriguingly, when scaling up the size of teacher models or their pre-training datasets, our method showcases the desired scalable properties, bringing increasingly larger gains to student models. The student backbones trained by our method transfer well on downstream MS-COCO and ADE20K datasets. More importantly, our method could be used as a more efficient alternative to the time-intensive pre-training paradigm for any target student model if a strong pre-trained ViT is available, reducing the amount of viewed training samples up to 195x.

PaperPDFCode

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

Code

deep-optimization/scalekd officialmentioned in papermentioned on GitHubpytorchApache-2.0 report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

Image ClassificationKnowledge Distillationimage-classification

Results from the paper archive 2025-07-28

TaskDatasetModelMetricValueRank at snapshotLeaderboardReport
Knowledge Distillation ImageNet ScaleKD (T:BEiT-L S:ViT-B/14) CRD training setting ✘ #1 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:BEiT-L S:ViT-B/14) Top-1 accuracy % 86.43 #1 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:BEiT-L S:ViT-B/14) model size 87M #1 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-B/16) CRD training setting ✘ #2 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-B/16) Top-1 accuracy % 85.53 #2 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-B/16) model size 87M #2 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-S/16) CRD training setting ✘ #3 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-S/16) Top-1 accuracy % 83.93 #3 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ViT-S/16) model size 22M #3 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:Swin-T) CRD training setting ✘ #4 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:Swin-T) Top-1 accuracy % 83.8 #4 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:Swin-T) model size 27M #4 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ResNet-50) CRD training setting ✘ #8 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ResNet-50) Top-1 accuracy % 82.55 #8 of 52 Archive leaderboard report
Knowledge Distillation ImageNet ScaleKD (T:Swin-L S:ResNet-50) model size 22M #8 of 52 Archive leaderboard report

Ranks are positions in the archive's leaderboards as they stood at the 2025-07-28 snapshot. Results published since then are not among these rows, so a rank here is not a current standing.

Methods

ALIGNAttentionDense ConnectionsKnowledge DistillationLayer NormalizationLinear LayerMulti-Head AttentionResidual ConnectionSoftmaxVision Transformer

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections