arrow
Return

TCKD: Task-Oriented Contrastive Knowledge Distillation for Object Detection

delete2026-03-12
delete0
PRE
AI
Z
Zeyuan Liu
H
He Huang
S
Siwei Sun
K
Kaibing Zhang
杨
杨澜 (Lan Yang)
T
Tao Gao
DOI:10.1109/tmm.2026.3673559delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Most existing prediction-based distillation methods for object detection suffer from misalignment between distillation losses and optimization objectives, which hinders the effective transfer of critical knowledge and severely limits the student’s improvement. To address this challenge, we propose a novel task-oriented knowledge distillation (TCKD) method that transfers both classification and regression knowledge through two efficient distillation losses: class-wise contrastive distillation loss (CCDL) for classification and point-wise contrastive distillation loss (PCDL) for regression. Specifically, CCDL constructs positive and negative samples based on class predictions and establishes intra-class consistency and inter-class discrepancy between samples. This guides the model to effectively focus on regions where the objects align with their corresponding classes. PCDL generates positive and negative samples by leveraging sample points at different spatial positions and captures finer-grained localization differences through similarity computation. Moreover, the proposed foreground-focused strategy used for PCDL effectively reduces background noise and alleviates the distribution imbalance between the foreground and background. Extensive experiments on the MS COCO2017, visDrone2019, and TinyPerson benchmarks demonstrate that TCKD not only achieves superior performance on large-scale datasets and challenging small object datasets but is also adaptable to various detectors.
Keywords:
Contrastive learning
knowledge distillation
object detection
prediction imitation

Journal

IEEE Transactions on Multimedia cover
IEEE Transactions on Multimedia
IF:
9.7
Papers:
4.7K
Citations:
2.4W

Organization

C
Chang'an University
Scholars:
4.1K
Papers: 1.5K
Citations: 1.3W
X
xi'an polytechnic university
Scholars:
1.2K
Papers: 359
Citations: 0
Cited Papers

Cited Papers

Context Matters: Distilling Knowledge Graph for Enhanced Object Detection
err2024-01-01
err0
PREAI
errAijia Yang; Sihao Lin; Chung-Hsing Yeh; Minglei Shu; Yi Yang; Xiaojun Chang
errShare
errSave
Variational Information Distillation for Knowledge Transfer
err2019-06-01
err0
errOAAI
errSungsoo Ahn; Shell Xu Hu; Andreas Damianou; Neil D. Lawrence; Zhenwen Dai
errShare
errSave
Masked Generative Distillation
err2022-11-03
err0
PREAI
errZhendong Yang; Zhe Li; Mingqi Shao; Dachuan Shi; Zehuan Yuan; Chun Yuan
errShare
errSave
One-for-All: Bridge the Gap Between Heterogeneous Architectures in Knowledge Distillation
err2023-01-01
err0
PREAI
errGuo,Jianyuan; Han,Kai; Hao,Zhiwei; Hu,Han; Tang,Yehui; Wang,Yunhe; Xu,Chang
errShare
errSave
Distilling Object Detectors via Decoupled Features
err2021-06-01
err0
errOAAI
errJianyuan Guo; Kai Han; Yunhe Wang; Han Wu; Xinghao Chen; Chunjing Xu; Chang Xu
errShare
errSave
Instance-Aware Distillation for Efficient Object Detection in Remote Sensing Images
err2023-01-01
err93
PREAI
errLi, Cong; Cheng, Gong; Wang, Guangxing; Zhou, Peicheng; Han, Junwei
errShare
errSave
researcher View more