DePT: Decoupled Prompt Tuning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Ji, Wu, Shihan, Gao, Lianli, Shen, Heng Tao, Song, Jingkuan
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911802192822272
author Zhang, Ji
Wu, Shihan
Gao, Lianli
Shen, Heng Tao
Song, Jingkuan
author_facet Zhang, Ji
Wu, Shihan
Gao, Lianli
Shen, Heng Tao
Song, Jingkuan
contents This work breaks through the Base-New Tradeoff (BNT)dilemma in prompt tuning, i.e., the better the tuned model generalizes to the base (or target) task, the worse it generalizes to new tasks, and vice versa. Specifically, through an in-depth analysis of the learned features of the base and new tasks, we observe that the BNT stems from a channel bias issue, i.e., the vast majority of feature channels are occupied by base-specific knowledge, resulting in the collapse of taskshared knowledge important to new tasks. To address this, we propose the Decoupled Prompt Tuning (DePT) framework, which decouples base-specific knowledge from feature channels into an isolated feature space during prompt tuning, so as to maximally preserve task-shared knowledge in the original feature space for achieving better zero-shot generalization on new tasks. Importantly, our DePT is orthogonal to existing prompt tuning methods, hence it can improve all of them. Extensive experiments on 11 datasets show the strong flexibility and effectiveness of DePT. Our code and pretrained models are available at https://github.com/Koorye/DePT.
format Preprint
id arxiv_https___arxiv_org_abs_2309_07439
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle DePT: Decoupled Prompt Tuning
Zhang, Ji
Wu, Shihan
Gao, Lianli
Shen, Heng Tao
Song, Jingkuan
Computer Vision and Pattern Recognition
This work breaks through the Base-New Tradeoff (BNT)dilemma in prompt tuning, i.e., the better the tuned model generalizes to the base (or target) task, the worse it generalizes to new tasks, and vice versa. Specifically, through an in-depth analysis of the learned features of the base and new tasks, we observe that the BNT stems from a channel bias issue, i.e., the vast majority of feature channels are occupied by base-specific knowledge, resulting in the collapse of taskshared knowledge important to new tasks. To address this, we propose the Decoupled Prompt Tuning (DePT) framework, which decouples base-specific knowledge from feature channels into an isolated feature space during prompt tuning, so as to maximally preserve task-shared knowledge in the original feature space for achieving better zero-shot generalization on new tasks. Importantly, our DePT is orthogonal to existing prompt tuning methods, hence it can improve all of them. Extensive experiments on 11 datasets show the strong flexibility and effectiveness of DePT. Our code and pretrained models are available at https://github.com/Koorye/DePT.
title DePT: Decoupled Prompt Tuning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2309.07439