Abstract: For long-tailed classification, most works often pretrain a big model on a large-scale dataset, and then fine-tune the whole model for adapting to long-tailed data. Though promising, fine-tuning the whole pretrained model tends to suffer from high cost in computation and deployment of different models for different tasks, as well as weakened generalization ability for overfitting to certain features of long-tailed data. To alleviate these issues, we propose an effective Long-tailed Prompt Tuning method for long-tailed classification. LPT introduces several trainable prompts into a frozen pretrained model to adapt it to long-tailed data. For better effectiveness, we divide prompts into two groups: 1) a shared prompt for the whole long-tailed dataset to learn general features and to adapt a pretrained model into target domain; and 2) group-specific prompts to gather group-specific features for the samples which have similar features and also to empower the pretrained model with discrimination ability. Then we design a two-phase training paradigm to learn these prompts. In phase 1, we train the shared prompt via supervised prompt tuning to adapt a pretrained model to the desired long-tailed domain. In phase 2, we use the learnt shared prompt as query to select a small best matched set for a group of similar samples from the group-specific prompt set to dig the common features of these similar samples, then optimize these prompts with dual sampling strategy and asymmetric GCL loss. By only fine-tuning a few prompts while fixing the pretrained model, LPT can reduce training and deployment cost by storing a few prompts, and enjoys a strong generalization ability of the pretrained model. Experiments show that on various long-tailed benchmarks, with only ~1.1% extra parameters, LPT achieves comparable performance than previous whole model fine-tuning methods, and is more robust to domain-shift.

Making Pretrained Language Models Good Long-tailed Learners.

LPT: Long-tailed Prompt Tuning for Image Classification

What Makes Pre-trained Language Models Better Zero/Few-shot Learners?

PPT: Pre-trained Prompt Tuning for Few-shot Learning

Parameter-Efficient Tuning Makes a Good Classification Head.

Making Pre-trained Language Models End-to-end Few-shot Learners with Contrastive Prompt Tuning

Exploring Lottery Prompts for Pre-trained Language Models

Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners

Don't Stop Pretraining? Make Prompt-based Fine-tuning Powerful Learner

APrompt: Attention Prompt Tuning for Efficient Adaptation of Pre-trained Language Models

GPT understands, too

Towards Unified Prompt Tuning for Few-shot Text Classification

SGL-PT: A Strong Graph Learner with Graph Prompt Tuning

Prompt-aligned Gradient for Prompt Tuning

Knowledgeable Prompt-tuning: Incorporating Knowledge into Prompt Verbalizer for Text Classification

Masked Siamese Prompt Tuning for Few-Shot Natural Language Understanding

Tuning Language Models as Training Data Generators for Augmentation-Enhanced Few-Shot Learning

Q-Tuning: Queue-based Prompt Tuning for Lifelong Few-shot Language Learning