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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Bui, Phuoc-Nguyen, Nguyen, Khanh-Binh, Choo, Hyunseung
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_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