paper-with-me

홈 › Papers

SKDF: A Simple Knowledge Distillation Framework for Distilling Open-Vocabulary Knowledge to Open-world Object Detector

2023-12-14 · Shuailei Ma, Yuefeng Wang, Ying WEI, Jiaqi Fan, Enming Zhang, Xinyu Sun, Peihao Chen

In this paper, we attempt to specialize the VLM model for OWOD tasks by distilling its open-world knowledge into a language-agnostic detector. Surprisingly, we observe that the combination of a simple \textbf{knowledge distillation} approach and the automatic pseudo-labeling mechanism in OWOD can achieve better performance for unknown object detection, even with a small amount of data. Unfortunately, knowledge distillation for unknown objects severely affects the learning of detectors with conventional structures for known objects, leading to catastrophic forgetting. To alleviate these problems, we propose the \textbf{down-weight loss function} for knowledge distillation from vision-language to single vision modality. Meanwhile, we propose the \textbf{cascade decouple decoding structure} that decouples the learning of localization and recognition to reduce the impact of category interactions of known and unknown objects on the localization learning process. Ablation experiments demonstrate that both of them are effective in mitigating the impact of open-world knowledge distillation on the learning of known objects. Additionally, to alleviate the current lack of comprehensive benchmarks for evaluating the ability of the open-world detector to detect unknown objects in the open world, we propose two benchmarks, which we name "\textbf{StandardSet}$\heartsuit$" and "\textbf{IntensiveSet}$\spadesuit$" respectively, based on the complexity of their testing scenarios. Comprehensive experiments performed on OWOD, MS-COCO, and our proposed benchmarks demonstrate the effectiveness of our methods. The code and proposed dataset are available at \url{https://github.com/xiaomabufei/SKDF}.

📄 PDF Abstract BibTeX arXiv:2312.08653

Code (1)

xiaomabufei/skdf 공식 구현 pytorch

Tasks

Knowledge DistillationObjectobject-detectionObject DetectionOpen World Object DetectionWorld Knowledge

Methods 이 논문이 사용한 방법론

Knowledge Distillation A very simple way to improve the performance of almost any machine learning algorithm is to train many different models on the same data and then to average their predictions.…

Similar Papers 제목 키워드 기반

Dynamic Rectification Knowledge Distillation

2022-01-27 · Fahad Rahman Amik, Ahnaf Ismat Tasin, Silvia Ahmed, M. M. Lutfe Elahi 외

Knowledge Distillation is a technique which aims to utilize dark knowledge to compress and transfer information from a vast, well-trained neural network (teacher model) to a smaller, less capable neural network (student …

Edge-computingKnowledge DistillationObject Recognition

Masked Autoencoders Enable Efficient Knowledge Distillers

2022-08-25 · CVPR 2023 1 · Yutong Bai, Zeyu Wang, Junfei Xiao, Chen Wei 외

This paper studies the potential of distilling knowledge from pre-trained models, especially Masked Autoencoders. Our approach is simple: in addition to optimizing the pixel reconstruction loss on masked inputs, we minim…

Knowledge Distillation

Distilling Out-of-Distribution Robustness from Vision-Language Foundation Models

2023-11-02 · NeurIPS 2023 11 · Andy Zhou, Jindong Wang, Yu-Xiong Wang, Haohan Wang

We propose a conceptually simple and lightweight framework for improving the robustness of vision models through the combination of knowledge distillation and data augmentation. We address the conjecture that larger mode…

Data AugmentationDomain GeneralizationImage ClassificationKnowledge Distillation

DLIP: Distilling Language-Image Pre-training

2023-08-24 · Huafeng Kuang, Jie Wu, Xiawu Zheng, Ming Li 외

Vision-Language Pre-training (VLP) shows remarkable progress with the assistance of extremely heavy parameters, which challenges deployment in real applications. Knowledge distillation is well recognized as the essential…

Image CaptioningImage-text RetrievalKnowledge DistillationModel Compression+4

Distilling Image Classifiers in Object Detectors

2021-06-09 · NeurIPS 2021 12 · Shuxuan Guo, Jose M. Alvarez, Mathieu Salzmann

Knowledge distillation constitutes a simple yet effective way to improve the performance of a compact student network by exploiting the knowledge of a more powerful teacher. Nevertheless, the knowledge distillation liter…

Knowledge DistillationObjectobject-detectionObject Detection+1