Stream-based active distillation for scalable model deployment

D Manjah, D Cacciarelli, B Standaert… - Proceedings of the …, 2023 - openaccess.thecvf.com
Proceedings of the IEEE/CVF Conference on Computer Vision and …, 2023openaccess.thecvf.com
This paper proposes a scalable technique for developing lightweight yet powerful models for
object detection in videos using self-training with knowledge distillation. This approach
involves training a compact student model using pseudo-labels generated by a
computationally complex but generic teacher model, which can help to reduce the need for
massive amounts of data and computational power. However, model-based annotations in
large-scale applications may propagate errors or biases. To address these issues, our paper …
Abstract
This paper proposes a scalable technique for developing lightweight yet powerful models for object detection in videos using self-training with knowledge distillation. This approach involves training a compact student model using pseudo-labels generated by a computationally complex but generic teacher model, which can help to reduce the need for massive amounts of data and computational power. However, model-based annotations in large-scale applications may propagate errors or biases. To address these issues, our paper introduces Stream-Based Active Distillation (SBAD) to endow pre-trained students with effective and efficient fine-tuning methods that are robust to teacher imperfections. The proposed pipeline:(i) adapts a pre-trained student model to a specific use case, based on a set of frames whose pseudo-labels are predicted by the teacher, and (ii) selects on-the-fly, along a streamed video, the images that should be considered to fine-tune the student model. Various selection strategies are compared, demonstrating: 1) the effectiveness of implementing distillation with pseudo-labels, and 2) the importance of selecting images for which the pre-trained student detects with a high confidence.
openaccess.thecvf.com
Showing the best result for this search. See all results