Accelerating Conditional Prompt Learning via Masked Image Modeling for Vision-Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bui, Phuoc-Nguyen, Nguyen, Khanh-Binh, Choo, Hyunseung
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916884954218496
author Bui, Phuoc-Nguyen
Nguyen, Khanh-Binh
Choo, Hyunseung
author_facet Bui, Phuoc-Nguyen
Nguyen, Khanh-Binh
Choo, Hyunseung
contents Vision-language models (VLMs) like CLIP excel in zero-shot learning but often require resource-intensive training to adapt to new tasks. Prompt learning techniques, such as CoOp and CoCoOp, offer efficient adaptation but tend to overfit to known classes, limiting generalization to unseen categories. We introduce ProMIM, a plug-and-play framework that enhances conditional prompt learning by integrating masked image modeling (MIM) into existing VLM pipelines. ProMIM leverages a simple yet effective masking strategy to generate robust, instance-conditioned prompts, seamlessly augmenting methods like CoOp and CoCoOp without altering their core architectures. By masking only visible image patches and using these representations to guide prompt generation, ProMIM improves feature robustness and mitigates overfitting, all while introducing negligible additional computational cost. Extensive experiments across zero-shot and few-shot classification tasks demonstrate that ProMIM consistently boosts generalization performance when plugged into existing approaches, providing a practical, lightweight solution for real-world vision-language applications.
format Preprint
id arxiv_https___arxiv_org_abs_2508_04942
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Accelerating Conditional Prompt Learning via Masked Image Modeling for Vision-Language Models
Bui, Phuoc-Nguyen
Nguyen, Khanh-Binh
Choo, Hyunseung
Computer Vision and Pattern Recognition
Vision-language models (VLMs) like CLIP excel in zero-shot learning but often require resource-intensive training to adapt to new tasks. Prompt learning techniques, such as CoOp and CoCoOp, offer efficient adaptation but tend to overfit to known classes, limiting generalization to unseen categories. We introduce ProMIM, a plug-and-play framework that enhances conditional prompt learning by integrating masked image modeling (MIM) into existing VLM pipelines. ProMIM leverages a simple yet effective masking strategy to generate robust, instance-conditioned prompts, seamlessly augmenting methods like CoOp and CoCoOp without altering their core architectures. By masking only visible image patches and using these representations to guide prompt generation, ProMIM improves feature robustness and mitigates overfitting, all while introducing negligible additional computational cost. Extensive experiments across zero-shot and few-shot classification tasks demonstrate that ProMIM consistently boosts generalization performance when plugged into existing approaches, providing a practical, lightweight solution for real-world vision-language applications.
title Accelerating Conditional Prompt Learning via Masked Image Modeling for Vision-Language Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2508.04942