Select and Distill: Selective Dual-Teacher Knowledge Transfer for Continual Learning on Vision-Language Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Yu, Yu-Chu, Huang, Chi-Pin, Chen, Jr-Jen, Chang, Kai-Po, Lai, Yung-Hsuan, Yang, Fu-En, Wang, Yu-Chiang Frank
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929423441199104
author Yu, Yu-Chu
Huang, Chi-Pin
Chen, Jr-Jen
Chang, Kai-Po
Lai, Yung-Hsuan
Yang, Fu-En
Wang, Yu-Chiang Frank
author_facet Yu, Yu-Chu
Huang, Chi-Pin
Chen, Jr-Jen
Chang, Kai-Po
Lai, Yung-Hsuan
Yang, Fu-En
Wang, Yu-Chiang Frank
contents Large-scale vision-language models (VLMs) have shown a strong zero-shot generalization capability on unseen-domain data. However, adapting pre-trained VLMs to a sequence of downstream tasks often leads to the forgetting of previously learned knowledge and a reduction in zero-shot classification performance. To tackle this problem, we propose a unique Selective Dual-Teacher Knowledge Transfer framework that leverages the most recent fine-tuned and the original pre-trained VLMs as dual teachers to preserve the previously learned knowledge and zero-shot capabilities, respectively. With only access to an unlabeled reference dataset, our proposed framework performs a selective knowledge distillation mechanism by measuring the feature discrepancy from the dual-teacher VLMs. Consequently, our selective dual-teacher knowledge distillation mitigates catastrophic forgetting of previously learned knowledge while preserving the zero-shot capabilities of pre-trained VLMs. Extensive experiments on benchmark datasets demonstrate that our framework is favorable against state-of-the-art continual learning approaches for preventing catastrophic forgetting and zero-shot degradation. Project page: https://chuyu.org/research/snd
format Preprint
id arxiv_https___arxiv_org_abs_2403_09296
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Select and Distill: Selective Dual-Teacher Knowledge Transfer for Continual Learning on Vision-Language Models
Yu, Yu-Chu
Huang, Chi-Pin
Chen, Jr-Jen
Chang, Kai-Po
Lai, Yung-Hsuan
Yang, Fu-En
Wang, Yu-Chiang Frank
Computer Vision and Pattern Recognition
Large-scale vision-language models (VLMs) have shown a strong zero-shot generalization capability on unseen-domain data. However, adapting pre-trained VLMs to a sequence of downstream tasks often leads to the forgetting of previously learned knowledge and a reduction in zero-shot classification performance. To tackle this problem, we propose a unique Selective Dual-Teacher Knowledge Transfer framework that leverages the most recent fine-tuned and the original pre-trained VLMs as dual teachers to preserve the previously learned knowledge and zero-shot capabilities, respectively. With only access to an unlabeled reference dataset, our proposed framework performs a selective knowledge distillation mechanism by measuring the feature discrepancy from the dual-teacher VLMs. Consequently, our selective dual-teacher knowledge distillation mitigates catastrophic forgetting of previously learned knowledge while preserving the zero-shot capabilities of pre-trained VLMs. Extensive experiments on benchmark datasets demonstrate that our framework is favorable against state-of-the-art continual learning approaches for preventing catastrophic forgetting and zero-shot degradation. Project page: https://chuyu.org/research/snd
title Select and Distill: Selective Dual-Teacher Knowledge Transfer for Continual Learning on Vision-Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.09296