atlas

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

KD-DETR: Knowledge Distillation for Detection Transformer with Consistent Distillation Points Sampling

CVPR 2024 · Label-efficient and robust learning › Knowledge distillation · approach: Knowledge distillation for detectors · 26 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 KD-DETR, a logit-based knowledge distillation paradigm for DETR that uses shared, unlearnable specialized object queries as consistent distillation points, improving DAB/Deformable/DINO students on COCO by 2.6-5.2 AP.

Problem

Knowledge distillation works poorly on DETR-style detectors because object queries in teacher and student are egocentric and inconsistent in number and correspondence, so the distillation points (inputs whose predictions are mimicked) are insufficient and unreliable.

Approach

Decouple detection and distillation by introducing a set of shared, unlearnable distillation-point queries (general random-initialized queries plus teacher's own queries) fed to both teacher and student decoders, weighted by teacher foreground probability, so the student mimics consistent teacher predictions; extended to DETR-to-CNN heterogeneous distillation via IoU-selected anchors.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use KD-DETR distillation from DAB-DETR ResNet-50DAB-DETR ResNet-18 · MS COCO2017 valAP: 36.2 -> 41.4Helped
Use KD-DETR distillation from Deformable DETR ResNet-101Deformable DETR ResNet-50 · MS COCO2017 valAP: 44.5 -> 48.3Helped
Distill from DINO ResNet-50 using KD-DETRFaster R-CNN ResNet-50 · MS COCO2017 valAP: 38.4 -> 40.5Helped
Use inconsistent object-query permutations for distillation (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 35.1Hurt
Use matched foreground queries as distillation points (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 37.2Helped
Add an average negative-query background distillation point (ablation)DAB-DETR ResNet-18 with Similar Foreground points · MS COCO2017AP: 37.2 -> 37.4Helped
Increase object queries from 300 to 600 without distillation points (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 37.7Helped
Add 300 distillation points (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 40.2Helped
Use randomly initialized general distillation queries (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 38.7Helped
Use teacher queries as specific distillation points (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 39.9Helped
Add foreground rebalance weight to general sampling (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 38.7 -> 40.0Helped
Combine general sampling, specific sampling, and foreground rebalance weight (ablation)DAB-DETR ResNet-18 · MS COCO2017AP: 36.2 -> 41.4Helped

Surprising findings

Limitations

Experimental setup

Datasets: MS COCO2017 ImageNet1K (backbone pretraining)

Benchmarks: MS COCO2017

Metrics: AP AP50 AP75 APS APM APL

Baselines: DETRDistill FGD FitNet FGFI GID Baseline student without distillation

Builds on: DETR DAB-DETR Deformable DETR DINO logit-based distillation (Chen et al. 2017 style, cited as [4])

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