atlas

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

Knowledge Distillation Based on Transformed Teacher Matching

ICLR 2024 · Label-efficient and robust learning › Knowledge distillation · approach: Distillation with causal and adaptive guidance · 5 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.

Proposes transformed teacher matching (TTM), which drops temperature scaling on the student side of knowledge distillation, showing it equals KD plus a Rényi entropy regularizer, and a sample-adaptive variant WTTM that reaches state-of-the-art on CIFAR-100 and ImageNet.

Problem

Knowledge distillation conventionally applies temperature scaling to both teacher and student logits, and the role of this temperature on the student side is not well understood or justified.

Approach

Drop temperature scaling on the student side; reinterpret temperature scaling of the teacher as a power transform of its distribution, which shows TTM equals KD plus a Rényi entropy regularizer. Further weight each sample's distillation term by the power sum of the teacher distribution (WTTM).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Remove student-side temperature scaling from KD, adding TTM’s Rényi entropy regularizationKD student · CIFAR-100, WRN-40-2 → WRN-16-2Top-1 accuracy (%): 74.92 -> 76.23Helped
Add sample-adaptive weighting to TTM to form WTTM (ablation)TTM · CIFAR-100, WRN-40-2 → WRN-16-2Top-1 accuracy (%): 76.23 -> 76.37Helped
Add sample-adaptive weighting to TTM to form WTTM (ablation)TTM · CIFAR-100, ResNet-50 → MobileNetV2Top-1 accuracy (%): 69.24 -> 69.59Helped
Add ITRD to WTTMWTTM · CIFAR-100, WRN-40-2 → WRN-16-2Top-1 accuracy (%): 76.37 -> 76.65Helped
Add CRD to WTTMWTTM · CIFAR-100, WRN-40-2 → WRN-16-2Top-1 accuracy (%): 76.37 -> 76.61Helped
Remove the cross-entropy loss from WTTM (ablation)WTTM · CIFAR-100, WRN-40-2 → WRN-16-2Top-1 accuracy (%): 74.92 -> 75.11Mixed
Distill ResNet-18 from increasingly accurate ResNet teachers using WTTMWTTM · ImageNet, ResNet-18Top-1 accuracy (%): 72.19 -> 72.26 -> 72.34 -> 72.39Helped
Add ViTKD to WTTMWTTM · ImageNet, DeiT III-Small → DeiT-TinyTop-1 accuracy (%): 77.03 -> 78.04Helped

Surprising findings

Limitations

Experimental setup

Datasets: CIFAR-100 ImageNet

Benchmarks: CIFAR-100 ImageNet

Metrics: top-1 accuracy

Baselines: KD FitNet AT VID RKD PKT CRD DIST DKD ITRD SRRL ReviewKD

Builds on: knowledge distillation (Hinton et al.) confidence penalty label smoothing CRD ITRD

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