atlas

Research map / Label-efficient and robust learning / Knowledge distillation

Fantastic Gains and Where to Find Them: On the Existence and Prospect of General Knowledge Transfer between Any Pretrained Model

ICLR 2024 · Label-efficient and robust learning › Knowledge distillation · approach: Patient consistent distillation · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that arbitrary pretrained ImageNet models hold complementary knowledge for each other and proposes KL distillation with confidence-based data partitioning (KL+DP) to transfer it without performance degradation, raising transfer success from under 40% to 92.5%.

Problem

Transferring complementary knowledge between arbitrary already-trained models (including weaker or equal-performing teachers) without degrading the student's existing performance, lacking external test rankings.

Approach

Treat transfer between pretrained models as continual learning and partition transfer data per sample by teacher-vs-frozen-student confidence, distilling from the teacher only on samples where it is more confident and from the frozen student elsewhere; this needs no extra hyperparameters and can be done without labels.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use data partitioning instead of standard KL distillationpretrained ImageNet students · ImageNettransfer success rate: 32.5% -> 92.5%Helped
Add momentum weight interpolation (MCL) to XE-KL distillationpretrained ImageNet students · 10% ImageNet subsettransfer success rate: 80%Helped
Replace standard distillation with confidence-based data partitioningpretrained ImageNet students · ImageNettransfer delta: 0.13 ($\pm$1.01) -> 0.65 ($\pm$0.26); 0.33 ($\pm$0.86) -> 0.55 ($\pm$0.25); 0.10 ($\pm$0.49) -> 0.40 ($\pm$0.12); -0.05 Helped
Use unsupervised confidence-based selection instead of supervised data partitioning (ablation)KL+DP transfer · ImageNettransfer delta: 0.65 ($\pm$0.26) -> 0.96 ($\pm$0.42); 0.55 ($\pm$0.25) -> 0.95 ($\pm$0.52); 0.40 ($\pm$0.12) -> 0.35 ($\pm$0.18); 0.27 (No clear effect
Remove the auxiliary cross-entropy classification objective from KL+DP (ablation)KL+DP transfer · ImageNet transfer pairstransfer performanceHelped
Replace KL divergence with cross-entropy loss (ablation)distillation transfer · 60 teacher-student pairstransfer delta: 1.2 percentage points lowerHurt
Add a warmup epoch training only the student's linear layers (ablation)KL distillation transfer · teacher-student transfer pairstransfer performanceNo clear effect
Restrict the KL loss to the top-10 teacher-predicted classes (ablation)distillation transfer · six teacher-student pairstransfer delta: 0.20 percentage pointsHelped
Use contrastive distillation (CD or CRD) instead of KL-based transfer (ablation)pretrained expert models · 10% ImageNet subsettransfer success rateHurt
Transfer sequentially from multiple teachers instead of the best single teacherKL+DP students · ImageNettransfer delta: 0.26 -> 0.65; 0.86 -> 1.04Helped
Transfer from multiple teachers in parallel instead of the best single teacherKL+DP students · ImageNettransfer deltaNo clear effect
Average separately distilled student variants into a model soupKL+DP students · ImageNettransfer deltaHurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet CUB200 StanfordCars Caltech256

Benchmarks: ImageNet CUB200 StanfordCars Caltech256

Metrics: top-1 accuracy transfer delta success rate (share of teachers with positive transfer) positive/negative prediction flips

Baselines: KL-Dist XE-KL XE-KL+MCL label smoothing CD distillation CRD distillation Model soups

Builds on: knowledge distillation (Hinton et al.) momentum continual learning (MCL) model soups CRD contrastive distillation ffcv

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy