Saved in:
Bibliographic Details
Main Authors: Dong, Songlin, Zhou, Zhengdong, Ding, Chenhao, Gao, Xinyuan, Kot, Alex, Gong, Yihong
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.01531
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929739585814528
author Dong, Songlin
Zhou, Zhengdong
Ding, Chenhao
Gao, Xinyuan
Kot, Alex
Gong, Yihong
author_facet Dong, Songlin
Zhou, Zhengdong
Ding, Chenhao
Gao, Xinyuan
Kot, Alex
Gong, Yihong
contents Prompt tuning can further enhance the performance of visual-language models across various downstream tasks (e.g., few-shot learning), enabling them to better adapt to specific applications and needs. In this paper, we present a Diversity Covariance-Aware framework that learns distributional information from the data to enhance the few-shot ability of the prompt model. First, we propose a covariance-aware method that models the covariance relationships between visual features and uses anisotropic Mahalanobis distance, instead of the suboptimal cosine distance, to measure the similarity between two modalities. We rigorously derive and prove the validity of this modeling process. Then, we propose the diversity-aware method, which learns multiple diverse soft prompts to capture different attributes of categories and aligns them independently with visual modalities. This method achieves multi-centered covariance modeling, leading to more diverse decision boundaries. Extensive experiments on 11 datasets in various tasks demonstrate the effectiveness of our method.
format Preprint
id arxiv_https___arxiv_org_abs_2503_01531
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Diversity Covariance-Aware Prompt Learning for Vision-Language Models
Dong, Songlin
Zhou, Zhengdong
Ding, Chenhao
Gao, Xinyuan
Kot, Alex
Gong, Yihong
Computer Vision and Pattern Recognition
Prompt tuning can further enhance the performance of visual-language models across various downstream tasks (e.g., few-shot learning), enabling them to better adapt to specific applications and needs. In this paper, we present a Diversity Covariance-Aware framework that learns distributional information from the data to enhance the few-shot ability of the prompt model. First, we propose a covariance-aware method that models the covariance relationships between visual features and uses anisotropic Mahalanobis distance, instead of the suboptimal cosine distance, to measure the similarity between two modalities. We rigorously derive and prove the validity of this modeling process. Then, we propose the diversity-aware method, which learns multiple diverse soft prompts to capture different attributes of categories and aligns them independently with visual modalities. This method achieves multi-centered covariance modeling, leading to more diverse decision boundaries. Extensive experiments on 11 datasets in various tasks demonstrate the effectiveness of our method.
title Diversity Covariance-Aware Prompt Learning for Vision-Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.01531